Mods/interface/GeneratorMod.h

//--------------------------------------------------------------------------------------------------
// $Id: GeneratorMod.h,v 1.3 2008/11/11 21:22:58 ceballos Exp $
//
// GeneratorMod
//
// This module collects interesting generator information and publishes collections
// for subsequent modules.
//
// Authors: G.Ceballos
//--------------------------------------------------------------------------------------------------

#ifndef MITPHYSICS_MODS_GENERATORMOD_H
#define MITPHYSICS_MODS_GENERATORMOD_H

#include "MitAna/TreeMod/interface/BaseMod.h" 
#include "MitAna/DataTree/interface/Collections.h"

class TH1D;
class TH2D;

namespace mithep 
{
  class GeneratorMod : public BaseMod
  {
    public:
      GeneratorMod(const char *name="GeneratorMod", 
                   const char *title="Generator information module");
      ~GeneratorMod() {}

      void           SetFillHist(bool b)                 { fFillHist         = b; }
      void           SetMCPartName(const char *s)        { fMCPartName       = s; }     
      void           SetMCLeptonsName(const char * s)    { fMCLeptonsName    = s; }     
      void           SetMCAllLeptonsName(const char * s) { fMCAllLeptonsName = s; }     
      void           SetMCTausName(const char *s)        { fMCTausName       = s; }     
      void           SetMCNeutrinosName(const char *s)   { fMCNeutrinosName  = s; }     
      void           SetMCQuarksName(const char *s)      { fMCQuarksName     = s; }     
      void           SetMCqqHsName(const char *s)        { fMCqqHsName       = s; }     
      void           SetMCBosonsName(const char *s)      { fMCBosonsName     = s; }     

    protected:
      Bool_t         fFillHist;                 //=true then fill histos (def=0)
      TString        fMCPartName;               //name of MCParticle branch
      TString        fMCLeptonsName ;           //name of lepton coll (from W)
      TString        fMCAllLeptonsName ;        //name of lepton coll (all)
      TString        fMCTausName;               //name of tau coll
      TString        fMCNeutrinosName;          //name of neutrinos coll
      TString        fMCQuarksName;             //name of quarks coll
      TString        fMCqqHsName;               //name of qqH coll
      TString        fMCBosonsName;             //name of bosons coll
      MCParticleCol *fParticles;                //MCParticle branch
      Int_t          fNEventsProcessed;         //number of events
      TH1D          *hDGenLeptons[20];          //!histos for W leptons
      TH1D          *hDGenAllLeptons[20];       //!histos for all leptons
      TH1D          *hDGenTaus[20];             //!histos for taus
      TH1D          *hDGenNeutrinos[20];        //!histos for neutrinos
      TH1D          *hDGenQuarks[20];           //!histos for quarks
      TH1D          *hDGenWBF[20];              //!histos for WBF
      TH1D          *hDGenBosons[20];           //!histos for bosons

      void           Process();
      void           SlaveBegin();
    
      ClassDef(GeneratorMod,1) // Module to gather generator information
  };
}
#endif
Revision:	1.4
Committed:	Wed Nov 19 15:44:34 2008 UTC (16 years, 5 months ago) by loizides
Content type:	text/plain
Branch:	MAIN
Changes since 1.3:	+37 -44 lines
Log Message:	Cleaned plus coding conventions
#	User	Rev	Content
1	loizides	1.1	//--------------------------------------------------------------------------------------------------
2	loizides	1.4	// $Id: GeneratorMod.h,v 1.3 2008/11/11 21:22:58 ceballos Exp $
3	loizides	1.1	//
4			// GeneratorMod
5			//
6	loizides	1.4	// This module collects interesting generator information and publishes collections
7			// for subsequent modules.
8	loizides	1.1	//
9	loizides	1.4	// Authors: G.Ceballos
10	loizides	1.1	//--------------------------------------------------------------------------------------------------
11
12	loizides	1.4	#ifndef MITPHYSICS_MODS_GENERATORMOD_H
13			#define MITPHYSICS_MODS_GENERATORMOD_H
14	loizides	1.1
15			#include "MitAna/TreeMod/interface/BaseMod.h"
16			#include "MitAna/DataTree/interface/Collections.h"
17
18			class TH1D;
19			class TH2D;
20
21			namespace mithep
22			{
23			class GeneratorMod : public BaseMod
24			{
25			public:
26			GeneratorMod(const char *name="GeneratorMod",
27	loizides	1.4	const char *title="Generator information module");
28	loizides	1.1	~GeneratorMod() {}
29	loizides	1.4
30			void SetFillHist(bool b) { fFillHist = b; }
31			void SetMCPartName(const char *s) { fMCPartName = s; }
32			void SetMCLeptonsName(const char * s) { fMCLeptonsName = s; }
33			void SetMCAllLeptonsName(const char * s) { fMCAllLeptonsName = s; }
34			void SetMCTausName(const char *s) { fMCTausName = s; }
35			void SetMCNeutrinosName(const char *s) { fMCNeutrinosName = s; }
36			void SetMCQuarksName(const char *s) { fMCQuarksName = s; }
37			void SetMCqqHsName(const char *s) { fMCqqHsName = s; }
38			void SetMCBosonsName(const char *s) { fMCBosonsName = s; }
39
40	loizides	1.1	protected:
41	loizides	1.4	Bool_t fFillHist; //=true then fill histos (def=0)
42			TString fMCPartName; //name of MCParticle branch
43			TString fMCLeptonsName ; //name of lepton coll (from W)
44			TString fMCAllLeptonsName ; //name of lepton coll (all)
45			TString fMCTausName; //name of tau coll
46			TString fMCNeutrinosName; //name of neutrinos coll
47			TString fMCQuarksName; //name of quarks coll
48			TString fMCqqHsName; //name of qqH coll
49			TString fMCBosonsName; //name of bosons coll
50			MCParticleCol *fParticles; //MCParticle branch
51			Int_t fNEventsProcessed; //number of events
52			TH1D *hDGenLeptons[20]; //!histos for W leptons
53			TH1D *hDGenAllLeptons[20]; //!histos for all leptons
54			TH1D *hDGenTaus[20]; //!histos for taus
55			TH1D *hDGenNeutrinos[20]; //!histos for neutrinos
56			TH1D *hDGenQuarks[20]; //!histos for quarks
57			TH1D *hDGenWBF[20]; //!histos for WBF
58			TH1D *hDGenBosons[20]; //!histos for bosons
59	loizides	1.1
60			void Process();
61			void SlaveBegin();
62
63	loizides	1.4	ClassDef(GeneratorMod,1) // Module to gather generator information
64	loizides	1.1	};
65			}
66			#endif