LauGenNtuple.hh

Go to the documentation of this file.

// Copyright University of Warwick 2006 - 2013.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// Authors:
// Thomas Latham
// John Back
// Paul Harrison

#ifndef LAU_GEN_NTUPLE
#define LAU_GEN_NTUPLE

#include <map>

#include "TString.h"

class TFile;
class TTree;


class LauGenNtuple {

        public:

                LauGenNtuple(const TString& rootFileName, const TString& rootTreeName);
                
                virtual ~LauGenNtuple();



                const TString& fileName() const {return rootFileName_;}
                

                const TString& treeName() const {return rootTreeName_;}


                void addIntegerBranch(const TString& name);


                void addDoubleBranch(const TString& name);
                

                void setIntegerBranchValue(const TString& name, Int_t value);
                

                void setDoubleBranchValue(const TString& name, Double_t value);
                

                Int_t getIntegerBranchValue(const TString& name) const;
                

                Double_t getDoubleBranchValue(const TString& name) const;
                
                void fillBranches();
                
                void deleteAndRecreateTree();
                

                Int_t buildIndex(const TString& majorName, const TString& minorName = "0");
                
                void writeOutGenResults();



                void addFriendTree(const TString& rootFileName, const TString& rootTreeName);

        protected:
                void createFileAndTree();
                
                void defineBranches();
                

                Bool_t definedBranches() const {return definedBranches_;}
                

                void definedBranches(Bool_t defined) {definedBranches_ = defined;}

        private:
                LauGenNtuple(const LauGenNtuple& rhs);

                LauGenNtuple& operator=(const LauGenNtuple& rhs);

                TString rootFileName_;
                TString rootTreeName_;
                TFile* rootFile_;
                TTree* rootTree_;

                Bool_t definedBranches_;

                typedef std::map<TString, Int_t>    IntVarMap;
                typedef std::map<TString, Double_t> DoubleVarMap;

                IntVarMap intVars_;
                DoubleVarMap doubleVars_;

                ClassDef(LauGenNtuple,0) // Generated toyMC ntuple

};

#endif