Process.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "Data.hpp"
#include "Params.hpp"
#include <Eigen/Dense>
#include <random>

class Process {
protected:
  Data &data;

  Params &params;

  Eigen::VectorXi old_allocations;

  int idx_i;

  int idx_j;

  const double log_a = log(params.a);
 
public:
  Process(Data &d, Params &p) : data(d), params(p) {
    old_allocations = data.get_allocations();
  };
 
  // ========== Gibbs Sampling Methods ==========

  virtual double gibbs_prior_existing_cluster(int cls_idx, int obs_idx) const = 0;

 
  virtual Eigen::VectorXd gibbs_prior_existing_clusters(int obs_idx) const = 0;

  virtual double gibbs_prior_new_cluster() const = 0;
 
  // ========== Split-Merge Algorithm Methods ==========

  virtual double prior_ratio_split(int ci, int cj) const = 0;

  virtual double prior_ratio_merge(int size_old_ci, int size_old_cj) const = 0;

  virtual double prior_ratio_shuffle(int size_old_ci, int size_old_cj, int ci,
                                     int cj) const = 0;
 
  // ========== State Management Methods ==========

  void set_old_allocations(const Eigen::VectorXi &new_allocations) {
    old_allocations = new_allocations;
  };

  void set_idx_i(int i) { idx_i = i; };

  void set_idx_j(int j) { idx_j = j; };
 
  // ========== Parameter Update Methods ==========

  virtual void update_params() = 0;

  virtual ~Process() {};
};