Generic3D.h

Go to the documentation of this file.

#ifndef GAUDISVC_GENERIC3D_H
#define GAUDISVC_GENERIC3D_H 1
#include "AIDA_visibility_hack.h"

#include "Axis.h"
#include "TFile.h"
#include "Annotation.h"
#include "GaudiKernel/HistogramBase.h"
#include "AIDA/IHistogram3D.h"
#include <stdexcept>

/*
 *    Gaudi namespace
 */
namespace Gaudi {

  template<typename INTERFACE, typename IMPLEMENTATION>
  class GAUDI_API Generic3D : virtual public INTERFACE, virtual public HistogramBase  {
  public:
    typedef Generic3D<INTERFACE,IMPLEMENTATION> Base;
    Generic3D() : m_rep(0)  {}
    virtual ~Generic3D()    { delete m_rep; }
    TObject* representation() const                      { return m_rep;                       }
    virtual void adoptRepresentation(TObject* rep);

    virtual int  dimension() const  { return 3; }
    virtual std::string title() const                    {  return m_annotation.value("Title");}
    virtual bool setTitle(const std::string & title);
    virtual std::string name() const                     { return m_annotation.value("Name"); }
    bool setName( const std::string& newName);
    virtual AIDA::IAnnotation & annotation()             { return m_annotation;               }
    virtual const AIDA::IAnnotation & annotation() const { return m_annotation;               }

    virtual int entries() const;
    virtual int allEntries() const;
    virtual double sumBinHeights() const;
    virtual double sumAllBinHeights() const;
    virtual double  sumExtraBinHeights (  ) const  { return  sumAllBinHeights()-sumBinHeights(); }
    virtual double minBinHeight() const;
    virtual double maxBinHeight() const;

    int rIndexX(int index) const { return m_xAxis.rIndex(index);}
    int rIndexY(int index) const { return m_yAxis.rIndex(index);}
    int rIndexZ(int index) const { return m_zAxis.rIndex(index);}

    double binMeanX(int indexX,int ,int ) const
    { return (m_rep->GetXaxis())->GetBinCenter( rIndexX(indexX) ); }
    double binMeanY(int,int indexY,int  ) const
    { return (m_rep->GetYaxis())->GetBinCenter( rIndexY(indexY) ); }
    double binMeanZ(int ,int ,int indexZ) const
    { return (m_rep->GetYaxis())->GetBinCenter( rIndexY(indexZ) ); }
    int binEntries(int indexX,int indexY,int indexZ) const    {
      if (binHeight(indexX, indexY, indexZ)<=0) return 0;
      double xx =  binHeight(indexX, indexY, indexZ)/binError(indexX, indexY, indexZ);
      return int(xx*xx+0.5);
    }
    virtual int binEntriesX(int index) const    {
      int n = 0;
      for (int i = -2; i < yAxis().bins(); ++i)
        for (int j = -2; j < zAxis().bins(); ++j)
          n += binEntries(index,i,j);
      return n;

    }
    virtual int binEntriesY(int index) const    {
      int n = 0;
      for (int i = -2; i < xAxis().bins(); ++i)
        for (int j = -2; j < zAxis().bins(); ++j)
          n += binEntries(i,index,j);
      return n;
    }

    virtual int binEntriesZ(int index) const    {
      int n = 0;
      for (int i = -2; i < xAxis().bins(); ++i)
        for (int j = -2; j < yAxis().bins(); ++j)
          n += binEntries(i,j,index);
      return n;
    }

    double  binHeight ( int indexX,int indexY,int indexZ ) const
    { return m_rep->GetBinContent ( rIndexX(indexX), rIndexY(indexY), rIndexZ(indexZ) ); }

    virtual double binHeightX(int index) const    {
      double s = 0;
      for (int i = -2; i < yAxis().bins(); ++i)
        for (int j = -2; j < zAxis().bins(); ++j)
          s += binHeight(index,i,j);
      return s;
    }
    virtual double binHeightY(int index) const    {
      double s = 0;
      for (int i = -2; i < xAxis().bins(); ++i)
        for (int j = -2; j < zAxis().bins(); ++j)
          s += binHeight(i,index,j);
      return s;
    }
    virtual double binHeightZ(int index) const    {
      double s = 0;
      for (int i = -2; i < xAxis().bins(); ++i)
        for (int j = -2; j < yAxis().bins(); ++j)
          s += binHeight(i,j,index);
      return s;
    }
    virtual double  binError ( int indexX,int indexY,int indexZ ) const
    { return m_rep->GetBinError ( rIndexX(indexX), rIndexY(indexY ), rIndexZ(indexZ ) ); }
    virtual double  meanX (  ) const  { return m_rep->GetMean ( 1); }

    virtual double  meanY (  ) const  { return m_rep->GetMean ( 2 ); }
    virtual double  meanZ (  ) const  { return m_rep->GetMean ( 3 ); }
    virtual double  rmsX (  ) const  { return m_rep->GetRMS( 1 ); }
    virtual double  rmsY (  ) const  { return m_rep->GetRMS( 2 ); }
    virtual double  rmsZ (  ) const  { return m_rep->GetRMS( 3 ); }
    virtual const AIDA::IAxis & xAxis (  ) const  { return m_xAxis; }
    virtual const AIDA::IAxis & yAxis (  ) const  { return m_yAxis; }
    virtual const AIDA::IAxis & zAxis (  ) const  { return m_zAxis; }
    virtual int  coordToIndexX ( double coord ) const { return xAxis().coordToIndex(coord);}
    virtual int  coordToIndexY ( double coord ) const { return yAxis().coordToIndex(coord);}
    virtual int  coordToIndexZ ( double coord ) const { return zAxis().coordToIndex(coord);}

