//-----------------------
// This Class's Header --
//-----------------------
#include "CalorObjects/CESQ_Storable.hh"

//-------------------------------
// Collaborating Class Headers --
//-------------------------------
#include "Experiment/Experiment.hh"
#include "AbsEnv/AbsEnv.hh"
#include "ErrorLogger_i/gERRLOG.hh"

// ROOT Headers :
#include "TBuffer.h"
#include "EdmUtilities/CdfClassImp.hh"

//-----------------------------------------------------------------------------
// Creation
//-----------------------------------------------------------------------------

CESQ_Storable::CESQ_Storable() : StorableObject() {}

CESQ_Storable::CESQ_Storable(const CESQ_Storable& rhs) :
  StorableObject(rhs),
  _container(rhs._container) {}

//-----------------------------------------------------------------------------
// Destruction
//-----------------------------------------------------------------------------

CESQ_Storable::~CESQ_Storable() {}

void CESQ_Storable::destroy()
{ CESQ_Storable::deallocate();
  delete this;
}

void CESQ_Storable::deallocate()
{ StorableObject::deallocate();
}

//-----------------------------------------------------------------------------
// EDM:  functions required by StorableObject
//-----------------------------------------------------------------------------
  
std::string CESQ_Storable::class_name() const
{
  return CESQ_Storable::Class_Name() ;
}

Version_t CESQ_Storable::class_version() const
{
  return CESQ_Storable::Class_Version();
}

void CESQ_Storable::print(std::ostream& os) const
{
  StorableObject::print(os);
  _container.print(os);
}

bool CESQ_Storable::postread(EventRecord* record)
{
  return true;
}

bool CESQ_Storable::prewrite(EventRecord* record)
{
  // Note : ValueVector does not currently call prewrite on its elements,
  // so do so here :
  bool status = StorableObject::prewrite(record);
  status = _container.prewrite(record) && status;
  for (CESQ_Storable::iterator i = _container.contents().begin();
       i != _container.contents().end(); ++i)
    {
      status = (*i).prewrite(record) && status;
    }
  return status;
}

bool CESQ_Storable::activate(EventRecord* record)
{
  bool status = StorableObject::activate(record);
  return status;
}

bool CESQ_Storable::deactivate(EventRecord* record)
{
  bool status = StorableObject::deactivate(record);
  return status;
}

void CESQ_Storable::Streamer(TBuffer& iobuffer)
{
  unsigned int start = 0 ;
  unsigned int byte_count = 0 ;
  
  //-----------------------------------------------------------------------------
  // Read object from buffer
  //-----------------------------------------------------------------------------
  
  if (iobuffer.IsReading())
    {
      Version_t original_version = iobuffer.ReadVersion(&start, &byte_count) ;
      if (original_version == 2)
	{ 
	  StorableObject::Streamer(iobuffer) ;
	  _container.Streamer(iobuffer); 
	}
      else
	{ 
	  ERRLOG( ELerror, "Unsupported CESQ_Storable version" )
	    << "@SUB=CESQ_Storable::Streamer(<isReading>)"
	    << "CESQ_Storable cannot be read."
	    << endmsg;
	}
      iobuffer.CheckByteCount(start, byte_count, CESQ_Storable::IsA()) ;
    }
  
  //-----------------------------------------------------------------------------
  // Write object to buffer
  //-----------------------------------------------------------------------------
  else if (iobuffer.IsWriting())
    { 
      byte_count = iobuffer.WriteVersion(CESQ_Storable::IsA(), kTRUE) ;
      Version_t current_version = CESQ_Storable::class_version() ;
      if (current_version == 2)
	{ 
	  StorableObject::Streamer(iobuffer) ;
	  _container.Streamer(iobuffer); 
	}
      else
	{ 
	  ERRLOG( ELerror, "Unsupported CESQ_Storable version" )
	    << "@SUB=CESQ_Storable::Streamer(<isWriting>)"
	    << "CESQ_Storable cannot be written."
	    << endmsg;
	}
      iobuffer.SetByteCount(byte_count, kTRUE) ;
    }
  
  //-----------------------------------------------------------------------------
  // Unanticipated action
  //-----------------------------------------------------------------------------
  else
    { 
      ERRLOG( ELerror, "CESQ_Storable::Streamer(<unknown action>)" )
	<< "@SUB=CESQ_Storable::Streamer"
	<< "NOTHING DONE."
	<< endmsg;
    }
}

//-----------------------------------------------------------------------------
// Functions to retrieve the collection from the event
//-----------------------------------------------------------------------------

CESQ_Storable::Error CESQ_Storable::find(CESQ_Storable_ch& cHandle)
{
  EventRecord::ConstIterator ciEvent(AbsEnv::theEvent(),"CESQ_Storable");
  
  if (ciEvent.is_valid())
    {
      cHandle = ciEvent;
      return OK;
    }
  else
    {
      cHandle.set_null();
      return ERROR;
    } 

}

CESQ_Storable::Error CESQ_Storable::find(CESQ_Storable_ch& cHandle,
					   const StorableObject::Selector & selector)
{
  EventRecord::ConstIterator ciEvent(AbsEnv::theEvent(),"CESQ_Storable");
  while (ciEvent.is_valid() && !selector(*ciEvent))
    {
      ++ciEvent;
    }
  if (ciEvent.is_valid()) 
    {
      cHandle = ciEvent;
      return OK;      
    } 
  else 
    {
      cHandle.set_null();
      return ERROR;
    }
}


//=============================================================================
// ROOT I/O Hook
//=============================================================================
CdfClassImp(CESQ_Storable)