MatrixConnectome.h

#pragma once
 
#include "eigen_matrix_defs.h"
#include <map>
#include <set>
#include <vector>
 
namespace CoRiBoDynamics
{
class MatrixConnectome
{
public:
    struct CRix {
        int cr; // column index = (cr>>16), row_index = (cr&0xffff)
        int ix; // matrix index
    };
 
    explicit MatrixConnectome(int drop_size);
    ~MatrixConnectome();
 
    void Update(const bool* diagonal_connection, int num, std::vector<CRix> extra_connections);
 
    void set_drop_size(int drop_size);
 
    struct segment {
        int ix0; 
        int ix1; 
        segment(int a, int b){ix0 = a; ix1 = b;}
    };
 
    std::vector<segment> m_tridiag_segments;
    std::vector<segment> m_null_tridiag_segments;
 
 
    struct colored_segment {
        int ix0; int ix1; int color;
        colored_segment(){}
        colored_segment(int a,int b,int c){ix0=a;ix1=b;color=c;}
    };
 
    std::vector<colored_segment> all_segments;
    std::vector<int> color_groups; // color merging lookup table. color_groups[c] contains the actual color group value that 'c' should be merged into
 
protected:
 
bool contains(const segment& o, const segment& i);
int m_drop_size;
segment drop_expand(const bool* diagonal_connection, int num, int expand_center);
 
};
}