TRestDetectorPositionMappingProcess.h

 
#ifndef RestCore_TRestDetectorPositionMappingProcess
#define RestCore_TRestDetectorPositionMappingProcess
 
#include <TH1D.h>
#include <TRestEventProcess.h>
 
#include "TRestDetectorGainMap.h"
#include "TRestDetectorGas.h"
#include "TRestDetectorHitsEvent.h"
#include "TRestDetectorReadout.h"
#include "TRestDetectorSignalEvent.h"
 
class TRestDetectorPositionMappingProcess : public TRestEventProcess {
   private:
    TRestDetectorHitsEvent* fHitsEvent;  
    TRestDetectorReadout* fReadout;      
    TRestDetectorGainMap* fCalib;        
    TRestDetectorGas* fGas;              
 
    void InitFromConfigFile() override;
 
    void Initialize() override;
    // parameters
    bool fApplyGainCorrection;
    bool fCreateGainMap;
    TVector2 fEnergyCutRange;
    TVector2 fNHitsCutRange;
    std::string fMappingSave;
 
    double fNBinsX;
    double fNBinsY;
    double fNBinsZ;
 
    // temp data
    TH2D* fAreaThrIntegralSum;  
    TH2D* fAreaCounts;          
 
    // analysis result
    TH2F* fAreaGainMap;  
 
   public:
    RESTValue GetInputEvent() const override { return fHitsEvent; }
    RESTValue GetOutputEvent() const override { return fHitsEvent; }
 
    void InitProcess() override;
    TRestEvent* ProcessEvent(TRestEvent* inputEvent) override;
    void EndProcess() override;
 
    double GetCorrection2(double x, double y);
    double GetCorrection3(double x, double y, double z);
 
    void PrintMetadata() override {
        BeginPrintProcess();
 
        RESTMetadata << "the mode is:" << RESTendl;
        RESTMetadata << (fApplyGainCorrection ? ">   " : "    ")
                     << "Apply position correction std::map for spectrum " << RESTendl;
        RESTMetadata << (fCreateGainMap ? ">   " : "    ")
                     << "Create new correction std::map for each position" << RESTendl;
        RESTMetadata << "output mapping file: " << fMappingSave << RESTendl;
        RESTMetadata << "Energy cut for Threshold integral: " << RESTValue(fEnergyCutRange) << RESTendl;
        RESTMetadata << "Energy cut for NGoodSignals: " << RESTValue(fNHitsCutRange) << RESTendl;
        RESTMetadata << "Binning: " << fNBinsX << ", " << fNBinsY << ", " << fNBinsZ << RESTendl;
 
        EndPrintProcess();
    }
 
    const char* GetProcessName() const override { return "readoutAnalysis"; }
 
    TRestDetectorPositionMappingProcess();
    TRestDetectorPositionMappingProcess(const char* configFilename);
 
    ~TRestDetectorPositionMappingProcess() override;
 
    ClassDefOverride(TRestDetectorPositionMappingProcess, 1);
};
#endif