AtmosShowerListModule.cxx

Go to the documentation of this file.

// Package: AtNuReco
//
// AtmosShowerListModule.cxx
//
// chapman@hep.phy.cam.ac.uk
// blake@hep.phy.cam.ac.uk
#include <cassert>

// For measuring CPU time
extern "C" {
#include <unistd.h>    // sysconf
#include <sys/times.h> // times()
}

// Job Control includes
#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "JobControl/JobCModuleRegistry.h" // For JOBMODULE macro

// Algorithm includes
#include "Algorithm/AlgFactory.h"
#include "Algorithm/AlgHandle.h"
#include "Algorithm/AlgConfig.h"

// Candidate includes
#include "CandData/CandRecord.h"
#include "Candidate/CandContext.h"

#include "RecoBase/CandSliceHandle.h"
#include "RecoBase/CandSliceListHandle.h"
#include "RecoBase/CandTrack.h"
#include "RecoBase/CandTrackHandle.h"
#include "RecoBase/CandTrackListHandle.h"

#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"

// This package includes
#include "AtNuReco/AtmosShowerListModule.h"
#include "AtNuReco/CandShowerAtNuListHandle.h"
#include "AtNuReco/CandShowerAtNuList.h"

JOBMODULE(AtmosShowerListModule, "AtmosShowerListModule", "Shower finder");
CVSID("$Id: AtmosShowerListModule.cxx,v 1.3 2006/04/25 17:17:32 chapman Exp $");

AtmosShowerListModule::AtmosShowerListModule()
{
  fAlgName="AlgAtmosShowerList";
  fAlgConfig="default";

  fListIn="CandSliceList";
  fTrackListIn="CandFitTrackCamList";
}


AtmosShowerListModule::~AtmosShowerListModule()
{
}


void AtmosShowerListModule::BeginJob()
{
  // Register algorithms
  AlgFactory &af = AlgFactory::GetInstance();
  af.Register("AlgAtmosShowerList", "default", "libAtNuReco.so", "AlgConfig");
  af.Register("AlgShowerCam", "default", "libAtNuReco.so", "AlgConfig");

  MSG("AtmosShowerListModule", Msg::kInfo) << " Shower finder startup complete " << endl;
}


void AtmosShowerListModule::EndJob()
{
}


