GeneratorFrame.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 GeneratorFrame_HH
#define GeneratorFrame_HH
 
#include <TRandom.h>
 
#include "RestFrames/RestFrame.hh"
 
namespace RestFrames {
 
  // GeneratorFrame class
  class GeneratorFrame : public RestFrame {
  public:
 
    GeneratorFrame(const std::string& sname, const std::string& stitle);
 
    GeneratorFrame();
    virtual ~GeneratorFrame();
 
    virtual void Clear();
 
    virtual void AddChildFrame(RestFrame& frame);
 
    virtual void SetParentFrame(RestFrame& frame = 
                RestFrame::Empty());
 
    virtual GeneratorFrame const& GetParentFrame() const;
 
    virtual GeneratorFrame& GetChildFrame(int i = 0) const;
 
    void SetPCut(double cut);
 
    void SetPtCut(double cut);
 
    void SetEtaCut(double cut);
 
    void SetMassWindowCut(double min, double max);
 
    void RemovePCut();
 
    void RemovePtCut();
 
    void RemoveEtaCut();
 
    void RemoveMassWindowCut();
 
    void PrintGeneratorEfficiency() const;
    
    virtual double GetMass() const;
 
 
    bool IsVariableMassMCMC() const;
 
    virtual double GetMinimumMassMCMC() const;
 
    virtual void GenerateMassMCMC(double& mass, double& prob, 
                  double max = -1.) const;
    
    virtual double GetProbMCMC(double mass = -1.) const;
 
    double GetRandom() const;
    
    double GetGaus(double mu, double sig) const;
 
 
    static GeneratorFrame& Empty();
 
  protected:
    double m_Mass;
 
    bool InitializeAnalysisRecursive();
    bool AnalyzeEventRecursive();
    bool ClearEventRecursive();
 
    virtual void ResetGenFrame() = 0;
    virtual bool GenerateFrame() = 0;
 
    void SetChildren(const std::vector<TLorentzVector>& P_children);
    virtual bool InitializeGenAnalysis();
 
    virtual bool IterateMCMC();
    bool IterateRecursiveMCMC();
 
    void SetVariableMassMCMC(bool var = true);
    virtual void SetMassMCMC(double mass);
    void SetMassMCMC(double mass, GeneratorFrame& frame) const;
 
    bool EventInAcceptance() const;
 
  private:
    TRandom *m_Random;
    bool m_VarMassMCMC;
 
    mutable long m_Ngen;
    mutable long m_Npass;
 
    double m_PCut;
    double m_PtCut;
    double m_EtaCut;
    double m_minMassCut;
    double m_maxMassCut;
 
    bool m_doCuts;
    bool m_doPCut;
    bool m_doPtCut;
    bool m_doEtaCut;
    bool m_dominMassCut;
    bool m_domaxMassCut;
   
  };
 
}
 
#endif