NICE_VisLearning/mrf/mrfmin/MaxProdBP.h

#ifndef __MAXPRODBP_H__
#define __MAXPRODBP_H__

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "mrf.h"
#include "LinkedBlockList.h"
#include "regions-new.h"

namespace OBJREC {

#define FloatType float
#define FLOATTYPE float
class MaxProdBP;

class MaxProdBP : public MRF {
  public:
    MaxProdBP(int width, int height, int nLabels, EnergyFunction *eng);
    MaxProdBP(int nPixels, int nLabels, EnergyFunction *eng);
    ~MaxProdBP();
    void setNeighbors(int pix1, int pix2, CostVal weight);
    Label getLabel(int pixel) {
      return(m_answer[pixel]);
    };
    void setLabel(int pixel, Label label) {
      m_answer[pixel] = label;
    };
    Label* getAnswerPtr() {
      return(m_answer);
    };
    void clearAnswer();
    void setParameters(int , void *) {
      printf("No optional parameters to set");
    }
    EnergyVal smoothnessEnergy();
    EnergyVal dataEnergy();
    EnergyFunction *getEnergyFunction();
    int getWidth();
    int getHeight();
    FLOATTYPE *getScratchMatrix();
    int getNLabels();
    bool varWeights();
    void setExpScale(int expScale);
    friend void getPsiMat(OneNodeCluster &cluster, FLOATTYPE *&destMatrix,
                          int r, int c, MaxProdBP *mrf, int direction, FLOATTYPE &var_weight);

    InputType getSmoothType();
    FLOATTYPE getExpV(int i);
    FLOATTYPE *getExpV();

    CostVal getHorizWeight(int r, int c);
    CostVal getVertWeight(int r, int c);


    CostVal m_lambda;
    CostVal m_smoothMax;
    int m_smoothExp;

    enum //Borrowed from BP-S.h by vnk
    {
      NONE,
      L1,
      L2,
      FIXED_MATRIX,
      GENERAL,
      BINARY,
    } m_type;

  protected:
    void setData(DataCostFn dcost);
    void setData(CostVal* data);
    void setSmoothness(SmoothCostGeneralFn cost);
    void setSmoothness(CostVal* V);
    void setSmoothness(int smoothExp, CostVal smoothMax, CostVal lambda);
    void setCues(CostVal* hCue, CostVal* vCue);
    void initializeAlg();
    void BPinitializeAlg();
    void optimizeAlg(int nIterations);

  private:
    Label *m_answer;
    CostVal *m_V;
    CostVal *m_D;
    CostVal *m_horizWeights;
    CostVal *m_vertWeights;
    FLOATTYPE m_exp_scale;
    DataCostFn m_dataFn;
    SmoothCostGeneralFn m_smoothFn;
    bool m_needToFreeV;
    FLOATTYPE *m_scratchMatrix;
    FLOATTYPE *m_ExpData;
    FLOATTYPE *m_message_chunk;
    OneNodeCluster *nodeArray;
    typedef struct NeighborStruct {
      int     to_node;
      CostVal weight;
    } Neighbor;

    LinkedBlockList *m_neighbors;
};

}


#endif /*  __ICM_H__ */