ophDepthMap.h

Go to the documentation of this file.

/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install, copy or use the software.
//
//
//                           License Agreement
//                For Open Source Digital Holographic Library
//
// Openholo library is free software;
// you can redistribute it and/or modify it under the terms of the BSD 2-Clause license.
//
// Copyright (C) 2017-2024, Korea Electronics Technology Institute. All rights reserved.
// E-mail : contact.openholo@gmail.com
// Web : http://www.openholo.org
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//  1. Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//  2. Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the copyright holder or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
// This software contains opensource software released under GNU Generic Public License,
// NVDIA Software License Agreement, or CUDA supplement to Software License Agreement.
// Check whether software you use contains licensed software.
//
//M*/

#ifndef __ophDepthMap_h
#define __ophDepthMap_h

#include "ophGen.h"
#include <cufft.h>
#include "include.h"

//Build Option : Multi Core Processing (OpenMP)
#ifdef _OPENMP
#include <omp.h>
#endif
using namespace oph;




class GEN_DLL ophDepthMap : public ophGen {
public:
    enum IMAGE_TYPE {
        COLOR = 0,
        DEPTH = 1
    };

    explicit ophDepthMap();

protected:
    virtual ~ophDepthMap();

public:
    bool readConfig(const char* fname);

    bool readImage(const char* fname, IMAGE_TYPE type = COLOR);

    bool readImageDepth(const char* source_folder, const char* img_prefix, const char* depth_img_prefix);
    //bool readImageDepth(const char* rgb, const char* depth);
    
    Real generateHologram(void);

    virtual void encoding(unsigned int ENCODE_FLAG);
    virtual void encoding(unsigned int ENCODE_FLAG, unsigned int SSB_PASSBAND);
    
    void setViewingWindow(bool is_ViewingWindow);

    ivec2 getRGBImgSize() { return m_vecRGBImg; };
    ivec2 getDepthImgSize() { return m_vecDepthImg; };

    void setRGBImgSize(ivec2 size) {
        m_vecRGBImg[_X] = size[_X];
        m_vecRGBImg[_Y] = size[_Y];
    };

    void setDepthImgSize(ivec2 size) {
        m_vecDepthImg[_X] = size[_X];
        m_vecDepthImg[_Y] = size[_Y];
    };

    void setConfig(OphDepthMapConfig config) {
        dm_config_ = config;
    };
    void setResolution(ivec2 resolution);
    uint* getProgress() { return &m_nProgress; }


    void normalize();

    inline void setFieldLens(Real fieldlens) { dm_config_.fieldLength = fieldlens; }
    inline void setNearDepth(Real neardepth) { dm_config_.near_depthmap = neardepth; }
    inline void setFarDepth(Real fardetph) { dm_config_.far_depthmap = fardetph; }
    inline void setNumOfDepth(uint numofdepth) { 
        dm_config_.default_depth_quantization = numofdepth;
        dm_config_.num_of_depth_quantization = numofdepth;
        dm_config_.num_of_depth = numofdepth;
        dm_config_.render_depth.clear();
        dm_config_.change_depth_quantization = true;
        for(int i = 1; i <= (int)numofdepth; i++)
            dm_config_.render_depth.push_back(i);
    }
    inline void setRGBImageBuffer(int idx, unsigned char* buffer, unsigned long long size)
    {
        if (idx < 0 || idx > 2) return;
        if (m_vecRGB.size() > idx)
        {
            if (m_vecRGB[idx] != nullptr)
            {
                delete[] m_vecRGB[idx];
                m_vecRGB[idx] = new uchar[size];
                memcpy(m_vecRGB[idx], buffer, size);
            }
        }
        else
        {
            uchar* pImg = new uchar[size];
            memcpy(pImg, buffer, size);
            m_vecRGB.push_back(pImg);
        }
    }

    inline void setDepthImageBuffer(unsigned char* buffer, unsigned long long size)
    {
        if (depth_img != nullptr)
        {
            delete[] depth_img;
            depth_img = nullptr;
        }
        depth_img = new unsigned char[size];
        memcpy(depth_img, buffer, size);
    }

    inline Real getFieldLens(void) { return dm_config_.fieldLength; }
    inline Real getNearDepth(void) { return dm_config_.near_depthmap; }
    inline Real getFarDepth(void) { return dm_config_.far_depthmap; }
    inline uint getNumOfDepth(void) { return dm_config_.num_of_depth; }
    inline void getRenderDepth(std::vector<int>& renderdepth) { renderdepth = dm_config_.render_depth; }
    inline unsigned char* getRGBImageBuffer(int idx) { return m_vecRGB[idx]; }
    inline unsigned char* getDepthImageBuffer() { return depth_img; }

    inline const OphDepthMapConfig& getConfig() { return dm_config_; }
    
private:
    void initCPU();

    void initGPU();

    bool prepareInputdataCPU();
    
    bool prepareInputdataGPU();

    void getDepthValues();

    void changeDepthQuanCPU();

    void changeDepthQuanGPU();

    void transVW();
    
    void calcHoloCPU();
    
    void calcHoloGPU();

    bool convertImage();

protected:
    void free_gpu(void);

    void ophFree(void);

private:
    bool                    is_ViewingWindow;
    unsigned char*          depth_img;
    vector<uchar*>          m_vecRGB;
    ivec2                   m_vecRGBImg;
    ivec2                   m_vecDepthImg;
    unsigned char*          img_src_gpu;                        
    unsigned char*          dimg_src_gpu;                       
    Real*                   depth_index_gpu;                    
    
    vector<Real *>          m_vecImgSrc;
    vector<int *>           m_vecAlphaMap;
    Real*                   dmap_src;                           
    uint*                   depth_index;                        
    vector<short>           depth_fill;
    Real*                   dmap;                               

    Real                    dstep;                              
    vector<Real>            dlevel;                             
    vector<Real>            dlevel_transform;                   

    OphDepthMapConfig       dm_config_;                         

    cufftDoubleComplex* u_o_gpu_;
    cufftDoubleComplex* u_complex_gpu_;
    cufftDoubleComplex* k_temp_d_;

    cudaStream_t    stream_;

    uint m_nProgress;
};

#endif //>__ophDepthMap_h