pdune::RawWaveformDump Class Reference

Inheritance diagram for pdune::RawWaveformDump:

Public Member Functions
	RawWaveformDump (fhicl::ParameterSet const &p)
	RawWaveformDump (RawWaveformDump const &)=delete
	RawWaveformDump (RawWaveformDump &&)=delete
RawWaveformDump &	operator= (RawWaveformDump const &)=delete
RawWaveformDump &	operator= (RawWaveformDump &&)=delete
void	analyze (art::Event const &e) override
void	reconfigure (fhicl::ParameterSet const &p)
void	beginJob () override
void	endJob () override
Public Member Functions inherited from art::EDAnalyzer
	EDAnalyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDAnalyzer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Analyzer
virtual	~Analyzer () noexcept
	Analyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	Analyzer (Table< Config > const &config)
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Observer
	~Observer () noexcept
	Observer (Observer const &)=delete
	Observer (Observer &&)=delete
Observer &	operator= (Observer const &)=delete
Observer &	operator= (Observer &&)=delete
void	registerProducts (ProductDescriptions &, ModuleDescription const &)
void	fillDescriptions (ModuleDescription const &)
fhicl::ParameterSetID	selectorConfig () const
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Attributes
std::string	fDumpWaveformsFileName
std::string	fSimulationProducerLabel
	The name of the producer that tracked simulated particles through the detector. More...
std::string	fDigitModuleLabel
	module that made digits More...
std::string	fSelectGenLabel
std::string	fSelectProcID
int	fSelectPDGCode
std::string	fPlaneToDump
double	fMinParticleEnergyGeV
double	fMinEnergyDepositedMeV
int	fMinNumberOfElectrons
int	fMaxNumberOfElectrons
art::ServiceHandle< geo::Geometry >	fgeom
art::ServiceHandle< cheat::ParticleInventoryService >	PIS
std::default_random_engine	rndm_engine
c2numpy_writer	npywriter

Additional Inherited Members
Public Types inherited from art::EDAnalyzer
using	WorkerType = WorkerT< EDAnalyzer >
using	ModuleType = EDAnalyzer
Protected Member Functions inherited from art::Observer
std::string const &	processName () const
bool	wantAllEvents () const noexcept
bool	wantEvent (ScheduleID id, Event const &e) const
Handle< TriggerResults >	getTriggerResults (Event const &e) const
	Observer (fhicl::ParameterSet const &config)
	Observer (std::vector< std::string > const &select_paths, std::vector< std::string > const &reject_paths, fhicl::ParameterSet const &config)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 56 of file RawWaveformDump_module.cc.

Constructor & Destructor Documentation

pdune::RawWaveformDump::RawWaveformDump ( fhicl::ParameterSet const &  p )

explicit

Definition at line 132 of file RawWaveformDump_module.cc.

  : EDAnalyzer{p}
{
  this->reconfigure(p);
}

pdune::RawWaveformDump::RawWaveformDump ( RawWaveformDump const &  )

delete

pdune::RawWaveformDump::RawWaveformDump ( RawWaveformDump &&  )

delete

Member Function Documentation

void pdune::RawWaveformDump::analyze ( art::Event const &  e )

overridevirtual

Implements art::EDAnalyzer.

Definition at line 221 of file RawWaveformDump_module.cc.

