PndDiscTaskDigitization.h

Go to the documentation of this file.

//-------------------------------------------------------------------------
// Author:      Oliver Merle (Oliver.Merle@exp2.physik.uni-giessen.de)
// Changes:     Mustafa Schmidt (Mustafa.A.Schmidt@physik.uni-giessen.de)
// Date:        30.11.2015
// Description: Digitization of Monte Carlo hits
//-------------------------------------------------------------------------

#ifndef PNDDISCTASKDIGITIZATION_H
#define PNDDISCTASKDIGITIZATION_H

#define USESENSORGRID

// project specific
#include "PndDiscNoiseGeneration.h"
#include "PndDiscPhotodetector.h"
#include "PndDiscSensorGrid.h"

// FAIR/PROOT
#include <PndPersistencyTask.h>
#include "PndDetectorList.h"
#include "FairMCEventHeader.h"

// ROOT
#include "TString.h"
#include <Math/Interpolator.h> // ROOT imports from GSL -> GPL License !!!

// cpp
#include <set>



//class TClonesArray;



class PndDiscTaskDigitization : public PndPersistencyTask
{
public:
    PndDiscTaskDigitization();
    PndDiscTaskDigitization(const char* name);
    virtual ~PndDiscTaskDigitization();

    // Implement FairTask interface:
    virtual InitStatus Init();
    virtual InitStatus ReInit();
    virtual void Exec(Option_t* opt);
    virtual void FinishEvent();
    virtual void FinishTask();

    void RunTimeBased(Bool_t time_based) { is_time_based = time_based; }

    // Additional functionality
    void FilterParticleSpecies(Int_t pdg);
    void SetRunMixed(Bool_t is_mixed=kTRUE)     
         { is_run_mixed = is_mixed; }

protected:
    TString branch_name_mc_point;               
    TString branch_name_digits;                 
    TString folder_name_digits;                 

    Int_t mc_point_branch_id;                   
    TClonesArray * tclarr_mc_points;            
    TClonesArray * tclarr_particle_tracks_in;   
    TClonesArray * tclarr_particle_tracks_out;  
    TClonesArray * array;
    std::set<int>  particle_types;              

    FairWriteoutBuffer * writeout_buffer;       

    FairMCEventHeader * fMcEventHeader;         
    
    Bool_t is_run_mixed;
    Bool_t is_time_based;                       
    Bool_t is_persistent;                       
  

#ifdef USESENSORGRID
    DiscDIRC_Photodetector*  photo_detector;    
#else
    //ROOT::Math::Interpolator pde_interpolator;  //!
    DiscDIRC_NoiseGeneration noise_generator;   
#endif
    
    ClassDef(PndDiscTaskDigitization, 1)
};

#endif // PNDDISCTASKDIGITIZATION_H