TreePhaseSpace.h

Go to the documentation of this file.

#ifndef AMPGEN_TREEPHASESPACE_H
#define AMPGEN_TREEPHASESPACE_H
 
#include <memory.h>
#include <stddef.h>
#include <algorithm>
#include <memory>
#include <ostream>
#include <string>
#include <vector>
#include <utility>
#include <random>
 
#include "AmpGen/EventType.h"
#include "AmpGen/Particle.h"
#include "AmpGen/DiscreteDistribution.h"
#include "AmpGen/Configurable.h"
#include "AmpGen/Property.h"
#include <TRandom3.h>
 
namespace AmpGen {
  class Particle;
  class Event;
  class DecayChainStackBase;
  class EventListSIMD;
  class TreePhaseSpace : public Configurable<TreePhaseSpace> {
  public:
    TreePhaseSpace() = default;
    explicit TreePhaseSpace(const EventType &type);
    TreePhaseSpace(const Particle &decayChain, const EventType &type, TRandom *rndm = nullptr);
    TreePhaseSpace(const std::vector<Particle> &decayChains, const EventType &type, TRandom *rndm = nullptr);
    ~TreePhaseSpace();
 
    void setRandom(TRandom *rand);
    Event makeEvent();
    size_t size() const;
    EventType eventType() const;
    const DecayChainStackBase *operator[](const unsigned i) const { return m_gen[i]; }
    void recalculate_weights(const EventListSIMD &events);
 
  private:
    void initialise_weights();
    std::vector<DecayChainStackBase *> m_gen;
    TRandom3 *m_rand{nullptr};
    EventType m_type; 
    DiscreteDistribution m_dice;
    std::vector<unsigned> m_generatorRecord;
    double m_wmax{0};
    Property<bool> m_aggressiveOptimisation{this, "TreePhaseSpace::AggressiveOptimisation", false};
  };
} // namespace AmpGen
 
#endif