spatial_module_cache.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "../../utils/Data.hpp"
#include "../../utils/Module.hpp"
#include "../caches/spatial_cache.hpp"

class SpatialModuleCache : public Module {
protected:
    const Data &data_module;           
    const double spatial_weight = 1.0; 
 
    SpatialCache &cache; 
 
public:
    SpatialModuleCache(const Data &data_, SpatialCache &spatial_cache_, double spatial_coeff,
                       const Eigen::VectorXi *old_alloc_provider = nullptr,
                       const std::unordered_map<int, std::vector<int>> *old_cluster_members_provider_ = nullptr)
        : data_module(data_), cache(spatial_cache_), spatial_weight(spatial_coeff),
          Module(old_alloc_provider, old_cluster_members_provider_) {
        cache.set_allocation_ptr(&data_module.get_allocations());
    }


    Eigen::VectorXd compute_similarity_obs(int obs_idx) const override;

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

    double compute_similarity_obs(int obs_idx, int cls_idx) const override;
};