Process.hpp

Go to the documentation of this file.

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

class Process {
protected:
    Data &data;

    const Params &params;

    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;

    const double log_a = log(params.a);
 
public:
    Process(Data &d, const Params &p) : data(d), params(p) {
        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() {};
};