Data.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <Eigen/Dense>
#include <Rcpp.h>
#include <RcppEigen.h>
#include <unordered_map>
#include "Params.hpp"
 
// compilation time verbosity level
#ifndef VERBOSITY_LEVEL
#define VERBOSITY_LEVEL 0
#endif

class Data {
protected:
    const Params &params; 
 
    Eigen::VectorXi allocations; 
    int K;                       

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

    void compact_cluster(int old_cluster);

    void set_allocation_wo_compaction(int index, int cluster);
 
public:
    Data(const Params &p, const Eigen::VectorXi &initial_allocations = Eigen::VectorXi());


    double get_distance(int i, int j) const;

    int get_n() const { return params.n; }

    int get_K() const { return K; }

    const Eigen::VectorXi &get_allocations() const { return allocations; }

    int get_cluster_size(unsigned cluster_index) const {
        return (cluster_index < K) ? cluster_members.at(cluster_index).size() : 0;
    }

    int get_cluster_assignment(int index) const {
        if (index < 0 || index >= params.n) {
            throw std::out_of_range("Index out of bounds in get_cluster_assignment");
        }
        return allocations(index);
    }

    Eigen::VectorXi get_cluster_assignments(int cluster) const;

    Eigen::Map<const Eigen::VectorXi> get_cluster_assignments_ref(int cluster) const;

    std::unordered_map<int, std::vector<int>> get_cluster_map_copy() const { return cluster_members; }



    virtual void set_allocation(int index, int cluster);

    virtual void set_allocations(const Eigen::VectorXi &new_allocations);

    virtual void restore_state(Eigen::VectorXi &old_allocations,
                               std::unordered_map<int, std::vector<int>> &old_cluster_members, int old_K);

    const std::unordered_map<int, std::vector<int>> &get_cluster_map() const { return cluster_members; }

 
    virtual ~Data() = default;
};