SubEventModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/core/EventProcessor.h>
#include <framework/datastore/StoreArray.h>
 
#include <memory>
#include <optional>
 
namespace Belle2 {
  class Path;
  class SubEventModule : public Module, public EventProcessor {
  public:
    enum EModes {
      c_ForEach = 0,
      c_DoWhile = 1,
    };
 
    SubEventModule();
    ~SubEventModule();
 
    void initSubEvent(const std::string& objectName, const std::string& loopOver, std::shared_ptr<Path> path);
    void initSubLoop(std::shared_ptr<Path> path, const std::string& condition, unsigned int maxIterations);
 
    virtual void initialize() override;
    virtual void beginRun() override;
    virtual void endRun() override;
    virtual void event() override;
    virtual void terminate() override;
 
  private:
    void setProperties();
    void setDoWhileConditions();
    std::optional<std::string> m_objectName{std::nullopt};
    std::optional<std::string> m_loopOverName{std::nullopt};
    StoreArray<TObject> m_loopOver;
    std::shared_ptr<Path> m_path;
    int m_processID{ -1};
    unsigned int m_maxIterations{10000};
    std::optional<std::string> m_loopConditionString{std::nullopt};
    std::unique_ptr<ModuleCondition> m_loopCondition;
    Module* m_loopConditionModule{nullptr};
    int m_mode{c_ForEach};
  };
}