Datax.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "Data.hpp"
#include "ClusterInfo.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 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;
};