LauGenNtuple.hh

Go to the documentation of this file.

/*
Copyright 2006 University of Warwick

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

/*
Laura++ package authors:
John Back
Paul Harrison
Thomas Latham
*/

#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