spatial_cache.hpp

Go to the documentation of this file.

#pragma once

 
#include "../../utils/ClusterInfo.hpp"
#include <Eigen/Dense>
#include <vector>

 
class SpatialCache : public ClusterInfo {
public:
    // Important to declare the struct useful.
    // HINT: If the name changed modified the file which uses it accordingly.
    struct ClusterStats {
        int spatial_sum = 0;
    };
 
    std::vector<std::vector<int>> neighbor_cache;
 
private:
    std::vector<ClusterStats> cluster_stats;
    Eigen::VectorXi* allocations_ptr;

    void neighbor_cache_compute();
 
public:
    const Eigen::MatrixXi W;
 
    SpatialCache(const Eigen::VectorXi &allocations_ref, const Eigen::MatrixXi &W)
        : W(W) {
 
        const int K = allocations_ref.maxCoeff() + 1;
        neighbor_cache_compute();
        recompute(K > 0 ? K : 0, allocations_ref);
    }

    void set_allocation(int index, int cluster, int old_cluster) override;
 
    void set_allocation_ptr(const Eigen::VectorXi *new_allocations) {
        allocations_ptr = const_cast<Eigen::VectorXi *>(new_allocations);
    }

    inline ClusterStats get_cluster_stats(int cluster) const { return cluster_stats[cluster]; }

    inline const ClusterStats &get_cluster_stats_ref(int cluster) const { return cluster_stats[cluster]; }

    void recompute(const int K, const Eigen::VectorXi &allocations_in) override;

    inline void move_cluster_info(int from_cluster, int to_cluster) override {
        cluster_stats[to_cluster] = std::move(cluster_stats[from_cluster]);
    };

    void remove_info(int cluster) override {
        if (cluster == static_cast<int>(cluster_stats.size()) - 1) {
            cluster_stats.pop_back();
        } else {
            cluster_stats.erase(cluster_stats.begin() + cluster);
        }
    }
};