ran.h

Go to the documentation of this file.

/*
  This file is part of MADNESS.

  Copyright (C) 2007,2010 Oak Ridge National Laboratory

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

  For more information please contact:

  Robert J. Harrison
  Oak Ridge National Laboratory
  One Bethel Valley Road
  P.O. Box 2008, MS-6367

  email: harrisonrj@ornl.gov
  tel:   865-241-3937
  fax:   865-572-0680

  $Id$
*/
#ifndef MADNESS_MISC_RAN_H__INCLUDED
#define MADNESS_MISC_RAN_H__INCLUDED

#include <madness/madness_config.h>
#include <madness/world/worldthread.h>

#include <complex>
typedef std::complex<float> float_complex;
typedef std::complex<double> double_complex;


namespace madness {

    struct RandomState {
        int cur;
        double u[1279];
    };


    class Random : private Mutex {
    private:
        const int r;
        const int s;
        const double beta;
        volatile int cur;
        volatile double* const u;
        unsigned int simple_state;

        void generate();

        unsigned int simple();

    public:
        Random(unsigned int seed = 5461);

        virtual ~Random();

        double get() {
            ScopedMutex<Mutex> safe(this);
            if (cur >= r) generate();
            return u[cur++];
        }

        template <typename T>
        void getv(int n, T * restrict v) {
            ScopedMutex<Mutex> safe(this);
            while (n) {
                if (cur >= r) generate();
                int ndo = std::min(n,r-cur);
                const double* ucur = const_cast<const double*>(u) + cur;
                for (int i=0; i<ndo; ++i) v[i] = (T)(ucur[i]);
                n -= ndo;
                v += ndo;
                cur += ndo;
            }
        }

        void getbytes(int n, unsigned char * restrict v);

        RandomState getstate() const;

        void setstate(const RandomState &s);

        void setstate(unsigned int seed);

        static void test();
    };


    extern Random default_random_generator;

    template <class T> T RandomValue();

    template <> double RandomValue<double> ();

    template <> float RandomValue<float> ();

    template <> int RandomValue<int> ();

    template <> long RandomValue<long> ();

    template <> double_complex RandomValue<double_complex> ();

    template <> float_complex RandomValue<float_complex> ();

    template <class T> void RandomVector(int n, T* t) {
        for (int i=0; i<n; ++i) t[i] = RandomValue<T>();
    }

    template <> void RandomVector<double>(int n, double* t);

    template <> void RandomVector<float>(int n, float* t);

    template <> void RandomVector<double_complex>(int n, double_complex* t);

    template <> void RandomVector<float_complex>(int n, float_complex* t);
}

#endif // MADNESS_MISC_RAN_H__INCLUDED