ParticlePropertySvc.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2019 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#ifndef PARTICLEPROPERTYSVC_PARTICLEPROPERTYSVC_H
#define PARTICLEPROPERTYSVC_PARTICLEPROPERTYSVC_H
// ============================================================================
// Include Files
// ============================================================================
// STD&STL
// ============================================================================
#include <map>
#include <set>
// ============================================================================
// GaudiKernel
// ============================================================================
#include "GaudiKernel/IParticlePropertySvc.h"
#include "GaudiKernel/Service.h"

class IFileAccess;

namespace Gaudi {
  // ============================================================================
  class ParticlePropertySvc : public extends<Service, IParticlePropertySvc> {
  public:
    // typedefs for container type
    typedef ParticleProperty*                  mapped_type;
    typedef std::map<int, mapped_type>         MapID;
    typedef std::map<std::string, mapped_type> MapName;
    typedef std::map<int, mapped_type>         MapStdHepID;
    typedef std::map<int, mapped_type>         MapPythiaID;
    using IParticlePropertySvc::const_iterator;
    using IParticlePropertySvc::iterator;
    using IParticlePropertySvc::VectPP;
    // Inherited Service overrides:

    StatusCode initialize() override;
    StatusCode finalize() override;
    StatusCode push_back( const std::string& particle, int geantId, int jetsetId, double charge, double mass,
                          double tlife, const std::string& evtName, int pythiaId, double maxWidth ) override;
    StatusCode push_back( ParticleProperty* pp ) override;
    const_iterator begin() const override { return m_vectpp.begin(); }
    const_iterator end() const override { return m_vectpp.end(); }
    int size() const override { return m_vectpp.size(); };
    ParticleProperty* find( int geantId ) override { return m_idmap[geantId]; }
    ParticleProperty* find( const std::string& name ) override { return m_namemap[name]; }
    ParticleProperty* findByStdHepID( int stdHepId ) override { return m_stdhepidmap[stdHepId]; }
    ParticleProperty* findByPythiaID( int pythiaID ) override { return m_pythiaidmap[pythiaID]; }
    StatusCode erase( int geantId ) override { return erase( find( geantId ) ); }
    StatusCode erase( const std::string& name ) override { return erase( find( name ) ); }
    StatusCode eraseByStdHepID( int stdHepId ) override { return erase( findByStdHepID( stdHepId ) ); }
    ParticlePropertySvc( const std::string& name, ISvcLocator* svc );

  protected:
    const ParticleProperty* anti( const ParticleProperty* pp ) const;
    StatusCode setAntiParticles();
    StatusCode rebuild();
    StatusCode erase( const ParticleProperty* pp );
    StatusCode parse( const std::string& file );
    // Text file parsing routine.
    StatusCode parse();
    // treat additional particles
    StatusCode addParticles();
    // compare 2 entries
    bool diff( const ParticleProperty* o, const ParticleProperty* n, const MSG::Level l = MSG::DEBUG ) const;

  private:
    Gaudi::Property<std::string>              m_filename{this, "ParticlePropertiesFile", "ParticleTable.txt",
                                            "Filename of the particle properties file"};
    Gaudi::Property<std::vector<std::string>> m_other{this, "OtherFiles", {}, "additional file names"};
    Gaudi::Property<std::vector<std::string>> m_particles{
        this, "Particles", {}, "properties to be redefined explicitly"};

    VectPP      m_vectpp;      
    MapID       m_idmap;       
    MapName     m_namemap;     
    MapStdHepID m_stdhepidmap; 
    MapPythiaID m_pythiaidmap; 

    // local storage of ALL properties
    std::set<std::unique_ptr<ParticleProperty>> m_owned;
    std::set<std::string>                       m_replaced;

    SmartIF<IFileAccess> m_fileAccess;
  };
} // namespace Gaudi
// =============================================================================
// The END
// =============================================================================
#endif