HepMC2DataLoader.cpp

// (c) Copyright 2022 CERN for the benefit of the LHCb Collaboration.
//
// This software is distributed under the terms of the GNU General Public
// Licence version 3 (GPL Version 3), copied verbatim in the file "LICENCE".
//
// In applying this licence, CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
// STL
#include <iostream>
#include <algorithm>
 
// Local
#include "SQLamarr/HepMC2DataLoader.h"
#include "SQLamarr/db_functions.h"
#include "SQLamarr/preprocessor_symbols.h"
 
namespace SQLamarr
{
  //==========================================================================
  // load
  //==========================================================================
  void HepMC2DataLoader::load (
      const std::string& file_path, 
      size_t run_number, 
      size_t evt_number
      )
  {
    begin_transaction();
    const int ds_id = insert_event(file_path, run_number, evt_number);
 
    HepMC3::ReaderAsciiHepMC2 reader(file_path.c_str());
    while ( !reader.failed() ) 
    {
      HepMC3::GenEvent evt(HepMC3::Units::MEV, HepMC3::Units::MM);
      reader.read_event(evt);
 
      auto pos = evt.event_pos();
      const int event_id = insert_collision(
          ds_id,
          evt.event_number(),
          pos.t(),
          pos.x(),
          pos.y(),
          pos.z()
          );
 
      std::vector<int> pvs;
      for (auto& bp: evt.beams())
        if (bp->end_vertex())
          pvs.push_back(bp->end_vertex()->id());
 
      std::unordered_map<int, int> vtxid_mapping;
      for (auto vertex: evt.vertices())
        vtxid_mapping[vertex->id()] = insert_vertex(
              event_id,
              vertex->id(),
              vertex->status(),
              vertex->position().t(),
              vertex->position().x(),
              vertex->position().y(),
              vertex->position().z(),
              (std::find(pvs.begin(), pvs.end(), vertex->id()) != pvs.end())
            );
 
      for (auto particle: evt.particles())
      {
        auto pv = particle->production_vertex();
        auto ev = particle->end_vertex();
 
 
        insert_particle(
              event_id,
              particle->id(),
              (pv ? vtxid_mapping[pv->id()] : LAMARR_BAD_INDEX),
              (ev ? vtxid_mapping[ev->id()] : LAMARR_BAD_INDEX),
              particle->pid(),
              particle->status(),
              particle->momentum().e(),
              particle->momentum().px(),
              particle->momentum().py(),
              particle->momentum().pz(),
              particle->generated_mass()
            );
      }
    }
 
    end_transaction();
  }
}