void processEvent(podio::EventStore& store, bool verbose,
podio::ROOTReader& reader) {
const ExampleClusterCollection* clusters = nullptr;
//std::cout << "Fetching collection 'clusters'" << std::endl;
bool is_available = store.get("clusters",clusters);
if(is_available){
auto cluster = (*clusters)[0];
//std::cout << "Cluster has an energy of " << cluster.energy() << std::endl;
for (auto i = cluster.Hits_begin(), end = cluster.Hits_end(); i!=end; ++i){
//std::cout << " Referenced hit has an energy of " << i->energy() << std::endl;
glob++;
}
}
const ExampleReferencingTypeCollection* refs = nullptr;
//std::cout << "Fetching collection 'refs'" << std::endl;
is_available = store.get("refs",refs);
if(is_available){
auto ref = (*refs)[0];
for (auto j = ref.Clusters_begin(), end = ref.Clusters_end(); j!=end; ++j){
for (auto i = j->Hits_begin(), end = j->Hits_end(); i!=end; ++i){
//std::cout << " Referenced object has an energy of " << i->energy() << std::endl;
glob++;
}
}
}
const ExampleWithOneRelationCollection* rels = nullptr;
//std::cout << "Fetching collection 'OneRelation'" << std::endl;
is_available = store.get("OneRelation",rels);
if(is_available) {
//std::cout << "Referenced object has an energy of " << (*rels)[0].cluster().energy() << std::endl;
glob++;
}
const ExampleWithVectorMemberCollection* vecs = nullptr;
// std::cout << "Fetching collection 'WithVectorMember'" << std::endl;
is_available = store.get("WithVectorMember",vecs);
if(is_available) {
// auto item = (*vecs)[0];
// std::cout << (*vecs).size() << std::endl;
// for (auto c = item.count_begin(), end = item.count_end(); c!=end; ++c){
// std::cout << " Counter value " << (*c) << std::endl;
// glob++;
// }
}
}
TEST(podio,referencing) {
bool success = true;
auto store = podio::EventStore();
auto& hits = store.create<ExampleHitCollection>("hits");
auto hit1 = hits.create(0.,0.,0.,0.);
auto hit2 = hits.create(1.,1.,1.,1.);
auto& clusters = store.create<ExampleClusterCollection>("clusters");
auto cluster = clusters.create();
cluster.addHits(hit1);
cluster.addHits(hit2);
int index = 0;
for (auto i = cluster.Hits_begin(), end = cluster.Hits_end(); i!=end; ++i){
if( i->energy() != index) success = false;
++index;
}
EXPECT_EQ(true, success);
}
void processEvent(podio::EventStore& store, bool verboser, unsigned eventNum) {
auto& failing = store.get<ExampleClusterCollection>("notthere");
if(failing.isValid() == true) {
throw std::runtime_error("Collection 'notthere' should not be valid");
};
auto& strings = store.get<ExampleWithStringCollection>("strings");
if(strings.isValid()){
auto string = strings[0];
if (string.theString() != "SomeString") {
throw std::runtime_error("Couldn't read string properly");
}
} else {
throw std::runtime_error("Collection 'strings' should be present.");
}
auto& clusters = store.get<ExampleClusterCollection>("clusters");
if(clusters.isValid()){
auto cluster = clusters[0];
for (auto i = cluster.Hits_begin(), end = cluster.Hits_end(); i!=end; ++i){
std::cout << " Referenced hit has an energy of " << i->energy() << std::endl;
glob++;
}
} else {
throw std::runtime_error("Collection 'clusters' should be present");
}
auto& mcps = store.get<ExampleMCCollection>("mcparticles");
if( mcps.isValid() ){
// check that we can retrieve the correct parent daughter relation
// set in write.cpp :
// particle 0 has particles 2,3,4 and 5 as daughters:
auto p = mcps[0] ;
//-------- print relations for debugging:
for( auto p : mcps ){
std::cout << " particle " << p.getObjectID().index << " has daughters: " ;
for(auto it = p.daughters_begin(), end = p.daughters_end() ; it!=end ; ++it ){
std::cout << " " << it->getObjectID().index ;
}
std::cout << " and parents: " ;
for(auto it = p.parents_begin(), end = p.parents_end() ; it!=end ; ++it ){
std::cout << " " << it->getObjectID().index ;
}
std::cout << std::endl ;
}
auto d0 = p.daughters(0) ;
auto d1 = p.daughters(1) ;
auto d2 = p.daughters(2) ;
auto d3 = p.daughters(3) ;
if( ! ( d0 == mcps[2] ) ) throw std::runtime_error(" error: 1. daughter of particle 0 is not particle 2 ");
if( ! ( d1 == mcps[3] ) ) throw std::runtime_error(" error: 2. daughter of particle 0 is not particle 3 ");
if( ! ( d2 == mcps[4] ) ) throw std::runtime_error(" error: 3. daughter of particle 0 is not particle 4 ");
if( ! ( d3 == mcps[5] ) ) throw std::runtime_error(" error: 4. daughter of particle 0 is not particle 5 ");
// particle 3 has particles 6,7,8 and 9 as daughters:
p = mcps[3] ;
d0 = p.daughters(0) ;
d1 = p.daughters(1) ;
d2 = p.daughters(2) ;
d3 = p.daughters(3) ;
if( ! ( d0 == mcps[6] ) ) throw std::runtime_error(" error: 1. daughter of particle 3 is not particle 6 ");
if( ! ( d1 == mcps[7] ) ) throw std::runtime_error(" error: 2. daughter of particle 3 is not particle 7 ");
if( ! ( d2 == mcps[8] ) ) throw std::runtime_error(" error: 3. daughter of particle 3 is not particle 8 ");
if( ! ( d3 == mcps[9] ) ) throw std::runtime_error(" error: 4. daughter of particle 3 is not particle 9 ");
} else {
throw std::runtime_error("Collection 'mcparticles' should be present");
}
//std::cout << "Fetching collection 'refs'" << std::endl;
auto& refs = store.get<ExampleReferencingTypeCollection>("refs");
if(refs.isValid()){
auto ref = refs[0];
for (auto j = ref.Clusters_begin(), end = ref.Clusters_end(); j!=end; ++j){
for (auto i = j->Hits_begin(), end = j->Hits_end(); i!=end; ++i){
//std::cout << " Referenced object has an energy of " << i->energy() << std::endl;
glob++;
}
}
} else {
throw std::runtime_error("Collection 'refs' should be present");
}
//std::cout << "Fetching collection 'OneRelation'" << std::endl;
auto& rels = store.get<ExampleWithOneRelationCollection>("OneRelation");
if(rels.isValid()) {
//std::cout << "Referenced object has an energy of " << (*rels)[0].cluster().energy() << std::endl;
glob++;
} else {
throw std::runtime_error("Collection 'OneRelation' should be present");
}
// std::cout << "Fetching collection 'WithVectorMember'" << std::endl;
auto& vecs = store.get<ExampleWithVectorMemberCollection>("WithVectorMember");
if(vecs.isValid()) {
std::cout << vecs.size() << std::endl;
for( auto item : vecs )
for (auto c = item.count_begin(), end = item.count_end(); c!=end; ++c){
std::cout << " Counter value " << (*c) << std::endl;
glob++;
}
} else {
throw std::runtime_error("Collection 'WithVectorMember' should be present");
}
auto& comps = store.get<ExampleWithComponentCollection>("Component");
if (comps.isValid()) {
auto comp = comps[0];
int a = comp.component().data.x + comp.component().data.z;
}
auto& arrays = store.get<ExampleWithArrayCollection>("arrays");
if (arrays.isValid() && arrays.size() != 0) {
auto array = arrays[0];
if (array.myArray(1) != eventNum) {
throw std::runtime_error("Array not properly set.");
}
if (array.arrayStruct().data.p.at(2) != 2*eventNum) {
throw std::runtime_error("Array not properly set.");
}
if (array.structArray(0).x != eventNum) {
throw std::runtime_error("Array of struct not properly set.");
}
} else {
throw std::runtime_error("Collection 'arrays' should be present");
}
auto& nmspaces = store.get<ex::ExampleWithARelationCollection>("WithNamespaceRelation");
auto& copies = store.get<ex::ExampleWithARelationCollection>("WithNamespaceRelationCopy");
auto& cpytest = store.create<ex::ExampleWithARelationCollection>("TestConstCopy");
if (nmspaces.isValid() && copies.isValid()) {
for (int j = 0; j < nmspaces.size(); j++) {
auto nmsp = nmspaces[j];
auto cpy = copies[j];
cpytest.push_back(nmsp.clone());
if (nmsp.ref().isAvailable()) {
if (nmsp.ref().data().x != cpy.ref().data().x || nmsp.ref().data().y != cpy.ref().data().y) {
throw std::runtime_error("Copied item has differing data in OneToOne referenced item.");
}
// check direct accessors of POD sub members
if (nmsp.ref().x() != cpy.ref().x()) {
throw std::runtime_error("Getting wrong values when using direct accessors for sub members.");
}
if (nmsp.number() != cpy.number()) {
throw std::runtime_error("Copied item has differing member.");
}
if (!(nmsp.ref().getObjectID() == cpy.ref().getObjectID())) {
throw std::runtime_error("Copied item has wrong OneToOne references.");
}
}
auto cpy_it = cpy.refs_begin();
for (auto it = nmsp.refs_begin(); it != nmsp.refs_end(); ++it, ++cpy_it) {
if (it->data().x != cpy_it->data().x || it->data().y != cpy_it->data().y) {
throw std::runtime_error("Copied item has differing data in OneToMany referenced item.");
}
if (!(it->getObjectID() == cpy_it->getObjectID())) {
throw std::runtime_error("Copied item has wrong OneToMany references.");
}
}
}
} else {
throw std::runtime_error("Collection 'WithNamespaceRelation' and 'WithNamespaceRelationCopy' should be present");
}
}
