Peter M. Groen / openbr

Commit d807738201e3e9211b58e9e857cfa3aa5113b248

Authored by Josh Klontz
1 parent 43fab3dc

removed seq gallery

Inline Side-by-side
Showing 1 changed file with 0 additions and 249 deletions
openbr/plugins/gallery/seq.cpp deleted
View file @43fab3d
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 <fstream>
18   -#include <QQueue>
19   -#include <opencv2/highgui/highgui.hpp>
20   -
21   -#include <openbr/plugins/openbr_internal.h>
22   -#include <openbr/core/qtutils.h>
23   -
24   -namespace br
25   -{
26   -
27   -/*!
28   - * \brief DOCUMENT ME
29   - * \author Unknown \cite unknown
30   - */
31   -class seqGallery : public Gallery
32   -{
33   -public:
34   - Q_OBJECT
35   -
36   - bool open()
37   - {
38   - seqFile.open(QtUtils::getAbsolutePath(file.name).toStdString().c_str(), std::ios::in | std::ios::binary | std::ios::ate);
39   - if (!isOpen()) {
40   - qDebug("Failed to open file %s for reading", qPrintable(file.name));
41   - return false;
42   - }
43   -
44   - int headSize = 1024;
45   - // start at end of file to get full size
46   - int fileSize = seqFile.tellg();
47   - if (fileSize < headSize) {
48   - qDebug("No header in seq file");
49   - return false;
50   - }
51   -
52   - // first 4 bytes store 0xEDFE, next 24 store 'Norpix seq '
53   - char firstFour[4];
54   - seqFile.seekg(0, std::ios::beg);
55   - seqFile.read(firstFour, 4);
56   - char nextTwentyFour[24];
57   - readText(24, nextTwentyFour);
58   - if (firstFour[0] != (char)0xED || firstFour[1] != (char)0xFE || strncmp(nextTwentyFour, "Norpix seq", 10) != 0) {
59   - qDebug("Invalid header in seq file");
60   - return false;
61   - }
62   -
63   - // next 8 bytes for version (skipped below) and header size (1024), then 512 for descr
64   - seqFile.seekg(4, std::ios::cur);
65   - int hSize = readInt();
66   - if (hSize != headSize) {
67   - qDebug("Invalid header size");
68   - return false;
69   - }
70   - char desc[512];
71   - readText(512, desc);
72   - file.set("Description", QString(desc));
73   -
74   - width = readInt();
75   - height = readInt();
76   - // get # channels from bit depth
77   - numChan = readInt()/8;
78   - int imageBitDepthReal = readInt();
79   - if (imageBitDepthReal != 8) {
80   - qDebug("Invalid bit depth");
81   - return false;
82   - }
83   - // the size of just the image part of a raw img
84   - imgSizeBytes = readInt();
85   -
86   - int imgFormatInt = readInt();
87   - if (imgFormatInt == 100 || imgFormatInt == 200 || imgFormatInt == 101) {
88   - imgFormat = "raw";
89   - } else if (imgFormatInt == 102 || imgFormatInt == 201 || imgFormatInt == 103 ||
90   - imgFormatInt == 1 || imgFormatInt == 2) {
91   - imgFormat = "compressed";
92   - } else {
93   - qFatal("unsupported image format");
94   - }
95   -
96   - numFrames = readInt();
97   - // skip empty int
98   - seqFile.seekg(4, std::ios::cur);
99   - // the size of a full raw file, with extra crap after img data
100   - trueImgSizeBytes = readInt();
101   -
102   - // gather all the frame positions in an array
103   - seekPos.reserve(numFrames);
104   - // start at end of header
105   - seekPos.append(headSize);
106   - // extra 8 bytes at end of img
107   - int extra = 8;
108   - for (int i=1; i<numFrames; i++) {
109   - int s;
110   - // compressed images have different sizes
111   - // the first byte at the beginning of the file
112   - // says how big the current img is
113   - if (imgFormat == "compressed") {
114   - int lastPos = seekPos[i-1];
115   - seqFile.seekg(lastPos, std::ios::beg);
116   - int currSize = readInt();
117   - s = lastPos + currSize + extra;
118   -
119   - // but there might be 16 extra bytes instead of 8...
120   - if (i == 1) {
121   - seqFile.seekg(s, std::ios::beg);
122   - char zero;
123   - seqFile.read(&zero, 1);
124   - if (zero == 0) {
125   - s += 8;
126   - extra += 8;
127   - }
128   - }
129   - }
130   - // raw images are all the same size
131   - else {
132   - s = headSize + (i*trueImgSizeBytes);
133   - }
134   -
135   - seekPos.enqueue(s);
136   - }
137   -
138   -#ifdef CVMATIO
139   - if (basis.file.contains("vbb")) {
140   - QString vbb = basis.file.get<QString>("vbb");
141   - annotations = TemplateList::fromGallery(File(vbb));
142   - }
143   -#else
144   - qWarning("cvmatio not installed, bounding boxes will not be available. Add -DBR_WITH_CVMATIO cmake flag to install.");
145   -#endif
146   -
147   - return true;
148   - }
149   -
150   - bool isOpen()
151   - {
152   - return seqFile.is_open();
153   - }
154   -
155   - void close()
156   - {
157   - seqFile.close();
158   - }
159   -
160   - TemplateList readBlock(bool *done)
161   - {
162   - if (!isOpen()) {
163   - if (!open())
164   - qFatal("Failed to open file %s for reading", qPrintable(file.name));
165   - else
166   - idx = 0;
167   - }
168   -
169   - // if we've reached the last frame, we're done
170   - if (seekPos.size() == 0) {
171   - *done = true;
172   - return TemplateList();
173   - }
174   -
175   - seqFile.seekg(seekPos.dequeue(), std::ios::beg);
176   -
177   - cv::Mat temp;
178   - // let imdecode do all the work to decode the compressed img
179   - if (imgFormat == "compressed") {
180   - int imgSize = readInt() - 4;
181   - std::vector<char> imgBuf(imgSize);
182   - seqFile.read(&imgBuf[0], imgSize);
183   - // flags < 0 means load image as-is (keep color info if available)
184   - cv::imdecode(imgBuf, -1, &temp);
185   - }
186   - // raw images can be loaded straight into a Mat
187   - else {
188   - char *imgBuf = new char[imgSizeBytes];
189   - seqFile.read(imgBuf, imgSizeBytes);
190   - int type = (numChan == 1 ? CV_8UC1 : CV_8UC3);
191   - temp = cv::Mat(height, width, type, imgBuf);
192   - }
193   - Template output;
194   - output.file = file;
195   - if (!annotations.empty()) {
196   - output.file.setRects(annotations.first().file.rects());
197   - annotations.removeFirst();
198   - }
199   - output.m() = temp;
200   - output.file.set("position",idx);
201   - idx++;
202   -
203   - *done = false;
204   - TemplateList rVal;
205   - rVal.append(output);
206   -
207   - return rVal;
208   - }
209   -
210   - void write(const Template &t)
211   - {
212   - (void) t;
213   - qFatal("Not implemented.");
214   - }
215   -
216   -private:
217   - qint64 idx;
218   - int readInt()
219   - {
220   - int num;
221   - seqFile.read((char*)&num, 4);
222   - return num;
223   - }
224   -
225   - // apparently the text in seq files is 16 bit characters (UTF-16?)
226   - // since we don't really need the last byte, snad since it gets interpreted as
227   - // a terminating char, let's just grab the first byte for storage
228   - void readText(int bytes, char *buffer)
229   - {
230   - seqFile.read(buffer, bytes);
231   - for (int i=0; i<bytes; i+=2) {
232   - buffer[i/2] = buffer[i];
233   - }
234   - buffer[bytes/2] = '\0';
235   - }
236   -
237   -protected:
238   - std::ifstream seqFile;
239   - QQueue<int> seekPos;
240   - int width, height, numChan, imgSizeBytes, trueImgSizeBytes, numFrames;
241   - QString imgFormat;
242   - TemplateList annotations;
243   -};
244   -
245   -BR_REGISTER(Gallery, seqGallery)
246   -
247   -} // namespace br
248   -
249   -#include "gallery/seq.moc"