CalorimeterSIO.cc

Go to the documentation of this file.

//  CalorimeterSIO.cxx
//

//#include <iostream>
//using namespace std;

#include "SIO_recordManager.h"
#include "SIO_record.h"
#include "SIO_definitions.h"
#include "SIO_functions.h"

#include "CalorimeterSIO.h"
#include <stdio.h>
#include <vector>
using std::vector;

// set unit conversion factors here
const double cm = 10.0; // mm, which is the default in Geant4
const double GeV = 1000.0; // MeV, which is the default in Geant4

// constructor
CalorimeterSIO::CalorimeterSIO(  const char* i_name, vector<CalorimeterHit> calHits) : SIO_block(i_name), _calHits(calHits)
{
  // SIO init --  join the event record.
  SIO_record* record;
  record = SIO_recordManager::get( "LCD_LCDG4_Event" );
  if (record != NULL)
    record->connect(this);
}

//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
void CalorimeterSIO::clear()
  //-----------------------
{
  _calHits.erase(_calHits.begin(),_calHits.end());
}

void CalorimeterSIO::addHit(CalorimeterHit hit)
{
  _calHits.push_back(hit);
}

// ----------------------------------------------------------------------------
// Perform SIO transfer.
// ----------------------------------------------------------------------------
unsigned int CalorimeterSIO::xfer
(
 SIO_stream*     stream,
 SIO_operation   op,
 unsigned int    version
 )
{
  // Make compiler happy about unused vars
  if(op || version) ;

  unsigned int status;

  // write out the number of calorimeter hits...
  int nHits = _calHits.size();
  status = SIO_DATA( stream, &nHits, 1 );

  // loop over calorimeter hits and write them out...

  std::vector<CalorimeterHit>::const_iterator it;
  for(it=_calHits.begin(); it!=_calHits.end(); ++it)
    {
      CalorimeterHit hit = *it;
      // cell ID
      unsigned int id = hit.id();
      SIO_DATA(stream, &id,     1);
      // energy
      float energy = hit.energy()/GeV;
      SIO_DATA(stream, &energy, 1);

      // For debugging
      //printf("CalorimeterSIO::hit.energy()=%f\n",hit.energy());

      // now loop over the individual contributions to this cell...
      CalorimeterHit::McEnergyMap mcmap = hit.getMcEnergyMap();
      CalorimeterHit::McEnergyMap::const_iterator mapit;
      int mapsize = mcmap.size();
      SIO_DATA(stream, &mapsize, 1)
        for(mapit = mcmap.begin(); mapit!=mcmap.end(); ++mapit)
          {
            //MCParticle*
            SIO_PNTR(stream, &((*mapit).first)                         );
            // energy
            float energy = mapit->second / GeV;
            SIO_DATA(stream, &energy, 1);
//          SIO_DATA(stream, const_cast<float *>(&((*mapit).second)), 1);
          }
    }

  //
  // That's all folks!
  //
  return( SIO_BLOCK_SUCCESS );
}

// ----------------------------------------------------------------------------
// Return SIO version number.
// ----------------------------------------------------------------------------

#define SIO_CALORIMETER_MAJOR 1
#define SIO_CALORIMETER_MINOR 0

unsigned int CalorimeterSIO::version()
{ return( SIO_VERSION_ENCODE( SIO_CALORIMETER_MAJOR,
                              SIO_CALORIMETER_MINOR ) ); }

---