Skip to content
Snippets Groups Projects

Added dummy to dataframe

@@ -12,6 +12,8 @@ using json = nlohmann::json;
@@ -12,6 +12,8 @@ using json = nlohmann::json;
#include "eicd/TrackParametersCollection.h"
#include "eicd/TrackParametersCollection.h"
#include "eicd/ClusterCollection.h"
#include "eicd/ClusterCollection.h"
#include "eicd/ClusterData.h"
#include "eicd/ClusterData.h"
 
#include "eicd/ReconstructedParticleCollection.h"
 
#include "eicd/ReconstructedParticleData.h"
using ROOT::RDataFrame;
using ROOT::RDataFrame;
using namespace ROOT::VecOps;
using namespace ROOT::VecOps;
@@ -56,6 +58,15 @@ auto fourvec = [](ROOT::VecOps::RVec<dd4pod::Geant4ParticleData> const& in) {
@@ -56,6 +58,15 @@ auto fourvec = [](ROOT::VecOps::RVec<dd4pod::Geant4ParticleData> const& in) {
}
}
return result;
return result;
};
};
 
auto dumfourvec = [](ROOT::VecOps::RVec<eic::ReconstructedParticleData> const& in) {
 
std::vector<ROOT::Math::PxPyPzMVector> result;
 
ROOT::Math::PxPyPzMVector lv;
 
for (size_t i = 0; i < in.size(); ++i) {
 
lv.SetCoordinates(in[i].p.x, in[i].p.y, in[i].p.z, in[i].mass);
 
result.push_back(lv);
 
}
 
return result;
 
};
auto delta_p = [](const std::vector<double>& tracks, const std::vector<double>& thrown) {
auto delta_p = [](const std::vector<double>& tracks, const std::vector<double>& thrown) {
std::vector<double> res;
std::vector<double> res;
@@ -95,6 +106,7 @@ void dvcs_tests(const char* fname = "rec_dvcs.root"){
@@ -95,6 +106,7 @@ void dvcs_tests(const char* fname = "rec_dvcs.root"){
auto df0 = df.Define("isThrown", "mcparticles2.genStatus == 1")
auto df0 = df.Define("isThrown", "mcparticles2.genStatus == 1")
.Define("thrownParticles", "mcparticles2[isThrown]")
.Define("thrownParticles", "mcparticles2[isThrown]")
.Define("thrownP", fourvec, {"thrownParticles"})
.Define("thrownP", fourvec, {"thrownParticles"})
 
.Define("dumRec", dumfourvec, {"DummyReconstructedParticles"})
.Define("p_thrown", momentum, {"thrownP"})
.Define("p_thrown", momentum, {"thrownP"})
.Define("nTracks", "outputTrackParameters.size()")
.Define("nTracks", "outputTrackParameters.size()")
.Define("p_track", p_track, {"outputTrackParameters"})
.Define("p_track", p_track, {"outputTrackParameters"})
Loading