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"
#include "AmpGen/enum.h"
#include "AmpGen/Property.h"
#include "AmpGen/Configurable.h"
 
namespace AmpGen {
  class Projection;
  class EventType;
  std::ostream &operator<<(std::ostream &os, const EventType &type);
  class EventType : public Configurable<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 std::string &var = "mass2") 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(unsigned seed = 1) 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;             
    Property<bool> m_altParticleNames{this, "EventType::AlternativeParticleNames", true,
                                      "Alternative naming in ouput tree (e.g. Xi- pi+ pi+ becomes Xim pip0 pip1 rather than _1_Xi# _2_pi~ _3_pi~)"};
    Property<bool> m_includeEnergy{this, "EventType::IncludeEnergy", true, "Include energy as part of the event (otherwise calculate from assumed mass)"};
    // TO DO: these are actually related to the plotting only
  };
 
  extern "C" unsigned python__EventType__dim(const char *eventType);
} // namespace AmpGen
 
#endif