rmidas.h

Go to the documentation of this file.

/********************************************************************\
 
  Name:         RMIDAS.H
  Created by:   Stefan Ritt & Konstantin Olchanski
 
  Contents:     ROOT definitions for MIDAS applications
 
\********************************************************************/
 
#ifndef _RMIDAS_H_
#define _RMIDAS_H_
 
#include <TObjArray.h>
#include <TFolder.h>
#include <TCutG.h>
 
/* root functions really are C++ functions */ extern TFolder *gManaHistosFolder;
extern TObjArray *gHistoFolderStack;
 
// book functions put a root object in a suitable folder
// for histos, there are a lot of types, so we use templates.
// for other objects we have one function per object
template < typename TH1X >
TH1X EXPRT * h1_book(const char *name, const char *title,
                     int bins, double min, double max)
{
   TH1X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH1X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH1X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH1X(name, title, bins, min, max);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
template < typename TH1X >
TH1X EXPRT * h1_book(const char *name, const char *title, int bins, double edges[])
{
   TH1X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH1X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH1X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH1X(name, title, bins, edges);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
template < typename TH2X >
TH2X EXPRT * h2_book(const char *name, const char *title,
                     int xbins, double xmin, double xmax,
                     int ybins, double ymin, double ymax)
{
   TH2X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH2X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH2X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH2X(name, title, xbins, xmin, xmax, ybins, ymin, ymax);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
template < typename TH2X >
TH2X EXPRT * h2_book(const char *name, const char *title,
                     int xbins, double xmin, double xmax, int ybins, double yedges[])
{
   TH2X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH2X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH2X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH2X(name, title, xbins, xmin, xmax, ybins, yedges);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
template < typename TH2X >
TH2X EXPRT * h2_book(const char *name, const char *title,
                     int xbins, double xedges[], int ybins, double ymin, double ymax)
{
   TH2X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH2X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH2X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH2X(name, title, xbins, xedges, ybins, ymin, ymax);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
template < typename TH2X >
TH2X EXPRT * h2_book(const char *name, const char *title,
                     int xbins, double xedges[], int ybins, double yedges[])
{
   TH2X *hist;
 
   /* check if histo already exists */
   if (!gHistoFolderStack->Last())
      hist = (TH2X *) gManaHistosFolder->FindObjectAny(name);
   else
      hist = (TH2X *) ((TFolder *) gHistoFolderStack->Last())->FindObjectAny(name);
 
   if (hist == NULL) {
      hist = new TH2X(name, title, xbins, xedges, ybins, yedges);
      if (!gHistoFolderStack->Last())
         gManaHistosFolder->Add(hist);
      else
         ((TFolder *) gHistoFolderStack->Last())->Add(hist);
   }
 
   return hist;
}
 
/*
 * the following two macros allow for simple fortran like usage
 * for the most common histo types
 */
#define H1_BOOK(n,t,b,min,max) (h1_book<TH1F>(n,t,b,min,max))
#define H2_BOOK(n,t,xb,xmin,xmax,yb,ymin,ymax) (h2_book<TH2F>(n,t,xb,xmin,xmax,yb,ymin,ymax))
 
TCutG *cut_book(const char *name);
 
#endif