MCSampler.h

 
#ifndef __MCSAMPLER_H
#define __MCSAMPLER_H
 
#include "KVBase.h"
#include "TClonesArray.h"
#include "TTree.h"
#include "TLegend.h"
#include "MicroStat/StatWeight.h"
 
namespace MicroStat {
 
   class MCSampler : public KVBase {
   private:
      void init();
      StatWeight*              fLastPicked;
      Double_t                 fLegendProbaMin;
      TLegend*                 fTheLegend;
 
      void SetBranch(TTree* theTree, const TString& name, void* variable, const TString& vartype);
 
   protected:
      Long64_t                 fPartitions;
      TBranch*                 fBranch;    
      KVEvent*                 fPartition;     
      TClass*                  fWeight;    
      TClonesArray*            fWeightList;
      Double_t                 fSumWeights;
      Double_t                 ESTAR;
      Double_t                 EDISP;
      Long64_t                 IPART;
 
      Bool_t                   fModifyMasses;
 
      void initialiseWeightList();
 
   public:
      MCSampler();
      MCSampler(const Char_t* name, const Char_t* title = "");
      virtual ~MCSampler();
 
      void SetEventList(TTree* t, const TString& branchname);
      void SetStatWeight(const TString&);
 
      KVEvent* GetPartition(Long64_t i)
      {
 
         fBranch->GetEntry(i);
         return fPartition;
      }
      void SetModifyMasses(Bool_t yes = kTRUE)
      {
         fModifyMasses = yes;
      }
      void UpdateMasses();
 
      void CalculateWeights(Double_t excitation_energy);
      TClonesArray* GetWeights() const
      {
         return fWeightList;
      }
      StatWeight* GetWeight(Int_t i) const
      {
         return (StatWeight*)(*fWeightList)[i];
      }
      Double_t GetSumWeights() const
      {
         return fSumWeights;
      }
 
      Long64_t PickRandomChannel();
      StatWeight* GetRandomChannelWeight() const
      {
         return fLastPicked;
      }
 
      void SetUpTreeBranches(KVEvent*& event, TTree* theTree, const TString& bname);
      void GenerateEvents(TTree*, KVEvent* event, Double_t, Long64_t npartitions, Long64_t nev_part = 10);
      Bool_t SetExcitationEnergy(Double_t Exx);
      Bool_t SetDecayChannel();
      void GenerateEvent(TTree* theTree, KVEvent* event);
 
      void PlotMultiplicities(double emin = 0., double emax = 100., double estep = 1., Option_t* opt = "");
      void PlotProbabilities(double emin = 0., double emax = 100., double estep = 1., Option_t* opt = "");
      void SetLegendProbaMin(Double_t p)
      {
         fLegendProbaMin = p;
      };
      Double_t GetLegendProbaMin() const
      {
         return fLegendProbaMin;
      };
      void ShowLegend()
      {
         if (fTheLegend) fTheLegend->Draw();
      };
 
      ClassDef(MCSampler, 1) //Monte-Carlo sampling of events with statistical weights
   };
 
}
#endif