io/podio/JEventProcessorPODIO.h

0001
0002 #pragma once
0003
0004 #include <JANA/JEvent.h>
0005 #include <JANA/JEventProcessor.h>
0006 #include <podio/ROOTWriter.h>
0007 #include <spdlog/logger.h>
0008 #include <memory>
0009 #include <mutex>
0010 #include <set>
0011 #include <string>
0012 #include <vector>
0013
0014 class JEventProcessorPODIO : public JEventProcessor {
0015
0016 public:
0017   JEventProcessorPODIO();
0018   virtual ~JEventProcessorPODIO() = default;
0019
0020   void Init() override;
0021   void Process(const std::shared_ptr<const JEvent>& event) override;
0022   void Finish() override;
0023
0024   void FindCollectionsToWrite(const std::shared_ptr<const JEvent>& event);
0025
0026   std::unique_ptr<podio::ROOTWriter> m_writer;
0027   std::mutex m_mutex;
0028   std::once_flag m_is_first_event;
0029   std::shared_ptr<spdlog::logger> m_log;
0030   bool m_output_include_collections_set = false;
0031
0032   std::string m_output_file          = "podio_output.root";
0033   std::string m_output_file_copy_dir = "";
0034   std::set<std::string> m_output_collections;         // config. parameter
0035   std::set<std::string> m_output_exclude_collections; // config. parameter
0036   std::vector<std::string> m_collections_to_write;    // derived from above config. parameters
0037   std::vector<std::string> m_collections_to_print;
0038 };