JobCResult AtmosShowerListModule::Reco(MomNavigator* mom)
{
  JobCResult result(JobCResult::kPassed);
  
  const char* alg_name = 0;
  const char* alg_config_name = 0;
  
  // Get CandRecord
  CandRecord *candrec = dynamic_cast<CandRecord *> (mom->GetFragment("CandRecord", "PrimaryCandidateRecord"));
  if(!candrec)
    {
      MSG("AtmosShowerListModule", Msg::kWarning) << " Failed to Find CandRecord " << endl;
      return result.SetFailed();
    }

  // Get RawRecord
  RawRecord *rr = dynamic_cast<RawRecord *>(mom->GetFragment("RawRecord"));
  if(rr==0){
    MSG("AtmosShowerListModule", Msg::kWarning) << " Failed to Find RawRecord " << endl;
    return result.SetFailed();
  }
  
  // Get RawHeader
  else {
    const RawDaqSnarlHeader* hdr = dynamic_cast<const RawDaqSnarlHeader*>(rr->GetRawHeader());
    if(hdr){
      MSG("AtmosShowerListModule", Msg::kVerbose) << "AtmosShowerListModule:Reco Run:Snarl "<<hdr->GetRun()<<":"<< hdr->GetSnarl()<<endl; 
    }
  }

  // Setup CandContext
  CandContext cx(this, mom);
  cx.SetCandRecord(candrec);
  TObjArray shwlistcxt;

  // Get CandSliceListHandle
  CandSliceListHandle* slice_list = dynamic_cast<CandSliceListHandle*>(candrec->FindCandHandle("CandSliceListHandle",fListIn.Data()));
  if( !slice_list || (slice_list && slice_list->GetNDaughters()<1) )
    {
      // Require number of CandSlices to be non-zero.
      MSG("AtmosShowerListModule", Msg::kWarning) << " Failed to Find " << fListIn.Data() << endl;
      //return JobCResult::kPassed;
      return JobCResult::kAOK;     // Otherwise no further processing of record
    }
  else { shwlistcxt.Add(slice_list); MSG("AtmosShowerListModule", Msg::kDebug) << "Found " << fListIn.Data() << endl; }

  // Get CandTrackListHandle
  CandTrackListHandle* track_list = dynamic_cast<CandTrackListHandle*>(candrec->FindCandHandle("CandTrackListHandle",fTrackListIn.Data()));
  if( !track_list || (track_list && track_list->GetNDaughters()<1) )
    {
      // Require number of CandSlices to be non-zero.
      MSG("AtmosShowerListModule", Msg::kWarning) << " Failed to Find " << fTrackListIn.Data() << endl;
    }
  else { shwlistcxt.Add(track_list); MSG("AtmosShowerListModule", Msg::kDebug) << "Found " << fTrackListIn.Data() << endl; }

  AlgFactory &af = AlgFactory::GetInstance();
  alg_name=fAlgName.Data(); alg_config_name=fAlgConfig.Data();
  AlgHandle adlh = af.GetAlgHandle(alg_name,alg_config_name);

  cx.SetDataIn(&shwlistcxt);

  clock_t dummyt;
  struct tms t1;
  struct tms t2;
  static double ticksPerSecond = sysconf(_SC_CLK_TCK);
  
  dummyt = times(&t1);
  CandShowerAtNuListHandle cslh = CandShowerAtNuList::MakeCandidate(adlh, cx);
  dummyt = times(&t2);
  //cslh.SetCPUTime((Double_t)(t2.tms_utime+t2.tms_stime-t1.tms_utime-t1.tms_stime)/ticksPerSecond);

  MSG("AtmosShowerListModule", Msg::kInfo) << "CandShowerAtNu CPUTime " << (Double_t)(t2.tms_utime+t2.tms_stime-t1.tms_utime-t1.tms_stime)/ticksPerSecond << endl;
  
  cslh.SetName(fListOut.Data());
  cslh.SetTitle(TString("Created by AtmosShowerListModule"));
  candrec->SecureCandHandle(cslh);
  
  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}


const char* AtmosShowerListModule::GetName() const
{
  const char* tmps = "AtmosShowerListModule";
  return tmps;
}


const Registry& AtmosShowerListModule::DefaultConfig() const
{
  //======================================================================
  // Supply the default configuration for the module
  //======================================================================
  static Registry r; // Default configuration for module

  // Set name of config
  std::string name = this->GetName();
  name += ".config.default";
  r.SetName(name.c_str());

  // Set values in configuration
  r.UnLockValues();
  r.Set("NameListIn","CandSliceList");
  r.Set("NameTrackListIn","CandFitTrackCamList");
  r.Set("NameListOut","CandShowerAtNuList");
  r.Set("AtmosShowerAlgorithm","AlgAtmosShowerList");
  r.Set("AtmosShowerAlgConfig","default");
  r.LockValues();

  return r;
}


void AtmosShowerListModule::Config(const Registry& r)
{
  //======================================================================
  // Configure the module given the Registry r
  //======================================================================
  const char* tmps;
  if (r.Get("NameListIn",tmps)) { fListIn = tmps; }
  if (r.Get("NameTrackListIn",tmps)) { fTrackListIn = tmps; }
  if (r.Get("NameListOut",tmps)) { fListOut = tmps; }
  if (r.Get("AtmosShowerAlgorithm",tmps)) { fAlgName = tmps; }
  if (r.Get("AtmosShowerAlgConfig",tmps)) { fAlgConfig = tmps; }
}

---