Datax.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "ClusterInfo.hpp"
#include "Data.hpp"
#include <memory>

class Datax : public Data {
 
protected:
  std::vector<std::shared_ptr<ClusterInfo>> cluster_info;
 
  void compact_cluster(int old_cluster);
 
public:
  Datax(const utils_params &p, std::vector<std::shared_ptr<ClusterInfo>> ci,
        const Eigen::VectorXi &initial_allocations = Eigen::VectorXi())
      : Data(p, initial_allocations), cluster_info(std::move(ci)) {}

  void set_allocation(int index, int cluster) override;

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

  void
  restore_state(Eigen::VectorXi &old_allocations,
                std::unordered_map<int, std::vector<int>> &old_cluster_members,
                int old_K) override;
 
  virtual ~Datax() = default;
};