CudaVecRandom.h

#ifndef PSCF_CUDA_RANDOM_H
#define PSCF_CUDA_RANDOM_H
 
#include "DeviceArray.h"
 
#include <curand.h>
 
/*
* PSCF - Polymer Self-Consistent Field
*
* Copyright 2015 - 2025, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
namespace Pscf {

   class CudaVecRandom 
   {
 
   public:
   
      CudaVecRandom();
   
      virtual ~CudaVecRandom();
   
      void setSeed(unsigned long long seed);

      void uniform(DeviceArray<float>& data);

      void uniform(DeviceArray<double>& data);
   
      void uniform(DeviceArray<float>& data, float min, float max);

      void uniform(DeviceArray<double>& data, double min, double max);

      void normal(DeviceArray<float>& data, 
                  float stddev, float mean = 0.0);

      void normal(DeviceArray<double>& data, 
                  double stddev, double mean = 0.0);
   
      long seed();
   
   private:

      curandGenerator_t gen_;

      unsigned long long seed_;

      bool isInitialized_;
 
      void errorCheck(curandStatus_t const & error);
   
   };
 
   /* 
   * Returns value of random seed (private member variable idum)
   */
   inline long CudaVecRandom::seed() 
   {  return seed_; }
 
} 
#endif