BGOverlayInputModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/DataStore.h>
#include <string>
#include "TChain.h"
 
namespace Belle2 {
  class BGOverlayInputModule : public Module {
 
  public:
 
    BGOverlayInputModule();
 
    virtual ~BGOverlayInputModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    typedef std::vector<DataStore::StoreEntry*> StoreEntries;   
    bool connectBranches();
 
 
    std::vector<std::string> m_inputFileNames; 
    std::string m_extensionName; 
    std::string m_BackgroundInfoInstanceName = ""; 
    TChain* m_tree = 0;            
    unsigned m_numEvents = 0;      
    unsigned m_eventCount = 0;     
    unsigned m_firstEvent = 0;     
    bool m_start = true;           
    StoreEntries m_storeEntries;   
    unsigned m_index = 0; 
  };
 
} // Belle2 namespace