defm-bones.hpp

Go to the documentation of this file.

#ifndef DEFM_BONES_HPP
#define DEFM_BONES_HPP 1
 
class DEFM : public DEFMModel {
private:
 
    int * Y           = nullptr; 
    int * ID          = nullptr; 
    double * X        = nullptr; 
    bool column_major = true;    
 
    // In case we need a copy of the data
    std::shared_ptr<std::vector< int >> Y_shared;    
    std::shared_ptr<std::vector< int >> ID_shared;   
    std::shared_ptr<std::vector< double >> X_shared; 
    
    size_t N;         
    size_t ID_length; 
    size_t Y_ncol;    
    size_t Y_length;  
    size_t X_ncol;    
    size_t X_length;  
    size_t M_order;   
 
    std::vector< std::string > Y_names; 
    std::vector< std::string > X_names; 
    std::vector< size_t > start_end;    
    std::vector< size_t > model_ord;    
 
public:
 
    DEFM(
        int * id,
        int * y,
        double * x,
        size_t id_length,
        size_t y_ncol,
        size_t x_ncol,
        size_t m_order,
        bool copy_data = true,
        bool column_major = true
    );
 
    DEFMModel & get_model() {
        return *this;
    };
 
    void init();
 
    void simulate(std::vector< double > par, int * y_out);
 
    size_t get_n_y() const;
    size_t get_n_obs() const;
    size_t get_n_covars() const;
    size_t get_m_order() const;
    size_t get_n_rows() const;
 
    const int * get_Y() const;
    const int * get_ID() const;
    const double * get_X() const;
 
    barry::FreqTable<int> motif_census(
        std::vector< size_t > idx
    );
 
    std::vector< double > logodds(
        const std::vector< double > & par,
        size_t i,
        size_t j
    );
 
    void set_names(
        std::vector< std::string > Y_names_,
        std::vector< std::string > X_names_
    );
 
    const std::vector< std::string > & get_Y_names() const;
    const std::vector< std::string > & get_X_names() const;
 
    void print() const;
 
    std::vector< bool > is_motif();
 
    bool get_column_major() const noexcept;
 
};
 
#endif