ParticleMCDecayStringModule.h

/* BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Thomas Keck, Matt Barrett                                *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.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:
 
    std::string m_listName; 
    std::string m_fileName; 
    std::string m_treeName; 
    TFile* m_file; 
    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