Peter M. Groen / openbr

Browse Code »

Commit 2f3a2fa283abfcb587ef7d269e1790dc8246930e

Authored by Scott Klum 2018-08-28 15:27:43 -0600

1 parent 78eb9f12

Change qtutils to unix encoding from dos

Inline Side-by-side

Showing 2 changed files with 810 additions and 810 deletions

openbr/core/qtutils.cpp

View file @2f3a2fa

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 <QCryptographicHash>
18		-#include <QDebug>
19		-#ifndef BR_EMBEDDED
20		-#include <QDesktopServices>
21		-#endif // BR_EMBEDDED
22		-#include <QFile>
23		-#include <QFileInfo>
24		-#include <QProcess>
25		-#include <QProcessEnvironment>
26		-#include <QRegExp>
27		-#include <QRegularExpression>
28		-#include <QStack>
29		-#include <QUrl>
30		-#include <openbr/openbr_plugin.h>
31		-
32		-#include "alphanum.hpp"
33		-#include "qtutils.h"
34		-#include "opencvutils.h"
35		-
36		-using namespace br;
37		-
38		-namespace QtUtils
39		-{
40		-
41		-QStringList readLines(const QString &file)
42		-{
43		- QStringList lines;
44		- readFile(file, lines);
45		- return lines;
46		-}
47		-
48		-void readFile(const QString &file, QStringList &lines)
49		-{
50		- QByteArray data;
51		- readFile(file, data);
52		- lines = QString(data).split(QRegularExpression("[\n\|\r\n\|\r]"), QString::SkipEmptyParts);
53		- for (int i=0; i<lines.size(); i++)
54		- lines[i] = lines[i].simplified();
55		-}
56		-
57		-void readFile(const QString &file, QByteArray &data, bool uncompress)
58		-{
59		- QFile f(file);
60		- if (!f.open(QFile::ReadOnly)) {
61		- if (f.exists()) qFatal("Unable to open %s for reading. Check file permissions.", qPrintable(file));
62		- else qFatal("Unable to open %s for reading. File does not exist.", qPrintable(file));
63		- }
64		- data = f.readAll();
65		- if (uncompress) data = qUncompress(data);
66		- f.close();
67		-}
68		-
69		-void writeFile(const QString &file, const QStringList &lines)
70		-{
71		- if (file.isEmpty()) return;
72		- const QString baseName = QFileInfo(file).baseName();
73		-
74		- if (baseName == "terminal") {
75		- printf("%s\n", qPrintable(lines.join("\n")));
76		- } else if (baseName == "buffer") {
77		- Globals->buffer = lines.join("\n").toStdString().c_str();
78		- } else {
79		- QFile f(file);
80		- touchDir(f);
81		-
82		- if (!f.open(QFile::WriteOnly))
83		- qFatal("Failed to open %s for writing.", qPrintable(file));
84		-
85		- foreach (const QString &line, lines)
86		- f.write((line+"\n").toLocal8Bit());
87		-
88		- f.close();
89		- }
90		-}
91		-
92		-void writeFile(const QString &file, const QString &data)
93		-{
94		- writeFile(file, data.toLocal8Bit());
95		-}
96		-
97		-void writeFile(const QString &file, const QByteArray &data, int compression)
98		-{
99		- if (file.isEmpty()) return;
100		- const QString baseName = QFileInfo(file).baseName();
101		- const QByteArray contents = (compression == 0) ? data : qCompress(data, compression);
102		- if (baseName == "terminal") {
103		- printf("%s\n", qPrintable(contents));
104		- } else if (baseName == "buffer") {
105		- Globals->buffer = data;
106		- } else {
107		- QFile f(file);
108		- touchDir(f);
109		- if (!f.open(QFile::WriteOnly))
110		- qFatal("Failed to open %s for writing.", qPrintable(file));
111		- f.write(contents);
112		- f.close();
113		- }
114		-}
115		-
116		-void copyFile(const QString &src, const QString &dst)
117		-{
118		- touchDir(QFileInfo(dst));
119		- if (!QFile::copy(src, dst)) {
120		- if (QFileInfo(src).exists()) qFatal("Unable to copy %s to %s. Check file permissions.", qPrintable(src), qPrintable(dst));
121		- else qFatal("Unable to copy %s to %s. File does not exist.", qPrintable(src), qPrintable(dst));
122		- }
123		-}
124		-
125		-void touchDir(const QDir &dir)
126		-{
127		- if (dir.exists(".")) return;
128		- if (!dir.mkpath("."))
129		- qFatal("Unable to create path to dir %s", qPrintable(dir.absolutePath()));
130		-}
131		-
132		-void touchDir(const QFile &file)
133		-{
134		- touchDir(QFileInfo(file));
135		-}
136		-
137		-void touchDir(const QFileInfo &fileInfo)
138		-{
139		- touchDir(fileInfo.dir());
140		-}
141		-
142		-void emptyDir(QDir &dir)
143		-{
144		- foreach (const QString &folder, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot \| QDir::NoSymLinks)) {
145		- QDir subdir(dir);
146		- bool success = subdir.cd(folder); if (!success) qFatal("cd failure.");
147		- emptyDir(subdir);
148		- }
149		-
150		- foreach (const QString &file, dir.entryList(QDir::Files))
151		- dir.remove(file);
152		-
153		- foreach (const QString &folder, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot \| QDir::NoSymLinks))
154		- dir.rmdir(folder);
155		-
156		- foreach (const QString &symlink, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot))
157		- dir.remove(symlink);
158		-}
159		-
160		-void deleteDir(QDir &dir)
161		-{
162		- emptyDir(dir);
163		- dir.rmdir(".");
164		-}
165		-
166		-QString find(const QString &file, const QString &alt)
167		-{
168		- if (QFileInfo(file).exists()) return file;
169		- if (QFileInfo(alt).exists()) return alt;
170		- qFatal("Can't find file %s or alt %s\n", qPrintable(file), qPrintable(alt));
171		- return "";
172		-}
173		-
174		-bool toBool(const QString &string)
175		-{
176		- bool ok;
177		- bool result = (string.toFloat(&ok) != 0.f);
178		- if (ok) return result;
179		- else return (string != "FALSE") && (string != "false") && (string != "F") && (string != "f");
180		-}
181		-
182		-int toInt(const QString &string)
183		-{
184		- bool ok;
185		- int result = string.toInt(&ok); if (!ok) qFatal("Expected integer value, got %s.", qPrintable(string));
186		- return result;
187		-}
188		-
189		-float toFloat(const QString &string)
190		-{
191		- bool ok;
192		- float result = string.toFloat(&ok); if (!ok) qFatal("Expected floating point value, got %s.", qPrintable(string));
193		- return result;
194		-}
195		-
196		-QList<float> toFloats(const QStringList &strings)
197		-{
198		- QList<float> floats;
199		- bool ok;
200		- foreach (const QString &string, strings) {
201		- floats.append(string.toFloat(&ok));
202		- if (!ok) qFatal("Failed to convert %s to floating point format.", qPrintable(string));
203		- }
204		- return floats;
205		-}
206		-
207		-QStringList toStringList(const QList<float> &values)
208		-{
209		- QStringList result; result.reserve(values.size());
210		- foreach (float value, values)
211		- result.append(QString::number(value));
212		- return result;
213		-}
214		-
215		-QStringList toStringList(const std::vector<std::string> &string_list)
216		-{
217		- QStringList result;
218		- foreach (const std::string &string, string_list)
219		- result.append(QString::fromStdString(string));
220		- return result;
221		-}
222		-
223		-QStringList toStringList(int num_strings, const char *strings[])
224		-{
225		- QStringList result;
226		- for (int i=0; i<num_strings; i++)
227		- result.append(strings[i]);
228		- return result;
229		-}
230		-
231		-QString shortTextHash(QString string)
232		-{
233		- string.remove(QRegExp("[{}<>&]"));
234		- return QString(QCryptographicHash::hash(qPrintable(string), QCryptographicHash::Md5).toBase64()).remove(QRegExp("[^a-zA-Z1-9]")).left(6);
235		-}
236		-
237		-QStringList parse(QString args, char split, bool *ok)
238		-{
239		- if (args.isEmpty()) return QStringList();
240		-
241		- QStringList words;
242		- int start = 0;
243		- bool inQuote = false;
244		- QStack<QChar> subexpressions;
245		- for (int i=0; i<args.size(); i++) {
246		- if (inQuote) {
247		- if (args[i] == '\'')
248		- inQuote = false;
249		- } else {
250		- if (args[i] == '\'') {
251		- inQuote = true;
252		- } else if ((args[i] == '(') \|\| (args[i] == '[') \|\| (args[i] == '<') \|\| (args[i] == '{')) {
253		- subexpressions.push(args[i]);
254		- } else if (args[i] == ')') {
255		- if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '(')) {
256		- if (ok) *ok = false;
257		- else qFatal("Unexpected ')'.");
258		- return words;
259		- }
260		- } else if (args[i] == ']') {
261		- if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '[')) {
262		- if (ok) *ok = false;
263		- else qFatal("Unexpected ']'.");
264		- return words;
265		- }
266		- } else if (args[i] == '>') {
267		- if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '<')) {
268		- if (ok) *ok = false;
269		- else qFatal("Unexpected '>'.");
270		- return words;
271		- }
272		- } else if (args[i] == '}') {
273		- if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '{')) {
274		- if (ok) *ok = false;
275		- else qFatal("Unexpected '}'.");
276		- return words;
277		- }
278		- } else if (subexpressions.isEmpty() && (args[i] == split)) {
279		- words.append(args.mid(start, i-start).trimmed());
280		- start = i+1;
281		- }
282		- }
283		- }
284		-
285		- if (ok) *ok = true;
286		- words.append(args.mid(start).trimmed());
287		- return words;
288		-}
289		-
290		-void checkArgsSize(const QString &name, const QStringList &args, int min, int max)
291		-{
292		- if (max == -1) max = std::numeric_limits<int>::max();
293		- if (max == 0) max = min;
294		- if (args.size() < min) qFatal("%s expects at least %d arguments, got %d", qPrintable(name), min, args.size());
295		- if (args.size() > max) qFatal("%s expects no more than %d arguments, got %d", qPrintable(name), max, args.size());
296		-}
297		-
298		-QPointF toPoint(const QString &string, bool *ok)
299		-{
300		- if (string.startsWith('(') && string.endsWith(')')) {
301		- bool okParse;
302		- const QStringList words = parse(string.mid(1, string.size()-2), ',', &okParse);
303		- if (okParse && (words.size() == 2)) {
304		- float x, y;
305		- bool okX, okY;
306		- x = words[0].toFloat(&okX);
307		- y = words[1].toFloat(&okY);
308		- if (okX && okY) {
309		- if (ok) *ok = true;
310		- return QPointF(x, y);
311		- }
312		- }
313		- }
314		-
315		- if (ok) *ok = false;
316		- return QPointF();
317		-}
318		-
319		-QRectF toRect(const QString &string, bool *ok)
320		-{
321		- if (string.startsWith('(') && string.endsWith(')')) {
322		- bool okParse;
323		- const QStringList words = parse(string.mid(1, string.size()-2), ',', &okParse);
324		- if (okParse && (words.size() == 4)) {
325		- float x, y, width, height;
326		- bool okX, okY, okWidth, okHeight;
327		- x = words[0].toFloat(&okX);
328		- y = words[1].toFloat(&okY);
329		- width = words[2].toFloat(&okWidth);
330		- height = words[3].toFloat(&okHeight);
331		- if (okX && okY && okWidth && okHeight) {
332		- if (ok) *ok = true;
333		- return QRectF(x, y, width, height);
334		- }
335		- }
336		- }
337		-
338		- if (ok) *ok = false;
339		- return QRectF();
340		-}
341		-
342		-QStringList naturalSort(const QStringList &strings)
343		-{
344		- QList<std::string> stdStrings; stdStrings.reserve(strings.size());
345		- foreach (const QString &string, strings)
346		- stdStrings.append(string.toStdString());
347		-
348		- std::sort(stdStrings.begin(), stdStrings.end(), doj::alphanum_less<std::string>());
349		-
350		- QStringList result; result.reserve(strings.size());
351		- foreach (const std::string &stdString, stdStrings)
352		- result.append(QString::fromStdString(stdString));
353		-
354		- return result;
355		-}
356		-
357		-bool runRScript(const QString &file)
358		-{
359		-// iOS doesn't support QProcess
360		-#ifdef QT_NO_PROCESS
361		- (void) file;
362		- return false;
363		-#else // !QT_NO_PROCESS
364		- QProcess RScript;
365		- RScript.start("Rscript", QStringList() << file);
366		- RScript.waitForFinished(-1);
367		- bool result = ((RScript.exitCode() == 0) && (RScript.error() == QProcess::UnknownError));
368		- if (!result) qDebug("Failed to run 'Rscript', did you forget to install R? "
369		- "See online documentation of 'br_plot' for required R packages. "
370		- "Otherwise, try running Rscript on %s to get the exact error.", qPrintable(file));
371		- return result;
372		-#endif // QT_NO_PROCESS
373		-}
374		-
375		-bool runDot(const QString &file)
376		-{
377		-// iOS doesn't support QProcess
378		-#ifdef QT_NO_PROCESS
379		- (void) file;
380		- return false;
381		-#else // !QT_NO_PROCESS
382		- QProcess dot;
383		- dot.start("dot -Tpdf -O " + file);
384		- dot.waitForFinished(-1);
385		- return ((dot.exitCode() == 0) && (dot.error() == QProcess::UnknownError));
386		-#endif // QT_NO_PROCESS
387		-}
388		-
389		-void showFile(const QString &file)
390		-{
391		-#ifndef BR_EMBEDDED
392		- (void) file;
393		- // A bug in Qt5 currently prevents us from doing this:
394		- // QDesktopServices::openUrl(QUrl::fromLocalFile(file));
395		-#else // BR_EMBEDDED
396		- (void) file;
397		-#endif // BR_EMBEDDED
398		-}
399		-
400		-QString toString(const QVariant &variant)
401		-{
402		- if (variant.canConvert(QVariant::List)) return toString(qvariant_cast<QVariantList>(variant));
403		- else if (variant.canConvert(QVariant::String)) return variant.toString();
404		- else if (variant.canConvert(QVariant::PointF)) {
405		- QPointF point = qvariant_cast<QPointF>(variant);
406		- return QString("(%1,%2)").arg(QString::number(point.x()),QString::number(point.y()));
407		- } else if (variant.canConvert(QVariant::RectF)) {
408		- QRectF rect = qvariant_cast<QRectF>(variant);
409		- return QString("(%1,%2,%3,%4)").arg(QString::number(rect.x()),
410		- QString::number(rect.y()),
411		- QString::number(rect.width()),
412		- QString::number(rect.height()));
413		- } else if (variant.canConvert<cv::Mat>()) {
414		- return OpenCVUtils::matrixToString(variant.value<cv::Mat>());
415		- } else if (variant.canConvert<cv::RotatedRect>()) {
416		- return OpenCVUtils::rotatedRectToString(variant.value<cv::RotatedRect>());
417		- }
418		-
419		- return QString();
420		-}
421		-
422		-QString toString(const QVariantList &variantList)
423		-{
424		- QStringList variants;
425		-
426		- foreach (const QVariant &variant, variantList)
427		- variants.append(toString(variant));
428		-
429		- if (!variants.isEmpty()) return "[" + variants.join(", ") + "]";
430		-
431		- return QString();
432		-}
433		-
434		-QString toString(const QMap<QString,QVariant> &variantMap)
435		-{
436		- QStringList variants = toStringList(variantMap);
437		-
438		- if (!variants.isEmpty()) return "[" + variants.join(", ") + "]";
439		-
440		- return QString();
441		-}
442		-
443		-QStringList toStringList(const QMap<QString,QVariant> &variantMap)
444		-{
445		- QStringList variants;
446		-
447		- QMapIterator<QString, QVariant> i(variantMap);
448		- while (i.hasNext()) {
449		- i.next();
450		- variants.append(i.key() + "=" + toString(i.value()));
451		- }
452		-
453		- return variants;
454		-}
455		-
456		-QString toTime(int s)
457		-{
458		- int h = s / (60*60);
459		- int m = (s - h6060) / 60;
460		- s = (s - h6060 - m*60);
461		-
462		- const QChar fillChar = QLatin1Char('0');
463		-
464		- return QString("%1:%2:%3").arg(h,2,10,fillChar).arg(m,2,10,fillChar).arg(s,2,10,fillChar);
465		-}
466		-
467		-float euclideanLength(const QPointF &point)
468		-{
469		- return sqrt(pow(point.x(), 2) + pow(point.y(), 2));
470		-}
471		-
472		-float orientation(const QPointF &pointA, const QPointF &pointB)
473		-{
474		- return atan2(pointB.y() - pointA.y(), pointB.x() - pointA.x());
475		-}
476		-
477		-float overlap(const QRectF &r, const QRectF &s) {
478		- QRectF intersection = r & s;
479		-
480		- return (intersection.width()intersection.height())/(r.width()r.height());
481		-}
482		-
483		-
484		-QString getAbsolutePath(const QString &filename)
485		-{
486		- // Try adding the global path, if present
487		- QString withPath = (Globals->path.isEmpty() ? "" : Globals->path + "/") + filename;
488		-
489		- // we weren't necessarily using it to begin with, so see if that file
490		- // exists
491		- QFileInfo wpInfo(withPath);
492		- if (wpInfo.exists() )
493		- return wpInfo.absoluteFilePath();
494		-
495		- // If no, just use the nominal filename
496		- return QFileInfo(filename).absoluteFilePath();
497		-}
498		-
499		-const int base_block = 100000000;
500		-
501		-BlockCompression::BlockCompression(QIODevice *_basis)
502		-{
503		- blockSize = base_block;
504		- setBasis(_basis);
505		-}
506		-
507		-BlockCompression::BlockCompression() { blockSize = base_block;};
508		-
509		-bool BlockCompression::open(QIODevice::OpenMode mode)
510		-{
511		- this->setOpenMode(mode);
512		- bool res = basis->open(mode);
513		-
514		- if (!res)
515		- return false;
516		-
517		- blockReader.setDevice(basis);
518		- blockWriter.setDevice(basis);
519		-
520		- if (mode & QIODevice::WriteOnly) {
521		- precompressedBlockWriter.open(QIODevice::WriteOnly);
522		- }
523		- else if (mode & QIODevice::ReadOnly) {
524		-
525		- // Read an initial compressed block from the underlying QIODevice,
526		- // decompress, and set up a reader on it
527		- QByteArray compressedBlock;
528		- quint32 block_size;
529		- blockReader >> block_size;
530		- compressedBlock.resize(block_size);
531		- int read_count = blockReader.readRawData(compressedBlock.data(), block_size);
532		- if (read_count != int(block_size))
533		- qFatal("Failed to read initial block");
534		-
535		- decompressedBlock = qUncompress(compressedBlock);
536		-
537		- decompressedBlockReader.setBuffer(&decompressedBlock);
538		- decompressedBlockReader.open(QIODevice::ReadOnly);
539		- }
540		-
541		- return true;
542		-}
543		-
544		-void BlockCompression::close()
545		-{
546		- // flush output buffer, since we may have a partial block which hasn't been
547		- // written to disk yet.
548		- if ((openMode() & QIODevice::WriteOnly) && precompressedBlockWriter.isOpen()) {
549		- QByteArray compressedBlock = qCompress(precompressedBlockWriter.buffer());
550		- precompressedBlockWriter.close();
551		-
552		- quint32 bsize= compressedBlock.size();
553		- blockWriter << bsize;
554		- blockWriter.writeRawData(compressedBlock.constData(), compressedBlock.size());
555		- }
556		- // close the underlying device.
557		- basis->close();
558		-}
559		-
560		-void BlockCompression::setBasis(QIODevice *_basis)
561		-{
562		- basis = _basis;
563		- blockReader.setDevice(basis);
564		- blockWriter.setDevice(basis);
565		-}
566		-
567		-// read from current decompressed block, if out of space, read and decompress another
568		-// block from basis
569		-qint64 BlockCompression::readData(char *data, qint64 remaining)
570		-{
571		- qint64 read = 0;
572		- while (remaining > 0) {
573		- // attempt to read the target amount of data
574		- qint64 single_read = decompressedBlockReader.read(data, remaining);
575		- if (single_read == -1)
576		- qFatal("miss read");
577		-
578		- remaining -= single_read;
579		- read += single_read;
580		- data += single_read;
581		-
582		- // need a new block if we didn't get enough bytes from the previous read
583		- if (remaining > 0) {
584		- QByteArray compressedBlock;
585		-
586		- // read the size of the next block
587		- quint32 block_size;
588		- blockReader >> block_size;
589		- if (block_size == 0)
590		- break;
591		-
592		- compressedBlock.resize(block_size);
593		- int actualRead = blockReader.readRawData(compressedBlock.data(), block_size);
594		- if (actualRead != int(block_size))
595		- qFatal("Bad read on nominal block size: %d, only got %d", block_size, int(remaining));
596		-
597		- decompressedBlock = qUncompress(compressedBlock);
598		-
599		- decompressedBlockReader.close();
600		- decompressedBlockReader.setBuffer(&decompressedBlock);
601		- decompressedBlockReader.open(QIODevice::ReadOnly);
602		- }
603		- }
604		-
605		- bool condition = blockReader.atEnd() && !basis->isReadable() ;
606		- if (condition)
607		- qWarning("Returning -1 from read");
608		-
609		- return condition ? -1 : read;
610		-}
611		-
612		-bool BlockCompression::isSequential() const
613		-{
614		- return true;
615		-}
616		-
617		-qint64 BlockCompression::writeData(const char *data, qint64 remaining)
618		-{
619		- const char * endPoint = data + remaining;
620		- qint64 initial = remaining;
621		-
622		- qint64 written = 0;
623		-
624		- while (remaining > 0) {
625		- // how much more can be put in this buffer?
626		- qint64 capacity = blockSize - precompressedBlockWriter.pos();
627		- if (capacity < 0)
628		- qFatal("Negative capacity!!!");
629		-
630		- // don't try to write beyond capacity
631		- qint64 write_size = qMin(capacity, remaining);
632		-
633		- qint64 singleWrite = precompressedBlockWriter.write(data, write_size);
634		-
635		- if (singleWrite == -1)
636		- qFatal("matrix write failure?");
637		-
638		- remaining -= singleWrite;
639		- data += singleWrite;
640		- written += singleWrite;
641		- if (data > endPoint)
642		- qFatal("Wrote past the end");
643		-
644		- if (remaining > 0) {
645		- QByteArray compressedBlock = qCompress(precompressedBlockWriter.buffer(), -1);
646		-
647		- if (precompressedBlockWriter.buffer().size() != 0) {
648		- quint32 block_size = compressedBlock.size();
649		- blockWriter << block_size;
650		-
651		- int write_count = blockWriter.writeRawData(compressedBlock.constData(), block_size);
652		- if (write_count != int(block_size))
653		- qFatal("Didn't write enough data");
654		- }
655		- else
656		- qFatal("serialized empty compressed block (?)");
657		-
658		- precompressedBlockWriter.close();
659		- precompressedBlockWriter.open(QIODevice::WriteOnly);
660		- }
661		- }
662		-
663		- if (written != initial)
664		- qFatal("didn't write enough bytes");
665		-
666		- bool condition = basis->isWritable();
667		- if (!condition)
668		- qWarning("Returning -1 from write");
669		-
670		- return basis->isWritable() ? written : -1;
671		-}
672		-
673		-
674		-
675		-} // namespace QtUtils
676		-
	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 <QCryptographicHash>
	18	+#include <QDebug>
	19	+#ifndef BR_EMBEDDED
	20	+#include <QDesktopServices>
	21	+#endif // BR_EMBEDDED
	22	+#include <QFile>
	23	+#include <QFileInfo>
	24	+#include <QProcess>
	25	+#include <QProcessEnvironment>
	26	+#include <QRegExp>
	27	+#include <QRegularExpression>
	28	+#include <QStack>
	29	+#include <QUrl>
	30	+#include <openbr/openbr_plugin.h>
	31	+
	32	+#include "alphanum.hpp"
	33	+#include "qtutils.h"
	34	+#include "opencvutils.h"
	35	+
	36	+using namespace br;
	37	+
	38	+namespace QtUtils
	39	+{
	40	+
	41	+QStringList readLines(const QString &file)
	42	+{
	43	+ QStringList lines;
	44	+ readFile(file, lines);
	45	+ return lines;
	46	+}
	47	+
	48	+void readFile(const QString &file, QStringList &lines)
	49	+{
	50	+ QByteArray data;
	51	+ readFile(file, data);
	52	+ lines = QString(data).split(QRegularExpression("[\n\|\r\n\|\r]"), QString::SkipEmptyParts);
	53	+ for (int i=0; i<lines.size(); i++)
	54	+ lines[i] = lines[i].simplified();
	55	+}
	56	+
	57	+void readFile(const QString &file, QByteArray &data, bool uncompress)
	58	+{
	59	+ QFile f(file);
	60	+ if (!f.open(QFile::ReadOnly)) {
	61	+ if (f.exists()) qFatal("Unable to open %s for reading. Check file permissions.", qPrintable(file));
	62	+ else qFatal("Unable to open %s for reading. File does not exist.", qPrintable(file));
	63	+ }
	64	+ data = f.readAll();
	65	+ if (uncompress) data = qUncompress(data);
	66	+ f.close();
	67	+}
	68	+
	69	+void writeFile(const QString &file, const QStringList &lines)
	70	+{
	71	+ if (file.isEmpty()) return;
	72	+ const QString baseName = QFileInfo(file).baseName();
	73	+
	74	+ if (baseName == "terminal") {
	75	+ printf("%s\n", qPrintable(lines.join("\n")));
	76	+ } else if (baseName == "buffer") {
	77	+ Globals->buffer = lines.join("\n").toStdString().c_str();
	78	+ } else {
	79	+ QFile f(file);
	80	+ touchDir(f);
	81	+
	82	+ if (!f.open(QFile::WriteOnly))
	83	+ qFatal("Failed to open %s for writing.", qPrintable(file));
	84	+
	85	+ foreach (const QString &line, lines)
	86	+ f.write((line+"\n").toLocal8Bit());
	87	+
	88	+ f.close();
	89	+ }
	90	+}
	91	+
	92	+void writeFile(const QString &file, const QString &data)
	93	+{
	94	+ writeFile(file, data.toLocal8Bit());
	95	+}
	96	+
	97	+void writeFile(const QString &file, const QByteArray &data, int compression)
	98	+{
	99	+ if (file.isEmpty()) return;
	100	+ const QString baseName = QFileInfo(file).baseName();
	101	+ const QByteArray contents = (compression == 0) ? data : qCompress(data, compression);
	102	+ if (baseName == "terminal") {
	103	+ printf("%s\n", qPrintable(contents));
	104	+ } else if (baseName == "buffer") {
	105	+ Globals->buffer = data;
	106	+ } else {
	107	+ QFile f(file);
	108	+ touchDir(f);
	109	+ if (!f.open(QFile::WriteOnly))
	110	+ qFatal("Failed to open %s for writing.", qPrintable(file));
	111	+ f.write(contents);
	112	+ f.close();
	113	+ }
	114	+}
	115	+
	116	+void copyFile(const QString &src, const QString &dst)
	117	+{
	118	+ touchDir(QFileInfo(dst));
	119	+ if (!QFile::copy(src, dst)) {
	120	+ if (QFileInfo(src).exists()) qFatal("Unable to copy %s to %s. Check file permissions.", qPrintable(src), qPrintable(dst));
	121	+ else qFatal("Unable to copy %s to %s. File does not exist.", qPrintable(src), qPrintable(dst));
	122	+ }
	123	+}
	124	+
	125	+void touchDir(const QDir &dir)
	126	+{
	127	+ if (dir.exists(".")) return;
	128	+ if (!dir.mkpath("."))
	129	+ qFatal("Unable to create path to dir %s", qPrintable(dir.absolutePath()));
	130	+}
	131	+
	132	+void touchDir(const QFile &file)
	133	+{
	134	+ touchDir(QFileInfo(file));
	135	+}
	136	+
	137	+void touchDir(const QFileInfo &fileInfo)
	138	+{
	139	+ touchDir(fileInfo.dir());
	140	+}
	141	+
	142	+void emptyDir(QDir &dir)
	143	+{
	144	+ foreach (const QString &folder, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot \| QDir::NoSymLinks)) {
	145	+ QDir subdir(dir);
	146	+ bool success = subdir.cd(folder); if (!success) qFatal("cd failure.");
	147	+ emptyDir(subdir);
	148	+ }
	149	+
	150	+ foreach (const QString &file, dir.entryList(QDir::Files))
	151	+ dir.remove(file);
	152	+
	153	+ foreach (const QString &folder, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot \| QDir::NoSymLinks))
	154	+ dir.rmdir(folder);
	155	+
	156	+ foreach (const QString &symlink, dir.entryList(QDir::Dirs \| QDir::NoDotAndDotDot))
	157	+ dir.remove(symlink);
	158	+}
	159	+
	160	+void deleteDir(QDir &dir)
	161	+{
	162	+ emptyDir(dir);
	163	+ dir.rmdir(".");
	164	+}
	165	+
	166	+QString find(const QString &file, const QString &alt)
	167	+{
	168	+ if (QFileInfo(file).exists()) return file;
	169	+ if (QFileInfo(alt).exists()) return alt;
	170	+ qFatal("Can't find file %s or alt %s\n", qPrintable(file), qPrintable(alt));
	171	+ return "";
	172	+}
	173	+
	174	+bool toBool(const QString &string)
	175	+{
	176	+ bool ok;
	177	+ bool result = (string.toFloat(&ok) != 0.f);
	178	+ if (ok) return result;
	179	+ else return (string != "FALSE") && (string != "false") && (string != "F") && (string != "f");
	180	+}
	181	+
	182	+int toInt(const QString &string)
	183	+{
	184	+ bool ok;
	185	+ int result = string.toInt(&ok); if (!ok) qFatal("Expected integer value, got %s.", qPrintable(string));
	186	+ return result;
	187	+}
	188	+
	189	+float toFloat(const QString &string)
	190	+{
	191	+ bool ok;
	192	+ float result = string.toFloat(&ok); if (!ok) qFatal("Expected floating point value, got %s.", qPrintable(string));
	193	+ return result;
	194	+}
	195	+
	196	+QList<float> toFloats(const QStringList &strings)
	197	+{
	198	+ QList<float> floats;
	199	+ bool ok;
	200	+ foreach (const QString &string, strings) {
	201	+ floats.append(string.toFloat(&ok));
	202	+ if (!ok) qFatal("Failed to convert %s to floating point format.", qPrintable(string));
	203	+ }
	204	+ return floats;
	205	+}
	206	+
	207	+QStringList toStringList(const QList<float> &values)
	208	+{
	209	+ QStringList result; result.reserve(values.size());
	210	+ foreach (float value, values)
	211	+ result.append(QString::number(value));
	212	+ return result;
	213	+}
	214	+
	215	+QStringList toStringList(const std::vector<std::string> &string_list)
	216	+{
	217	+ QStringList result;
	218	+ foreach (const std::string &string, string_list)
	219	+ result.append(QString::fromStdString(string));
	220	+ return result;
	221	+}
	222	+
	223	+QStringList toStringList(int num_strings, const char *strings[])
	224	+{
	225	+ QStringList result;
	226	+ for (int i=0; i<num_strings; i++)
	227	+ result.append(strings[i]);
	228	+ return result;
	229	+}
	230	+
	231	+QString shortTextHash(QString string)
	232	+{
	233	+ string.remove(QRegExp("[{}<>&]"));
	234	+ return QString(QCryptographicHash::hash(qPrintable(string), QCryptographicHash::Md5).toBase64()).remove(QRegExp("[^a-zA-Z1-9]")).left(6);
	235	+}
	236	+
	237	+QStringList parse(QString args, char split, bool *ok)
	238	+{
	239	+ if (args.isEmpty()) return QStringList();
	240	+
	241	+ QStringList words;
	242	+ int start = 0;
	243	+ bool inQuote = false;
	244	+ QStack<QChar> subexpressions;
	245	+ for (int i=0; i<args.size(); i++) {
	246	+ if (inQuote) {
	247	+ if (args[i] == '\'')
	248	+ inQuote = false;
	249	+ } else {
	250	+ if (args[i] == '\'') {
	251	+ inQuote = true;
	252	+ } else if ((args[i] == '(') \|\| (args[i] == '[') \|\| (args[i] == '<') \|\| (args[i] == '{')) {
	253	+ subexpressions.push(args[i]);
	254	+ } else if (args[i] == ')') {
	255	+ if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '(')) {
	256	+ if (ok) *ok = false;
	257	+ else qFatal("Unexpected ')'.");
	258	+ return words;
	259	+ }
	260	+ } else if (args[i] == ']') {
	261	+ if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '[')) {
	262	+ if (ok) *ok = false;
	263	+ else qFatal("Unexpected ']'.");
	264	+ return words;
	265	+ }
	266	+ } else if (args[i] == '>') {
	267	+ if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '<')) {
	268	+ if (ok) *ok = false;
	269	+ else qFatal("Unexpected '>'.");
	270	+ return words;
	271	+ }
	272	+ } else if (args[i] == '}') {
	273	+ if (subexpressions.isEmpty() \|\| (subexpressions.pop() != '{')) {
	274	+ if (ok) *ok = false;
	275	+ else qFatal("Unexpected '}'.");
	276	+ return words;
	277	+ }
	278	+ } else if (subexpressions.isEmpty() && (args[i] == split)) {
	279	+ words.append(args.mid(start, i-start).trimmed());
	280	+ start = i+1;
	281	+ }
	282	+ }
	283	+ }
	284	+
	285	+ if (ok) *ok = true;
	286	+ words.append(args.mid(start).trimmed());
	287	+ return words;
	288	+}
	289	+
	290	+void checkArgsSize(const QString &name, const QStringList &args, int min, int max)
	291	+{
	292	+ if (max == -1) max = std::numeric_limits<int>::max();
	293	+ if (max == 0) max = min;
	294	+ if (args.size() < min) qFatal("%s expects at least %d arguments, got %d", qPrintable(name), min, args.size());
	295	+ if (args.size() > max) qFatal("%s expects no more than %d arguments, got %d", qPrintable(name), max, args.size());
	296	+}
	297	+
	298	+QPointF toPoint(const QString &string, bool *ok)
	299	+{
	300	+ if (string.startsWith('(') && string.endsWith(')')) {
	301	+ bool okParse;
	302	+ const QStringList words = parse(string.mid(1, string.size()-2), ',', &okParse);
	303	+ if (okParse && (words.size() == 2)) {
	304	+ float x, y;
	305	+ bool okX, okY;
	306	+ x = words[0].toFloat(&okX);
	307	+ y = words[1].toFloat(&okY);
	308	+ if (okX && okY) {
	309	+ if (ok) *ok = true;
	310	+ return QPointF(x, y);
	311	+ }
	312	+ }
	313	+ }
	314	+
	315	+ if (ok) *ok = false;
	316	+ return QPointF();
	317	+}
	318	+
	319	+QRectF toRect(const QString &string, bool *ok)
	320	+{
	321	+ if (string.startsWith('(') && string.endsWith(')')) {
	322	+ bool okParse;
	323	+ const QStringList words = parse(string.mid(1, string.size()-2), ',', &okParse);
	324	+ if (okParse && (words.size() == 4)) {
	325	+ float x, y, width, height;
	326	+ bool okX, okY, okWidth, okHeight;
	327	+ x = words[0].toFloat(&okX);
	328	+ y = words[1].toFloat(&okY);
	329	+ width = words[2].toFloat(&okWidth);
	330	+ height = words[3].toFloat(&okHeight);
	331	+ if (okX && okY && okWidth && okHeight) {
	332	+ if (ok) *ok = true;
	333	+ return QRectF(x, y, width, height);
	334	+ }
	335	+ }
	336	+ }
	337	+
	338	+ if (ok) *ok = false;
	339	+ return QRectF();
	340	+}
	341	+
	342	+QStringList naturalSort(const QStringList &strings)
	343	+{
	344	+ QList<std::string> stdStrings; stdStrings.reserve(strings.size());
	345	+ foreach (const QString &string, strings)
	346	+ stdStrings.append(string.toStdString());
	347	+
	348	+ std::sort(stdStrings.begin(), stdStrings.end(), doj::alphanum_less<std::string>());
	349	+
	350	+ QStringList result; result.reserve(strings.size());
	351	+ foreach (const std::string &stdString, stdStrings)
	352	+ result.append(QString::fromStdString(stdString));
	353	+
	354	+ return result;
	355	+}
	356	+
	357	+bool runRScript(const QString &file)
	358	+{
	359	+// iOS doesn't support QProcess
	360	+#ifdef QT_NO_PROCESS
	361	+ (void) file;
	362	+ return false;
	363	+#else // !QT_NO_PROCESS
	364	+ QProcess RScript;
	365	+ RScript.start("Rscript", QStringList() << file);
	366	+ RScript.waitForFinished(-1);
	367	+ bool result = ((RScript.exitCode() == 0) && (RScript.error() == QProcess::UnknownError));
	368	+ if (!result) qDebug("Failed to run 'Rscript', did you forget to install R? "
	369	+ "See online documentation of 'br_plot' for required R packages. "
	370	+ "Otherwise, try running Rscript on %s to get the exact error.", qPrintable(file));
	371	+ return result;
	372	+#endif // QT_NO_PROCESS
	373	+}
	374	+
	375	+bool runDot(const QString &file)
	376	+{
	377	+// iOS doesn't support QProcess
	378	+#ifdef QT_NO_PROCESS
	379	+ (void) file;
	380	+ return false;
	381	+#else // !QT_NO_PROCESS
	382	+ QProcess dot;
	383	+ dot.start("dot -Tpdf -O " + file);
	384	+ dot.waitForFinished(-1);
	385	+ return ((dot.exitCode() == 0) && (dot.error() == QProcess::UnknownError));
	386	+#endif // QT_NO_PROCESS
	387	+}
	388	+
	389	+void showFile(const QString &file)
	390	+{
	391	+#ifndef BR_EMBEDDED
	392	+ (void) file;
	393	+ // A bug in Qt5 currently prevents us from doing this:
	394	+ // QDesktopServices::openUrl(QUrl::fromLocalFile(file));
	395	+#else // BR_EMBEDDED
	396	+ (void) file;
	397	+#endif // BR_EMBEDDED
	398	+}
	399	+
	400	+QString toString(const QVariant &variant)
	401	+{
	402	+ if (variant.canConvert(QVariant::List)) return toString(qvariant_cast<QVariantList>(variant));
	403	+ else if (variant.canConvert(QVariant::String)) return variant.toString();
	404	+ else if (variant.canConvert(QVariant::PointF)) {
	405	+ QPointF point = qvariant_cast<QPointF>(variant);
	406	+ return QString("(%1,%2)").arg(QString::number(point.x()),QString::number(point.y()));
	407	+ } else if (variant.canConvert(QVariant::RectF)) {
	408	+ QRectF rect = qvariant_cast<QRectF>(variant);
	409	+ return QString("(%1,%2,%3,%4)").arg(QString::number(rect.x()),
	410	+ QString::number(rect.y()),
	411	+ QString::number(rect.width()),
	412	+ QString::number(rect.height()));
	413	+ } else if (variant.canConvert<cv::Mat>()) {
	414	+ return OpenCVUtils::matrixToString(variant.value<cv::Mat>());
	415	+ } else if (variant.canConvert<cv::RotatedRect>()) {
	416	+ return OpenCVUtils::rotatedRectToString(variant.value<cv::RotatedRect>());
	417	+ }
	418	+
	419	+ return QString();
	420	+}
	421	+
	422	+QString toString(const QVariantList &variantList)
	423	+{
	424	+ QStringList variants;
	425	+
	426	+ foreach (const QVariant &variant, variantList)
	427	+ variants.append(toString(variant));
	428	+
	429	+ if (!variants.isEmpty()) return "[" + variants.join(", ") + "]";
	430	+
	431	+ return QString();
	432	+}
	433	+
	434	+QString toString(const QMap<QString,QVariant> &variantMap)
	435	+{
	436	+ QStringList variants = toStringList(variantMap);
	437	+
	438	+ if (!variants.isEmpty()) return "[" + variants.join(", ") + "]";
	439	+
	440	+ return QString();
	441	+}
	442	+
	443	+QStringList toStringList(const QMap<QString,QVariant> &variantMap)
	444	+{
	445	+ QStringList variants;
	446	+
	447	+ QMapIterator<QString, QVariant> i(variantMap);
	448	+ while (i.hasNext()) {
	449	+ i.next();
	450	+ variants.append(i.key() + "=" + toString(i.value()));
	451	+ }
	452	+
	453	+ return variants;
	454	+}
	455	+
	456	+QString toTime(int s)
	457	+{
	458	+ int h = s / (60*60);
	459	+ int m = (s - h6060) / 60;
	460	+ s = (s - h6060 - m*60);
	461	+
	462	+ const QChar fillChar = QLatin1Char('0');
	463	+
	464	+ return QString("%1:%2:%3").arg(h,2,10,fillChar).arg(m,2,10,fillChar).arg(s,2,10,fillChar);
	465	+}
	466	+
	467	+float euclideanLength(const QPointF &point)
	468	+{
	469	+ return sqrt(pow(point.x(), 2) + pow(point.y(), 2));
	470	+}
	471	+
	472	+float orientation(const QPointF &pointA, const QPointF &pointB)
	473	+{
	474	+ return atan2(pointB.y() - pointA.y(), pointB.x() - pointA.x());
	475	+}
	476	+
	477	+float overlap(const QRectF &r, const QRectF &s) {
	478	+ QRectF intersection = r & s;
	479	+
	480	+ return (intersection.width()intersection.height())/(r.width()r.height());
	481	+}
	482	+
	483	+
	484	+QString getAbsolutePath(const QString &filename)
	485	+{
	486	+ // Try adding the global path, if present
	487	+ QString withPath = (Globals->path.isEmpty() ? "" : Globals->path + "/") + filename;
	488	+
	489	+ // we weren't necessarily using it to begin with, so see if that file
	490	+ // exists
	491	+ QFileInfo wpInfo(withPath);
	492	+ if (wpInfo.exists() )
	493	+ return wpInfo.absoluteFilePath();
	494	+
	495	+ // If no, just use the nominal filename
	496	+ return QFileInfo(filename).absoluteFilePath();
	497	+}
	498	+
	499	+const int base_block = 100000000;
	500	+
	501	+BlockCompression::BlockCompression(QIODevice *_basis)
	502	+{
	503	+ blockSize = base_block;
	504	+ setBasis(_basis);
	505	+}
	506	+
	507	+BlockCompression::BlockCompression() { blockSize = base_block;};
	508	+
	509	+bool BlockCompression::open(QIODevice::OpenMode mode)
	510	+{
	511	+ this->setOpenMode(mode);
	512	+ bool res = basis->open(mode);
	513	+
	514	+ if (!res)
	515	+ return false;
	516	+
	517	+ blockReader.setDevice(basis);
	518	+ blockWriter.setDevice(basis);
	519	+
	520	+ if (mode & QIODevice::WriteOnly) {
	521	+ precompressedBlockWriter.open(QIODevice::WriteOnly);
	522	+ }
	523	+ else if (mode & QIODevice::ReadOnly) {
	524	+
	525	+ // Read an initial compressed block from the underlying QIODevice,
	526	+ // decompress, and set up a reader on it
	527	+ QByteArray compressedBlock;
	528	+ quint32 block_size;
	529	+ blockReader >> block_size;
	530	+ compressedBlock.resize(block_size);
	531	+ int read_count = blockReader.readRawData(compressedBlock.data(), block_size);
	532	+ if (read_count != int(block_size))
	533	+ qFatal("Failed to read initial block");
	534	+
	535	+ decompressedBlock = qUncompress(compressedBlock);
	536	+
	537	+ decompressedBlockReader.setBuffer(&decompressedBlock);
	538	+ decompressedBlockReader.open(QIODevice::ReadOnly);
	539	+ }
	540	+
	541	+ return true;
	542	+}
	543	+
	544	+void BlockCompression::close()
	545	+{
	546	+ // flush output buffer, since we may have a partial block which hasn't been
	547	+ // written to disk yet.
	548	+ if ((openMode() & QIODevice::WriteOnly) && precompressedBlockWriter.isOpen()) {
	549	+ QByteArray compressedBlock = qCompress(precompressedBlockWriter.buffer());
	550	+ precompressedBlockWriter.close();
	551	+
	552	+ quint32 bsize= compressedBlock.size();
	553	+ blockWriter << bsize;
	554	+ blockWriter.writeRawData(compressedBlock.constData(), compressedBlock.size());
	555	+ }
	556	+ // close the underlying device.
	557	+ basis->close();
	558	+}
	559	+
	560	+void BlockCompression::setBasis(QIODevice *_basis)
	561	+{
	562	+ basis = _basis;
	563	+ blockReader.setDevice(basis);
	564	+ blockWriter.setDevice(basis);
	565	+}
	566	+
	567	+// read from current decompressed block, if out of space, read and decompress another
	568	+// block from basis
	569	+qint64 BlockCompression::readData(char *data, qint64 remaining)
	570	+{
	571	+ qint64 read = 0;
	572	+ while (remaining > 0) {
	573	+ // attempt to read the target amount of data
	574	+ qint64 single_read = decompressedBlockReader.read(data, remaining);
	575	+ if (single_read == -1)
	576	+ qFatal("miss read");
	577	+
	578	+ remaining -= single_read;
	579	+ read += single_read;
	580	+ data += single_read;
	581	+
	582	+ // need a new block if we didn't get enough bytes from the previous read
	583	+ if (remaining > 0) {
	584	+ QByteArray compressedBlock;
	585	+
	586	+ // read the size of the next block
	587	+ quint32 block_size;
	588	+ blockReader >> block_size;
	589	+ if (block_size == 0)
	590	+ break;
	591	+
	592	+ compressedBlock.resize(block_size);
	593	+ int actualRead = blockReader.readRawData(compressedBlock.data(), block_size);
	594	+ if (actualRead != int(block_size))
	595	+ qFatal("Bad read on nominal block size: %d, only got %d", block_size, int(remaining));
	596	+
	597	+ decompressedBlock = qUncompress(compressedBlock);
	598	+
	599	+ decompressedBlockReader.close();
	600	+ decompressedBlockReader.setBuffer(&decompressedBlock);
	601	+ decompressedBlockReader.open(QIODevice::ReadOnly);
	602	+ }
	603	+ }
	604	+
	605	+ bool condition = blockReader.atEnd() && !basis->isReadable() ;
	606	+ if (condition)
	607	+ qWarning("Returning -1 from read");
	608	+
	609	+ return condition ? -1 : read;
	610	+}
	611	+
	612	+bool BlockCompression::isSequential() const
	613	+{
	614	+ return true;
	615	+}
	616	+
	617	+qint64 BlockCompression::writeData(const char *data, qint64 remaining)
	618	+{
	619	+ const char * endPoint = data + remaining;
	620	+ qint64 initial = remaining;
	621	+
	622	+ qint64 written = 0;
	623	+
	624	+ while (remaining > 0) {
	625	+ // how much more can be put in this buffer?
	626	+ qint64 capacity = blockSize - precompressedBlockWriter.pos();
	627	+ if (capacity < 0)
	628	+ qFatal("Negative capacity!!!");
	629	+
	630	+ // don't try to write beyond capacity
	631	+ qint64 write_size = qMin(capacity, remaining);
	632	+
	633	+ qint64 singleWrite = precompressedBlockWriter.write(data, write_size);
	634	+
	635	+ if (singleWrite == -1)
	636	+ qFatal("matrix write failure?");
	637	+
	638	+ remaining -= singleWrite;
	639	+ data += singleWrite;
	640	+ written += singleWrite;
	641	+ if (data > endPoint)
	642	+ qFatal("Wrote past the end");
	643	+
	644	+ if (remaining > 0) {
	645	+ QByteArray compressedBlock = qCompress(precompressedBlockWriter.buffer(), -1);
	646	+
	647	+ if (precompressedBlockWriter.buffer().size() != 0) {
	648	+ quint32 block_size = compressedBlock.size();
	649	+ blockWriter << block_size;
	650	+
	651	+ int write_count = blockWriter.writeRawData(compressedBlock.constData(), block_size);
	652	+ if (write_count != int(block_size))
	653	+ qFatal("Didn't write enough data");
	654	+ }
	655	+ else
	656	+ qFatal("serialized empty compressed block (?)");
	657	+
	658	+ precompressedBlockWriter.close();
	659	+ precompressedBlockWriter.open(QIODevice::WriteOnly);
	660	+ }
	661	+ }
	662	+
	663	+ if (written != initial)
	664	+ qFatal("didn't write enough bytes");
	665	+
	666	+ bool condition = basis->isWritable();
	667	+ if (!condition)
	668	+ qWarning("Returning -1 from write");
	669	+
	670	+ return basis->isWritable() ? written : -1;
	671	+}
	672	+
	673	+
	674	+
	675	+} // namespace QtUtils
	676	+
