LauMergeDataFiles.hh

Go to the documentation of this file.

/*
Copyright 2008 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
*/

#include <map>
#include <cstdlib>

#include "TFile.h"
#include "TString.h"
#include "TTree.h"

class LauMergeDataFiles
{
        public:

                LauMergeDataFiles(const TString& fileName1, const TString& fileName2, const TString& treeName);

                virtual ~LauMergeDataFiles();


                void process(const TString& fileName);

        protected:
                typedef std::map<TString,Double_t> LeafDoubleMap;
                typedef std::map<TString,Int_t> LeafIntegerMap;
                typedef std::map< Int_t,std::pair<Int_t,Int_t> > ExptsMap;

                void openInputFiles();
                void setupInputTrees();
                void setupOutputTree();
                void findExperiments(TTree* tree, ExptsMap& exptsMap);
                Bool_t checkExperimentMaps() const;
                void readExperiment(TTree* tree, const ExptsMap::const_iterator& exptsMap, Int_t offset);
                void writeFile();

        private:
                TString fileName1_;
                TString fileName2_;
                TString treeName_;

                TFile * inputFile1_;
                TFile * inputFile2_;

                TTree * inputTree1_;
                TTree * inputTree2_;

                TFile * outputFile_;
                TTree * outputTree_;

                // Tree variables
                Int_t iExpt_;
                Int_t iEvtWithinExpt_;

                LeafDoubleMap doubleVars_;
                LeafIntegerMap integerVars_;

                ExptsMap tree1Expts_;
                ExptsMap tree2Expts_;

                ClassDef(LauMergeDataFiles,0)
};