EuclideanDistanceRandomConnections.cpp

Go to the documentation of this file.

#include "EuclideanDistanceRandomConnections.h"
#include <iostream>
using std::cerr;
using std::endl;

#include <typeinfo>
using std::type_info;

EuclideanDistanceRandomConnections::EuclideanDistanceRandomConnections( const double C, const double lambda )
        : maxConnProb(C), lambda(lambda), unirnd(UniformDistribution(0,1)) {
    /* NOOP */
}


EuclideanDistanceRandomConnections::EuclideanDistanceRandomConnections( const double C, const double lambda, MPI::Intracomm const& mpiComm)
        : maxConnProb(C), lambda(lambda), unirnd(0,1) {
    /* NOOP */
}

void EuclideanDistanceRandomConnections::init(const SimObjectPopulation &srcPopulation, const SimObjectPopulation &dstPopulation) {
    src_popul  = dynamic_cast<const SpatialSimObjectPopulation*>( &srcPopulation  );
    dst_popul = dynamic_cast<const SpatialSimObjectPopulation*>( &dstPopulation );
    if (!src_popul || !dst_popul) {
        throw PCSIM::ConstructionException("EuclideanDistanceRandomConnections::initPopulations", "EuclideanDistanceRandomConnections can operate only with SpatialSimObjectPopulation populations");
    }
}

void EuclideanDistanceRandomConnections::reset(
    SimObject::ID::SortedVector::const_iterator src_begin_it,
    SimObject::ID::SortedVector::const_iterator src_end_it,
    SimObject::ID::SortedVector::const_iterator dst_begin_it,
    SimObject::ID::SortedVector::const_iterator dst_end_it) {

    curr_src_idx = 0;
    curr_dst_idx = -1;

    this->src_begin_it = src_begin_it;
    this->dst_begin_it = dst_begin_it;

    max_src_idx = src_end_it - src_begin_it;
    max_dst_idx = dst_end_it - dst_begin_it;

}

bool EuclideanDistanceRandomConnections::next( pair<SimObject::ID, SimObject::ID> &conn_pair )
{
    last_conn_valid = false;

    int src, dst;

    do {
        if (++curr_dst_idx == max_dst_idx) {
            curr_dst_idx = 0;
            if (++curr_src_idx == max_src_idx)
                return false;
        }
        src = *(src_begin_it + curr_src_idx);
        dst = *(dst_begin_it + curr_dst_idx);

    } while ( unirnd( *m_rnd_eng ) > maxConnProb *
              exp( - sqr_distance( src_popul->getLocation(src), dst_popul->getLocation(dst)) / (lambda*lambda) )
              || ((*src_popul).getID(src) == (*dst_popul).getID(dst)));

    conn_pair.first = (*src_popul).getID(src);
    conn_pair.second = (*dst_popul).getID(dst);

    last_conn_idx.first = src;
    last_conn_idx.second = dst;

    last_conn_valid = true;

    return true;
}

---