SliceFromChopModule.cxx

Go to the documentation of this file.

// $Id: SliceFromChopModule.cxx,v 1.5 2007/03/01 17:34:30 rhatcher Exp $
//
// FILL_IN: [Document your code!!]
//
// n.tagg1@physics.ox.ac.uk
#include "SliceFromChopModule.h"
#include "DigitVector.h"
#include "CandChopListHandle.h"
#include "CandChopList.h"

#include "RecoBase/CandSliceListHandle.h"
#include "RecoBase/CandSliceList.h"
#include "RecoBase/CandStripListHandle.h"
#include "RecoBase/CandStripList.h"

#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "JobControl/JobCModuleRegistry.h" // For JOBMODULE macro
#include "Algorithm/AlgFactory.h"
#include "CandData/CandRecord.h"
#include "Candidate/CandContext.h"
#include "Navigation/XxxItr.h"

#include <TFolder.h>
#include <TROOT.h>

JOBMODULE(SliceFromChopModule, "SliceFromChopModule",
          "Selects the biggest slice and replaces the CandDigitList with it");
CVSID("$Id: SliceFromChopModule.cxx,v 1.5 2007/03/01 17:34:30 rhatcher Exp $");
//......................................................................

SliceFromChopModule::SliceFromChopModule()
{

  // A copy of Peter's StripStat code, copied from th StripSRListModule constructor.
  // Jan, 2006
  TFolder *lf = dynamic_cast<TFolder *>(gROOT->FindObject("Loon"));
  if (lf==0) {
    MSG("Chop", Msg::kDebug) << "Creating Loon TFolder" << endl;
    lf = gROOT->GetRootFolder()->AddFolder("Loon", "Loon analysis");
    gROOT->GetListOfBrowsables()->Add(lf, "Loon");
  }
  TFolder* statFolder = lf->AddFolder("CandStripSR", "CandStripSR statistics");
  statFolder->Add(&fStats);

}

//......................................................................

SliceFromChopModule::~SliceFromChopModule()
{
}

//......................................................................
void SliceFromChopModule::EndJob() 
{ 
  // For Peter's stats.
  fStats.Print();
}


//......................................................................

JobCResult SliceFromChopModule::Reco(MomNavigator* mom)
{
  JobCResult result = JobCResult::kPassed;

  // config.
  Registry& cfg = this->GetConfig();
  const char* chopListIn = cfg.GetCharString("ChopListIn");
  const char* sliceListOut = cfg.GetCharString("SliceListOut");
  const char* stripListOut = cfg.GetCharString("StripListOut");

  // Find PrimaryCandidateRecord fragment in MOM.
  CandRecord *candrec = dynamic_cast<CandRecord *>
    (mom->GetFragment("CandRecord", "PrimaryCandidateRecord"));
  if (candrec == 0) {
    MSG("SliceFromChop", Msg::kWarning) << "No PrimaryCandidateRecord in MOM."
                                        << endl;
    result.SetWarning().SetFailed();
    return result;
  }

  // find the list of slices.
  CandChopListHandle *chopList = dynamic_cast<CandChopListHandle*>
    (candrec->FindCandHandle("CandChopListHandle", chopListIn));

  // Set up the restripper algorithm
  AlgFactory &af = AlgFactory::GetInstance();
  AlgHandle adlh = af.GetAlgHandle( cfg.GetCharString("SliceFromChopAlgorithm"),
                                    cfg.GetCharString("SliceFromChopAlgConfig") );
  CandContext cx(this, mom);
  cx.SetDataIn(chopList);
  cx.SetCandRecord(candrec);

  CandSliceListHandle outSliceList = CandSliceList::MakeCandidate(adlh,cx);
  outSliceList.SetName(sliceListOut);
  outSliceList.SetTitle("Slices created from Chops"); 
  outSliceList.SetCandRecord(candrec);
  candrec->SecureCandHandle(outSliceList);
  

  // We now have a list of slices. Now also make a global strip list.
  {
  adlh = af.GetAlgHandle( cfg.GetCharString("StripFromSliceAlgorithm"),
                          cfg.GetCharString("StripFromSliceAlgConfig") );
  cx.SetDataIn(&outSliceList);
  cx.SetCandRecord(candrec);

  CandStripListHandle outStrips = CandStripList::MakeCandidate(adlh,cx);
  outStrips.SetName(stripListOut);
  outStrips.SetTitle("Strips aggregated from Slices"); 
  outStrips.SetCandRecord(candrec);
  candrec->SecureCandHandle(outStrips);
  
  }

  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}

//......................................................................



JobCResult SliceFromChopModule::Ana(const MomNavigator* /*mom*/)
{
  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}

//......................................................................

const Registry& SliceFromChopModule::DefaultConfig() const
{
  //======================================================================
  // Supply the default configuration for the module
  //======================================================================
  static Registry r; // Default configuration for module
  
  // Set name of config
  std::string name = this->JobCModule::GetName();
  name += ".config.default";
  r.SetName(name.c_str());
  
  // Set values in configuration
  r.UnLockValues();
  r.Set("SliceFromChopAlgorithm","AlgSliceListFromChopList");
  r.Set("SliceFromChopAlgConfig","default");
  r.Set("StripFromSliceAlgorithm","AlgStripListFromSliceList");
  r.Set("StripFromSliceAlgConfig","default");
  r.Set("ChopListIn",     "candchoplist");
  r.Set("SliceListOut",   "CandSliceList");
  r.Set("StripListOut",   "candstriplist");

  r.LockValues();

  return r;
}

---