VirtualRangeCam.h

Go to the documentation of this file.

/*
 * Copyright 2017 Fraunhofer Institute for Manufacturing Engineering and Automation (IPA)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */



#ifndef __IPA_VIRTUALRANGECAM_H__
#define __IPA_VIRTUALRANGECAM_H__

#include "StdAfx.h"

#ifdef __LINUX__
        #include <cob_camera_sensors/AbstractRangeImagingSensor.h>
#else
        #include <cob_driver/cob_camera_sensors/common/include/cob_camera_sensors/AbstractRangeImagingSensor.h>
#endif

#include <stdio.h>
#include <math.h>
#include <assert.h>
#include <sstream>

#include <boost/filesystem.hpp>

namespace ipa_CameraSensors {

static const int SWISSRANGER_COLUMNS = 176;
static const int SWISSRANGER_ROWS = 144;

class __DLL_LIBCAMERASENSORS__ VirtualRangeCam : public AbstractRangeImagingSensor
{
public:

        VirtualRangeCam();
        ~VirtualRangeCam();

        //*******************************************************************************
        // AbstractRangeImagingSensor interface implementation
        //*******************************************************************************

        unsigned long Init(std::string directory, int cameraIndex = 0);

        unsigned long Open();
        unsigned long Close();

        unsigned long SetProperty(t_cameraProperty* cameraProperty);
        unsigned long SetPropertyDefaults();
        unsigned long GetProperty(t_cameraProperty* cameraProperty);

        unsigned long AcquireImages(int widthStepRange, int widthStepGray, int widthStepCartesian, char* rangeImage=NULL, char* intensityImage=NULL,
                char* cartesianImage=NULL, bool getLatestFrame=true, bool undistort=true,
                ipa_CameraSensors::t_ToFGrayImageType grayImageType = ipa_CameraSensors::INTENSITY_32F1);
        unsigned long AcquireImages(cv::Mat* rangeImage = 0, cv::Mat* intensityImage = 0,
                cv::Mat* cartesianImage = 0, bool getLatestFrame = true, bool undistort = true,
                ipa_CameraSensors::t_ToFGrayImageType grayImageType = ipa_CameraSensors::INTENSITY_32F1);

        unsigned long GetCalibratedUV(double x, double y, double z, double& u, double& v);

        unsigned long SaveParameters(const char* filename);

        bool isInitialized() {return m_initialized;}
        bool isOpen() {return m_open;}

        int GetNumberOfImages();

        unsigned long SetPathToImages(std::string path);

private:
        //*******************************************************************************
        // Camera specific members
        //*******************************************************************************

        inline void UpdateImageDimensionsOnFirstImage(std::string filename, std::string ext=".xml");

        inline void FindSourceImageFormat(std::map<std::string, int>::iterator& itCounter, std::string& ext);

        unsigned long GetCalibratedZMatlab(int u, int v, float zRaw, float& zCalibrated);
        unsigned long GetCalibratedXYMatlab(int u, int v, float z, float& x, float& y);

        unsigned long LoadParameters(const char* filename, int cameraIndex);

        bool m_CoeffsInitialized;

        cv::Mat m_CoeffsA0; 
        cv::Mat m_CoeffsA1; 
        cv::Mat m_CoeffsA2; 
        cv::Mat m_CoeffsA3; 
        cv::Mat m_CoeffsA4; 
        cv::Mat m_CoeffsA5; 
        cv::Mat m_CoeffsA6; 

        std::string m_CameraDataDirectory; 
        int m_CameraIndex; 

        std::vector<std::string> m_AmplitudeImageFileNames;
        std::vector<std::string> m_IntensityImageFileNames;
        std::vector<std::string> m_RangeImageFileNames ;
        std::vector<std::string> m_CoordinateImageFileNames ;

        int m_ImageWidth;  
        int m_ImageHeight; 

        double m_k1, m_k2, m_p1, m_p2; 
};

__DLL_LIBCAMERASENSORS__ AbstractRangeImagingSensorPtr CreateRangeImagingSensor_VirtualCam();

} // end namespace ipa_CameraSensors
#endif // __IPA_VIRTUALRANGECAM_H__