ParticleLoaderModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2019 - Belle II Collaboration                        *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric, Anze Zupanc                                *
 *               Sam Cunliffe, Torben Ferber                              *
 *               Frank Meier                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
 
#include <analysis/dataobjects/RestOfEvent.h>
#include <analysis/DecayDescriptor/DecayDescriptor.h>
#include <analysis/VariableManager/Utility.h>
 
#include <vector>
#include <tuple>
#include <string>
#include <memory>
 
namespace Belle2 {
  class ParticleLoaderModule : public Module {
 
    typedef std::tuple<int, std::string, std::string, bool, std::shared_ptr<Variable::Cut>> PList;
    enum PListIndex {
      c_PListPDGCode, c_PListName, c_AntiPListName, c_IsPListSelfConjugated, c_CutPointer
    };
 
  public:
 
    ParticleLoaderModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  private:
 
    void mcParticlesToParticles();
 
    void mdstToParticle();
 
    void tracksToParticles();
 
    void eclClustersToParticles();
 
    void klmClustersToParticles();
 
    void v0sToParticles();
 
    void roeToParticles();
 
    void addROEToParticleList(RestOfEvent* roe, int pdgCode = 0, bool isSelfConjugatedParticle = true);
 
    bool isValidPDGCode(const int pdgCode);
 
    void appendDaughtersRecursive(Particle* mother);
 
    bool m_useMCParticles;  
    bool m_useROEs;  
    DecayDescriptor m_decaydescriptor; 
    std::vector<std::tuple<std::string, std::string>>
                                                   m_decayStringsWithCuts; 
    std::vector<PList> m_MCParticles2Plists; 
    std::vector<PList> m_Tracks2Plists; 
    std::vector<PList> m_V02Plists; 
    std::vector<PList> m_ROE2Plists; 
    std::vector<PList> m_ECLClusters2Plists; 
    std::vector<PList> m_KLMClusters2Plists; 
    bool m_writeOut;  
    bool m_addDaughters; 
    bool m_skipNonPrimaryDaughters; 
    std::string m_roeMaskName; 
    std::string m_sourceParticleListName; 
    bool m_useMissing; 
    int m_trackHypothesis; 
    bool m_enforceFitHypothesis =
      false; 
    std::vector<int> m_chargeZeroTrackCounts; 
    std::vector<int> m_sameChargeDaughtersV0Counts; 
  };
 
} // Belle2 namespace