Analyzer.cc

Go to the documentation of this file.

#include <evt/Event.h>
#include <sevt/SEvent.h>
#include <sevt/PMTSimData.h>
#include <utl/ErrorLogger.h>
#include <utl/Branch.h>
#include <utl/TimeDistribution.h>
#include <utl/Particle.h>
#include <utl/ParticlePropertiesFactory.h>
#include <fwk/CentralConfig.h>

#include "Analyzer.h"

using namespace std;
using namespace evt;
using namespace sevt;
using namespace fwk;    
using namespace utl;


namespace AnalyzerNS {

  VModule::ResultFlag
  Analyzer::Init()
  {
    // sneak-peek into other module's config
    Branch topB = CentralConfig::GetInstance()->GetTopBranch("ParticleInjector");

    const int stationId = topB.GetChild("StationId").Get<int>();

    ostringstream info;
    info << "\n  StationId = " << stationId << '\n';

    const int particleType = topB.GetChild("NumberOfParticles").GetChild("Type").Get<int>();
    const string particleName = ParticlePropertiesFactory::Create(particleType)->GetName();
    info << "  Particle = " << particleType << " (" << particleName << ")\n";

    const double energy = topB.GetChild("Energy").GetChild("Fixed").Get<double>();
    info << "  Energy = " << energy/GeV << " GeV\n";

    ostringstream filename("PE_");
    filename << particleName << '_' << energy/GeV << "GeV_3x800ns.txt";

    fOutput.open(filename.str().c_str());

    info << "  Filename = " << filename.str();
    INFO(info);

    return eSuccess;
  }


  VModule::ResultFlag
  Analyzer::Run(evt::Event& event)
  {
    if (!event.HasSEvent()) {
      ERROR("Event should have SEvent!");
      return eFailure;
    }

    const SEvent& sEvent = event.GetSEvent();

    int nStations = 0;

    for (const auto& st : sEvent.StationsRange()) {

      int bins800ns[3] = { 0 };
      int bins800nsSSD[3] = { 0 };

      for (const auto& pmt : st.PMTsRange()) {

        if (!pmt.HasSimData())
          continue;

        const auto& sim = pmt.GetSimData();
        if (!sim.HasPETimeDistribution())
          continue;

        cerr << "PE " << st.GetId() << ' ' << pmt.GetId() << "->";

        int peTotal = 0;

        const auto& peTD = sim.GetPETimeDistribution();

        for (const auto& tb : peTD.SparseRange()) {
          const int t = tb.get<0>();
          const int npe = tb.get<1>();
          const int bin =
            t < 800 ? 0 :
            t < 1600 ? 1 : 2;
          if (pmt.GetId() < 3)
            bins800ns[bin] += npe;
          else
            bins800nsSSD[bin] += npe;
          peTotal += npe;
          cerr << ' ' << t << ':' << npe;
        }

        cerr << " total: " << peTotal << '\n';

      }

      fOutput << bins800ns[0] << ' ' << bins800ns[1] << ' ' << bins800ns[2] << "  "
              << bins800nsSSD[0] << ' ' << bins800nsSSD[1] << ' ' << bins800nsSSD[2]
              << '\n';

      ++nStations;
    }

    cerr << "** nStations " << nStations << endl;

    return eSuccess;
  }


  VModule::ResultFlag
  Analyzer::Finish()
  {
    fOutput.close();
    return eSuccess;
  }

}