dr_program/time_utils.h at master · precious/dr_program · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#ifndef TIME_UTILS_H
#define TIME_UTILS_H

#include <unistd.h>
#include <ctime>
#include <cstdlib>
#include <iostream>
#include <random>
#include <functional>
#include <iomanip>
#include "constants.h"

using namespace std;

#define RAND(n) rand()%n

extern int CLOCK_ID;

typedef uniform_real_distribution<double> UniformDistribution;
typedef normal_distribution<double> GaussianDistribution;

typedef mt19937 Engine;

template <typename Eng,typename Distrib>
class Generator {
private:
    Eng engine;
    Distrib distribution;
public:
    Generator(Eng e,Distrib d): engine(e), distribution(d) {}
    double operator() () {
        return distribution(engine);
    }
};

typedef Generator<Engine,UniformDistribution> UniformDistributionGenerator;
typedef Generator<Engine,GaussianDistribution> GaussianDistributionGenerator;
typedef function<velocity ()> MaxwellDistributionSpeedGenerator;

namespace Time {
    void printTimespec(timespec*);

    UniformDistributionGenerator getUniformDistributionGenerator(double,double);
    GaussianDistributionGenerator getGaussianDistributionGenerator(double,double);
    MaxwellDistributionSpeedGenerator getMaxwellDistributionSpeedGenerator(double,double);

    timespec* getTimespecDelta(timespec*,timespec*);

    double getRandom();

    double getRandom(double,double);
}

#endif // TIME_UTILS_H