add first information to digitizer spec and test reader

Author: maciacco

Steer/DigitizerWorkflow/src/IOTOFDigitizerSpec.cxx

@@ -36,6 +36,8 @@
 #include <memory>
 #include <string>
 
+using namespace o2::framework;
+
 namespace o2::iotof
 {
 
@@ -49,14 +51,100 @@ class IOTOFDPLDigitizerTask : o2::base::BaseDPLDigitizer
 
   void initDigitizerTask(framework::InitContext& ic) override
   {
+    mDisableQED = ic.options().get<bool>("disable-qed");
+    mDigits.resize(mLayers);
+    mROFRecords.resize(mLayers);
+    mROFRecordsAccum.resize(mLayers);
+    mLabels.resize(mLayers);
+    mLabelsAccum.resize(mLayers);
   }
 
   void run(framework::ProcessingContext& pc)
   {
+    if (mFinished) {
+      return;
+    }
+    mFirstOrbitTF = pc.services().get<o2::framework::TimingInfo>().firstTForbit;
+    const o2::InteractionRecord firstIR(0, mFirstOrbitTF);
+
+    // read collision context from input
+    auto context = pc.inputs().get<o2::steer::DigitizationContext*>("collisioncontext");
+    context->initSimChains(mID, mSimChains);
+    const bool withQED = context->isQEDProvided() && !mDisableQED;
+    auto& timesview = context->getEventRecords();
+    LOG(info) << "GOT " << timesview.size() << " COLLISION TIMES";
+    LOG(info) << "SIMCHAINS " << mSimChains.size();
+
+    // if there is nothing to do ... return
+    if (timesview.empty()) {
+      return;
+    }
+
+    TStopwatch timer;
+    timer.Start();
+    LOG(info) << " CALLING TF3 DIGITIZATION ";
+
+    auto& eventParts = context->getEventParts(withQED);
+    uint64_t nDigits{0};
+    for (uint32_t iLayer = 0; iLayer < static_cast<uint32_t>(mLayers); ++iLayer) {
+      mDigits[iLayer].clear();
+      // mROFRecords[iLayer].clear();
+      // mROFRecordsAccum[iLayer].clear();
+      if (mWithMCTruth) {
+        // mLabels[iLayer].clear();
+        // mLabelsAccum[iLayer].clear();
+        // mMC2ROFRecordsAccum[iLayer].clear();
+      }
+
+      mDigitizer.setDigits(&mDigits[iLayer]);
+      // mDigitizer.setROFRecords(&mROFRecords[iLayer]);
+      // mDigitizer.setMCLabels(&mLabels[iLayer]);
+
+      for (int collID = 0; collID < timesview.size(); ++collID) {
+        for (const auto& part : eventParts[collID]) {
+
+          // get the hits for this event and this source
+          mHits.clear();
+          context->retrieveHits(mSimChains, o2::detectors::SimTraits::DETECTORBRANCHNAMES[mID][0].c_str(), part.sourceID, part.entryID, &mHits);
+
+          if (mHits.size() > 0) {
+            LOG(debug) << "For collision " << collID << " eventID " << part.entryID << " found " << mHits.size() << " hits ";
+            // mDigitizer.process(&mHits, part.entryID, part.sourceID, layer); // call actual digitization procedure
+          }
+        }
+      }
+
+    }
+
+    timer.Stop();
+    LOG(info) << "Digitization took " << timer.CpuTime() << "s";
+    LOG(info) << "Produced " << nDigits << " digits";
+
+    // we should be only called once; tell DPL that this process is ready to exit
+    pc.services().get<ControlService>().readyToQuit(QuitRequest::Me);
+
+    mFinished = true;
   }
 
  private:
+  bool mDisableQED = false;
   bool mWithMCTruth{true};
+  bool mFinished{false};
+  int mLayers{2};
+  unsigned long mFirstOrbitTF = 0x0;
+  const o2::detectors::DetID mID{o2::detectors::DetID::TF3};
+  const o2::header::DataOrigin mOrigin{o2::header::gDataOriginTF3};
+  o2::iotof::Digitizer mDigitizer{};
+  std::vector<std::vector<o2::iotof::Digit>> mDigits{};
+  std::vector<std::vector<o2::itsmft::ROFRecord>> mROFRecords{};
+  std::vector<std::vector<o2::itsmft::ROFRecord>> mROFRecordsAccum{};
+  std::vector<o2::itsmft::Hit> mHits{};
+  std::vector<o2::itsmft::Hit>* mHitsP{&mHits};
+  std::vector<o2::dataformats::MCTruthContainer<o2::MCCompLabel>> mLabels{};
+  std::vector<o2::dataformats::MCTruthContainer<o2::MCCompLabel>> mLabelsAccum{};
+  std::vector<std::vector<o2::itsmft::MC2ROFRecord>> mMC2ROFRecordsAccum{};
+  std::vector<TChain*> mSimChains{};
+  o2::parameters::GRPObject::ROMode mROMode = o2::parameters::GRPObject::PRESENT; // readout mode
 };
 
 std::vector<o2::framework::OutputSpec> makeOutChannels(o2::header::DataOrigin detOrig, bool mctruth)