BeliefPropagator.h

/*
 * BeliefPropagator.h
 *
 *  Created on: Dec 2, 2010
 *      Author: bert
 *
 *      Objects of this class execute belief propagation
 *      for b-matching.
 *
 */

#ifndef BELIEFPROPAGATOR_H_
#define BELIEFPROPAGATOR_H_

#include "utils.h"
#include "IntSet.h"
#include "IndexHeap.h"
#include "SparseMatrix.h"
#include "WeightOracle.h"
#include "OscillationDetector.h"

#define DEFAULT_MAX_ITERS 20000

#define REQUIRED_ITERATIONS 5

class BeliefPropagator {
public:
        BeliefPropagator();
        virtual ~BeliefPropagator();

        void setWeightOracle(WeightOracle * wo);

        bool checkConvergence();

        void updateBeliefs();

        SparseMatrix<bool> * getEstimate();

        int getIteration();

        void setMaxIter(int i);

        void setB(int * degrees);

        void setB(int * brows, int * bcols);

        void updateRow(double * newAlpha, double * newBeta, int i);

        int getViolatedColumns() { return violatedColumns; }

        double getIterationTime() { return iterationTime; }

        double getChangeX() { return changex; }

private:
        int maxIter;
        int iter;
        int lookups;
        double * alpha;
        double * beta;
        int * betaIndex;
        int ** rowMatches;
        int ** colMatches;
        int * colMatchCount;
        int converged;
        int numThreads;
        int * b;
        bool * dontcare;
        WeightOracle * weightOracle;
        OscillationDetector * detector;

        // last iteration stats
        int violatedColumns;
        double iterationTime;
        double changex;

        pthread_t * threads;

        void initBetas();
        int updateScore(double score, double *bestScores, int *bestIndex,
                int minIndex, int column, int b);
};



typedef struct {
        int start;
        int interval;
        int size;
        double * newAlpha;
        double * newBeta;
        BeliefPropagator * bp;
} threadParam;

void * threadedUpdateRows(void * v);

#endif /* BELIEFPROPAGATOR_H_ */