PhaseSpaceAnalysisModule.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
 
// framework
#include <framework/core/Module.h>
#include <mdst/dataobjects/MCParticle.h>
#include <framework/datastore/StoreArray.h>
#include <genfit/TrackCand.h>
#include <tracking/spacePointCreation/SpacePointTrackCand.h>
 
// ROOT
#include <TFile.h>
#include <TTree.h>
 
// other stuff
#include <string>
 
// boost
#include <boost/any.hpp>
 
namespace Belle2 {
  class PhaseSpaceAnalysisModule : public Module {
 
  public:
    PhaseSpaceAnalysisModule(); 
    void initialize() override; 
    void event() override; 
    void terminate() override; 
    struct RootVariables {
      std::vector<double> MomX; 
      std::vector<double> MomY; 
      std::vector<double> MomZ; 
      std::vector<double> Eta;
      std::vector<double> pT;
 
      std::vector<double> VertX; 
      std::vector<double> VertY; 
      std::vector<double> VertZ; 
      std::vector<int> PDG; 
      std::vector<double> Charge; 
      std::vector<double> Energy; 
      std::vector<double> Mass; 
//       std::vector<unsigned short int> RefereeStatus; /**< refereeStatus of a SPTC (if available) */
    };
 
    enum e_trackCandType {
      c_gftc, 
      c_sptc, 
    };
 
  protected:
 
    // ======================================== PARAMETERS ========================================================================
    std::vector<std::string> m_PARAMcontainerNames; 
    std::vector<std::string> m_PARAMtrackCandTypes; 
    std::vector<std::string> m_PARAMrootFileName; 
    bool m_PARAMdiffAnalysis;
 
    // ======================================== INTERNALLY USED MEMBERS ===========================================================
    std::vector<std::pair<boost::any, e_trackCandType> > m_tcStoreArrays; 
    RootVariables m_rootVariables; 
    std::vector<TTree*> m_treePtrs; 
    TFile* m_rootFilePtr; 
    std::vector<std::string> m_treeNames;  
    // ==================================================== COUNTERS ==============================================================
    unsigned int m_noMcPartCtr; 
    unsigned int m_skippedTCsCtr; 
    std::vector<unsigned int> m_mcPartCtr; 
    void initializeCounters(size_t nContainers)
    {
      m_noMcPartCtr = 0;
      m_skippedTCsCtr = 0;
 
      m_mcPartCtr = std::vector<unsigned int>(nContainers, 0);
    }
 
    void initializeRootFile(std::string fileName, std::string writeOption, std::vector<std::string> treeNames);
    void getValuesForRoot(Belle2::MCParticle* mcParticle, RootVariables& rootVariables); 
    template<typename TrackCandType>
    std::vector<int> getMCParticleIDs(Belle2::StoreArray<TrackCandType> trackCands);
 
//     /** get all referee statuses from all SpacePointTrackCand */
//     std::vector<unsigned short int> getRefereeStatuses(Belle2::StoreArray<Belle2::SpacePointTrackCand> trackCands);
 
    std::vector<std::vector<int> > getDiffIds(const std::vector<std::vector<int> >& allIDs);
 
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<genfit::TrackCand> m_GenfitTrackCands; 
    StoreArray<SpacePointTrackCand> m_SpacePointTrackCands; 
  };
}