pottsmrf.cpp

来自「The package includes 3 Matlab-interfaces」· C++ 代码 · 共 90 行

#include "PottsMRF.h"
#include <math.h>
#include "mex.h"

PottsMRF::~PottsMRF() {
  for (int i=0; i<N; i++) {
    delete[] lambdaValues[i];
  }
  delete[] lambdaValues;
  lambdaValues = 0;
}

double PottsMRF::pairPotential (int i, int n, int xi, int xj) const {

  if (xi!=xj) {
    return exp(-lambdaValues[i][n] / T);
  }
  else {
    return 1.0;
  }
}

double PottsMRF::pairEnergy (int i, int n, int xi, int xj) const {

  if (xi!=xj) {
    return lambdaValues[i][n];
  }
  else {
    return 0.0;
  }
}

void PottsMRF::initLambdaValues() {
  lambdaValues = new double*[N];
  for (int i=0; i<N; i++) {
    lambdaValues[i] = new double[neighbNum(i)];
    for (int n=0; n<neighbNum(i); n++) {
      lambdaValues[i][n] = 0.0;
    }
  }
}

double PottsMRF::getEnergy(int const* assignment) const {
  double energy = 0.0;
  for (int i=0; i<N; i++) {
    int xi = assignment[i];
    if (logspace) {
      energy += localMat[i][xi];
    }
    else {
      energy -= (log(localMat[i][xi]) * T);
    }
    for (int n=0; n<neighbNum(i); n++) {
      int j = adjMat[i][n];
      if (i<j) {
	int xj = assignment[j];
	if (xi != xj) {
	  double lambda_ij = lambdaValues[i][n];
	  energy -= lambda_ij;
	}
      }
    }
  }
  return energy;
}

void PottsMRF::setTemperature(double sT) {
  if (localMat != 0 || logspace) {
    if (sT < 0.01 && !logspace) {
      mexWarnMsgTxt("temperature must be >= 0.01 for numeric stability. updated temperature to 0.01 instead of the given one\n");
      sT = 0.01;
    }
    if (T != sT && !logspace) {
      // the power in which the potentials should be raised
      double power = T / sT;
      
      for (int i=0; i<N; i++) {
	// raise local potentials in power
	for (int xi=0; xi<V[i]; xi++) {
	  localMat[i][xi] = pow(localMat[i][xi],power);
	}
      }
    }
    T = sT;
  }
  else {
    mexWarnMsgTxt("local potentials must be given before updating temperature, cannot update temperature yet\n");
  }
}