OraReferenceStreamer.cc

Go to the documentation of this file.

#include "CondCore/ORA/interface/Exception.h"
#include "CondCore/ORA/interface/Reference.h"
#include "CondCore/ORA/interface/IReferenceHandler.h"
#include "OraReferenceStreamer.h"
#include "DataElement.h"
#include "MappingElement.h"
#include "ContainerSchema.h"
#include "RelationalOperation.h"
#include "ClassUtils.h"
// externals
#include "CoralBase/Attribute.h"
#include "FWCore/Utilities/interface/MemberWithDict.h"

ora::OraReferenceStreamerBase::OraReferenceStreamerBase( const edm::TypeWithDict& objectType,
                                                         MappingElement& mapping,
                                                         ContainerSchema& schema):
  m_objectType( objectType ),
  m_mapping( mapping ),
  m_schema( schema ),
  m_dataElement( 0 ),
  m_dataElemOId0( 0 ),
  m_dataElemOId1( 0 ),
  m_relationalData( 0 ){
  m_columnIndexes[0] = -1;
  m_columnIndexes[1] = -1;
}

ora::OraReferenceStreamerBase::~OraReferenceStreamerBase(){
}


bool ora::OraReferenceStreamerBase::buildDataElement(DataElement& dataElement,
                                                     IRelationalData& relationalData){
  m_dataElement = &dataElement;
  // first resolve the oid0 and oid2 data elements...
  edm::TypeWithDict refType( typeid(Reference) );
  //edm::TypeWithDict oidType = edm::TypeWithDict::ByTypeInfo( typeid(OId) );
  size_t baseOffsetFunc = 0;
  if( m_objectType != refType ){
    bool foundRef = ClassUtils::findBaseType( m_objectType, refType, baseOffsetFunc );
    if(!foundRef){
      throwException("Type \""+m_objectType.cppName()+"\" is not an Ora Reference.",
                     "OraReferenceStreamerBase::buildDataElement");
    } 
  }
  edm::MemberWithDict contIdMember = refType.dataMemberByName("m_containerId");
  edm::MemberWithDict itemIdMember = refType.dataMemberByName("m_itemId");
  if( !contIdMember || !itemIdMember ){
    throwException("Data members for class OId not found.",
                   "OraReferenceStreamerBase::buildDataElement");
  }
  m_dataElemOId0 = &dataElement.addChild( contIdMember.offset(), baseOffsetFunc );
  m_dataElemOId1 = &dataElement.addChild( itemIdMember.offset(), baseOffsetFunc);
  // then book the columns in the data attribute... 
  const std::vector<std::string>& columns =  m_mapping.columnNames();
  if( columns.size() < 2 ){
      throwException("Expected column names have not been found in the mapping.",
                     "OraReferenceStreamerBase::buildDataElement");    
  }
  const std::type_info& attrType = typeid(int);
  for( size_t i=0; i<2; i++ ){
    m_columnIndexes[i] = relationalData.addData( columns[i],attrType ); 
  }
  m_relationalData = &relationalData;
  return true;
}


void ora::OraReferenceStreamerBase::bindDataForUpdate( const void* data ){
  if(!m_relationalData){
    throwException("The streamer has not been built.",
                   "OraReferenceStreamerBase::bindDataForUpdate");
  }
  
  void* oid0Address = m_dataElemOId0->address( data );
  coral::Attribute& oid0Attr = m_relationalData->data()[ m_columnIndexes[0] ];
  oid0Attr.data<int>()= *static_cast<int*>(oid0Address);
  void* oid1Address = m_dataElemOId1->address( data );
  coral::Attribute& oid1Attr = m_relationalData->data()[ m_columnIndexes[1] ];
  oid1Attr.data<int>()= *static_cast<int*>(oid1Address) ;
  IReferenceHandler* refHandler = m_schema.referenceHandler();
  void* refPtr = m_dataElement->address( data );
  if(refHandler) refHandler->onSave( *static_cast<Reference*>( refPtr ) );
}

