Create class VignetteCalib

Author: nynyg

modules/photometric_calib/include/opencv2/photometric_calib/VignetteCalib.hpp

@@ -0,0 +1,43 @@
+// This file is part of OpenCV project.
+// It is subject to the license terms in the LICENSE file found in the top-level directory
+// of this distribution and at http://opencv.org/license.html.
+
+#ifndef _OPENCV_VIGNETTECALIB_HPP
+#define _OPENCV_VIGNETTECALIB_HPP
+
+#include "opencv2/photometric_calib.hpp"
+#include "opencv2/photometric_calib/Reader.hpp"
+
+namespace cv { namespace photometric_calib {
+
+
+class CV_EXPORTS VignetteCalib
+{
+public:
+    VignetteCalib(std::string folderPath, std::string timePath);
+    VignetteCalib(std::string folderPath, std::string timePath, int imageSkip, int maxIterations, int outlierTh,
+                  int gridWidth, int gridHeight, float facW, float facH, int maxAbsGrad);
+
+private:
+    int _imageSkip;
+    int _maxIterations;
+    int _outlierTh;
+
+    // grid width for template image.
+    int _gridWidth;
+    int _gridHeight;
+
+    // width of grid relative to marker (fac times marker size)
+    float _facW;
+    float _facH;
+
+    // remove pixel with absolute gradient larger than this from the optimization.
+    int _maxAbsGrad;
+
+    Reader *imageReader;
+};
+
+}}
+
+
+#endif //_OPENCV_VIGNETTECALIB_HPP

modules/photometric_calib/src/VignetteCalib.cpp

@@ -0,0 +1,28 @@
+// This file is part of OpenCV project.
+// It is subject to the license terms in the LICENSE file found in the top-level directory
+// of this distribution and at http://opencv.org/license.html.
+
+#include "precomp.hpp"
+#include "opencv2/photometric_calib/VignetteCalib.hpp"
+
+#include <fstream>
+#include <iostream>
+
+namespace cv { namespace photometric_calib{
+
+
+VignetteCalib::VignetteCalib(std::string folderPath, std::string timePath) :
+        _imageSkip(1), _maxIterations(20), _outlierTh(15), _gridWidth(1000), _gridHeight(1000), _facW(5), _facH(5),
+        _maxAbsGrad(255)
+{
+    imageReader = new Reader(folderPath, "png", timePath);
+}
+
+VignetteCalib::VignetteCalib(std::string folderPath, std::string timePath, int imageSkip, int maxIterations,
+                             int outlierTh, int gridWidth, int gridHeight, float facW, float facH, int maxAbsGrad) :
+        _imageSkip(imageSkip), _maxIterations(maxIterations), _outlierTh(outlierTh), _gridWidth(gridWidth), _gridHeight(gridHeight),
+        _facW(facW), _facH(facH), _maxAbsGrad(maxAbsGrad)
+{
+    imageReader = new Reader(folderPath, "png", timePath);
+}
+}}// namespace photometric_calib, cv