ThreadSpecificRandomDistribution.h

Go to the documentation of this file.

#ifndef THREADSPECIFICRANDOMDISTRIBUTION_H_
#define THREADSPECIFICRANDOMDISTRIBUTION_H_

#include "ThreadSpecificRandomEngine.h"
#include "RandomDistribution.h"

template< class Dist >
class ThreadSpecificRandomDistribution
{
public:

    ThreadSpecificRandomDistribution( void ) :
            tsp(cleanup),
            alloc_ptr(0)
    {
        init();
    }

    ThreadSpecificRandomDistribution( Dist const& d ) :
            tsp(cleanup),
            alloc_ptr(0),
            dist_model(d)
    {
        init();
    }

    virtual ~ThreadSpecificRandomDistribution()
    {
        free_ptr();
    }

    void free_ptr(void) {
        boost::mutex::scoped_lock scoped_lock(mutex);
        for( size_t i=0; i<alloc_ptr.size(); i++ ) {
            if( alloc_ptr[i] != NULL ) delete alloc_ptr[i];
            alloc_ptr[i] = NULL;
        }
    }

    void init(void)
    {
        Dist *var = tsp.get();
        if( var == NULL ) {
            boost::mutex::scoped_lock scoped_lock(mutex);
            var = new Dist( dist_model );
            tsp.reset( var );
            alloc_ptr.push_back( var );
            //cerr << "ThreadSpecificRandomDistribution::init: new distribution " << typeid( Dist ).name() << " allocated: ptr=" << var << ", size=" << alloc_ptr.size() << "\n\n" ;
            theThreadSpecificRandomEngine.init();
        }
    }

    void set( Dist const& d )
    {
        Dist *var = tsp.get();
        if( var != NULL ) {
            *var = dist_model = d; // the distribution pointed to by var is overwritten with d
            //cerr << "ThreadSpecificRandomDistribution::set: dist " << var << " set\n\n" ;
        } else {
            dist_model = d;
            init();
        }
        theThreadSpecificRandomEngine.init();
    }

    double operator()(void)
    {
        // cerr << "operator(): dist= " << tsp.get() <<  ", eng=" << theThreadSpecificRandomEngine.get() << "\n";
        return (*tsp.get())( *theThreadSpecificRandomEngine.get() );
    }

private:
    boost::thread_specific_ptr< Dist > tsp;
    boost::mutex mutex;
    std::vector< Dist * > alloc_ptr;
    static void cleanup( Dist *d ) { };
    Dist dist_model;
};

#endif /*THREADSPECIFICRANDOMDISTRIBUTION_H_*/

---