ParticleMCDecayStringModule.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 <analysis/dataobjects/ParticleList.h>
#include <analysis/dataobjects/StringWrapper.h>
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/pcore/SetMergeable.h>
#include <framework/pcore/RootMergeable.h>
 
#include <TTree.h>
#include <TFile.h>
 
#include <string>
#include <unordered_set>
#include <cstdint>
 
namespace Belle2 {
 
  class Particle;
  class MCParticle;

 
  class ParticleMCDecayStringModule : public Module {
  public:

    ParticleMCDecayStringModule();

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;
 
  private:

    std::string getDecayString(const Particle& p);

    std::string getDecayStringFromParticle(const Particle* p);

    std::string getMCDecayStringFromParticle(const Particle* p);

    std::string getMCDecayStringFromMCParticle(const MCParticle* mcPMatched);

    const MCParticle* getInitialParticle(const MCParticle* mcP);

    std::string buildMCDecayString(const MCParticle* mcPMother, const MCParticle* mcPMatched);

    void convertToConciseString(std::string& string);
 
 
  private:
 
    StoreObjPtr<ParticleList> m_pList; 
    StoreArray<StringWrapper> m_stringWrapperArray; 
 
    std::string m_listName; 
 
    std::string m_fileName; 
    std::string m_treeName; 
 
    std::shared_ptr<TFile> m_file{nullptr}; 
 
    StoreObjPtr<RootMergeable<TTree>> m_tree; 
 
    StoreObjPtr<SetMergeable<std::unordered_set<uint64_t>>> m_hashset; 
 
    const std::string c_ExtraInfoName = "DecayHash"; 
    const std::string c_ExtraInfoNameExtended = "DecayHashExtended"; 
 
    float m_decayHash; 
    float m_decayHashExtended; 
    std::string m_decayString; 
 
    bool m_useConciseString; 
    std::string m_identifiers; 
 
    std::hash<std::string> m_hasher; 
  };

} // Belle2 namespace