Data.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../params/utils-params.hpp"
#include <Eigen/Dense>
#include <unordered_map>
#include <vector>
 
// compilation time verbosity level
#ifndef VERBOSITY_LEVEL
#define VERBOSITY_LEVEL 0
#endif

class Data {
protected:
  const utils_params &p; 
 
  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 utils_params &p,
       const Eigen::VectorXi &initial_allocations = Eigen::VectorXi());


  double get_data(int i, int j = 0) const;

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

  int get_K() const { return K; }

  int get_p() const { return p.p; }

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

  int get_cluster_size(unsigned cluster_index) const {
    if (cluster_index >= static_cast<unsigned>(K)) {
      return 0;
    }
 
    const auto it = cluster_members.find(static_cast<int>(cluster_index));
    return (it != cluster_members.end()) ? static_cast<int>(it->second.size())
                                         : 0;
  }

  int get_cluster_assignment(int index) const {
    if (index < 0 || index >= p.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;
};