...	...

openbr/core/qtutils.h

View file @2f3a2fa

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		-#ifndef QTUTILS_QTUTILS_H
18		-#define QTUTILS_QTUTILS_H
19		-
20		-#include <QBuffer>
21		-#include <QByteArray>
22		-#include <QDataStream>
23		-#include <QDir>
24		-#include <QFile>
25		-#include <QFileInfo>
26		-#include <QFuture>
27		-#include <QFutureSynchronizer>
28		-#include <QMap>
29		-#include <QString>
30		-#include <QStringList>
31		-#include <QThreadPool>
32		-#include <string>
33		-#include <vector>
34		-
35		-#include <openbr/openbr_export.h>
36		-
37		-namespace QtUtils
38		-{
39		- /** File Utilities **/
40		- QStringList readLines(const QString &file);
41		- void readFile(const QString &file, QStringList &lines);
42		- void readFile(const QString &file, QByteArray &data, bool uncompress = false);
43		- void writeFile(const QString &file, const QStringList &lines);
44		- void writeFile(const QString &file, const QString &data);
45		- void writeFile(const QString &file, const QByteArray &data, int compression = 0);
46		- void copyFile(const QString &src, const QString &dst);
47		-
48		- /** Directory Utilities **/
49		- void touchDir(const QDir &dir);
50		- void touchDir(const QFile &file);
51		- void touchDir(const QFileInfo &fileInfo);
52		- void emptyDir(QDir &dir);
53		- void deleteDir(QDir &dir);
54		- QString find(const QString &file, const QString &alt);
55		- QString getAbsolutePath(const QString &filename);
56		-
57		- /** String Utilities **/
58		- bool toBool(const QString &string);
59		- int toInt(const QString &string);
60		- float toFloat(const QString &string);
61		- QList<float> toFloats(const QStringList &strings);
62		- QStringList toStringList(const QList<float> &values);
63		- QStringList toStringList(const std::vector<std::string> &string_list);
64		- QStringList toStringList(int num_strings, const char* strings[]);
65		- QString shortTextHash(QString string);
66		- QStringList parse(QString args, char split = ',', bool *ok = NULL);
67		- void checkArgsSize(const QString &name, const QStringList &args, int min, int max);
68		- BR_EXPORT QPointF toPoint(const QString &string, bool *ok = NULL);
69		- BR_EXPORT QRectF toRect(const QString &string, bool *ok = NULL);
70		- QStringList naturalSort(const QStringList &strings);
71		- QString toTime(int s);
72		-
73		- /** Process Utilities **/
74		- bool runRScript(const QString &file);
75		- bool runDot(const QString &file);
76		- void showFile(const QString &file);
77		-
78		- /** Variant Utilities **/
79		- BR_EXPORT QString toString(const QVariant &variant);
80		- QString toString(const QVariantList &variantList);
81		- BR_EXPORT QString toString(const QVariantMap &QVariantMap);
82		- BR_EXPORT QStringList toStringList(const QVariantMap &QVariantMap);
83		-
84		- template <typename T>
85		- QVariantList toVariantList(const QList<T> &list)
86		- {
87		- QVariantList variantList;
88		- foreach (const T &item, list)
89		- variantList << item;
90		-
91		- return variantList;
92		- }
93		-
94		- /** Point Utilities **/
95		- float euclideanLength(const QPointF &point);
96		- float orientation(const QPointF &pointA, const QPointF &pointB);
97		-
98		- /** Rect Utilities **/
99		- float overlap(const QRectF &r, const QRectF &s);
100		-
101		-
102		- class BlockCompression : public QIODevice
103		- {
104		- public:
105		- BlockCompression(QIODevice *_basis);
106		- BlockCompression();
107		- int blockSize;
108		- QIODevice *basis;
109		-
110		- bool open(QIODevice::OpenMode mode);
111		-
112		- void close();
113		-
114		- void setBasis(QIODevice *_basis);
115		-
116		- QDataStream blockReader;
117		- QByteArray decompressedBlock;
118		- QBuffer decompressedBlockReader;
119		-
120		- // read from current decompressed block, if out of space, read and decompress another
121		- // block from basis
122		- qint64 readData(char *data, qint64 remaining);
123		-
124		- bool isSequential() const;
125		-
126		- // write to a QByteArray, when max block sized is reached, compress and write
127		- // it to basis
128		- QBuffer precompressedBlockWriter;
129		- QDataStream blockWriter;
130		- qint64 writeData(const char *data, qint64 remaining);
131		- };
132		-}
133		-
134		-#endif // QTUTILS_QTUTILS_H
	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	+#ifndef QTUTILS_QTUTILS_H
	18	+#define QTUTILS_QTUTILS_H
	19	+
	20	+#include <QBuffer>
	21	+#include <QByteArray>
	22	+#include <QDataStream>
	23	+#include <QDir>
	24	+#include <QFile>
	25	+#include <QFileInfo>
	26	+#include <QFuture>
	27	+#include <QFutureSynchronizer>
	28	+#include <QMap>
	29	+#include <QString>
	30	+#include <QStringList>
	31	+#include <QThreadPool>
	32	+#include <string>
	33	+#include <vector>
	34	+
	35	+#include <openbr/openbr_export.h>
	36	+
	37	+namespace QtUtils
	38	+{
	39	+ /** File Utilities **/
	40	+ QStringList readLines(const QString &file);
	41	+ void readFile(const QString &file, QStringList &lines);
	42	+ void readFile(const QString &file, QByteArray &data, bool uncompress = false);
	43	+ void writeFile(const QString &file, const QStringList &lines);
	44	+ void writeFile(const QString &file, const QString &data);
	45	+ void writeFile(const QString &file, const QByteArray &data, int compression = 0);
	46	+ void copyFile(const QString &src, const QString &dst);
	47	+
	48	+ /** Directory Utilities **/
	49	+ void touchDir(const QDir &dir);
	50	+ void touchDir(const QFile &file);
	51	+ void touchDir(const QFileInfo &fileInfo);
	52	+ void emptyDir(QDir &dir);
	53	+ void deleteDir(QDir &dir);
	54	+ QString find(const QString &file, const QString &alt);
	55	+ QString getAbsolutePath(const QString &filename);
	56	+
	57	+ /** String Utilities **/
	58	+ bool toBool(const QString &string);
	59	+ int toInt(const QString &string);
	60	+ float toFloat(const QString &string);
	61	+ QList<float> toFloats(const QStringList &strings);
	62	+ QStringList toStringList(const QList<float> &values);
	63	+ QStringList toStringList(const std::vector<std::string> &string_list);
	64	+ QStringList toStringList(int num_strings, const char* strings[]);
	65	+ QString shortTextHash(QString string);
	66	+ QStringList parse(QString args, char split = ',', bool *ok = NULL);
	67	+ void checkArgsSize(const QString &name, const QStringList &args, int min, int max);
	68	+ BR_EXPORT QPointF toPoint(const QString &string, bool *ok = NULL);
	69	+ BR_EXPORT QRectF toRect(const QString &string, bool *ok = NULL);
	70	+ QStringList naturalSort(const QStringList &strings);
	71	+ QString toTime(int s);
	72	+
	73	+ /** Process Utilities **/
	74	+ bool runRScript(const QString &file);
	75	+ bool runDot(const QString &file);
	76	+ void showFile(const QString &file);
	77	+
	78	+ /** Variant Utilities **/
	79	+ BR_EXPORT QString toString(const QVariant &variant);
	80	+ QString toString(const QVariantList &variantList);
	81	+ BR_EXPORT QString toString(const QVariantMap &QVariantMap);
	82	+ BR_EXPORT QStringList toStringList(const QVariantMap &QVariantMap);
	83	+
	84	+ template <typename T>
	85	+ QVariantList toVariantList(const QList<T> &list)
	86	+ {
	87	+ QVariantList variantList;
	88	+ foreach (const T &item, list)
	89	+ variantList << item;
	90	+
	91	+ return variantList;
	92	+ }
	93	+
	94	+ /** Point Utilities **/
	95	+ float euclideanLength(const QPointF &point);
	96	+ float orientation(const QPointF &pointA, const QPointF &pointB);
	97	+
	98	+ /** Rect Utilities **/
	99	+ float overlap(const QRectF &r, const QRectF &s);
	100	+
	101	+
	102	+ class BlockCompression : public QIODevice
	103	+ {
	104	+ public:
	105	+ BlockCompression(QIODevice *_basis);
	106	+ BlockCompression();
	107	+ int blockSize;
	108	+ QIODevice *basis;
	109	+
	110	+ bool open(QIODevice::OpenMode mode);
	111	+
	112	+ void close();
	113	+
	114	+ void setBasis(QIODevice *_basis);
	115	+
	116	+ QDataStream blockReader;
	117	+ QByteArray decompressedBlock;
	118	+ QBuffer decompressedBlockReader;
	119	+
	120	+ // read from current decompressed block, if out of space, read and decompress another
	121	+ // block from basis
	122	+ qint64 readData(char *data, qint64 remaining);
	123	+
	124	+ bool isSequential() const;
	125	+
	126	+ // write to a QByteArray, when max block sized is reached, compress and write
	127	+ // it to basis
	128	+ QBuffer precompressedBlockWriter;
	129	+ QDataStream blockWriter;
	130	+ qint64 writeData(const char *data, qint64 remaining);
	131	+ };
	132	+}
	133	+
	134	+#endif // QTUTILS_QTUTILS_H
...	...