void ora::OraReferenceStreamerBase::bindDataForRead( void* data ){
  if(!m_relationalData){
    throwException("The streamer has not been built.",
                   "OraReferenceStreamerBase::bindDataForRead");
  }

  void* oid0Address = m_dataElemOId0->address( data );
  coral::Attribute& oid0Attr = m_relationalData->data()[ m_columnIndexes[0] ];
  *static_cast<int*>(oid0Address) = oid0Attr.data<int>();
  void* oid1Address = m_dataElemOId1->address( data );
  coral::Attribute& oid1Attr = m_relationalData->data()[ m_columnIndexes[1] ];
  *static_cast<int*>( oid1Address ) = oid1Attr.data<int>();
  IReferenceHandler* refHandler = m_schema.referenceHandler();
  void* refPtr = m_dataElement->address( data );
  if(refHandler) refHandler->onLoad( *static_cast<Reference*>( refPtr ) );
}

ora::OraReferenceWriter::OraReferenceWriter( const edm::TypeWithDict& objectType,
                                             MappingElement& mapping,
                                             ContainerSchema& schema  ):
  OraReferenceStreamerBase( objectType, mapping, schema ){
}

ora::OraReferenceWriter::~OraReferenceWriter(){
}

bool ora::OraReferenceWriter::build(DataElement& dataElement,
                                    IRelationalData& relationalData,
                                    RelationalBuffer&){
  return buildDataElement( dataElement, relationalData );
}

void ora::OraReferenceWriter::setRecordId( const std::vector<int>& ){
}

void ora::OraReferenceWriter::write( int,
                                     const void* data ){
  bindDataForUpdate( data );  
}

ora::OraReferenceUpdater::OraReferenceUpdater( const edm::TypeWithDict& objectType,
                                               MappingElement& mapping,
                                               ContainerSchema& schema):
  OraReferenceStreamerBase( objectType, mapping, schema ){
}

ora::OraReferenceUpdater::~OraReferenceUpdater(){
}

bool ora::OraReferenceUpdater::build(DataElement& dataElement,
                                     IRelationalData& relationalData,
                                     RelationalBuffer&){
  return buildDataElement( dataElement, relationalData );  
}

void ora::OraReferenceUpdater::setRecordId( const std::vector<int>& ){
}

void ora::OraReferenceUpdater::update( int,
                                       const void* data ){
  bindDataForUpdate( data );  
}

ora::OraReferenceReader::OraReferenceReader( const edm::TypeWithDict& objectType,
                                             MappingElement& mapping,
                                             ContainerSchema& schema ):
  OraReferenceStreamerBase( objectType, mapping, schema ){
}

ora::OraReferenceReader::~OraReferenceReader(){
}

bool ora::OraReferenceReader::build(DataElement& dataElement,
                                    IRelationalData& relationalData){
  return buildDataElement( dataElement, relationalData );  
}

void ora::OraReferenceReader::select( int ){
}

void ora::OraReferenceReader::setRecordId( const std::vector<int>& ){
}

void ora::OraReferenceReader::read( void* data ){
  bindDataForRead( data );
}

void ora::OraReferenceReader::clear(){
}

    
ora::OraReferenceStreamer::OraReferenceStreamer( const edm::TypeWithDict& objectType,
                                                 MappingElement& mapping,
                                                 ContainerSchema& schema ):
  m_objectType( objectType ),
  m_mapping( mapping ),
  m_schema( schema ){
}

ora::OraReferenceStreamer::~OraReferenceStreamer(){
}

ora::IRelationalWriter* ora::OraReferenceStreamer::newWriter(){
  return new OraReferenceWriter( m_objectType, m_mapping, m_schema );
}

ora::IRelationalUpdater* ora::OraReferenceStreamer::newUpdater(){
  return new OraReferenceUpdater( m_objectType, m_mapping, m_schema );
}

ora::IRelationalReader* ora::OraReferenceStreamer::newReader(){
  return new OraReferenceReader( m_objectType, m_mapping, m_schema );
}