arak.hpp

Go to the documentation of this file.

#ifndef _ARAK_HPP
#define _ARAK_HPP

#include "properties.hpp"
#include "coloring.hpp"

namespace CGAL {
  class Qt_widget;
}

namespace Arak {

  class ArakProcess {

  protected:

    const Coloring& c;

  public:

    ArakProcess(const Coloring& c) : c(c) { }

    virtual ~ArakProcess() { }

    const Coloring& getColoring() const { return c; }

    virtual double scale() const = 0;

    virtual double logMeasure() const = 0;

    virtual double potential() const = 0;

    double logLikelihood(double temp = 1.0) const {
      return logMeasure() - potential() / temp;
    }
    
    double logLikelihood(Coloring& c, double temp = 1.0) const {
      assert(c == this->c);
      return logLikelihood(temp);
    }

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

  inline CGAL::Qt_widget& 
  operator<<(CGAL::Qt_widget& widget, const ArakProcess& p) {
    p.visualize(widget);
    return widget;
  }

  class ArakPrior : public ArakProcess, public Coloring::Listener {

  protected:

    double p;

    double sumLogLengths;
  
    double sumLogSines;

    double sumLengths;
  
    void initialize();

  public:

    ArakPrior(Coloring& c, double p = 1.0) : ArakProcess(c), p(p) { 
      initialize();
    }

    ArakPrior(Coloring& c, const Arak::Util::PropertyMap& props) : 
      ArakProcess(c), p(1.0) { 
      using namespace Arak::Util;
      assert(parse(getp(props, "arak.scale"), p));
      initialize();
    }

    virtual ~ArakPrior() { }

    virtual double scale() const { return p; }

    virtual double logMeasure() const;

    virtual double potential() const {
      return 2.0 * p * sumLengths;
    }

    virtual void recolored(const Geometry::Point& a,
         const Geometry::Point& b,
         const Geometry::Point& c) { }
      
    virtual void recolored(const Geometry::Point& a,
         const Geometry::Point& b,
         const Geometry::Point& c,
         const Geometry::Point& d) { }
    
    virtual void vertexHasBeenAdded(Coloring::VertexHandle vh);
    
    virtual void vertexWillBeRemoved(Coloring::VertexHandle vh);

    virtual void edgeHasBeenAdded(Coloring::IntEdgeHandle eh);

    virtual void edgeWillBeRemoved(Coloring::IntEdgeHandle eh);
    
    virtual void visualize(CGAL::Qt_widget& widget) const { }
  };

}

#endif

---