FairEveRecoTracksExample.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/*
 * FairEveRecoTracksExample.h
 *
 *  Created on: 16 cze 2020
 *      Author: Daniel Wielanek
 *    E-mail: daniel.wielanek@gmail.com
 *    Warsaw University of Technology, Faculty of Physics
 */
#ifndef FAIREVERECOTRACKS_H_
#define FAIREVERECOTRACKS_H_

#include "FairEveTracks.h"
#include "FairTask.h"

#include <Rtypes.h>
#include <RtypesCore.h>

class FairRKPropagator;
class FairTutPropTrack;
class TDatabasePDG;

class FairEveRecoTracksExample : public FairEveTracks
{
    TClonesArray *fContainerReco;
    TClonesArray *fContainerSim;
    TClonesArray *fHits1, *fHits2;
    Bool_t fShowPrimary;
    Bool_t fShowSecondary;
    Bool_t fDrawMC;
    Bool_t fUsePdg;
    Int_t fPdgCut;
    FairRKPropagator *fRK;
    TDatabasePDG *fPDG;

  protected:
    Bool_t CheckCuts(FairTutPropTrack *tr);
    void DrawTrack(Int_t id);

  public:
    FairEveRecoTracksExample();
    void Repaint();
    void SetDrawMC(Bool_t draw);
    void SetPdgCut(Int_t pdg, Bool_t use)
    {
        fPdgCut = pdg;
        fUsePdg = use;
    };
    void SetShowPrimSec(Bool_t prim, Bool_t sec)
    {
        fShowPrimary = prim;
        fShowSecondary = sec;
    }
    virtual InitStatus Init();
    virtual ~FairEveRecoTracksExample();
    ClassDef(FairEveRecoTracksExample, 0)
};

#endif /* FAIREVERECOTRACKS_H_ */