EventType.h

Go to the documentation of this file.

#ifndef AMPGEN_EVENTTYPE_H
#define AMPGEN_EVENTTYPE_H
 
#include <functional>
#include <iostream>
#include <map>
#include <vector>
#include <initializer_list>
#include "AmpGen/Event.h"
 
namespace AmpGen
{
  class Projection;
  class EventType; 
  std::ostream& operator<<( std::ostream& os, const EventType& type );
 
  class EventType
  {
    public:
      EventType() = default;

      EventType( const std::vector<std::string>&, const bool& isTD = false );

      EventType conj( const bool& headOnly = 0, const bool& dontConjHead = 0 ) const;

      std::map<std::string, unsigned> getEventFormat( const bool& outputNames = false ) const;

      std::pair<unsigned,unsigned>  count(const unsigned& index) const;
      std::pair<double, double> minmax( const std::vector<unsigned>& indices) const;
      std::vector<double> masses() const;
      std::string mother() const;
      double mass( const unsigned& index ) const;
      double motherMass() const;
      std::vector<std::string> finalStates() const;
      bool isTimeDependent() const;
      unsigned eventSize() const;
      unsigned size()      const;
      unsigned dof()       const;
      std::string operator[]( const unsigned& index ) const;
      std::string decayDescriptor() const;
      std::string label( const unsigned& index, bool isRoot = true ) const;
      std::string label( const std::vector<unsigned>& index, bool isRoot = true ) const;
      std::vector<Projection> defaultProjections(const unsigned& nBins=100) const;
      Projection projection(const unsigned& nBins, const std::vector<unsigned>& indices, const std::string& observable = "mass2") const;
 
      bool operator==( const EventType& other ) const;
      bool has( const std::string& name ) const;
 
      void extendEventType( const std::string& branch );

      std::function<void( Event& )> symmetriser() const;
      std::function<bool( Event&, const std::vector<int>& ids)> automaticOrdering() const;

      std::pair<unsigned, unsigned> dim() const;
  
      friend std::ostream& AmpGen::operator<<( std::ostream& os, const EventType& type );
      
      Event makeEvent() const; 
 
      std::string constructor_string() const;  
    private:
      std::string               m_mother;               
      double                    m_motherMass;           
      std::vector<std::string>  m_particleNames;        
      std::vector<std::string>  m_particleNamesPickled; 
      std::vector<double>       m_particleMasses;       
      std::vector<bool>         m_ignore;               
      bool                      m_timeDependent;        
      std::vector<std::string>  m_eventTypeExtensions;  
      std::pair<unsigned, unsigned> m_dim;              
      bool                      m_alt_part_names;       
  };
 
 
  extern "C" unsigned python__EventType__dim( const char* eventType ); 
} // namespace AmpGen
 
#endif