Peter M. Groen / openbr

Commit b423c5321d1417bd027ba079d6540487f2759baa

Authored by Josh Klontz
1 parent e9ff5ce4

remove download transform

Inline Side-by-side
Showing 1 changed file with 0 additions and 126 deletions
openbr/plugins/io/download.cpp deleted
View file @e9ff5ce
1   -/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
2   - * Copyright 2012 The MITRE Corporation *
3   - * *
4   - * Licensed under the Apache License, Version 2.0 (the "License"); *
5   - * you may not use this file except in compliance with the License. *
6   - * You may obtain a copy of the License at *
7   - * *
8   - * http://www.apache.org/licenses/LICENSE-2.0 *
9   - * *
10   - * Unless required by applicable law or agreed to in writing, software *
11   - * distributed under the License is distributed on an "AS IS" BASIS, *
12   - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
13   - * See the License for the specific language governing permissions and *
14   - * limitations under the License. *
15   - * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
16   -
17   -#include <QtNetwork>
18   -#include <opencv2/highgui/highgui.hpp>
19   -
20   -#include <openbr/plugins/openbr_internal.h>
21   -
22   -using namespace cv;
23   -
24   -namespace br
25   -{
26   -
27   -/*!
28   - * \ingroup transforms
29   - * \brief Downloads an image from a URL
30   - * \author Josh Klontz \cite jklontz
31   - */
32   -class DownloadTransform : public UntrainableMetaTransform
33   -{
34   - Q_OBJECT
35   - Q_ENUMS(Mode)
36   - Q_PROPERTY(Mode mode READ get_mode WRITE set_mode RESET reset_mode STORED false)
37   -
38   -public:
39   - enum Mode { Permissive,
40   - Encoded,
41   - Decoded };
42   -private:
43   - BR_PROPERTY(Mode, mode, Encoded)
44   -
45   - // The reasons for this data structure are as follows:
46   - // 1) The QNetworkAccessManager must be used in the thread that _created_ it,
47   - // hence the use of `QThreadStorage`.
48   - // 2) The QThreadStorage must be deleted _after_ the threads that added QNetworkAccessManager
49   - // to it are deleted, hence the `static` ensuring that `nam` is deleted at program termination,
50   - // long after the threads that created QNetworkAccessManager are deleted.
51   - static QThreadStorage<QNetworkAccessManager*> nam;
52   -
53   - void project(const Template &src, Template &dst) const
54   - {
55   - dst.file = src.file;
56   - QString url = src.file.get<QString>("URL", src.file.name).simplified();
57   - if (!url.contains("://"))
58   - url = "file://" + url;
59   - dst.file.set("URL", url);
60   -
61   - static const QRegularExpression regExp("file:///[A-Z]:/");
62   -
63   - if (url.contains(regExp))
64   - url = url.mid(8);
65   - else if (url.startsWith("file://"))
66   - url = url.mid(7);
67   -
68   - QIODevice *device = NULL;
69   - if (QFileInfo(url).exists()) {
70   - device = new QFile(url);
71   - device->open(QIODevice::ReadOnly);
72   - } else {
73   - if (!nam.hasLocalData())
74   - nam.setLocalData(new QNetworkAccessManager());
75   - const QUrl qURL(url, QUrl::StrictMode);
76   - if (qURL.isValid() && !qURL.isRelative()) {
77   - QNetworkRequest req = QNetworkRequest(qURL);
78   - req.setRawHeader("User-Agent", "br");
79   - QNetworkReply *reply = nam.localData()->get(req);
80   -
81   - reply->waitForReadyRead(-1);
82   - while (!reply->isFinished())
83   - QCoreApplication::processEvents();
84   -
85   - if (reply->error() != QNetworkReply::NoError) {
86   - qDebug() << reply->errorString() << url;
87   - reply->deleteLater();
88   - } else {
89   - device = reply;
90   - }
91   - }
92   - }
93   -
94   - QByteArray data;
95   - if (device) {
96   - data = device->readAll();
97   - delete device;
98   - device = NULL;
99   - }
100   -
101   - if (!data.isEmpty()) {
102   - Mat encoded(1, data.size(), CV_8UC1, (void*)data.data());
103   - encoded = encoded.clone();
104   - if (mode == Permissive) {
105   - dst += encoded;
106   - } else {
107   - Mat decoded = imdecode(encoded, IMREAD_UNCHANGED);
108   - if (!decoded.empty())
109   - dst += (mode == Encoded) ? encoded : decoded;
110   - }
111   -
112   - dst.file.set("ImageID", QVariant(QCryptographicHash::hash(data, QCryptographicHash::Md5).toHex()));
113   - dst.file.set("AlgorithmID", data.isEmpty() ? 0 : (mode == Decoded ? 5 : 3));
114   - } else {
115   - dst.file.fte = true;
116   - qWarning("Error opening %s", qPrintable(url));
117   - }
118   - }
119   -};
120   -QThreadStorage<QNetworkAccessManager*> DownloadTransform::nam;
121   -
122   -BR_REGISTER(Transform, DownloadTransform)
123   -
124   -} // namespace br
125   -
126   -#include "io/download.moc"