ophPointCloud_GPU.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 __ophPointCloud_GPU_h
#define __ophPointCloud_GPU_h

#include "ophPointCloud.h"

#define __DEBUG_LOG_GPU_SPEC_

/* CUDA Library Include */
#include <cuda_runtime_api.h>
#include <cuComplex.h>

#define __CUDA_INTERNAL_COMPILATION__ //for CUDA Math Module
#include <math_constants.h>
//#include <math_functions_dbl_ptx3.h> //Double Precision Floating
#include <vector_functions.h> //Vector Processing Function
#undef __CUDA_INTERNAL_COMPILATION__

#define OPH_CUDA_N_STREAM 100
static void HandleError(cudaError_t err,
    const char *file,
    int line) {
    if (err != cudaSuccess) {
        printf("%s in %s at line %d\n", cudaGetErrorString(err),
            file, line);
        system("pause");
        exit(EXIT_FAILURE);
    }
}
#define HANDLE_ERROR( err ) (HandleError( err, __FILE__, __LINE__ ))


#define HANDLE_NULL( a ) {if (a == NULL) { \
                            printf( "Host memory failed in %s at line %d\n", \
                                    __FILE__, __LINE__ ); \
                            exit( EXIT_FAILURE );}} 
// for PointCloud only GPU
typedef struct _CudaPointCloudConfig {
    int n_points;   
    double scale_X;     
    double scale_Y;     
    double scale_Z;     

    double offset_depth;    

    int pn_X;       
    int pn_Y;       

    int offset_X; // x-axis start offset
    int offset_Y; // y-axis start offset

    double pp_X; 
    double pp_Y; 

    double half_ss_X; 
    double half_ss_Y; 

    double k;         
    double lambda;

    _CudaPointCloudConfig(
        const int &n_points,        
        const vec3 &scale_factor,   
        const Real &offset_depth,   
        const ivec2 &pixel_number,  
        const ivec2 &offset,        
        const vec2 &pixel_pitch,    
        const vec2 &ss,             
        const Real &k,              
        const Real &lambda          
    )
    {
        this->n_points = n_points;
        this->scale_X = scale_factor[_X];
        this->scale_Y = scale_factor[_Y];
        this->scale_Z = scale_factor[_Z];
        this->offset_depth = offset_depth;

        // Output Image Size
        this->pn_X = pixel_number[_X];
        this->pn_Y = pixel_number[_Y];

        // Start offset
        this->offset_X = offset[_X];
        this->offset_Y = offset[_Y];

        // Pixel pitch at eyepiece lens plane (by simple magnification) ==> SLM pitch
        this->pp_X = pixel_pitch[_X];
        this->pp_Y = pixel_pitch[_Y];

        // Length (Width) of complex field at eyepiece plane (by simple magnification)
        this->half_ss_X = ss[_X] / 2;
        this->half_ss_Y = ss[_Y] / 2;

        // Wave Number
        this->k = k;

        this->lambda = lambda;
    }
} CudaPointCloudConfig;

typedef struct _CudaPointCloudConfigRS : public _CudaPointCloudConfig {
    double det_tx;  
    double det_ty;  

    _CudaPointCloudConfigRS(
        const int &n_points,        
        const vec3 &scale_factor,   
        const Real &offset_depth,   
        const ivec2 &pixel_number,  
        const ivec2 &offset,        
        const vec2 &pixel_pitch,    
        const vec2 &ss,             
        const Real &k,              
        const Real &lambda          
    )
        : _CudaPointCloudConfig(n_points, scale_factor, offset_depth, pixel_number, offset, pixel_pitch, ss, k, lambda)
    {
        double tx = lambda / (2 * pixel_pitch[_X]);
        double ty = lambda / (2 * pixel_pitch[_Y]);
        
        this->det_tx = tx / sqrt(1 - tx * tx);
        this->det_ty = ty / sqrt(1 - ty * ty);
    }

    _CudaPointCloudConfigRS(_CudaPointCloudConfig &cuda_config)
        : _CudaPointCloudConfig(cuda_config)
    {
        double tx = lambda / (2 * cuda_config.pp_X);
        double ty = lambda / (2 * cuda_config.pp_Y);

        this->det_tx = tx / sqrt(1 - tx * tx);
        this->det_ty = ty / sqrt(1 - ty * ty);
    }
} CudaPointCloudConfigRS;


typedef struct _CudaPointCloudConfigFresnel : public _CudaPointCloudConfig {

    double tx;  
    double ty;  

    _CudaPointCloudConfigFresnel(
        const int &n_points,        
        const vec3 &scale_factor,   
        const Real &offset_depth,   
        const ivec2 &pixel_number,  
        const ivec2 &offset,        
        const vec2 &pixel_pitch,    
        const vec2 &ss,             
        const Real &k,              
        const Real &lambda          
    )
        : _CudaPointCloudConfig(n_points, scale_factor, offset_depth, pixel_number, offset, pixel_pitch, ss, k, lambda)
    {
        this->tx = lambda / (2 * pixel_pitch[_X]);
        this->ty = lambda / (2 * pixel_pitch[_Y]);
    }

    _CudaPointCloudConfigFresnel(_CudaPointCloudConfig& cuda_config)
        : _CudaPointCloudConfig(cuda_config)
    {
        this->tx = lambda / (2 * cuda_config.pp_X);
        this->ty = lambda / (2 * cuda_config.pp_Y);
    }
} CudaPointCloudConfigFresnel;


extern "C"
{   
    void sum_Kernel(
        const int& nBlocks, const int& nThreads, cuDoubleComplex* dst, cuDoubleComplex* src, int size
    );
    void cudaPointCloud_RS(
        const int& nBlocks, const int& nThreads, Vertex* cuda_vertex_data, cuDoubleComplex* cuda_dst,
        const CudaPointCloudConfigRS* cuda_config, const uint& iColor, const uint& mode
    );

    void cudaPointCloud_Fresnel(
        const int& nBlocks, const int& nThreads, Vertex* cuda_vertex_data, cuDoubleComplex* cuda_dst,
        const CudaPointCloudConfigFresnel* cuda_config, const uint& iColor, const uint& mode
    );
}

#endif