Skip to content

Commit

Permalink
Codify some of the implicitly assumed conventions (AIDASoft#439)
Browse files Browse the repository at this point in the history
  • Loading branch information
tmadlener authored and hegner committed Jul 27, 2023
1 parent 7fd6668 commit a83402e
Show file tree
Hide file tree
Showing 5 changed files with 56 additions and 12 deletions.
1 change: 1 addition & 0 deletions include/podio/Frame.h
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
#include "podio/CollectionBase.h"
#include "podio/CollectionBufferFactory.h"
#include "podio/CollectionIDTable.h"
#include "podio/FrameCategories.h" // mainly for convenience
#include "podio/GenericParameters.h"
#include "podio/ICollectionProvider.h"
#include "podio/SchemaEvolution.h"
Expand Down
43 changes: 43 additions & 0 deletions include/podio/FrameCategories.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
#ifndef PODIO_FRAMECATEGORIES_H
#define PODIO_FRAMECATEGORIES_H

#include <string>

namespace podio {

/**
* Create a parameterName that encodes the collection name and the parameter
* Name into one string.
*
* This codifies a convention that was decided on to store collection level
* parameters. These are parameters / metadata that are valid for all
* collections of a given name in a file, e.g. CellID encoding strings. These
* parameters are usually stored in a dedicated metadata Frame inside a file,
* see the predefined category names in the Cateogry namespace.
*
* @param collName the name of the collection
* @param paramName the name of the parameter
*
* @returns A single key string that combines the collection and parameter name
*/
inline std::string collMetadataParamName(const std::string& collName, const std::string& paramName) {
return collName + "__" + paramName;
}

/**
* This namespace mimics an enum (at least in its usage) and simply defines
* either commonly used category names, or category names that form a
* convention.
*/
namespace Category {
/// The event category
constexpr const auto Event = "events";
/// The run category
constexpr const auto Run = "runs";
/// The metadata cateogry that is used to store a single Frame that holds data
/// that is valid for a whole file, e.g. collection level parameters
constexpr const auto Metadata = "metadata";
} // namespace Category
} // namespace podio

#endif
18 changes: 9 additions & 9 deletions tests/read_frame.h
Original file line number Diff line number Diff line change
Expand Up @@ -77,22 +77,22 @@ int read_frames(const std::string& filename, bool assertBuildVersion = true) {
return 1;
}

if (reader.getEntries("events") != 10) {
if (reader.getEntries(podio::Category::Event) != 10) {
std::cerr << "Could not read back the number of events correctly. "
<< "(expected:" << 10 << ", actual: " << reader.getEntries("events") << ")" << std::endl;
<< "(expected:" << 10 << ", actual: " << reader.getEntries(podio::Category::Event) << ")" << std::endl;
return 1;
}

if (reader.getEntries("events") != reader.getEntries("other_events")) {
if (reader.getEntries(podio::Category::Event) != reader.getEntries("other_events")) {
std::cerr << "Could not read back the number of events correctly. "
<< "(expected:" << 10 << ", actual: " << reader.getEntries("other_events") << ")" << std::endl;
return 1;
}

// Read the frames in a different order than when writing them here to make
// sure that the writing/reading order does not impose any usage requirements
for (size_t i = 0; i < reader.getEntries("events"); ++i) {
auto frame = podio::Frame(reader.readNextEntry("events"));
for (size_t i = 0; i < reader.getEntries(podio::Category::Event); ++i) {
auto frame = podio::Frame(reader.readNextEntry(podio::Category::Event));
if (reader.currentFileVersion() > podio::version::Version{0, 16, 2}) {
if (frame.get("emptySubsetColl") == nullptr) {
std::cerr << "Could not retrieve an empty subset collection" << std::endl;
Expand All @@ -114,7 +114,7 @@ int read_frames(const std::string& filename, bool assertBuildVersion = true) {
}
}

if (reader.readNextEntry("events")) {
if (reader.readNextEntry(podio::Category::Event)) {
std::cerr << "Trying to read more frame data than is present should return a nullptr" << std::endl;
return 1;
}
Expand All @@ -127,10 +127,10 @@ int read_frames(const std::string& filename, bool assertBuildVersion = true) {

// Reading specific (jumping to) entry
{
auto frame = podio::Frame(reader.readEntry("events", 4));
auto frame = podio::Frame(reader.readEntry(podio::Category::Event, 4));
processEvent(frame, 4, reader.currentFileVersion());
// Reading the next entry after jump, continues from after the jump
auto nextFrame = podio::Frame(reader.readNextEntry("events"));
auto nextFrame = podio::Frame(reader.readNextEntry(podio::Category::Event));
processEvent(nextFrame, 5, reader.currentFileVersion());

auto otherFrame = podio::Frame(reader.readEntry("other_events", 4));
Expand All @@ -147,7 +147,7 @@ int read_frames(const std::string& filename, bool assertBuildVersion = true) {
}

// Trying to read a Frame that is not present returns a nullptr
if (reader.readEntry("events", 10)) {
if (reader.readEntry(podio::Category::Event, 10)) {
std::cerr << "Trying to read a specific entry that does not exist should return a nullptr" << std::endl;
return 1;
}
Expand Down
4 changes: 2 additions & 2 deletions tests/read_frame_auxiliary.h
Original file line number Diff line number Diff line change
Expand Up @@ -56,9 +56,9 @@ int test_frame_aux_info(const std::string& fileName) {
reader.openFile(fileName);

// Test on the first event only here. Additionally, also only testing the
// "events" category, since that is the one where not all collections are
// events category, since that is the one where not all collections are
// written
auto event = podio::Frame(reader.readEntry("events", 0));
auto event = podio::Frame(reader.readEntry(podio::Category::Event, 0));

auto collsToRead = collsToWrite;
if (reader.currentFileVersion() < podio::version::Version{0, 16, 3}) {
Expand Down
2 changes: 1 addition & 1 deletion tests/write_frame.h
Original file line number Diff line number Diff line change
Expand Up @@ -424,7 +424,7 @@ void write_frames(const std::string& filename) {

for (int i = 0; i < 10; ++i) {
auto frame = makeFrame(i);
writer.writeFrame(frame, "events", collsToWrite);
writer.writeFrame(frame, podio::Category::Event, collsToWrite);
}

for (int i = 100; i < 110; ++i) {
Expand Down

0 comments on commit a83402e

Please sign in to comment.