.. _program_listing_file_include_gwmodelpp_spatialweight_Weight.h:

Program Listing for File Weight.h
=================================

|exhale_lsh| :ref:`Return to documentation for file <file_include_gwmodelpp_spatialweight_Weight.h>` (``include/gwmodelpp/spatialweight/Weight.h``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   #ifndef WEIGHT_H
   #define WEIGHT_H
   
   #ifdef ENABLE_CUDA
   #include <cuda_runtime.h>
   #include "gwmodelpp/spatialweight/cuda/ISpatialCudaEnabled.h"
   #endif // ENABLE_CUDA
   
   #include <unordered_map>
   #include <string>
   #include "armadillo_config.h"
   
   
   namespace gwm
   {
   
   class Weight
   #ifdef ENABLE_CUDA
       : public ISpatialCudaEnabled
   #endif
   {
   public:
   
       enum WeightType
       {
           BandwidthWeight 
       };
   
       static std::unordered_map<WeightType, std::string> TypeNameMapper;
   
   public:
   
       Weight() {}
   
       virtual ~Weight() {}
   
       virtual Weight* clone() const = 0;
   
   public:
   
       virtual arma::vec weight(arma::vec dist) = 0;
   
   #ifdef ENABLE_CUDA
       bool useCuda() { return mUseCuda; }
   
       void setUseCuda(bool isUseCuda) { mUseCuda = isUseCuda; }
   
       virtual cudaError_t prepareCuda(size_t gpuId) override;
       
       virtual cudaError_t weight(double* d_dists, double* d_weights, size_t elems)
       {
           throw std::logic_error("Function not yet implemented");
       }
   #endif
   
   #ifdef ENABLE_CUDA
   protected:
       bool mUseCuda = false;  //<! \~english Whether to use CUDA \~chinese 是否使用 CUDA
       int mGpuID = 0;  //<! \~english The ID of selected GPU \~chinese 选择的 GPU 的索引
       bool mCudaPrepared = false;  //<! \~english Whether CUDA has been prepared \~chinese CUDA 环境是否已经准备
       size_t mCudaThreads = 0;  //<! \~english Number of GPU threads \~chinese GPU 线程数
   #endif // ENABLE_CUDA
   };
   
   }
   
   #endif // WEIGHT_H