Module.hpp

Go to the documentation of this file.

#pragma once

 
#include <Eigen/Dense>
#include <unordered_map>
#include <vector>

 
class Module {
protected:
    const Eigen::VectorXi *old_allocations_provider;

    const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider;
 
public:
    Module(const Eigen::VectorXi *old_allocations_provider_ = nullptr,
           const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider_ = nullptr)
        : old_allocations_provider(old_allocations_provider_),
          old_cluster_members_provider(old_cluster_members_provider_) {}
 
    void set_old_allocations_provider(const Eigen::VectorXi *provider) { old_allocations_provider = provider; }
 
    void set_old_cluster_members_provider(const std::unordered_map<int, std::vector<int>> *provider) {
        old_cluster_members_provider = provider;
    }


    virtual double compute_similarity_cls(int cls_idx, bool old_allo = false) const = 0;

    virtual double compute_similarity_obs(int obs_idx, int cls_idx) const = 0;

    virtual Eigen::VectorXd compute_similarity_obs(int obs_idx) const = 0;
 
    virtual ~Module() = default;
};