LauFitDataTree.hh

Go to the documentation of this file.

// Copyright University of Warwick 2004 - 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_FIT_DATA_TREE
#define LAU_FIT_DATA_TREE

#include <vector>
#include <map>

#include "TEventList.h"
#include "TTree.h"

class TFile;
class TLeaf;

typedef std::map<TString,Double_t> LauFitData;


class LauFitDataTree {

        public:

                LauFitDataTree(const TString& rootFileName, const TString& rootTreeName);

                virtual ~LauFitDataTree();


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


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


                Bool_t findBranches();

                void readAllData();


                void readExperimentData( UInt_t iExpt );


                void appendFakePoints(const std::vector<Double_t>& xCoords, const std::vector<Double_t>& yCoords);


                UInt_t nBranches() const;


                UInt_t nTreeEvents() const {return rootTree_ ? static_cast<UInt_t>(rootTree_->GetEntries()) : 0;}


                UInt_t nEvents() const {return eventList_ ? static_cast<UInt_t>(eventList_->GetN()) : this->nTreeEvents();}


                UInt_t nFakeEvents() const {return fakeEvents_.size();}


                UInt_t nTotalEvents() const {return this->nTreeEvents()+this->nFakeEvents();}


                Bool_t haveBranch(const TString& name) const;


                const LauFitData& getData(UInt_t iEvt) const;

                void disableAllBranches() const;

                void enableAllBranches() const;


                void enableBranch(const TString& name) const;


                void disableBranch(const TString& name) const;

        protected:
                void openFileAndTree();


                void loadData();

        private:
                typedef std::map<TString,UInt_t> LauNameIndexMap;

                typedef std::vector<Double_t> LauEventData;

                typedef std::vector<TLeaf*> LauLeafList;

                TString rootFileName_;

                TString rootTreeName_;

                TFile* rootFile_;

                TTree* rootTree_;

                TEventList* eventList_;

                LauNameIndexMap leafNames_;

                mutable LauEventData eventData_;

                mutable LauFitData eventDataOut_;

                LauLeafList leaves_;

                std::vector<LauEventData> treeEvents_;

                std::vector<LauEventData> fakeEvents_;

                ClassDef(LauFitDataTree, 0)
};

#endif