DecayGenFrame.hh

Go to the documentation of this file.

//   RestFrames: particle physics event analysis library
//   --------------------------------------------------------------------
//   Copyright (c) 2014-2016, Christopher Rogan
//   This file is part of RestFrames.
//
//   RestFrames is free software; you can redistribute it and/or modify
//   it under the terms of the GNU General Public License as published by
//   the Free Software Foundation; either version 2 of the License, or
//   (at your option) any later version.
// 
//   RestFrames is distributed in the hope that it will be useful,
//   but WITHOUT ANY WARRANTY; without even the implied warranty of
//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//   GNU General Public License for more details.
// 
//   You should have received a copy of the GNU General Public License
//   along with RestFrames. If not, see <http://www.gnu.org/licenses/>.
 
#ifndef DecayGenFrame_HH
#define DecayGenFrame_HH
 
#include "RestFrames/DecayFrame.hh"
#include "RestFrames/GeneratorFrame.hh"
 
namespace RestFrames {
 
  class ResonanceGenFrame;
 
  // DecayGenFrame class
  class DecayGenFrame : public DecayFrame<GeneratorFrame> {
  public:
 
    DecayGenFrame(const std::string& sname, const std::string& stitle);
 
    DecayGenFrame();
    virtual ~DecayGenFrame();
 
    virtual void SetMass(double val);
 
    virtual void SetVariableMass(bool varymass = true);
 
    // For two-body decays
 
    virtual void SetCosDecayAngle(double val);
 
    virtual void SetDeltaPhiDecayPlane(double val);
 
 
    virtual double GetProbMCMC(double mass = -1.) const;
 
    virtual void GenerateMassMCMC(double& mass, double& prob, 
                  double max = -1.) const;
 
 
  protected:
    double m_CosDecayAngle;
    double m_DeltaPhiDecayPlane;
 
    std::vector<int>    m_ChildIndexMCMC;
    std::vector<double> m_ChildMassMCMC;
    std::vector<double> m_ChildProbMCMC;
    std::vector<double> m_InterMassFracMCMC;
 
    virtual bool IsSoundBody() const;
 
    virtual void ResetGenFrame();
    virtual bool GenerateFrame();
   
    void GenerateTwoBodyRecursive(const std::vector<double>& M_parent, 
                  const std::vector<double>& M_child,
                  const TVector3& axis_par, 
                  const TVector3& axis_perp,
                  std::vector<TLorentzVector>& P_child);
 
    virtual bool InitializeGenAnalysis();
    virtual bool IterateMCMC();
 
  };
 
  int DoubleMax(const void *a, const void *b);
 
}
 
#endif