usPreScanToPostScan3DConverter.h

/****************************************************************************
 *
 * This file is part of the ustk software.
 * Copyright (C) 2016 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ustk with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at ustk@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Authors:
 * Jason Chevrie
 * Marc Pouliquen
 *
 *****************************************************************************/
 
#ifndef __usPreScanToPostScan3DConverter_h_
#define __usPreScanToPostScan3DConverter_h_
 
#include <cmath>
#include <vector>
 
#include <visp3/ustk_core/usConfig.h>
#include <visp3/ustk_core/usImagePostScan3D.h>
#include <visp3/ustk_core/usImagePreScan3D.h>
 
class VISP_EXPORT usPreScanToPostScan3DConverter
{
public:
  typedef enum {
    SINGLE_THREAD_DIRECT_CONVERSION,
    MULTI_THREAD_DIRECT_CONVERSION,
    GPU_DIRECT_CONVERSION,
    SINGLE_THREAD_FULL_LOOKUP_TABLE,
    MULTI_THREAD_FULL_LOOKUP_TABLE,
    GPU_FULL_LOOKUP_TABLE,
    SINGLE_THREAD_REDUCED_LOOKUP_TABLE,
    MULTI_THREAD_REDUCED_LOOKUP_TABLE,
    GPU_REDUCED_LOOKUP_TABLE
  } usConverterOptimizationMethod;
protected:
#ifndef DOXYGEN_SHOULD_SKIP_THIS
  class usVoxelWeightAndIndex
  {
    friend class usPreScanToPostScan3DConverter;
    unsigned int m_outputIndex;
    unsigned int m_inputIndex[8];
    double m_W[8];
  };
  class usVoxelWeightAndIndexReducedMemory
  {
      friend class usPreScanToPostScan3DConverter;
      unsigned int m_outputIndex;
      unsigned int m_inputIndex;
      double m_W[3];
  };
#endif
  
  usConverterOptimizationMethod m_converterOptimizationMethod;
  usConverterOptimizationMethod m_conversionOptimizationMethodUsedAtInit;
 
 
  std::vector<usVoxelWeightAndIndex> m_lookupTables[2];
  std::vector<usVoxelWeightAndIndexReducedMemory> m_reducedLookupTables[2];
  void *m_GPULookupTables[2];
  long int m_GPULookupTablesSize[2];
 
  usImagePreScan3D<unsigned char> m_VpreScan;
 
  double m_downSamplingFactor;
  double m_resolution;
  bool m_SweepInZdirection;
 
  unsigned int m_nbX;
  unsigned int m_nbY;
  unsigned int m_nbZ;
 
  bool m_initDone;
 
public:
  usPreScanToPostScan3DConverter();
  usPreScanToPostScan3DConverter(const usImagePreScan3D<unsigned char> &preScanImage, double down);
  virtual ~usPreScanToPostScan3DConverter();
 
  void convert(usImagePostScan3D<unsigned char> &postScanImage, const usImagePreScan3D<unsigned char> &preScanImage);
 
  void init(const usImagePreScan3D<unsigned char> &preScanImage, double down = 1);
 
  void setConverterOptimizationMethod(usConverterOptimizationMethod method);
  usConverterOptimizationMethod setConverterOptimizationMethod() const {return m_converterOptimizationMethod;}
  
  double getResolution() const { return m_resolution; }
 
  void SweepInZdirection(bool flag) { m_SweepInZdirection = flag; }
 
private:
  void convertPreScanCoordToPostScanCoord(double i_preScan, double j_preScan, double k_preScan,
                                          double *i_postScan = NULL, double *j_postScan = NULL,
                                          double *k_postScan = NULL, bool sweepInZdirection = true);
  void convertPostScanCoordToPreScanCoord(double x, double y, double z, double *i = NULL, double *j = NULL,
                                          double *k = NULL, bool sweepInZdirection = true);
#ifdef USTK_HAVE_CUDA
  void GPUDirectConversion(unsigned char *dataPost, const unsigned char *dataPre);
 
  void GPUFreeLookupTables();
 
  void GPUAllocateFullLookupTables();
  void GPUFillFullLookupTables();
  void GPUFullLookupTableConversion(unsigned char *dataPost, const unsigned char *dataPre);
 
  void GPUAllocateReducedLookupTables();
  void GPUFillReducedLookupTables();
  void GPUReducedLookupTableConversion(unsigned char *dataPost, const unsigned char *dataPre);
#endif
};
 
#endif // __usPreScanToPostScan3DConverter_h_