SimilarAttributeConnectionPredicate.h

Go to the documentation of this file.

#ifndef SIMILARATTRIBUTECONNECTIONPREDICATE_H
#define SIMILARATTRIBUTECONNECTIONPREDICATE_H

#include <ConnectionDecisionPredicateSpecialization.h>
#include <string>


class SimilarAttributeConnectionPredicate : public AugmentedConnectionDecisionPredicate
{
public:
    SimilarAttributeConnectionPredicate(string attr, double C=1.0, double tol=0.0)
            : attr(attr), C(C), tol(tol), unirnd(0.0, 1.0)
    {
    };


    virtual bool decide(size_t src, size_t dst, RandomEngine *rnd)
    {
        const SimObjectAttributes & dst_attr = m_destinationPopulation->getAttributes(dst);
        const SimObjectAttributes & src_attr = m_sourcePopulation->getAttributes(src);

        bool connect=false;
        try{
                connect = fabs(dst_attr.getDouble(attr) - src_attr.getDouble(attr)) <= tol; 
        }
        catch (PCSIM::Exception e)
        {
                // TODO: e.g. throw exception
        }

        return connect && (unirnd(*rnd) < C);
    };

private:
    string attr;
    double C;
    double tol;

    UniformDistribution unirnd;
};

#endif

---