MCDecayFinderModule.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 <analysis/modules/MCDecayFinder/DecayTree.h>
#include <analysis/DecayDescriptor/DecayDescriptor.h>
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/dataobjects/ParticleExtraInfoMap.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
 
namespace Belle2 {
  class MCDecayFinderModule : public Module {
  private:
 
    DecayDescriptor m_decaydescriptor; 
    std::string m_strDecay; 
    std::string m_listName; 
    std::string m_antiListName; 
    bool m_isSelfConjugatedParticle; 
    bool m_writeOut;  
    std::string m_particleStore; 
    bool m_skipNonPrimaryDaughters; 
    bool m_appendAllDaughters; 
    StoreObjPtr<ParticleList> m_outputList; 
    StoreObjPtr<ParticleList> m_antiOutputList; 
    StoreArray<Particle> m_particles; 
    StoreObjPtr<ParticleExtraInfoMap> m_extraInfoMap; 
    StoreArray<MCParticle> m_mcparticles; 
  public:
    MCDecayFinderModule();
    ~MCDecayFinderModule() {}
    virtual void initialize() override;
    virtual void event() override;
    DecayTree<MCParticle>* match(const MCParticle* mcp, const DecayDescriptor* d, bool isCC, int& arrayIndex);
    int write(DecayTree<MCParticle>* decay);
    void appendParticles(const MCParticle* gen, std::vector<const MCParticle*>& children);
    int getNDaughtersRecursive(const DecayDescriptor* d);
 
    void countMaxDepthOfNest(const DecayDescriptor* d, int& depth);
    bool performMCMatching(const DecayTree<MCParticle>* decay, const DecayDescriptor* dd);
    Particle* buildParticleFromDecayTree(const DecayTree<MCParticle>* decay, const DecayDescriptor* dd);
    Particle* createParticleRecursively(const MCParticle* mcp, bool skipNonPrimaryDaughters);
    void addUniqueParticleToList(Particle* newParticle);
  };
}