AtNuFitModule.cxx

Go to the documentation of this file.

#include  "AtNuFitModule.h"

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

#include "CandData/CandRecord.h"
#include "Candidate/CandContext.h"
#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCommand.h"
#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"

#include "RecoBase/CandTrackHandle.h"
#include "RecoBase/CandTrackListHandle.h"

#include "CandFitTrackAtNuHandle.h"
#include "CandFitTrackAtNuListHandle.h"

#include <sys/time.h>

//
// $Log $
//

ClassImp(AtNuFitModule)

CVSID("$Id: AtNuFitModule.cxx,v 1.7 2006/03/28 13:54:11 blake Exp $");
JOBMODULE(AtNuFitModule, "AtNuFitModule","AtNu Track/Shower Fitter Module");

AtNuFitModule::AtNuFitModule() :
  fListIn("CandTrackListHandle"),
  fListOut("CandFitTrackAtNuListHandle")
{
  
}

AtNuFitModule::~AtNuFitModule()
{
 
}

void AtNuFitModule::BeginJob()
{

  MSG("AtNuFitModule", Msg::kInfo) << " *** AtNuFitModule::BeginJob() *** " << endl;

  /*
  AlgFactory &af = AlgFactory::GetInstance();
  af.Register("AlgFitTrackAtNuList", "default", "libAtNuReco.so", "AlgConfig");
  af.Register("AlgFitTrackAtNu", "default", "libAtNuReco.so", "AlgConfig");
  */
}

JobCResult AtNuFitModule::Reco(MomNavigator *mom)
{

  // AtNuFitModule::Reco()
  MSG("AtNuFitModule", Msg::kInfo) << " *** AtNuFitModule::Reco() *** " << endl;
  JobCResult result(JobCResult::kPassed);

  // Start Clock
  struct timeval tpbefore;
  gettimeofday(&tpbefore,0);

  // Get CandRecord
  CandRecord *candrec = dynamic_cast<CandRecord *> (mom->GetFragment("CandRecord", "PrimaryCandidateRecord"));
  if(!candrec){ 
    MSG("AtNuFitModule", Msg::kWarning) << " Failed to Fit CandRecord " << endl;
    return result.SetFailed();
  }

  // Get CandTrackListHandle
  CandTrackListHandle* track_list = dynamic_cast<CandTrackListHandle*>(candrec->FindCandHandle("CandTrackListHandle",fListIn.Data()));
  if(!track_list){
    MSG("AtNuFitModule", Msg::kWarning) << " Failed to Find " << fListIn.Data() << endl;
    return result.SetAOK();
  }

  // Make CandFitTrackAtNuHandle
  MSG("AtNuFitModule", Msg::kDebug) << " Make CandFitTrackAtNuHandle " << endl;
 
  AlgFactory &af = AlgFactory::GetInstance();
  AlgHandle ah_atnu = af.GetAlgHandle("AlgFitTrackAtNuList", "default");
  CandContext cx_atnu(this, mom);
  cx_atnu.SetCandRecord(candrec);
  cx_atnu.SetDataIn(track_list);

  CandFitTrackAtNuListHandle candfittrack = CandFitTrackAtNuList::MakeCandidate(ah_atnu, cx_atnu);
  candfittrack.SetName(fListOut.Data());
  candfittrack.SetTitle(TString("Created by AtNuFitModule from ").Append(track_list->GetName()));

  // Stop Clock
  struct timeval tpafter;
  gettimeofday(&tpafter,0);

  Double_t dtime;
  dtime = 1000.0*(tpafter.tv_sec-tpbefore.tv_sec)+0.001*(tpafter.tv_usec-tpbefore.tv_usec);
  candfittrack.SetCPUTime(dtime);

  candrec->SecureCandHandle(candfittrack);

  MSG("AtNuFitModule", Msg::kInfo) << " *** AtNuFitModule::Reco() FINISHED *** " << endl;

  return result;

}

JobCResult AtNuFitModule::Ana(const MomNavigator * /*mom*/)
{

  JobCResult result(JobCResult::kPassed);


  return result;

}

const Registry& AtNuFitModule::DefaultConfig() const
{

  MSG("AtNuFitModule", Msg::kInfo) << " *** AtNuFitModule::DefaultConfig() *** " << endl;

  static Registry r;
  r.SetName("AtNuFitModule.config.default");
  r.UnLockValues();
  r.Set("ListIn",fListIn.Data());
  r.Set("ListOut",fListOut.Data());
  r.Set("BFieldMap",201);
  r.LockValues();

  AlgFactory &af = AlgFactory::GetInstance();

  af.Register("AlgFitTrackAtNu", "default");
  AlgHandle ah_atnu = af.GetAlgHandle("AlgFitTrackAtNu","default");
  AlgConfig &ac_atnu = ah_atnu.GetAlgConfig();
  ac_atnu.UnLockValues();
  ac_atnu.Set("BFieldMap",201);
  ac_atnu.LockValues();

  return r;
}

void AtNuFitModule::Config(const Registry& r)
{

  MSG("AtNuFitModule", Msg::kInfo) << " *** AtNuFitModule::Config() *** " << endl;

  Int_t tmpint=0;
  const char* tmpchar = 0;
  if(r.Get("ListIn",tmpchar)) fListIn = tmpchar;
  if(r.Get("ListOut",tmpchar)) fListOut = tmpchar;
  MSG("AtNuFitModule", Msg::kInfo) << "   ... input: " << fListIn.Data() << endl;
  MSG("AtNuFitModule", Msg::kInfo) << "   ... output: " << fListOut.Data() << endl;

  AlgFactory &af = AlgFactory::GetInstance();
    
  AlgHandle ah_atnu = af.GetAlgHandle("AlgFitTrackAtNu","default");
  AlgConfig &ac_atnu = ah_atnu.GetAlgConfig();
  ac_atnu.UnLockValues();
  if(r.Get("BFieldMap",tmpint)) ac_atnu.Set("BFieldMap",tmpint);
  ac_atnu.LockValues();


  return;
}

void AtNuFitModule::HandleCommand(JobCommand *command)
{
  TString cmd = command->PopCmd();
  if(cmd=="Set"){
    TString opt = command->PopOpt();

  }
}

void AtNuFitModule::EndJob()
{   

}     

---