plotAllIntensities.h

Go to the documentation of this file.

//
//    Copyright 2009 Sebastian Neubert
//
//    This file is part of rootpwa
//
//    rootpwa is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    rootpwa is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with rootpwa.  If not, see <http://www.gnu.org/licenses/>.
//
//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Draws intensity graph for single wave from tree.
//
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//
//
//-----------------------------------------------------------


//
// plots all waves of all trees sorted by JPC and intensity and
// returns a list of wave names and pointers to the pads they were
// drawn into
// the intensity sorting is performed w.r.t. the first tree
//


#ifndef PLOTALLINTENSITIES_HH
#define PLOTALLINTENSITIES_HH


#include <string>
#include <vector>

#include "TVirtualPad.h"
#include "TTree.h"

using namespace std;


std::vector<std::pair<std::string, TVirtualPad*> >
plotAllIntensities(const unsigned int nmbTrees,               // number of fitResult trees
                   TTree**            trees,                  // array of fitResult trees
                   const bool         createPsFile  = false,  // if true, plots are written to waves.ps
                   const std::string& outPath       = "./",   // path for output files
                   const int*         graphColors   = NULL,   // array of colors for graph line and marker
                   const double*      graphScales   = NULL,   // array of scales for graph scaling (scales all graphs of one tree)
                   const bool         drawLegend    = true,   // if set legend is drawn
                   const double       yAxisRangeMax = 0,      // if != 0; range of y-axis is limited to this value
                   const string&      branchName    = "fitResult_v2");

inline
std::vector<std::pair<std::string, TVirtualPad*> >
plotAllIntensities(TTree*             tree,                   // fitResult tree
                   const bool         createPsFile  = false,  // if true, plots are written to waves.ps
                   const std::string& outPath       = "./",   // path for output files
                   const double       yAxisRangeMax = 0,      // if != 0; range of y-axis is limited to this value
                   const string&      branchName    = "fitResult_v2")
{
  return plotAllIntensities(1, &tree, createPsFile, outPath, NULL, NULL, false, yAxisRangeMax, branchName);
}


inline
std::vector<std::pair<std::string, TVirtualPad*> >
plotAllIntensities(std::vector<TTree*>&    trees,                  // vector of fitResult trees
                   const bool                 createPsFile  = false,  // if true, plots are written to waves.ps
                   const std::string&         outPath       = "./",   // path for output files
                   const std::vector<int>&    graphColors   = std::vector<int>(),  // vector of colors for graph line and marker
                   const std::vector<double>& graphScales   = std::vector<double>(), // vector of scales for graphs
                   const bool              drawLegend    = true,   // if set legend is drawn
                   const double            yAxisRangeMax = 0,      // if != 0; range of y-axis is limited to this value
                   const string&           branchName    = "fitResult_v2")
{
  return plotAllIntensities(trees.size(), &(*(trees.begin())), createPsFile, outPath,
                            &(*(graphColors.begin())), &(*(graphScales.begin())), drawLegend, yAxisRangeMax, branchName);
}


#endif  // PLOTALLINTENSITIES_HH