{
  cout << "Event "
       << " " << evt.id().run()
       << " " << evt.id().subRun()
       << " " << evt.id().event() << endl;

  // ... Read in the digit List object(s).
  auto digitVecHandle = evt.getHandle< std::vector<raw::RawDigit> >(fDigitModuleLabel);

  if (!digitVecHandle || !digitVecHandle->size())  return;

  // ... Use the handle to get a particular (0th) element of collection.
  art::Ptr<raw::RawDigit> digitVec0(digitVecHandle, 0);
  unsigned int dataSize = digitVec0->Samples(); //size of raw data vectors
  if (dataSize!=6000){
    std::cout << "!!!!! Bad dataSize: " << dataSize << std::endl;
    return;
  }

  std::vector<short> rawadc(dataSize);  // vector to hold uncompressed adc values later

  // ... Build a map from channel number -> rawdigitVec
  std::map< raw::ChannelID_t, art::Ptr<raw::RawDigit> > rawdigitMap;
  raw::ChannelID_t chnum = raw::InvalidChannelID;       // channel number
  for ( size_t rdIter = 0; rdIter < digitVecHandle->size(); ++rdIter ) {
    art::Ptr<raw::RawDigit> digitVec(digitVecHandle, rdIter);
    chnum = digitVec->Channel();
    if(chnum==raw::InvalidChannelID)continue;
    rawdigitMap[chnum]=digitVec;
  }

  // ... Read in MC particle list
  auto particleHandle = evt.getHandle< std::vector<simb::MCParticle> >(fSimulationProducerLabel);
  if (!particleHandle){
    throw cet::exception("AnalysisExample")
      << " No simb::MCParticle objects in this event - "
      << " Line " << __LINE__ << " in file " << __FILE__ << std::endl;
  }

  // ... Read in sim channel list
  auto simChannelHandle = evt.getValidHandle<std::vector<sim::SimChannel>>(fSimulationProducerLabel);

  if (!simChannelHandle->size())  return;

  // ... Create a map of track IDs to generator labels
  //Get a list of generator names.
  auto mcHandles = evt.getMany<std::vector<simb::MCTruth>>();
  std::vector< std::pair<int, std::string>> track_id_to_label;

  for( auto const& mcHandle : mcHandles ){
    const std::string& sModuleLabel = mcHandle.provenance()->moduleLabel();
    art::FindManyP<simb::MCParticle> findMCParts(mcHandle, evt, "largeant");
    std::vector<art::Ptr<simb::MCParticle> > mcParts = findMCParts.at(0);
    for( const art::Ptr<simb::MCParticle> ptr : mcParts){
      int track_id = ptr->TrackId();
      gf->add(track_id, sModuleLabel);
    }
  }

  std::string dummystr6="none  ";
  std::string dummystr7="none   ";

  // .. create a channel number to trackid-wire signal info map
  std::map<raw::ChannelID_t,std::map<int,WireSigInfo>>Ch2TrkWSInfoMap;

  // .. create a track ID to vector of channel numbers (in w/c this track deposited energy) map
  std::map<int,std::vector<raw::ChannelID_t>>Trk2ChVecMap;

  auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService>()->DataFor(evt);
  // ... Loop over simChannels
  for ( auto const& channel : (*simChannelHandle) ){

    // .. get simChannel channel number
    const raw::ChannelID_t ch1 = channel.Channel();
    if(ch1==raw::InvalidChannelID)continue;
    if(geo::PlaneGeo::ViewName(fgeom->View(ch1))!=fPlaneToDump[0])continue;

    bool selectThisChannel=false;

    // .. create a track ID to wire signal info map
    std::map<int,WireSigInfo>Trk2WSInfoMap;

    // ... Loop over all ticks with ionization energy deposited
    auto const& timeSlices = channel.TDCIDEMap();
    for ( auto const& timeSlice : timeSlices ){

      auto const& energyDeposits = timeSlice.second;
      auto const tpctime = timeSlice.first;
      unsigned int tdctick = static_cast<unsigned int>(clockData.TPCTDC2Tick(double(tpctime)));
      if(tdctick!=tpctime)std::cout << "tpctime: " << tpctime << ", tdctick: " << tdctick << std::endl;
      if(tdctick<0||tdctick>(dataSize-1))continue;

      // ... Loop over all energy depositions in this tick
      for ( auto const& energyDeposit : energyDeposits ){

        if (!energyDeposit.trackID) continue;
        int trkid = energyDeposit.trackID;
        simb::MCParticle particle = PIS->TrackIdToMotherParticle(trkid);
        std::cout << energyDeposit.trackID << " " << trkid << " " << particle.TrackId() << std::endl;

        // .. ignore this energy deposition if incident particle energy below some threshold
        if ( particle.E() < fMinParticleEnergyGeV ) continue;

        int eve_id = PIS->TrackIdToEveTrackId(trkid);
        if (!eve_id) continue;
        std::string genlab = gf->get_gen(eve_id);

        if (Trk2WSInfoMap.find(trkid) == Trk2WSInfoMap.end()){
          WireSigInfo wsinf;
          wsinf.pdgcode = particle.PdgCode();
          wsinf.genlab = genlab;
          wsinf.procid = particle.Process();
          wsinf.tdcmin=dataSize-1;
          wsinf.tdcmax=0;
          wsinf.edep=0.;
          wsinf.numel=0;
          Trk2WSInfoMap.insert(std::pair<int,WireSigInfo>(trkid,wsinf));
        }
        if (tdctick<Trk2WSInfoMap.at(trkid).tdcmin)Trk2WSInfoMap.at(trkid).tdcmin=tdctick;
        if (tdctick>Trk2WSInfoMap.at(trkid).tdcmax)Trk2WSInfoMap.at(trkid).tdcmax=tdctick;
        Trk2WSInfoMap.at(trkid).edep +=energyDeposit.energy;
        Trk2WSInfoMap.at(trkid).numel+=energyDeposit.numElectrons;
      }
    }

    if(!Trk2WSInfoMap.empty()){
      for(std::pair<int,WireSigInfo> itmap : Trk2WSInfoMap){
        if (fSelectGenLabel!="ANY"){
          if(itmap.second.genlab!=fSelectGenLabel)continue;
        }
        if (fSelectProcID!="ANY"){
          if(itmap.second.procid!=fSelectProcID)continue;
        }
        if (fSelectPDGCode!=0){
          if(itmap.second.pdgcode!=fSelectPDGCode)continue;
        }
        itmap.second.genlab.resize(6,' ');
        itmap.second.procid.resize(7,' ');
        if(itmap.second.numel>=fMinNumberOfElectrons && itmap.second.edep>=fMinEnergyDepositedMeV){
          if(fMaxNumberOfElectrons>=0 && itmap.second.numel>=fMaxNumberOfElectrons){
            continue;
          } else {
            int trkid = itmap.first;
            if (Trk2ChVecMap.find(trkid) == Trk2ChVecMap.end()){
              std::vector<raw::ChannelID_t>chvec;
              Trk2ChVecMap.insert(std::pair<int,std::vector<raw::ChannelID_t>>(trkid,chvec));
            }
            Trk2ChVecMap.at(trkid).push_back(ch1);
            selectThisChannel=true;
          }
        }
      } // loop over Trk2WSinfoMap
      if(selectThisChannel){
        Ch2TrkWSInfoMap.insert(std::pair<raw::ChannelID_t,std::map<int,WireSigInfo>>(ch1,Trk2WSInfoMap));
      }
    } // if Trk2WSInfoMap not empty

  } // loop over SimChannels

  // ... Now write out the signal waveforms for each track
  if(!Trk2ChVecMap.empty()){
    for(auto const& ittrk : Trk2ChVecMap){
      std::uniform_int_distribution<int> rndm_dist(0,ittrk.second.size()-1);
      int i = rndm_dist(rndm_engine); // randomly select one channel with a signal from this particle
      chnum=ittrk.second[i];

      std::map<raw::ChannelID_t,std::map<int,WireSigInfo>>::iterator itchn;
      itchn = Ch2TrkWSInfoMap.find(chnum);
      if( itchn != Ch2TrkWSInfoMap.end() ){
        auto search = rawdigitMap.find(chnum);
        if ( search == rawdigitMap.end() ) continue;
        art::Ptr<raw::RawDigit> rawdig = (*search).second;
        raw::Uncompress(rawdig->ADCs(), rawadc, rawdig->GetPedestal(), rawdig->Compression());

        c2numpy_uint32(&npywriter, evt.id().event());
        c2numpy_uint32(&npywriter, chnum);
        c2numpy_string(&npywriter, geo::PlaneGeo::ViewName(fgeom->View(chnum)).c_str());
        c2numpy_uint16(&npywriter, itchn->second.size());  // size of Trk2WSInfoMap, or number of peaks

        // .. write out info for each peak
        unsigned int icnt=0;
        for ( auto const& it : itchn->second ){
          c2numpy_int32(&npywriter,   it.first);                 // trackid
          c2numpy_int32(&npywriter,   it.second.pdgcode);        // pdgcode
          c2numpy_string(&npywriter,  it.second.genlab.c_str()); // genlab
          c2numpy_string(&npywriter,  it.second.procid.c_str()); // procid
          c2numpy_float32(&npywriter, it.second.edep);           // edepo
          c2numpy_uint32(&npywriter,  it.second.numel);          // numelec
          c2numpy_uint16(&npywriter,  it.second.tdcmin);         // stck1
          c2numpy_uint16(&npywriter,  it.second.tdcmax);         // stc2
          icnt++;
          if(icnt==5)break;
        }

        // .. pad with 0's if number of peaks less than 5
        for ( unsigned int i=icnt; i<5; ++i){
          c2numpy_int32(&npywriter, 0);
          c2numpy_int32(&npywriter, 0);
          c2numpy_string(&npywriter, dummystr6.c_str());
          c2numpy_string(&npywriter, dummystr7.c_str());
          c2numpy_float32(&npywriter, 0.);
          c2numpy_uint32(&npywriter, 0);
          c2numpy_uint16(&npywriter, 0);
          c2numpy_uint16(&npywriter, 0);
        }

        // .. now write out the ADC values
        for ( unsigned int itck=0; itck<dataSize; ++itck ){
          rawadc[itck] -= rawdig->GetPedestal();
          c2numpy_int16(&npywriter, rawadc[itck]);
        }
      }
    }
  }

}

