ophRec.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 __OphReconstruction_h
#define __OphReconstruction_h

#include "Openholo.h"

#ifdef _WIN64
#ifdef RECON_EXPORT
#define RECON_DLL __declspec(dllexport)
#else
#define RECON_DLL __declspec(dllimport)
#endif
#else
#ifdef RECON_EXPORT
#define RECON_DLL __attribute__((visibility("default")))
#else
#define RECON_DLL
#endif
#endif




struct RECON_DLL OphRecConfig
{
    Real EyeLength;
    Real EyePupilDiaMeter;
    Real EyeBoxSizeScale;
    vec2 EyeBoxSize;
    int EyeBoxUnit;
    vec3 EyeCenter;
    Real EyeFocusDistance;
    Real ResultSizeScale;
    Real SimulationTo;
    Real SimulationFrom;
    int SimulationStep;
    int SimulationMode;
    Real RatioAtRetina;
    Real RatioAtPupil;
    bool CreatePupilFieldImg;
    bool CenteringRetinaImg;
    bool ViewingWindow;
    bool SimulationPos[3];

    OphRecConfig()
        : EyeLength(0.0), EyePupilDiaMeter(0.0), EyeBoxSizeScale(0.0), EyeBoxSize(0.0, 0.0), EyeBoxUnit(0),
        EyeCenter(0.0, 0.0, 0.0), EyeFocusDistance(0.0), ResultSizeScale(0.0), SimulationTo(0.0), SimulationFrom(0.0),
        SimulationStep(0), SimulationMode(0), RatioAtRetina(0.0), RatioAtPupil(0.0), CreatePupilFieldImg(false),
        CenteringRetinaImg(false), ViewingWindow(false)
    {
        SimulationPos[0] = SimulationPos[1] = SimulationPos[2] = false;
    }
};

class RECON_DLL ophRec : public Openholo
{
public:
    explicit ophRec(void);

protected:
    virtual ~ophRec(void);


private:
    vector<Real *>          m_vecEncoded;
    vector<uchar *>         m_vecNormalized;

    std::vector<Complex<Real>*> field_set_;
    std::vector<double*>    field_ret_set_;

    std::vector<double*>    res_set_;
    std::vector<double*>    res_set_norm_255_;

    std::vector<ivec2>      pn_set_;                        // Pixel number of output plane
    std::vector<vec2>       pp_set_;                        // Pixel pitch of output plane
    std::vector<ivec2>      pn_ret_set_;
    std::vector<vec2>       pp_ret_set_;
    std::vector<vec2>       ss_ret_set_;

    std::vector<Real*>      recon_set;
    std::vector<uchar*>     img_set;
    std::vector<ivec2>      img_size;
    std::vector<Real*>      focus_recon_set;
    std::vector<uchar*>     focus_img_set;
    std::vector<ivec2>      focus_img_size;

    int                     m_oldSimStep;


    OphRecConfig            rec_config;
    int                     m_nOldChannel;
    int                     m_idx;
    unsigned int            m_mode;

protected:
    virtual void ophFree(void);
    void Clear();
    void GetPupilFieldFromHologram();
    void GetPupilFieldFromVWHologram();
    void Propagation_Fresnel_FFT(int chnum);
    void ASM_Propagation();
    void ASM_Propagation_GPU();
    void GetPupilFieldImage(Complex<Real>* src, double* dst, int pnx, int pny, double ppx, double ppy, double scaleX, double scaleY);
    void getVarname(int vtr, vec3& var_vals, std::string& varname2);
public:
    void SaveImage(const char* path, const char* ext = "bmp");
    void setConfig(OphRecConfig config) { rec_config = config; }
    void SetMode(unsigned int mode) { m_mode = mode; }
    OphRecConfig& getConfig() { return rec_config; }
    bool ReconstructImage();
    bool readConfig(const char* fname);
    bool readImage(const char* path);
    bool readImagePNA(const char* phase, const char* amplitude);
    bool readImageRNI(const char* real, const char* imaginary);
    void Perform_Simulation();
    void Initialize();
    bool save(const char * fname, uint8_t bitsperpixel, uchar* src, uint px, uint py);
    template<typename T>
    void normalize(T* src, uchar* dst, int x, int y)
    {
        T maxVal = src[0];
        T minVal = src[0];
        int size = x * y;

        for (int i = 0; i < size; i++)
        {
            if (src[i] < minVal)
                minVal = src[i];
            if (src[i] > maxVal)
                maxVal = src[i];
        }
        T gap = maxVal - minVal;

        for (int i = 0; i < y; i++)
        {
            int base = i * x;
            //int base2 = (y - 1 - i) * x;

            for (int j = 0; j < x; j++)
            {
                dst[base + j] = (uchar)((src[base + j] - minVal) / gap * 255.0);
            }
        }
    }
    template<typename T>
    void normalize(T* src, uchar* dst, int x, int y, T max, T min)
    {
        T gap = max - min;

        for (int i = 0; i < y; i++)
        {
            int base = i * x;
            for (int j = 0; j < x; j++)
            {
                dst[base + j] = (uchar)((src[base + j] - min) / gap * 255.0);
            }
        }
    }
    vector<uchar*>& getNormalizedBuffer() { return m_vecNormalized; };

};


#endif // !__OphReconstruction_h