    virtual double equivalentBinEntries (  ) const;
    virtual bool scale( double scaleFactor );
    virtual bool  add ( const INTERFACE & hist ) {
      const Base* p = dynamic_cast<const Base*>(&hist);
      if ( p )  {
        m_rep->Add(p->m_rep);
        return true;
      }
      throw std::runtime_error("Cannot add profile histograms of different implementations.");
    }

    // overwrite extraentries
    int extraEntries() const {
      return
        binEntries(AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN) +
        binEntries(AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::OVERFLOW_BIN)  +
        binEntries(AIDA::IAxis::UNDERFLOW_BIN, AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN)  +
        binEntries(AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN) +
        binEntries(AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN,AIDA::IAxis::OVERFLOW_BIN) +
        binEntries(AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::UNDERFLOW_BIN)  +
        binEntries(AIDA::IAxis::OVERFLOW_BIN, AIDA::IAxis::OVERFLOW_BIN,AIDA::IAxis::OVERFLOW_BIN);
    }
    virtual std::ostream& print( std::ostream& s ) const;
    virtual std::ostream& write( std::ostream& s ) const;
    virtual int write( const char* file_name ) const;

  protected:
    Gaudi::Axis              m_xAxis;
    Gaudi::Axis              m_yAxis;
    Gaudi::Axis              m_zAxis;
    mutable AIDA::Annotation m_annotation;
    IMPLEMENTATION*          m_rep;
    // class type
    std::string              m_classType;
    // cache sumEntries (allEntries)   when setting contents since Root can't compute by himself
    int                      m_sumEntries;
  }; // end class IHistogram3D

  template <class INTERFACE, class IMPLEMENTATION>
  bool Generic3D<INTERFACE,IMPLEMENTATION>::setTitle(const std::string & title)  {
    m_rep->SetTitle(title.c_str());
    if ( !annotation().addItem( "Title", title ) )
      m_annotation.setValue( "Title" , title );
    if ( !annotation().addItem( "title", title ) )
      annotation().setValue( "title", title );
    return true;
  }

  template <class INTERFACE, class IMPLEMENTATION>
  bool Generic3D<INTERFACE,IMPLEMENTATION>::setName( const std::string& newName ) {
    m_rep->SetName(newName.c_str());
    m_annotation.setValue( "Name", newName );
    return true;
  }
  template <class INTERFACE, class IMPLEMENTATION>
  int Generic3D<INTERFACE,IMPLEMENTATION>::entries() const                       {
    return (int)m_rep->GetEntries();
  }

  template <class INTERFACE, class IMPLEMENTATION>
  int Generic3D<INTERFACE,IMPLEMENTATION>::allEntries (  ) const  {
    return int(m_rep->GetEntries());
  }

  template <class INTERFACE, class IMPLEMENTATION>
  double  Generic3D<INTERFACE,IMPLEMENTATION>::minBinHeight() const  {
    return m_rep->GetMinimum();
  }

  template <class INTERFACE, class IMPLEMENTATION>
  double  Generic3D<INTERFACE,IMPLEMENTATION>::maxBinHeight() const  {
    return m_rep->GetMaximum();
  }

  template <class INTERFACE, class IMPLEMENTATION>
  double Generic3D<INTERFACE,IMPLEMENTATION>::sumBinHeights () const  {
    return m_rep->GetSumOfWeights();
  }

  template <class INTERFACE, class IMPLEMENTATION>
  double Generic3D<INTERFACE,IMPLEMENTATION>::sumAllBinHeights () const  {
    return m_rep->GetSum();
  }

  template <class INTERFACE, class IMPLEMENTATION>
  double Generic3D<INTERFACE,IMPLEMENTATION>::equivalentBinEntries() const  {
    if (sumBinHeights() <= 0) return 0;
    Stat_t stats[11];   // cover up to 3D...
    m_rep->GetStats(stats);
    return stats[0]*stats[0]/stats[1];
  }

  template <class INTERFACE, class IMPLEMENTATION>
  bool Generic3D<INTERFACE,IMPLEMENTATION>::scale(double scaleFactor)   {
    m_rep->Scale ( scaleFactor );
    return true;
  }

  template <class INTERFACE, class IMPLEMENTATION>
  std::ostream& Generic3D<INTERFACE,IMPLEMENTATION>::print( std::ostream& s ) const
  {
    m_rep->Print("all");
    return s;
  }


  template <class INTERFACE, class IMPLEMENTATION>
  std::ostream& Generic3D<INTERFACE,IMPLEMENTATION>::write( std::ostream& s ) const
  {
    s << "\n3D Histogram Table: " << std::endl;
    s << "BinX, BinY, BinZ, Height, Error " << std::endl;
    for ( int i = 0; i < xAxis().bins(); ++i )
      for ( int j = 0; j < yAxis().bins(); ++j )
        for ( int k = 0; k < zAxis().bins(); ++k )
        s << binMeanX( i, j, k )    << ", "
      << binMeanY( i, j, k )    << ", "
      << binMeanZ( i, j, k )    << ", "
      << binHeight( i, j, k ) << ", "
      << binError ( i, j, k ) << std::endl;
    s << std::endl;
    return s;
  }

  template <class INTERFACE, class IMPLEMENTATION>
  int Generic3D<INTERFACE,IMPLEMENTATION>::write( const char* file_name ) const
  {
    TFile *f = TFile::Open(file_name,"RECREATE");
    Int_t nbytes = m_rep->Write();
    f->Close();
    return nbytes;
  }
} // end namespace AIDA
#endif // GAUDIPI_GENERIC3D_H