HitXPModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <tracking/dataobjects/hitXP.h>
#include <TTree.h>
#include <TFile.h>
 
#include <set>
 
namespace Belle2 {


  class HitXPModule : public Module {
  public:
 
    HitXPModule();
 
    ~HitXPModule();

    void initialize() override;

    void beginRun() override;

    void event() override;

    void endRun() override;

    void terminate() override;
 
  private:
    TTree* m_tree = nullptr; 
    TFile* m_outputFile = nullptr; 
    std::vector<hitXP> m_hitXP; 
    std::set<hitXP, hitXP::timeCompare> m_hitXPSet; 
    bool c_addTree; 
 
    int m_trackNumber = 0; 
    int m_eventNumber = 0; 
    int m_numberHitPerTrack = 0; 
    int m_trackIterator = 0; 
    int m_eventIterator = 0; 
 
    TTree* m_treeSel = nullptr; 
    TFile* m_outputFileSel = nullptr;
    std::vector<hitXP> m_hitXPSel; 
    int m_trackNumberSel = 0; 
    int m_eventNumberSel = 0; 
    int m_numberHitPerTrackSel = 0;
 
    TTree* m_treeTiSel = nullptr; 
    TFile* m_outputFileTiSel = nullptr; 
    std::vector<hitXP> m_hitXPTiSel; 
    int m_trackNumberTiSel = 0; 
    int m_eventNumberTiSel = 0; 
    int m_numberHitPerTrackTiSel = 0; 
 
    TFile* m_outputFileExt = nullptr; 
    TTree* m_treeExt = nullptr; 
    std::vector<double> m_EpositionEntryX; 
    std::vector<double> m_EpositionEntryY; 
    std::vector<double> m_EpositionEntryZ; 
    std::vector<double> m_EmomentumEntryX; 
    std::vector<double> m_EmomentumEntryY; 
    std::vector<double> m_EmomentumEntryZ; 
    std::vector<double> m_EpositionLocalEntryX; 
    std::vector<double> m_EpositionLocalEntryY; 
    std::vector<double> m_EpositionLocalEntryZ; 
    std::vector<int> m_EPDGID; 
    std::vector<double> m_Eposition0X; 
    std::vector<double> m_Eposition0Y; 
    std::vector<double> m_Eposition0Z; 
    std::vector<double> m_Emomentum0X; 
    std::vector<double> m_Emomentum0Y; 
    std::vector<double> m_Emomentum0Z; 
    std::vector<double> m_Etime; 
    std::vector<int> m_EsensorSensor; 
    std::vector<int> m_EsensorLayer; 
    std::vector<int> m_EsensorLadder; 
    std::vector<int> m_Ereconstructed; 
    std::vector<int> m_EclusterU; 
    std::vector<int> m_EclusterV; 
    std::vector<double> m_Echarge; 
    int m_EtrackNumber = 0; 
    int m_EeventNumber = 0; 
    int m_EnumberHitPerTrack = 0; 
    std::vector<double> m_EomegaEntry; 
    std::vector<double> m_Eomega0;  
    std::vector<double> m_Ed0Entry; 
    std::vector<double> m_Ed00; 
    std::vector<double> m_Ephi0Entry; 
    std::vector<double> m_Ephi00; 
    std::vector<double> m_Ez0Entry; 
    std::vector<double> m_Ez00; 
    std::vector<double> m_EtanlambdaEntry; 
    std::vector<double> m_Etanlambda0; 
    int m_Eprimary = 0; 
 
    StoreArray<MCParticle> m_MCParticles;   
 
  protected:
 
  };

}