NtpFitSAModule.cxx

Go to the documentation of this file.

//
// NtpFitSAModule.cxx
//
// A JobControl Module for filling an NtpFitSA
//
#include <iostream>
#include <cassert>

using namespace std;

#include "TParticle.h"
#include "TClonesArray.h"

#include "MessageService/MsgService.h"
#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCommand.h"
#include "MinosObjectMap/MomNavigator.h"

#include "Algorithm/AlgFactory.h"
#include "Algorithm/AlgConfig.h"
#include "Algorithm/AlgHandle.h"
#include "CandData/CandRecord.h"
#include "CandData/CandHeader.h"
#include "RawData/RawRecord.h"
#include "RecoBase/CandFitTrackListHandle.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "Candidate/CandContext.h"
#include "JobControl/JobCModuleRegistry.h"
#include "MCNtuple/NtpMCRecord.h"
#include "MCNtuple/NtpMCStdHep.h"

#include "CandFitTrackSA/CandFitTrackSAHandle.h"
#include "CandFitTrackSA/Ntp/NtpFitSARecord.h"
#include "CandFitTrackSA/Ntp/NtpFitSA.h"
#include "CandFitTrackSA/NtpModule/NtpFitSAModule.h"
#include "CandFitTrackSA/TracerSA.h"

//   Definition of static data members
//   *********************************

CVSID("$Id: NtpFitSAModule.cxx,v 1.5 2009/05/04 19:33:49 arms Exp $");
JOBMODULE(NtpFitSAModule, "NtpFitSAModule",
         "A module for filling FitSA ntuple records.");


// Definition of methods (alphabetical order)
// ***************************************************


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

const Registry& NtpFitSAModule::DefaultConfig() const {
  //
  // Purpose: Method to return default configuration.
  //
  // Arguments: none.
  //
  // Return: Registry containing default configuration
  //

  TracerSA trace("NtpFitSAModule::DefaultConfig()");;

  static Registry r;
  std::string name = this->JobCModule::GetName();
  name += ".config.default";
  r.SetName(name.c_str());

  r.UnLockValues();

  r.Set("ListOut",  "CandFitTrackSAList");      // Name of the output list

  r.LockValues();

  return r;
}

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

void NtpFitSAModule::Config(const Registry& /* r */) {
  //
  // Purpose: Configure the module given a registry.
  //
  // Arguments: Registry to use to configure the module.
  //
  // Return: none.
  //

  TracerSA trace("NtpFitSAModule::Config(const Registry&)");

  //  Int_t tmpi;
  //  Double_t tmpd;
  //  const Char_t* tmps;


}

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

JobCResult NtpFitSAModule::Reco(MomNavigator *mom) {
  //
  //  Purpose:  Create and fill ntuple record.
  //
  //  Arguments: mom.
  //
  //  Return: status.
  //

    TracerSA trace("NtpFitSAModule::Reco(MomNavigator*)");

    JobCResult result(JobCResult::kPassed); // The default result

    const RawRecord* rawrec = dynamic_cast<const RawRecord*>(mom->GetFragment("RawRecord","","DaqSnarl"));

    CandRecord* cndrec = dynamic_cast<CandRecord*>
                         (mom->GetFragment("CandRecord","PrimaryCandidateRecord"));

    // Check if there actually are records in mom before trying to use them!
    if ( !rawrec || !cndrec ) {
      return JobCResult::kFailed;
    }

    // Extract header from CandRecord and use this to create RecCandHeader
    // and NtpSRRecord.
    const CandHeader* cndhdr = cndrec -> GetCandHeader();
    const RawDaqSnarlHeader* rawhdr = dynamic_cast<const RawDaqSnarlHeader*> (rawrec -> GetRawHeader());
    RecCandHeader ntphdr(rawhdr->GetVldContext(),rawhdr->GetRun(),
                            rawhdr->GetSubRun(),rawhdr->GetRunType(),rawhdr->GetErrorCode(),
                         rawhdr->GetSnarl(),rawhdr->GetTrigSrc(),rawhdr->GetTimeFrameNum(), rawhdr->GetRemoteSpillType(),
                            cndhdr->GetEvent());

    NtpFitSARecord* ntprec = new NtpFitSARecord(ntphdr);

    this -> FillNtpFitSA(ntprec, cndrec);

    mom -> AdoptFragment(ntprec);  // pass record to mom to own

    return result;
}

//......................................................................
void NtpFitSAModule::FillNtpFitSA(NtpFitSARecord* ntprec, const CandRecord* cndrec) {
    //
    //
    TracerSA trace("NtpFitSA::FillNtpFitSA(NtpFitSARecord* ,const CandRecord*)");

    TClonesArray& fitsaarray = *(ntprec->fitsa);
    
    const CandFitTrackListHandle *fitsalisthandle = dynamic_cast <const CandFitTrackListHandle*>
                                                      (cndrec->FindCandHandle("",GetConfig().GetCharString("ListOut")));

    if ( fitsalisthandle ) {

        Int_t nfitsa = 0;

        TIter fitsaItr(fitsalisthandle->GetDaughterIterator());
        while ( const CandFitTrackSAHandle* fitsahandle = dynamic_cast<const CandFitTrackSAHandle*> (fitsaItr())) {
            // increment number of tracks
            (ntprec->ntrack)++;
            // get array from cand handle
            const TClonesArray& fits = fitsahandle->GetFits();
            NtpFitSA* ntpfitsa;
            for (Int_t i = 0; i<fits.GetEntriesFast(); i++) {
                // Uses new with placement to efficiently create strip ntp
                ntpfitsa = new((fitsaarray)[nfitsa++]) NtpFitSA(*(NtpFitSA*) fits[i]);
                // Transport information from CandFitTrackSA to fitsa ntp
                //(*ntpfitsa) = fitsahandle->GetNtpFitSA();
            }
        }
    }
}

---