UnitTestAlg1.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/

 
//
// includes
//
 
#include <EventLoopTest/UnitTestAlg1.h>
 
#include <EventLoop/Job.h>
#include <EventLoop/OutputStream.h>
#include <EventLoop/Worker.h>
#include <RootCoreUtils/Assert.h>
#include <TFile.h>
#include <TH1.h>
#include <TTree.h>
#include <TObjString.h>
 
#include <AsgMessaging/MsgStream.h>
#include <AsgMessaging/MsgStreamMacros.h>
 
//
// method implementations
//
 
ClassImp (EL::UnitTestAlg1)
 
namespace EL
{
  void UnitTestAlg1 ::
  testInvariant () const
  {}
 
 
 
  UnitTestAlg1 ::
  UnitTestAlg1 (const std::string& branchName)
    : makeOutput (true),
      m_name (branchName),
      m_branch (0),
      m_value (0),// m_hist (0),
      m_tree (0),
      m_hasInitialize (false),
      m_hasHistInitialize (false)
  {
    RCU_NEW_INVARIANT (this);
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  setupJob (Job& job)
  {
    RCU_CHANGE_INVARIANT (this);
 
    if (makeOutput)
    {
      OutputStream out ("out");
      job.outputAdd (out);
    }
    {
      OutputStream out ("out_empty");
      job.outputAdd (out);
    }
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  changeInput (bool firstFile)
  {
    RCU_CHANGE_INVARIANT (this);
 
    if (firstFile)
      getCallbacks()->Fill (CB_CHANGE_INPUT_FIRST);
    else
      getCallbacks()->Fill (CB_CHANGE_INPUT_OTHER);
 
    RCU_ASSERT (wk()->tree() != 0);
    m_branch = wk()->tree()->GetBranch (m_name.c_str());
    if (m_branch == 0)
    {
      ATH_MSG_ERROR ("failed to find branch " << m_name);
      return StatusCode::FAILURE;
    }
    m_branch->SetAddress (&m_value);
    RCU_ASSERT_SOFT (firstFile == m_fileName.empty());
    m_fileName = wk()->inputFile()->GetName();
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  histInitialize ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_HIST_INITIALIZE);
 
    RCU_ASSERT_SOFT (!m_hasHistInitialize);
 
    book (TH1F ((m_name + "2").c_str(), m_name.c_str(), 50, 0, 50));
    book (TH1F ("file_executes", "file executes", 1, 0, 1));
    m_hasHistInitialize = true;
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  initialize ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_INITIALIZE);
 
    RCU_ASSERT_SOFT (m_hasHistInitialize);
    RCU_ASSERT_SOFT (!m_hasInitialize);
 
    RCU_ASSERT_SOFT (wk()->tree()->GetEntries() > wk()->treeEntry());
    RCU_ASSERT_SOFT (m_fileName == wk()->inputFile()->GetName());
 
    wk()->addOutput (/*m_hist = */new TH1F (m_name.c_str(), m_name.c_str(),
                    50, 0, 50));
    if (makeOutput)
    {
      TFile *file = wk()->getOutputFile ("out");
      m_tree = new TTree ("tree", "test output");
      m_tree->SetDirectory (file);
      m_tree->Branch (m_name.c_str(), &m_value, (m_name + "/I").c_str());
    }
    m_hasInitialize = true;
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  fileExecute ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_FILE_EXECUTE);
 
    RCU_ASSERT_SOFT (m_hasHistInitialize);
    hist ("file_executes")->Fill (0);
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  execute ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_EXECUTE);
 
    RCU_ASSERT_SOFT (m_hasInitialize);
 
    m_branch->GetEntry (wk()->treeEntry());
    hist(m_name)->Fill (m_value);
    if (makeOutput)
      m_tree->Fill ();
 
    setMsgLevel (MSG::INFO);
    if (++ m_calls < 3)
      ATH_MSG_INFO ("message test");
 
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  finalize ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_FINALIZE);
 
    RCU_ASSERT_SOFT (m_hasInitialize);
    wk()->addOutput (new TH1F ("beta/dir/hist", "directory test", 10, 0, 10));
    wk()->addOutputList ("alpha", new TObjString ("alpha"));
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode UnitTestAlg1 ::
  histFinalize ()
  {
    RCU_CHANGE_INVARIANT (this);
 
    getCallbacks()->Fill (CB_HIST_FINALIZE);
 
    RCU_ASSERT_SOFT (m_hasHistInitialize);
    wk()->addOutput (new TH1F ("beta/dir/hist", "directory test", 10, 0, 10));
    wk()->addOutputList ("alpha", new TObjString ("alpha"));
    return StatusCode::SUCCESS;
  }
 
 
 
  TH1 *UnitTestAlg1 ::
  getCallbacks ()
  {
    if (m_callbacks == nullptr)
    {
      m_callbacks = new TH1F ("callbacks", "callbacks", CB_HIST_FINALIZE + 1,
                  0, CB_HIST_FINALIZE + 1);
      wk()->addOutput (m_callbacks);
    }
    return m_callbacks;
  }
}