MedLabels.h

Go to the documentation of this file.

#ifndef __MED_LABELS_H__
#define __MED_LABELS_H__
 
#include <vector>
#include <unordered_map>
#include "MedRegistryRecord.h"
#include "MedEnums.h"
#include "LabelParams.h"
#include <MedProcessTools/MedProcessTools/MedSamples.h>
#include "MedSamplingStrategy.h"
 
using namespace std;
 
class SamplingRes {
public:
    int done_cnt = 0;
    int conflict_cnt = 0;
    int no_rule_cnt = 0;
    int miss_pid_in_reg_cnt = 0;
};
 
class MedSamplingStrategy;
static unordered_set<int> default_empty_set;
class MedLabels {
private:
    vector<MedRegistryRecord> all_reg_records; 
    vector<MedRegistryRecord> all_censor_records; 
    unordered_map<int, vector<const MedRegistryRecord *>> pid_reg_records; 
    unordered_map<int, vector<const MedRegistryRecord *>> pid_censor_records; 
public:
    LabelParams labeling_params; 
 
    void prepare_from_registry(const vector<MedRegistryRecord> &reg_records, const vector<MedRegistryRecord> *censor_records = NULL);
 
    bool has_censor_reg() const;
 
    bool has_censor_reg(int pid) const;
 
    void get_pids(vector<int> &pids) const;
 
    void get_records(int pid, vector<const MedRegistryRecord *> &reg_records, vector<const MedRegistryRecord *> &censor_records) const;
 
    void calc_signal_stats(const string &repository_path, const string &signal_name,
        const string &signalHirerchyType, int ageBinValue, MedSamplingStrategy &sampler,
        const LabelParams &inc_labeling_params, map<float, map<float, vector<int>>> &maleSignalToStats,
        map<float, map<float, vector<int>>> &femaleSignalToStats,
        const string &debug_file = "", const unordered_set<int> &debug_vals = default_empty_set) const;
 
    void create_incidence_file(const string &file_path, const string &rep_path, int age_bin, int min_age,
        int max_age, bool use_kaplan_meir = false, const string &sampler_name = "yearly",
        const string &sampler_args = "day_jump=365;start_year=2007;end_year=2012;prediction_month_day=101",
        const string &debug_file = "") const;
 
    SamplingRes get_samples(int pid, int time, vector<MedSample> &samples, bool show_conflicts = false) const;
 
    SamplingRes get_samples(int pid, const vector<MedSample> &samples, vector<MedSample> &new_samples, bool show_conflicts = false) const;
 
    SamplingRes get_samples(int pid, const vector<int> &times, vector<MedSample> &samples, bool show_conflicts = false) const;
 
    void create_samples(const MedSamplingStrategy *sampler, MedSamples &samples, bool show_conflicts = true) const;
 
    void relabel_samples(MedSamples &samples, bool show_conflicts = true) const;
 
    MedLabels(const LabelParams &params);
 
    MedLabels(const string &labling_params);
};
 
#endif