void pdune::RawWaveformDump::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 160 of file RawWaveformDump_module.cc.

{
    std::random_device rndm_device;     // this will give us our seed
    rndm_engine.seed(rndm_device());

    c2numpy_init(&npywriter, fDumpWaveformsFileName, 50000);
    c2numpy_addcolumn(&npywriter, "evt",   C2NUMPY_UINT32);
    c2numpy_addcolumn(&npywriter, "chan",  C2NUMPY_UINT32);
    c2numpy_addcolumn(&npywriter, "view",  (c2numpy_type)((int)C2NUMPY_STRING + 1) );
    c2numpy_addcolumn(&npywriter, "ntrk",  C2NUMPY_UINT16);

    for(unsigned int i=0; i<5; i++){
      std::ostringstream name;

      name.str("");
      name << "tid" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_INT32);

      name.str("");
      name << "pdg" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_INT32);

      name.str("");
      name << "gen" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), (c2numpy_type)((int)C2NUMPY_STRING + 6));

      name.str("");
      name << "pid" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), (c2numpy_type)((int)C2NUMPY_STRING + 7));

      name.str("");
      name << "edp" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_FLOAT32);

      name.str("");
      name << "nel" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_UINT32);

      name.str("");
      name << "sti" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_UINT16);

      name.str("");
      name << "stf" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_UINT16);
    }

    for(unsigned int i=0; i<6000; i++){
      std::ostringstream name;
      name << "tck_" << i;
      c2numpy_addcolumn(&npywriter, name.str().c_str(), C2NUMPY_INT16);
    }
}

