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