From 8553b91befa4cd438c894c93fefe711f2a02c46b Mon Sep 17 00:00:00 2001 From: Bernhard Manfred Gruber Date: Tue, 20 Apr 2021 13:33:44 +0200 Subject: [PATCH] update names of recently renamed LLAMA constructs and reformat --- .../hep_dimuon_analysis_cms.cpp | 40 +++++++++++-------- 1 file changed, 23 insertions(+), 17 deletions(-) diff --git a/examples/hep_dimuon_analysis_cms/hep_dimuon_analysis_cms.cpp b/examples/hep_dimuon_analysis_cms/hep_dimuon_analysis_cms.cpp index 5e8b0cf0a2..03ec8c13a5 100644 --- a/examples/hep_dimuon_analysis_cms/hep_dimuon_analysis_cms.cpp +++ b/examples/hep_dimuon_analysis_cms/hep_dimuon_analysis_cms.cpp @@ -31,16 +31,16 @@ namespace tag struct Muon_mass{}; } -using Event = llama::DS< - llama::DE +using Event = llama::Record< + llama::Field >; -using Muon = llama::DS< - llama::DE, - llama::DE, - llama::DE, - llama::DE, - llama::DE +using Muon = llama::Record< + llama::Field, + llama::Field, + llama::Field, + llama::Field, + llama::Field >; // clang-format on @@ -97,7 +97,8 @@ auto buildRNTupleFileModel(const std::string& path) // we cannot copy the offsets stored in the RNTuple directly, because they are local to the cluster they reside in. // To correctly interpret this information, we would need access to the ClusterInfo stored in the RPage, which is // not reachable via RNTupleReader. - auto copyOffsets = [](ROOT::Experimental::RNTupleViewCollection& view, std::vector& dstPages) { + auto copyOffsets = [](ROOT::Experimental::RNTupleViewCollection& view, std::vector& dstPages) + { using FieldType = ROOT::Experimental::ClusterSize_t; FieldType* dst = nullptr; auto offset = FieldType{0}; @@ -118,7 +119,8 @@ auto buildRNTupleFileModel(const std::string& path) } }; - auto copy = [](ROOT::Experimental::RNTupleView& view, std::vector& dstPages) { + auto copy = [](ROOT::Experimental::RNTupleView& view, std::vector& dstPages) + { FieldType* dst = nullptr; std::size_t written = 0; for (auto i : view.GetFieldRange()) @@ -173,14 +175,16 @@ int main(int argc, const char* argv[]) auto& Muon_massPages = rntuple.at("Muon_mass"); start = std::chrono::steady_clock::now(); - auto viewEventPage = [&](std::size_t i) { + auto viewEventPage = [&](std::size_t i) + { return llama::View{ - llama::mapping::SoA, Event, std::true_type>{llama::ArrayDomain{elementsPerPage}}, + llama::mapping::SoA, Event, true>{llama::ArrayDims{elementsPerPage}}, llama::Array{Muons_endPages.at(i).data()}}; }; - auto viewMuonPage = [&](std::size_t i) { + auto viewMuonPage = [&](std::size_t i) + { return llama::View{ - llama::mapping::SoA, Muon, std::true_type>{llama::ArrayDomain{elementsPerPage}}, + llama::mapping::SoA, Muon, true>{llama::ArrayDims{elementsPerPage}}, llama::Array{ Muon_chargePages.at(i).data(), Muon_phiPages.at(i).data(), @@ -202,7 +206,8 @@ int main(int argc, const char* argv[]) const auto eventsOnThisPage = std::min(elementsPerPage, entries - ep * elementsPerPage); for (std::size_t e = 0; e < eventsOnThisPage; e++) { - const auto muonOffset = [&]() { + const auto muonOffset = [&]() + { if (e == 0) { if (ep == 0) @@ -222,7 +227,8 @@ int main(int argc, const char* argv[]) const auto muonPageInnerIndex = muonOffset % elementsPerPage; auto muonView = viewMuonPage(muonPageIndex); - auto processDimuons = [&](auto dimuonView) { + auto processDimuons = [&](auto dimuonView) + { if (dimuonView(0u)(tag::Muon_charge{}) == dimuonView(1u)(tag::Muon_charge{})) return; @@ -250,7 +256,7 @@ int main(int argc, const char* argv[]) processDimuons(llama::VirtualView{muonView, {muonPageInnerIndex}, {2}}); else { - constexpr auto mapping = llama::mapping::SoA, Muon>{{2}}; + constexpr auto mapping = llama::mapping::SoA, Muon>{{2}}; auto dimuonView = llama::allocView(mapping, llama::bloballoc::Stack{}); dimuonView(0u) = muonView(muonPageInnerIndex); dimuonView(1u) = viewMuonPage(muonPageIndex + 1)(0u);