PlaybackRawDataProvider.cc

Go to the documentation of this file.

//
// PlaybackRawDataProvider
// -----------------------
//
//            21/09/2006 Philipp Schieferdecker <philipp.schieferdecker@cern.ch>


#include "EventFilter/Playback/interface/PlaybackRawDataProvider.h"
#include "DataFormats/FEDRawData/interface/FEDNumbering.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h"
#include <cstring>
#include <iostream>

using namespace std;
using namespace edm;

// initialize static data members

PlaybackRawDataProvider* PlaybackRawDataProvider::instance_=nullptr;


// construction/destruction

//______________________________________________________________________________
PlaybackRawDataProvider::PlaybackRawDataProvider(const ParameterSet& iConfig)
  : queueSize_(0)
  , eventQueue_(nullptr)
  , runNumber_(nullptr)
  , evtNumber_(nullptr)
  , count_(0)
  , writeIndex_(0)
  , readIndex_(0)
  , freeToEof_(false)
  , filesClosed_(false)
  , destroying_(false)
{
  queueSize_=iConfig.getUntrackedParameter<unsigned int>("queueSize",32);
  sem_init(&lock_,0,1);
  sem_init(&writeSem_,0,queueSize_);
  sem_init(&readSem_,0,0);
  runNumber_ =new unsigned int[queueSize_];
  evtNumber_ =new unsigned int[queueSize_];
  eventQueue_=new FEDRawDataCollection*[queueSize_];
  for (unsigned int i=0;i<queueSize_;i++) eventQueue_[i]=nullptr;
  edm::LogInfo("PbImpl") << "Created Concrete RawData Provider 0x"<< hex << (unsigned long) this << dec << endl;
  instance_=this;
}


//______________________________________________________________________________
PlaybackRawDataProvider::~PlaybackRawDataProvider()
{
  if (nullptr!=runNumber_) delete [] runNumber_;
  if (nullptr!=evtNumber_) delete [] evtNumber_;
  if (nullptr!=eventQueue_) {
    for (unsigned int i=0;i<queueSize_;i++)
      if (nullptr!=eventQueue_[i]) delete eventQueue_[i];
    delete [] eventQueue_;
  }
  edm::LogInfo("PbImpl") << "Destroyed Concrete RawData Provider 0x"<< hex << (unsigned long) this << dec << endl;
  instance_=nullptr;

  destroying_=true;
  postReadSem();
  postWriteSem();
  unlock(); 
  sem_destroy(&lock_);
  sem_destroy(&writeSem_);
  sem_destroy(&readSem_);
  usleep(10000);
}


// implementation of member functions

//______________________________________________________________________________
void PlaybackRawDataProvider::analyze(const Event& iEvent,
                      const EventSetup& iSetup)
{
  if(freeToEof_) return;
  waitWriteSem();
  if(freeToEof_) return;
  runNumber_[writeIndex_]=iEvent.id().run();
  evtNumber_[writeIndex_]=iEvent.id().event();

  Handle<FEDRawDataCollection> pRawData;
  iEvent.getByLabel("rawDataCollector", pRawData);
  
  if (!pRawData.isValid()) {
    edm::LogError("InvalidHandle")<<"no raw data found!"<<endl;
    return;
  }
  
  
  // copy the raw data collection into rawData_, retrievable via getFEDRawData()
  assert(nullptr==eventQueue_[writeIndex_]);
  eventQueue_[writeIndex_]=new FEDRawDataCollection();
  for (unsigned int i=0;i<(unsigned int)FEDNumbering::MAXFEDID+1;i++) {
    unsigned int fedSize=pRawData->FEDData(i).size();
    if (fedSize>0) {
      FEDRawData& fedData=eventQueue_[writeIndex_]->FEDData(i);
      fedData.resize(fedSize);
      memcpy(fedData.data(),pRawData->FEDData(i).data(),fedSize);
    }
  }
  
  lock();
  writeIndex_=(writeIndex_+1)%queueSize_;
  count_++;
  unlock();
  
  postReadSem();

  return;
}


//______________________________________________________________________________
void PlaybackRawDataProvider::beginJob()
{
  
}

//______________________________________________________________________________
void PlaybackRawDataProvider::endJob()
{
  edm::LogInfo("Summary")<<count_<<" events read."<<endl;
}

//______________________________________________________________________________
void PlaybackRawDataProvider::respondToCloseInputFile(edm::FileBlock const& fb)
{
  filesClosed_ = true;
}

//______________________________________________________________________________
FEDRawDataCollection* PlaybackRawDataProvider::getFEDRawData()
{
  FEDRawDataCollection* result = nullptr;
  waitReadSem();
  //do not read data if destructor is called
  if (destroying_) return nullptr;
  lock();
  result = eventQueue_[readIndex_];
  eventQueue_[readIndex_]=nullptr;
  readIndex_=(readIndex_+1)%queueSize_;
  unlock();
  
  postWriteSem();
  return result;
}


//______________________________________________________________________________
FEDRawDataCollection* PlaybackRawDataProvider::getFEDRawData(unsigned int& runNumber,
                                 unsigned int& evtNumber)
{

  FEDRawDataCollection* result = nullptr;
  waitReadSem();
  //do not read data if destructor is called
  if (destroying_) return nullptr;
  lock();
  runNumber=runNumber_[readIndex_];
  evtNumber=evtNumber_[readIndex_];
  result=eventQueue_[readIndex_];
  //  assert(0!=result);
  eventQueue_[readIndex_]=nullptr;
  readIndex_=(readIndex_+1)%queueSize_;

  unlock();
  
  postWriteSem();
  return result;
}


//______________________________________________________________________________
void PlaybackRawDataProvider::sem_print()
{
  lock();
  int wsem,rsem;
  sem_getvalue(&writeSem_,&wsem);
  sem_getvalue(&readSem_,&rsem);
  cout<<"sem_print():"
      <<" wsem="<<wsem
      <<" rsem="<<rsem
      <<" writeIndex="<<writeIndex_
      <<" readIndex="<<readIndex_
      <<endl;
  unlock();
}

void PlaybackRawDataProvider::setFreeToEof()
{ 
  //  cout << "  PlaybackRawDataProvider::setFreeToEof()" << endl;
  freeToEof_ = true; 
  //  cout << "  PlaybackRawDataProvider::setFreeToEof() call postReadSem" << endl;  
  postWriteSem(); 
}

bool PlaybackRawDataProvider::areFilesClosed()
{
  return filesClosed_;
}
// framework module implementation macro