SusyNtuple/VarHandle.h

// Dear emacs, this is -*- c++ -*-
// $Id: CodeGenerator_v2_constants.h 510583 2012-07-18 09:12:16Z krasznaa $
#ifndef D3PDREADER_VARHANDLE_H
#define D3PDREADER_VARHANDLE_H

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

// Local include(s):
#include "SusyNtuple/D3PDReadStats.h"

// Forward declaration(s):
class TObject;
class TTree;
class TBranch;

namespace D3PDReader {

   class VarHandleBase {

   protected:
      enum BranchAvailability {
         UNKNOWN = 0, 
         AVAILABLE = 1, 
         UNAVAILABLE = 2 
      };

   public:
      VarHandleBase( ::TObject* parent = 0, const char* name = "",
                     const ::Long64_t* master = 0 );
      virtual ~VarHandleBase() {}

      ::TObject* GetParent() const;
      void SetParent( ::TObject* parent );

      const char* GetName() const;
      void SetName( const char* name );

      const char* GetType() const;
      void SetType( const char* type );

      const ::Long64_t* GetMaster() const;
      void SetMaster( const ::Long64_t* master );

      virtual void ReadFrom( ::TTree* tree ) = 0;
      virtual ::TBranch* WriteTo( ::TTree* tree ) = 0;

      ::Bool_t IsActive() const;
      void SetActive( ::Bool_t active = kTRUE );

      virtual ::Bool_t IsAvailable() const;

      virtual void ReadCurrentEntry() const = 0;

      virtual void Clear() = 0;

      virtual VariableStats GetStatistics() const;

   protected:
      ::Bool_t ConnectVariable( void* var, ::TClass* realClass,
                                EDataType dtype, Bool_t isptr ) const;
      void UpdateBranch() const;
      void UpdateStat( ::TBranch* br ) const;
      const char* RootType( const char* typeid_type ) const;
      const char* RootCppType( const char* typeid_type ) const;

      const ::Long64_t* fMaster; 
      ::TObject* fParent; 
      ::Bool_t fFromInput; 
      ::TTree* fInTree; 
      mutable ::TBranch* fInBranch; 
      mutable BranchAvailability fAvailable; 

   private:
      ::TString fName; 
      ::Bool_t fActive; 

      ::TString fType; 
      mutable std::vector< ::Long64_t > fEntriesRead; 
      mutable std::vector< ::Float_t > fBranchSize; 
      mutable std::vector< ::Float_t > fZippedSize; 

   }; // class VarHandleBase

   template< typename Type >
   class VarHandle : public VarHandleBase {

   public:
      typedef Type& result_type;
      typedef const Type& const_result_type;

      VarHandle( ::TObject* parent = 0, const char* name = "",
                 const ::Long64_t* master = 0 );
      ~VarHandle();

      virtual void ReadFrom( ::TTree* tree );
      virtual ::TBranch* WriteTo( ::TTree* tree );

      result_type operator()();
      const_result_type operator()() const;

      virtual void ReadCurrentEntry() const;

      virtual void Clear();

   private:
      mutable Type fVariable; 

   }; // class VarHandle

   template< typename Type >
   class VarHandle< Type* > : public VarHandleBase {

   public:
      typedef Type* result_type;
      typedef const Type* const_result_type;

      VarHandle( ::TObject* parent = 0, const char* name = "",
                 const ::Long64_t* master = 0 );
      ~VarHandle();

      virtual void ReadFrom( ::TTree* tree );
      virtual ::TBranch* WriteTo( ::TTree* tree );

      result_type operator()();
      const_result_type operator()() const;

      virtual void ReadCurrentEntry() const;

      virtual void Clear();

   private:
      mutable Type* fVariable; 

   }; // class VarHandle

} // namespace D3PDReader

// Include the implementation:
#ifndef __CLING__
#include "VarHandle.icc"
#endif // cling

#endif // D3PDREADER_VARHANDLE_H