ReadoutSim.cxx

Go to the documentation of this file.

#include "ReadoutSim/ReadoutSim.h"
#include "ReadoutSim/Simulator.h"
#include "RawData/DAQHeader.h"
#include "PhotonTransport/PhotonSignal.h"
#include "JobControl/Exception.h"
#include <TRandom3.h>
#include <string>
#include "Geometry/CellUniqueId.h"
#include "Geometry/PlaneGeo.h"
#include "Geometry/Geometry.h"
#include "RawData/RawDigit.h"
#include "Simulation/HitProfile.h"
#include "Simulation/MCDigitInfo.h"

#include "ReadoutSim/MsgLevel.h"

//#include "Simulation/FLSHit.h"
//#include "Simulation/FLSHitList.h"

#include "PhotonTransport/PhotonSignal.h"

using namespace rsim;

MODULE_DECL(ReadoutSim);

ClassImp(ReadoutSim);

//----------------------------------------------------------------------

ReadoutSim::ReadoutSim(const char* version) :
  jobc::Module("ReadoutSim")
{ 
  this->SetWatch("ReadoutSimConfig","default");
}

//----------------------------------------------------------------------

ReadoutSim::~ReadoutSim()
{
  this->RemoveAllWatches();
}

void ReadoutSim::Update(const cfg::Config& c)
{
  
  //choose which mySim to use
  
  //std::cout <<"ReadoutSim::Update called \n";
  std::string simtype;
  c("SimType").Get(simtype);
  c("MessageLevel").Get(msglevel);
  c("StoreMCInfo").Get(storemcinfo);
  
  ChooseSim(simtype);
  
  //vector<std::string> simconfig;
  //c("SimConfig").Get(simconfig);
  //mySim->Configure(simconfig);
  
  mySim->Configure(c);
  
}

void ReadoutSim::ChooseSim(std::string simtype="")
{
  
  //delete(mySim);
  
  if(!simtype.compare("SimpleReadout"))
  {
    mySim=new SimpleReadout();
  }
  else if(!simtype.compare("FarReadout"))
  {
    mySim=new FarReadout();
  } 
  else
  {
    mySim=new Simulator();
  }
}

jobc::Result ReadoutSim::Reco(edm::EventHandle& evt)
{
  
  if(msglevel>=DEBUG)   printf("ReadoutSim Reco called\n");
  if(mySim==NULL){
    if(msglevel>=FATAL)std::cout<<"Major Error: Simulator never chosen....\n";
    return jobc::kFailed;
  }
  
  /*
    vector<const sim::FLSHitList*> hitlist(0);
    assert_jobc(evt.DetSim().Get("./",hitlist),
              "No FLSHitLists found in DetSim() folder!");
  */
  
  vector<const photrans::PhotonSignal*> hitlist(0);
  // assert_jobc(evt.Raw().Get("./",hitlist),
  //      "No PhotonSignals found in DetSim() folder!");
  
  try{
    evt.Raw().Get("./",hitlist);
  }catch(...){return jobc::kFailed;}
  
  std::vector<rawdata::RawDigit> rawdigitout(0);
  std::vector<sim::MCDigitInfo> mcdigitout(0);
  
  // take the digit list and pass it to the simulator, which returns raw digits that have passed through the simulator
  if(msglevel>=INFO)  std::cout <<"using simulator "<<*mySim<<"\n";     
  if(msglevel>=INFO)  std::cout<<"evt with "<<hitlist.size()<<" events\n";
  
  std::vector<const rawdata::DAQHeader*> header;
  short int det = rawdata::kFar;
  try{ evt.DAQ().Get("./",header); }
  catch(edm::Exception e){
    std::cerr << "Error retrieving daq header, while looking "<<
      "in ReadoutSim::Reco(), using default fardet geomety" << std::endl;
  }
  if(header.size() > 0) det = header[0]->DetId();

  mySim->SetGeom( evt.Header().Run(), det);

  rawdigitout = mySim->Simulate(hitlist, mcdigitout);
  
  //remove the raw digits from the root file from before the readout sim
  //    evt.Raw().ClearArrays();
  
  if(msglevel>=INFO) printf("   %d raw hits after simulation\n",rawdigitout.size());
  
  //put the new rawdigits
  for (unsigned int i=0; i<rawdigitout.size(); ++i)
    evt.Raw().Put(rawdigitout[i],"");

  // store MC info
  if(storemcinfo==1) {
    if (evt.Raw().GetFolder("./mcinfo")==0) 
      evt.Raw().MakeFolder("./mcinfo");

    /* cause bad_alloc error?
    for (unsigned int i=0; i<mcdigitout.size(); ++i)
      evt.Raw().Put(mcdigitout[i],"./mcinfo");
    */
  }    

  return jobc::kPassed;
}

---