RHistogramCnv.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2019 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#ifndef ROOTHISTCNV_RHISTOGRAMCNV_H
#define ROOTHISTCNV_RHISTOGRAMCNV_H 1

// Include files
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/HistogramBase.h"
#include "GaudiKernel/IOpaqueAddress.h"
#include "GaudiKernel/IRegistry.h"
#include "GaudiKernel/ObjectFactory.h"
#include "GaudiKernel/SmartIF.h"
#include "RConverter.h"
#include "RootObjAddress.h"
#include "TArray.h"
#include <memory>

namespace RootHistCnv {

  template <typename T, typename S, typename Q>
  class RHistogramCnv : public RConverter {
    template <typename CLASS>
    class TTH {
    public:
      template <typename INPUT>
      TTH( INPUT* i ) : m_c( dynamic_cast<CLASS*>( i ) ) {}
      template <typename INPUT>
      bool CopyH( INPUT& i ) {
        if ( m_c ) { m_c->Copy( i ); }
        return m_c != nullptr;
      }

    private:
      CLASS* m_c = nullptr;
    };

  public:
    StatusCode createObj( IOpaqueAddress* pAddr, DataObject*& refpObj ) override {
      refpObj           = DataObjFactory::create( objType() ).release();
      RootObjAddress* r = dynamic_cast<RootObjAddress*>( pAddr );
      Q*              h = dynamic_cast<Q*>( refpObj );
      if ( r && h ) {
        // Need to flip representation .... clumsy for the time being, because
        // THXY constructor has no "generic" copy constructor
        auto s = dynamic_cast<S*>( r->tObj() );
        if ( s ) {
          auto a = dynamic_cast<TArray*>( s );
          if ( a ) {
            auto p  = std::make_unique<T>();
            auto ok = TTH<S>( s ).CopyH( *p );
            if ( ok ) {
              p->Set( a->GetSize() );
              p->Reset();
              p->Add( s );
              h->adoptRepresentation( p.release() );
              return StatusCode::SUCCESS;
            }
          }
        }
      }
      return error( "Cannot create histogram - invalid address." );
    }
    StatusCode updateObj( IOpaqueAddress* /* pAddr */, DataObject* /* pObj */ ) override { return StatusCode::SUCCESS; }
    TObject* createPersistent( DataObject* pObj ) override {
      auto h = dynamic_cast<Q*>( pObj );
      if ( h ) {
        auto r = dynamic_cast<T*>( h->representation() );
        if ( r ) {
          auto a = dynamic_cast<TArray*>( r );
          if ( a ) {
            auto c  = std::make_unique<T>();
            auto ok = TTH<S>( r ).CopyH( *c );
            if ( ok ) {
              c->Set( a->GetSize() );
              c->Reset();
              c->Add( r );
              c->SetName( pObj->registry()->name().c_str() + 1 );
              return c.release();
            }
          }
        }
      }
      error( "Histogram object is invalid!" );
      return nullptr;
    }
    static const CLID& classID();
    RHistogramCnv( ISvcLocator* svc ) : RConverter( classID(), svc ) {}
  };
} // namespace RootHistCnv
#endif // ROOTHISTCNV_RHISTOGRAMCNV_H