TRestDetectorSingleChannelAnalysisProcess.h

 
#ifndef RestCore_TRestDetectorSingleChannelAnalysisProcess
#define RestCore_TRestDetectorSingleChannelAnalysisProcess
 
#include <TH1D.h>
#include <TRestEventProcess.h>
 
#include "TRestDetectorGainMap.h"
#include "TRestDetectorGas.h"
#include "TRestDetectorHitsEvent.h"
#include "TRestDetectorReadout.h"
#include "TRestDetectorSignalEvent.h"
 
class TRestDetectorSingleChannelAnalysisProcess : public TRestEventProcess {
   private:
    TRestDetectorSignalEvent* fSignalEvent;  
    TRestDetectorReadout* fReadout;          
    TRestDetectorGainMap* fCalib;            
 
    void InitFromConfigFile() override;
 
    void Initialize() override;
    // parameters
    bool fApplyGainCorrection;
    bool fCreateGainMap;
    TVector2 fThrIntegralCutRange;
    TVector2 fNGoodSignalsCutRange;
    TVector2 fSpecFitRange;
    std::string fCalibSave;
 
    // analysis result
    std::map<int, TH1D*> fChannelThrIntegral;  //-> [channel id, sum]
    std::map<int, double> fChannelFitMean;     // [MM id, fitted position]
    std::map<int, double> fChannelGain;        // [MM id, channel gain]
    std::map<int, double> fChannelGainError;   // [MM id, channel gain error]
 
   public:
    RESTValue GetInputEvent() const override { return fSignalEvent; }
    RESTValue GetOutputEvent() const override { return fSignalEvent; }
 
    void FitChannelGain();
    // See comments on CXX
    void SaveGainMetadata(std::string filename);
    void InitProcess() override;
    TRestEvent* ProcessEvent(TRestEvent* inputEvent) override;
    void EndProcess() override;
    TH1D* GetChannelSpectrum(int id);
    void PrintChannelSpectrums(std::string filename);
 
    void PrintMetadata() override {
        BeginPrintProcess();
 
        RESTMetadata << "the mode is:" << RESTendl;
        RESTMetadata << (fApplyGainCorrection ? ">   " : "    ")
                     << "Apply channel correction std::map for spectrum " << RESTendl;
        RESTMetadata << (fCreateGainMap ? ">   " : "    ")
                     << "Create new correction std::map for each channel" << RESTendl;
        RESTMetadata << "output mapping file: " << fCalibSave << RESTendl;
        RESTMetadata << "Energy cut for Threshold integral: " << RESTValue(fThrIntegralCutRange) << RESTendl;
        RESTMetadata << "Energy cut for NGoodSignals: " << RESTValue(fNGoodSignalsCutRange) << RESTendl;
        RESTMetadata << "Fit range for the spectra: " << RESTValue(fSpecFitRange) << RESTendl;
 
        EndPrintProcess();
    }
 
    const char* GetProcessName() const override { return "readoutAnalysis"; }
 
    // Constructor
    TRestDetectorSingleChannelAnalysisProcess();
    TRestDetectorSingleChannelAnalysisProcess(const char* configFilename);
    // Destructor
    ~TRestDetectorSingleChannelAnalysisProcess();
 
    ClassDefOverride(TRestDetectorSingleChannelAnalysisProcess, 1);
};
#endif