void pdune::RawWaveformDump::endJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 215 of file RawWaveformDump_module.cc.

{
    c2numpy_close(&npywriter);
}

RawWaveformDump& pdune::RawWaveformDump::operator= ( RawWaveformDump const &  )

delete

RawWaveformDump& pdune::RawWaveformDump::operator= ( RawWaveformDump &&  )

delete

void pdune::RawWaveformDump::reconfigure

(

fhicl::ParameterSet const &

p

)

Definition at line 139 of file RawWaveformDump_module.cc.

{
  fDumpWaveformsFileName = p.get<std::string>("DumpWaveformsFileName","dumpwaveforms");

  fSimulationProducerLabel = p.get<std::string>("SimulationProducerLabel", "largeant");
  fDigitModuleLabel = p.get<std::string>("DigitModuleLabel", "daq");

  fSelectGenLabel = p.get<std::string>("SelectGenLabel","ANY");
  fSelectProcID   = p.get<std::string>("SelectProcID",  "ANY");
  fSelectPDGCode  = p.get<int>("SelectPDGCode", 0);

  fPlaneToDump = p.get<std::string>("PlaneToDump", "U");
  fMinParticleEnergyGeV = p.get<double>("MinParticleEnergyGeV",0.);
  fMinEnergyDepositedMeV = p.get<double>("MinEnergyDepositedMeV",0.);
  fMinNumberOfElectrons = p.get<int>("MinNumberOfElectrons",1000);
  fMaxNumberOfElectrons = p.get<int>("MaxNumberOfElectrons",100000);

  return;
}

Member Data Documentation

std::string pdune::RawWaveformDump::fDigitModuleLabel

private

module that made digits

Definition at line 81 of file RawWaveformDump_module.cc.

std::string pdune::RawWaveformDump::fDumpWaveformsFileName

private

Definition at line 78 of file RawWaveformDump_module.cc.

art::ServiceHandle<geo::Geometry> pdune::RawWaveformDump::fgeom

private

Definition at line 91 of file RawWaveformDump_module.cc.

int pdune::RawWaveformDump::fMaxNumberOfElectrons

private

Definition at line 90 of file RawWaveformDump_module.cc.

double pdune::RawWaveformDump::fMinEnergyDepositedMeV

private

Definition at line 88 of file RawWaveformDump_module.cc.

int pdune::RawWaveformDump::fMinNumberOfElectrons

private

Definition at line 89 of file RawWaveformDump_module.cc.

double pdune::RawWaveformDump::fMinParticleEnergyGeV

private

Definition at line 87 of file RawWaveformDump_module.cc.

std::string pdune::RawWaveformDump::fPlaneToDump

private

Definition at line 86 of file RawWaveformDump_module.cc.

std::string pdune::RawWaveformDump::fSelectGenLabel

private

Definition at line 83 of file RawWaveformDump_module.cc.

int pdune::RawWaveformDump::fSelectPDGCode

private

Definition at line 85 of file RawWaveformDump_module.cc.

std::string pdune::RawWaveformDump::fSelectProcID

private

Definition at line 84 of file RawWaveformDump_module.cc.

std::string pdune::RawWaveformDump::fSimulationProducerLabel

private

The name of the producer that tracked simulated particles through the detector.

Definition at line 80 of file RawWaveformDump_module.cc.

c2numpy_writer pdune::RawWaveformDump::npywriter

private

Definition at line 97 of file RawWaveformDump_module.cc.

art::ServiceHandle<cheat::ParticleInventoryService> pdune::RawWaveformDump::PIS

private

Definition at line 92 of file RawWaveformDump_module.cc.

std::default_random_engine pdune::RawWaveformDump::rndm_engine

private

Definition at line 95 of file RawWaveformDump_module.cc.

---

The documentation for this class was generated from the following file:

• protoduneana/protoduneana/DataDump/RawWaveformDump_module.cc