RandomFixNumberConnections.h

Go to the documentation of this file.

#ifndef RANDOMFIXNUMBERCONNECTIONS_H
#define RANDOMFIXNUMBERCONNECTIONS_H

#include <ConnectionIterator.h>
#include <set>
using std::set;


class RandomFixNumberConnections : public ConnectionIterator
{
public:
    typedef set< pair<int, int> > ConnectedPairsVector;

    RandomFixNumberConnections( const double conn_prob );

    RandomFixNumberConnections( const double conn_prob, MPI::Intracomm const& mpiComm );

    virtual ~RandomFixNumberConnections();

    virtual void init(const SimObjectPopulation &srcPopulation, const SimObjectPopulation &destPopulation);

    virtual void reset(SimObject::ID::SortedVector::const_iterator src_begin_it,
                       SimObject::ID::SortedVector::const_iterator src_end_it,
                       SimObject::ID::SortedVector::const_iterator dest_begin_it,
                       SimObject::ID::SortedVector::const_iterator dest_end_it);

    virtual bool next( pair<SimObject::ID, SimObject::ID> &connection );

    virtual size_t estimate();


protected:
    ConnectedPairsVector  connected_pairs;

    double conn_prob;
    const SimObjectPopulation * src_popul;
    const SimObjectPopulation * dest_popul;

    UniformDistribution *uni_rnd;

    int curr_src_idx;
    int curr_dest_idx;

    int src_max_idx;
    int dest_max_idx;

    unsigned int num_connections;

    SimObject::ID::SortedVector::const_iterator from_begin_it;
    SimObject::ID::SortedVector::const_iterator to_begin_it;
};

#endif

---