point_obs.hpp

Go to the documentation of this file.

#ifndef _POINT_OBS_HPP
#define _POINT_OBS_HPP

#include <iostream>
#include "properties.hpp"
#include "query_point.hpp"
#include "arak.hpp"

namespace Arak {

  class QueryPointPotential : public QueryPointListener {

    double& potential;

    Color color;

    const double blackVal;

    const double whiteVal;

  public:

    QueryPointPotential(double &potential,
      double blackVal,
      double whiteVal) : 
      potential(potential), 
      color(INVALID_COLOR), 
      blackVal(blackVal), 
      whiteVal(whiteVal) { }

    virtual void recolor(Color color) {
      if (color == this->color) return;
      // Subtract off the old potential.
      switch (this->color) {
      case BLACK:
  potential -= blackVal; 
  break;
      case WHITE:
  potential -= whiteVal; 
  break;
      default:
  break;
      }
      // Add in the new potential.
      switch (color) {
      case BLACK:
  potential += blackVal; 
  break;
      case WHITE:
  potential += whiteVal; 
  break;
      default:
  break;
      }
      // Update the current color.
      this->color = color;
    }
  };

  class ArakPosteriorGaussianObs : public ArakProcess {

  protected:

    const ArakProcess& prior;

    std::vector<Geometry::Point> points;

    std::vector<double> obs;

    std::vector<QueryPointPotential*> potentials;

    double whiteMean;
  
    double blackMean;

    double whiteVar;

    double blackVar;

    double lp;

  public:

    ArakPosteriorGaussianObs(const ArakProcess& prior,
           const Arak::Util::PropertyMap& props);

    virtual ~ArakPosteriorGaussianObs();

    virtual double scale() const { return prior.scale(); }

    virtual double logMeasure() const;

    virtual double potential() const;

    virtual void visualize(CGAL::Qt_widget& widget) const;
  };

  CGAL::Qt_widget& operator<<(CGAL::Qt_widget& widget, 
            ArakPosteriorGaussianObs& p);

  class ArakPosteriorBernoulliObs : public ArakProcess {

  protected:

    const ArakProcess& prior;

    std::vector<Geometry::Point> points;

    std::vector<QueryPointPotential*> potentials;

    std::vector<bool> obs;

    double blackProb;
  
    double whiteProb;

    double lp;

  public:

    ArakPosteriorBernoulliObs(const ArakProcess& prior,
            const Arak::Util::PropertyMap& props);

    virtual ~ArakPosteriorBernoulliObs();

    virtual double scale() const { return prior.scale(); }

    virtual double logMeasure() const;

    virtual double potential() const;

    virtual void visualize(CGAL::Qt_widget& widget) const;
  };

  CGAL::Qt_widget& operator<<(CGAL::Qt_widget& widget, 
            ArakPosteriorBernoulliObs& p);

}

#endif

---