DuplicateChecker.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
 
 
#ifndef EVENT_LOOP_ALGS__DUPLICATE_CHECKER_H
#define EVENT_LOOP_ALGS__DUPLICATE_CHECKER_H
 
#include <EventLoopAlgs/Global.h>
 
#include <EventLoop/Algorithm.h>
#include <SampleHandler/Global.h>
#include <map>
#include <set>
 
namespace xAOD
{
  class TEvent;
}
 
class TObjString;
class TTree;
 
namespace EL
{
 
  class DuplicateChecker : public Algorithm
  {
    //
    // public interface
    //
 
  private:
    typedef uint32_t number_type;
 
 
  public:
    void testInvariant () const;
 
 
  public:
    DuplicateChecker ();
 
 
  public:
    const std::string& eventInfoName () const;
 
  public:
    void setEventInfoName (const std::string& val_eventInfoName);
 
 
  public:
    const std::string& outputTreeName () const;
 
  public:
    void setOutputTreeName (const std::string& val_outputTreeName);
 
 
  public:
    void addKnownDuplicate (const std::string& sampleName,
                const std::string& fileName,
                Long64_t entry, number_type runNumber,
                number_type eventNumber);
 
  public:
    void addKnownDuplicatesFile (const std::string& duplicatesFile);
 
 
  public:
    static bool
    processSummary (const std::string& submitdir, const std::string& treeName);
 
 
  public:
    static bool
    processSummary (const SH::SampleHandler& sh, const std::string& outputFile);
 
 
 
    //
    // inherited interface
    //
 
  private:
    virtual StatusCode setupJob (Job& job) override;
 
  private:
    virtual StatusCode changeInput (bool firstFile) override;
 
  private:
    virtual StatusCode initialize () override;
 
  private:
    virtual StatusCode execute () override;
 
 
 
    //
    // private interface
    //
 
  private:
    std::string m_eventInfoName;
 
  private:
    std::string m_outputTreeName;
 
  private:
    std::map<std::pair<std::string,std::string>,std::map<Long64_t,std::pair<number_type,number_type> > > m_duplicates;
 
  private:
  std::map<Long64_t,std::pair<number_type,number_type> > *m_currentDuplicates = nullptr; 
 
 
  private:
    std::set<std::pair<number_type,number_type> > m_processed; 
 
  private:
    xAOD::TEvent *m_event = nullptr; 
 
 
  private:
    TTree *m_outputTree = nullptr; 
 
  private:
    std::string m_inputFileName; 
 
  private:
    Long64_t m_inputFileIndex; 
 
  private:
    number_type m_runNumber; 
 
  private:
    number_type m_eventNumber; 
 
  private:
    Bool_t m_processEvent; 
 
  private:
    void read_run_event_number ();
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wpragmas"
#pragma GCC diagnostic ignored "-Wunknown-pragmas"
#pragma GCC diagnostic ignored "-Winconsistent-missing-override"
    ClassDef(DuplicateChecker, 1);
#pragma GCC diagnostic pop
  };
}
 
#endif