Process.hpp

Go to the documentation of this file.

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

class Process {
protected:
  Data &data;

  Eigen::VectorXi old_allocations;

  std::unordered_map<int, std::vector<int>> old_cluster_members;

  int old_K = 0;

  [[nodiscard]] const std::unordered_map<int, std::vector<int>> &
  old_cluster_members_view() const {
    return old_cluster_members;
  }

  int idx_i;

  int idx_j;
 
public:
  Process(Data &d) : data(d) {
    old_allocations = data.get_allocations();
    old_cluster_members = data.get_cluster_map();
    old_K = data.get_K();
  };
 
  // ========== 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;

  virtual double gibbs_prior_new_cluster_obs(int obs_idx) const {
    (void)obs_idx;
    return gibbs_prior_new_cluster();
  }
 
  // ========== 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_old_cluster_members(
      const std::unordered_map<int, std::vector<int>> &new_cluster_members) {
    old_cluster_members = new_cluster_members;
  };

  void set_old_K(int new_K) { old_K = new_K; }

  void restore_state() {
    data.restore_state(old_allocations, old_cluster_members, old_K);
  }

  [[nodiscard]] const Eigen::VectorXi &old_allocations_view() const {
    return old_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() {};
};