portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-11-04 05:54:32 +00:00
Code Releases Activity

Bernoulli RNG

Browse Source
This commit is contained in:
Antonin Portelli
2016-04-30 00:14:13 -07:00
parent ca5eebe10c
commit 23b6172c31
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
lib/lattice/Lattice_rng.h
View File

@@ -75,7 +75,7 @@ namespace Grid {
std::seed_seq src; std::seed_seq src;
fixedSeed(std::vector<int> &seeds) : src(seeds.begin(),seeds.end()) {}; fixedSeed(const std::vector<int> &seeds) : src(seeds.begin(),seeds.end()) {};
result_type operator () (void){ result_type operator () (void){
@@ -119,9 +119,10 @@ namespace Grid {
typedef uint32_t RngStateType; typedef uint32_t RngStateType;
static const int RngStateCount = std::mt19937::state_size; static const int RngStateCount = std::mt19937::state_size;
#endif #endif
std::vector<RngEngine> _generators; std::vector<RngEngine> _generators;
std::vector<std::uniform_real_distribution<RealD> > _uniform; std::vector<std::uniform_real_distribution<RealD>> _uniform;
std::vector<std::normal_distribution<RealD> > _gaussian; std::vector<std::normal_distribution<RealD>> _gaussian;
std::vector<std::discrete_distribution<RealD>> _bernoulli;
void GetState(std::vector<RngStateType> & saved,int gen) { void GetState(std::vector<RngStateType> & saved,int gen) {
saved.resize(RngStateCount); saved.resize(RngStateCount);
@@ -161,6 +162,7 @@ namespace Grid {
_generators.resize(1); _generators.resize(1);
_uniform.resize(1,std::uniform_real_distribution<RealD>{0,1}); _uniform.resize(1,std::uniform_real_distribution<RealD>{0,1});
_gaussian.resize(1,std::normal_distribution<RealD>(0.0,1.0) ); _gaussian.resize(1,std::normal_distribution<RealD>(0.0,1.0) );
_bernoulli.resize(1,std::discrete_distribution<RealD>{1,1});
_seeded=0; _seeded=0;
} }
@@ -242,7 +244,7 @@ namespace Grid {
std::random_device rd; std::random_device rd;
Seed(rd); Seed(rd);
} }
void SeedFixedIntegers(std::vector<int> &seeds){ void SeedFixedIntegers(const std::vector<int> &seeds){
fixedSeed src(seeds); fixedSeed src(seeds);
Seed(src); Seed(src);
} }
@@ -266,6 +268,7 @@ namespace Grid {
_generators.resize(_vol); _generators.resize(_vol);
_uniform.resize(_vol,std::uniform_real_distribution<RealD>{0,1}); _uniform.resize(_vol,std::uniform_real_distribution<RealD>{0,1});
_gaussian.resize(_vol,std::normal_distribution<RealD>(0.0,1.0) ); _gaussian.resize(_vol,std::normal_distribution<RealD>(0.0,1.0) );
_bernoulli.resize(_vol,std::discrete_distribution<RealD>{1,1});
_seeded=0; _seeded=0;
} }
@@ -354,7 +357,7 @@ PARALLEL_FOR_LOOP
std::random_device rd; std::random_device rd;
Seed(rd); Seed(rd);
} }
void SeedFixedIntegers(std::vector<int> &seeds){ void SeedFixedIntegers(const std::vector<int> &seeds){
fixedSeed src(seeds); fixedSeed src(seeds);
Seed(src); Seed(src);
} }
@@ -368,14 +371,22 @@ PARALLEL_FOR_LOOP
template <class vobj> inline void gaussian(GridParallelRNG &rng,Lattice<vobj> &l){ template <class vobj> inline void gaussian(GridParallelRNG &rng,Lattice<vobj> &l){
rng.fill(l,rng._gaussian); rng.fill(l,rng._gaussian);
} }
template <class vobj> inline void bernoulli(GridParallelRNG &rng,Lattice<vobj> &l){
rng.fill(l,rng._bernoulli);
}
template <class sobj> inline void random(GridSerialRNG &rng,sobj &l){ template <class sobj> inline void random(GridSerialRNG &rng,sobj &l){
rng.fill(l,rng._uniform); rng.fill(l,rng._uniform);
} }
template <class sobj> inline void gaussian(GridSerialRNG &rng,sobj &l){ template <class sobj> inline void gaussian(GridSerialRNG &rng,sobj &l){
rng.fill(l,rng._gaussian); rng.fill(l,rng._gaussian);
} }
template <class sobj> inline void bernoulli(GridSerialRNG &rng,sobj &l){
rng.fill(l,rng._bernoulli);
}
} }
#endif #endif