add Reader class to read the imgaes used for photometric calibration

Author: nynyg

modules/pcalib/include/opencv2/pcalib.hpp

@@ -47,6 +47,9 @@
 #include <opencv2/imgproc.hpp>
 
 #include <vector>
+#include <string>
+#include <iostream>
+#include <fstream>
 
 namespace cv{ namespace pcalib{
 

modules/pcalib/include/opencv2/pcalib/Reader.hpp

@@ -0,0 +1,33 @@
+#ifndef _OPENCV_READER_HPP
+#define _OPENCV_READER_HPP
+
+#include "opencv2/pcalib.hpp"
+
+namespace cv { namespace pcalib{
+
+class Reader
+{
+public:
+    Reader(std::string folderPath, std::string timesPath);
+
+    unsigned long getNumImages();
+
+    double getTimestamp(unsigned long id);
+
+    float getExposureTime(unsigned long id);
+
+
+private:
+    inline void loadTimestamps(std::string timesFile);
+
+    std::vector<String> files;
+    std::vector<double> timeStamps;
+    std::vector<float> exposureTimes;
+
+    int width, height;
+
+    String path;
+};
+
+}} // namespace cv pcalib
+#endif //_OPENCV_READER_HPP

modules/pcalib/src/Reader.cpp

@@ -0,0 +1,81 @@
+//
+// Created by 杨楠 on 17/6/14.
+//
+
+#include "opencv2/pcalib/Reader.hpp"
+#include "precomp.hpp"
+
+namespace cv{ namespace pcalib{
+
+unsigned long Reader::getNumImages()
+{
+    return files.size();
+}
+
+void Reader::loadTimestamps(std::string timesFile)
+{
+    std::ifstream timesStream;
+    timesStream.open(timesFile);
+    timeStamps.clear();
+    exposureTimes.clear();
+    while (!timesStream.eof() && timesStream.good())
+    {
+        char buf[1000];
+        timesStream.getline(buf, 1000);
+
+        int id = 0;
+        double timeStamp = 0.0;
+        float exposureTime = 0.0;
+
+        CV_Assert(3 == scanf(buf, "%d %lf %f", &id, &timeStamp, &exposureTime));
+
+        timeStamps.push_back(timeStamp);
+        exposureTimes.push_back(exposureTime);
+    }
+    timesStream.close();
+
+    CV_Assert(timeStamps.size() == getNumImages() && exposureTimes.size() == getNumImages());
+}
+
+Reader::Reader(std::string folderPath, std::string timesPath)
+{
+    String cvFolderPath(folderPath);
+    glob(cvFolderPath, files);
+    CV_Assert(files.size() > 0);
+    std::sort(files.begin(), files.end());
+    loadTimestamps(timesPath);
+
+    width = 0;
+    height = 0;
+
+    for(unsigned long i = 0; i < files.size(); ++i)
+    {
+        Mat img = imread(files[i]);
+        CV_Assert(img.type() == CV_8U);
+        if(0 == i)
+        {
+            width = img.cols;
+            height = img.rows;
+        }
+        else
+        {
+            CV_Assert(width == img.cols && height == img.rows);
+        }
+    }
+
+    std::cout<<getNumImages()<<" imgases from"<<folderPath<<" loaded successfully!"<<std::endl;
+}
+
+double Reader::getTimestamp(unsigned long id)
+{
+    CV_Assert(id >= 0 && id < timeStamps.size());
+    return timeStamps[id];
+}
+
+float Reader::getExposureTime(unsigned long id)
+{
+    CV_Assert(id >= 0 && id < exposureTimes.size());
+    return exposureTimes[id];
+}
+
+}}

modules/pcalib/src/pcalib.cpp

@@ -46,11 +46,6 @@
 #include "precomp.hpp"
 #include "opencv2/pcalib.hpp"
 
-#include <opencv2/core.hpp>
-
-#include <vector>
-#include <iostream>
-
 namespace cv{ namespace pcalib{
 
 using namespace std;