core/include/SCycleBaseHist.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
// $Id: SCycleBaseHist.h 335 2012-11-21 14:11:47Z krasznaa $
/***************************************************************************
 * @Project: SFrame - ROOT-based analysis framework for ATLAS
 * @Package: Core
 *
 * @author Stefan Ask       <Stefan.Ask@cern.ch>           - Manchester
 * @author David Berge      <David.Berge@cern.ch>          - CERN
 * @author Johannes Haller  <Johannes.Haller@cern.ch>      - Hamburg
 * @author A. Krasznahorkay <Attila.Krasznahorkay@cern.ch> - CERN/Debrecen
 *
 ***************************************************************************/

#ifndef SFRAME_CORE_SCycleBaseHist_H
#define SFRAME_CORE_SCycleBaseHist_H

// STL include(s):
#include <map>
#include <string>

// ROOT include(s):
#include <TObject.h>
#include <TString.h>

// Local include(s):
#include "ISCycleBaseHist.h"
#include "SCycleBaseBase.h"
#include "SError.h"

// Forward declaration(s):
class TDirectory;
class TH1;
class TList;

class SCycleBaseHist : public virtual ISCycleBaseHist,
                       public virtual SCycleBaseBase {

public:
   SCycleBaseHist();

   virtual void SetHistOutput( TList* output );
   virtual TList* GetHistOutput() const;

   template< class T > T* Book( const T& histo,
                                const char* directory = 0,
                                Bool_t inFile = kFALSE ) throw( SError );
   template< class T > T* Retrieve( const char* name,
                                    const char* directory = 0,
                                    Bool_t outputOnly = kFALSE ) throw( SError );
   template< class T >
   std::vector< T* > RetrieveAll( const char* name,
                                  const char* directory = 0 ) throw( SError );
   void WriteObj( const TObject& obj,
                  const char* directory = 0,
                  Bool_t inFile = kFALSE ) throw( SError );

   TH1* Hist( const char* name, const char* dir = 0 ) throw( SError );

protected:
   virtual void SetHistInputFile( TDirectory* file );
   virtual TDirectory* GetHistInputFile() const;

   virtual void WriteHistObjects( TDirectory* output );

private:
   TDirectory* GetTempDir() const;

#ifndef __MAKECINT__

   std::map< std::pair< std::string, std::string >, TH1* > m_histoMap;
   TList m_fileOutput;
#endif // __MAKECINT__

   TList* m_proofOutput; 
   TDirectory* m_inputFile; 

#ifndef DOXYGEN_IGNORE
   ClassDef( SCycleBaseHist, 0 );
#endif // DOXYGEN_IGNORE

}; // class SCycleBaseHist

// Don't include the templated function(s) when we're generating
// a dictionary:
#ifndef __CINT__
#include "SCycleBaseHist.icc"
#endif

#endif // SFRAME_CORE_SCycleBaseHist_H