Module: store local ROOT directory in module object itself

src/core/module/Module.cpp

@@ -46,6 +46,24 @@ ModuleIdentifier Module::get_identifier() const {
     return identifier_;
 }
 
+/**
+ * @throws InvalidModuleActionException If this method is called from the constructor or destructor
+ * @warning Cannot be used from the constructor, because the instantiation logic has not finished yet
+ * @warning This method should not be accessed from the destructor (the file is then already closed)
+ * @note It is not needed to change directory to this file explicitly in the module, this is done automatically.
+ */
+TDirectory* Module::getROOTDirectory() const {
+    // The directory will only be a null pointer if this method is executed from the constructor or destructor
+    if(directory_ == nullptr) {
+        throw InvalidModuleActionException("Cannot access ROOT directory in constructor or destructor");
+    }
+
+    return directory_;
+}
+void Module::set_ROOT_directory(TDirectory* directory) {
+    directory_ = directory;
+}
+
 std::shared_ptr<Detector> Module::get_detector(std::string name) {
     auto it = find_if(
         m_detectors.begin(), m_detectors.end(), [&name](std::shared_ptr<Detector> obj) { return obj->name() == name; });

src/core/module/Module.hpp

@@ -120,6 +120,12 @@ namespace corryvreckan {
          */
         virtual void finalise(){};
 
+        /**
+         * @brief Get ROOT directory which should be used to output histograms et cetera
+         * @return ROOT directory for storage
+         */
+        TDirectory* getROOTDirectory() const;
+
     protected:
         // Configuration of this module
         Configuration m_config;
@@ -177,6 +183,13 @@ namespace corryvreckan {
         ModuleIdentifier get_identifier() const;
         ModuleIdentifier identifier_;
 
+        /**
+         * @brief Set the output ROOT directory for this module
+         * @param directory ROOT directory for storage
+         */
+        void set_ROOT_directory(TDirectory* directory);
+        TDirectory* directory_{nullptr};
+
         // Configure the reference detector:
         void setReference(std::shared_ptr<Detector> reference) { m_reference = reference; };
         std::shared_ptr<Detector> m_reference;