DrugIntakeGenerator.h

#pragma once
#include <MedProcessTools/MedProcessTools/FeatureGenerator.h>
#include <SerializableObject/SerializableObject/SerializableObject.h>
 
class DrugIntakeGenerator : public FeatureGenerator {
public:
 
    // Feature Descrption
    string signalName;
    int signalId;
 
    // parameters (should be serialized)
    int win_from = 0; 
    int win_to = 360000;            
    int time_unit_win = MedTime::Undefined;         
    vector<string> sets;                        
    int time_unit_sig = MedTime::Undefined;     
    string in_set_name = "";                    
    bool bound_outcomeTime = false;
 
 
    // Signal to determine allowed time-range (e.g. current stay/admission for inpatients)
    string timeRangeSignalName = "";
    int timeRangeSignalId;
    TimeRangeTypes timeRangeType = TIME_RANGE_CURRENT;
    int time_unit_range_sig = MedTime::Undefined;       
 
    // helpers
    vector<unsigned char> lut;                          
 
    // Constructor/Destructor
    DrugIntakeGenerator() : FeatureGenerator() { init_defaults(); };
    ~DrugIntakeGenerator() {};
 
    int init(map<string, string>& mapper);
    void init_defaults();
 
    // Naming
    void set_names();
 
    // Copy
    virtual void copy(FeatureGenerator *generator) { *this = *(dynamic_cast<DrugIntakeGenerator *>(generator)); }
 
    // Learn a generator
    int _learn(MedPidRepository& rep, const MedSamples& samples, vector<RepProcessor *> processors) { time_unit_sig = rep.sigs.Sid2Info[rep.sigs.sid(signalName)].time_unit; return 0; }
 
    // generate a new feature
    int _generate(PidDynamicRec& rec, MedFeatures& features, int index, int num, vector<float *> &_p_data);
    float get_value(PidDynamicRec &rec, int idx, int time, int sig_outcomeTime);
 
    // Signal Ids
    void set_signal_ids(MedSignals& sigs) { signalId = sigs.sid(signalName); }
 
    // Init required tables
    void init_tables(MedDictionarySections& dict);
 
    void get_required_signal_categories(unordered_map<string, vector<string>> &signal_categories_in_use) const;
 
    // Serialization
    ADD_CLASS_NAME(DrugIntakeGenerator)
    ADD_SERIALIZATION_FUNCS(generator_type, tags, serial_id, win_from, win_to,time_unit_win, signalName, sets, names, req_signals, in_set_name, iGenerateWeights, timeRangeSignalName, timeRangeType , bound_outcomeTime)
};
 
MEDSERIALIZE_SUPPORT(DrugIntakeGenerator);