Skip to content

Commit

Permalink
move things around
Browse files Browse the repository at this point in the history
  • Loading branch information
lobis committed Nov 30, 2023
1 parent aca1c84 commit a43bd90
Showing 1 changed file with 17 additions and 18 deletions.
35 changes: 17 additions & 18 deletions application/include/geant4/DataModel.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,23 @@ class G4Step;

namespace geant4::data {

using UserDefinedMap = std::map<std::size_t, std::string>;

template<std::size_t field_name, class BUILDER>
using RecordField = awkward::LayoutBuilder::Field<field_name, BUILDER>;

template<class... BUILDERS>
using RecordBuilder = awkward::LayoutBuilder::Record<UserDefinedMap, BUILDERS...>;

template<class PRIMITIVE, class BUILDER>
using ListOffsetBuilder = awkward::LayoutBuilder::ListOffset<PRIMITIVE, BUILDER>;

template<class PRIMITIVE, class BUILDER>
using IndexedBuilder = awkward::LayoutBuilder::IndexedOption<PRIMITIVE, BUILDER>;

template<class PRIMITIVE>
using NumpyBuilder = awkward::LayoutBuilder::Numpy<PRIMITIVE>;

enum Field : std::size_t {
runId,
eventId,
Expand Down Expand Up @@ -48,8 +65,6 @@ enum Field : std::size_t {
stepTrackKineticEnergy,
};

using UserDefinedMap = std::map<std::size_t, std::string>;

inline static const UserDefinedMap fieldToNameEvent = {
{Field::runId, "run_id"},
{Field::eventId, "event_id"},
Expand Down Expand Up @@ -84,21 +99,6 @@ inline static const UserDefinedMap fieldToNameStep = {
// {Field::stepVolume, "track.step.volume"}},
};

template<std::size_t field_name, class BUILDER>
using RecordField = awkward::LayoutBuilder::Field<field_name, BUILDER>;

template<class... BUILDERS>
using RecordBuilder = awkward::LayoutBuilder::Record<UserDefinedMap, BUILDERS...>;

template<class PRIMITIVE, class BUILDER>
using ListOffsetBuilder = awkward::LayoutBuilder::ListOffset<PRIMITIVE, BUILDER>;

template<class PRIMITIVE, class BUILDER>
using IndexedBuilder = awkward::LayoutBuilder::IndexedOption<PRIMITIVE, BUILDER>;

template<class PRIMITIVE>
using NumpyBuilder = awkward::LayoutBuilder::Numpy<PRIMITIVE>;

typedef unsigned int id;
using Builder = RecordBuilder<
RecordField<Field::runId, NumpyBuilder<id>>,
Expand Down Expand Up @@ -133,7 +133,6 @@ using Builder = RecordBuilder<

Builder MakeBuilder();


void InsertEventBegin(const G4Event* event, Builder& builder);
void InsertEventEnd(const G4Event* event, Builder& builder);
void InsertTrackBegin(const G4Track* track, Builder& builder);
Expand Down

0 comments on commit a43bd90

Please sign in to comment.