FitTrackCamListModule.cxx

Go to the documentation of this file.

// Package: CandFitTrackCam
//
// marshall@hep.phy.cam.ac.uk
#include <cassert>

// 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/CandTrack.h"
#include "RecoBase/CandTrackHandle.h"
#include "RecoBase/CandTrackListHandle.h"
#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
// This package includes
#include "CandFitTrackCam/FitTrackCamListModule.h"
#include "CandFitTrackCam/CandFitTrackCamListHandle.h"
#include "CandFitTrackCam/CandFitTrackCamList.h"

JOBMODULE(FitTrackCamListModule, "FitTrackCamListModule", "Track Fitter");
CVSID("$Id: FitTrackCamListModule.cxx,v 1.12 2010/01/06 18:44:00 rhatcher Exp $");

FitTrackCamListModule::FitTrackCamListModule()
{
  fAlgName="AlgFitTrackCamList";
  fAlgConfig="default";

  fAlgFitName="AlgFitTrackCam";

  fListIn="CandTrackSRList";

}


FitTrackCamListModule::~FitTrackCamListModule()
{
}


void FitTrackCamListModule::BeginJob()
{
  MSG("FitTrackCamListModule", Msg::kDebug) << " Track Fitter startup complete " << endl;
}


void FitTrackCamListModule::EndJob()
{
}


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

  // Get RawRecord, in particular a DaqSnarl record
  RawRecord *rr = dynamic_cast<RawRecord *>(mom->GetFragment("RawRecord",0,"DaqSnarl"));
  if(rr==0){
    MSG("FitTrackCamListModule", 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("FitTrackCamListModule", Msg::kDebug) << "Run, Snarl: "<<hdr->GetRun()<<", "<< hdr->GetSnarl()<<endl; 
    }
  }


  // Get CandTrackListHandle
  CandTrackListHandle* track_list = dynamic_cast<CandTrackListHandle*>(candrec->FindCandHandle("CandTrackListHandle",fListIn.Data()));
  //assert(track_list);
  if( !track_list || (track_list && track_list->GetNDaughters()<1) )
    {
      // Require number of CandTracks to be non-zero.
      if(!track_list) {
        MSG("FitTrackCamListModule", Msg::kDebug) << " Failed to Find " << fListIn.Data() << endl;}
      else if(track_list && track_list->GetNDaughters()<1) {
        MSG("FitTrackCamListModule", Msg::kDebug) << " No tracks found. Empty " << fListIn.Data() << endl;}

      //return JobCResult::kPassed;
      return JobCResult::kAOK;     // Otherwise no further processing of record
    }


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

  algfit_name=fAlgFitName.Data();
  AlgHandle adlhfit = af.GetAlgHandle(algfit_name,alg_config_name);
  
  AlgConfig &algfit = adlhfit.GetAlgConfig();

  algfit.UnLockValues();
  algfit.UnLockKeys();
  int fSet;
  if(!algfit.Get("UseGeoSwimmer",fSet)) {
    algfit.Set("UseGeoSwimmer",fUseGeoSwimmer);
  }
  algfit.LockValues();
  algfit.LockKeys();

  CandContext cx(this, mom);
  cx.SetCandRecord(candrec);
  cx.SetDataIn(track_list);
  CandFitTrackCamListHandle cftlh = CandFitTrackCamList::MakeCandidate(adlh, cx);
  cftlh.SetName(fListOut.Data());
  cftlh.SetTitle(TString("Created by FitTrackCamListModule"));
  candrec->SecureCandHandle(cftlh);
   
  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}


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


const Registry& FitTrackCamListModule::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","CandTrackSRList");
  r.Set("NameListOut","CandFitTrackCamList");
  r.Set("FitTrackCamAlgorithm","AlgFitTrackCamList");
  r.Set("FitTrackCamAlgConfig","default");
  r.Set("UseGeoSwimmer",1);
  r.LockValues();

  return r;
}


void FitTrackCamListModule::Config(const Registry& r)
{
//======================================================================
// Configure the module given the Registry r
//======================================================================

  const char* tmps;
  int   tmpi;
  if (r.Get("NameListIn",tmps)) { fListIn = tmps; }
  if (r.Get("NameListOut",tmps)) { fListOut = tmps; }
  if (r.Get("FitTrackCamAlgorithm",tmps)) { fAlgName = tmps; }
  if (r.Get("FitTrackCamAlgConfig",tmps)) { fAlgConfig = tmps; }
  if (r.Get("UseGeoSwimmer",tmpi)) { fUseGeoSwimmer = tmpi; }

}

---