binary_cache.hpp

Go to the documentation of this file.

#pragma once

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

 
class BinaryCache : public ClusterInfo {
public:
    // Important to declare the struct useful.
    // HINT: If the name changed modified the file which uses it accordingly.
    struct ClusterStats {
        int binary_sum = 0;
        int n = 0;
    };
 
private:
    std::vector<ClusterStats> cluster_stats;
 
public:
    const Eigen::VectorXi binary_covariates;
 
    BinaryCache(const Eigen::VectorXi &allocations_ref, const Eigen::VectorXi &binary_covariates)
        : binary_covariates(binary_covariates) {
 
        const int K = allocations_ref.maxCoeff() + 1;
        recompute(K > 0 ? K : 0, allocations_ref);
    }

    void set_allocation(int index, int cluster, int old_cluster) override;

    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);
        }
    }
};