SubEventModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2014 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Christian Pulvermacher                                   *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/core/EventProcessor.h>
#include <framework/datastore/StoreArray.h>
#include <boost/optional.hpp>
#include <memory>
 
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();
    boost::optional<std::string> m_objectName{boost::none};
    boost::optional<std::string> m_loopOverName{boost::none};
    StoreArray<TObject> m_loopOver;
    std::shared_ptr<Path> m_path;
    int m_processID{ -1};
    unsigned int m_maxIterations{10000};
    boost::optional<std::string> m_loopConditionString{boost::none};
    std::unique_ptr<ModuleCondition> m_loopCondition;
    Module* m_loopConditionModule{nullptr};
    int m_mode{c_ForEach};
  };
}