QVision: Qt's Image, Video and Computer Vision Library

00001 /*
00002  *   Copyright (C) 2007, 2008, 2009, 2010, 2011, 2012. PARP Research Group.
00003  *   <http://perception.inf.um.es>
00004  *   University of Murcia, Spain.
00005  *
00006  *   This file is part of the QVision library.
00007  *
00008  *   QVision is free software: you can redistribute it and/or modify
00009  *   it under the terms of the GNU Lesser General Public License as
00010  *   published by the Free Software Foundation, version 3 of the License.
00011  *
00012  *   QVision is distributed in the hope that it will be useful,
00013  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
00014  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
00015  *   GNU Lesser General Public License for more details.
00016  *
00017  *   You should have received a copy of the GNU Lesser General Public
00018  *   License along with QVision. If not, see <http://www.gnu.org/licenses/>.
00019  */
00020 
00024 
00025 #include <QVDirReader>
00026 
00027 #include <QUrl>
00028 
00029 #ifdef QVIPP
00030 #include <qvipp.h>
00031 #endif // QVIPP
00032 
00033 bool QVDirReader:: open(const QString & url_string,
00034                          unsigned int & suggested_cols, unsigned int & suggested_rows,
00035                          unsigned int & suggested_fps, QVVideoReader::OpenOptions & suggested_options,
00036                          QVVideoReader::TSourceMode & source_mode)
00037 {
00038 
00039         // Preferred mode: RGB
00040         source_mode = QVVideoReader::RGBMode;
00041 
00042         ended = false;
00043 
00044         // Treat both isolated directory names and path to files (with possible wildcards):
00045         QString dir_string = url_string;
00046         dir_string.remove("dir://");
00047         QStringList filters;
00048         QFileInfo info(dir_string);
00049         if(info.isDir()) { // Will read every image file in specified dir:
00050             dir = QDir(dir_string);
00051             filters << "*.jpg" << "*.png" << "*.bmp" << "*.pgm" << "*.gif" << "*.jpeg" << "*.pbm" << "*.ppm"
00052                     << "*.tiff" << "*.xbm" << "*.xpm";
00053         } else { // Will read only selected files (i.e. using wildcards) in specified dir:
00054             QString path = info.path(), files = info.fileName();
00055             QFileInfo info(path);
00056             if(not info.isDir())
00057                 return false;
00058             dir = QDir(path);
00059             filters << files;
00060         }
00061         dir.setNameFilters(filters);
00062         dir.setSorting(QDir::Name);
00063         filenames = dir.entryList();
00064 
00065         if(filenames.isEmpty())
00066             return false;
00067 
00068         // NoLoop & realTime flags:
00069         noLoop = (suggested_options & QVVideoReader::NoLoop);
00070         realTime = (suggested_options & QVVideoReader::RealTime);
00071 
00072         if(realTime)
00073             suggested_fps = 25;
00074 
00075         // std::cout << qPrintable(dir.absolutePath()) << "\n";
00076 
00077         // Will not rescale, answer size (0,0)
00078         suggested_cols = suggested_rows = 0;
00079 
00080         if(filenames.size() > 0) {
00081             index = 0;
00082             return true;
00083         } else
00084             return false;
00085 }
00086 
00087 bool QVDirReader::close()
00088 {
00089     return true;
00090 }
00091 
00092 bool QVDirReader::grab(QVImage<uChar,1> &imgR, QVImage<uChar,1> &imgG, QVImage<uChar,1> &imgB)
00093 {
00094         if(ended)
00095             return false;
00096 
00097         QString fullpathtofile = dir.absolutePath() + "/" + filenames.at(index);
00098 
00099         //std::cout << "QVDirReader::grab: Read file " << qPrintable(fullpathtofile) << "\n";
00100 
00101         QVImage<uchar,3> ColorImage(fullpathtofile);
00102 
00103         /* Not needed yet (all resizing in qvvideoreader.cpp):
00104         if((cols_out != 0 and rows_out !=0) || (ColorImage.getRows()%2==1) || (ColorImage.getCols()%2==1)) {
00105             QVImage<uchar,3> InterImage;
00106             if(cols_out == 0 || rows_out == 0)
00107                 InterImage.resize(ColorImage.getCols() & 0xfffffffe,ColorImage.getRows() & 0xfffffffe);
00108             else
00109                 InterImage.resize(cols_out,rows_out);
00110             InterImage.resetROI();
00111             Resize(ColorImage,InterImage);
00112             ColorImage = InterImage;
00113         }*/
00114 
00115         imgR = QVImage<uChar>(ColorImage.getCols(),ColorImage.getRows());
00116         imgG = QVImage<uChar>(ColorImage.getCols(),ColorImage.getRows());
00117         imgB = QVImage<uChar>(ColorImage.getCols(),ColorImage.getRows());
00118         Copy(ColorImage,0,imgR);
00119         Copy(ColorImage,1,imgG);
00120         Copy(ColorImage,2,imgB);
00121 
00122         index = index + 1;
00123         if(index == filenames.size()) {
00124             if(noLoop) {
00125                 ended = true;
00126                 return false;
00127             } else
00128                 index = 0;
00129         }
00130         return true;
00131 }
00132 
00133 int QVDirReader::getLength()
00134 {
00135     return static_cast<double>(filenames.size());
00136 }
00137 
00138 int QVDirReader::getPos()
00139 {
00140     return static_cast<double>(index);
00141 }
00142 
00143 bool QVDirReader::seek(int pos)
00144 {
00145     index = pos;
00146     return true;
00147 }
src/qvio/qvdirreader.cpp
