NtpCluster3DModule.cxx

Go to the documentation of this file.

//
// NtpCluster3DModule (based on NtpSRModule)
//
// A JobControl Module for filling an NtpCluster3DRecord
//
#include <iostream>
using namespace std;

#include "Conventions/CalStripType.h"
#include "TClonesArray.h"
#include "Cluster3D/NtpCluster3DModule.h"
#include "Cluster3D/NtpCluster3DRecord.h"
#include "Cluster3D/NtpCluster3D.h"
#include "MessageService/MsgService.h"
#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCommand.h"
#include "MinosObjectMap/MomNavigator.h"
#include "RawData/RawDigit.h"
#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "RawData/RawDigitDataBlock.h"
#include "Record/RecCandHeader.h"
#include "CandData/CandRecord.h"
#include "CandData/CandHeader.h"
#include "RecoBase/CandStripListHandle.h"
#include "RecoBase/CandStripHandle.h"
#include "Cluster3D/CandCluster3DListHandle.h"
#include "Cluster3D/CandCluster3DHandle.h"
#include "UgliGeometry/UgliGeomHandle.h"

#include <cassert>

ClassImp(NtpCluster3DModule)

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

CVSID("$Id: NtpCluster3DModule.cxx,v 1.5 2005/05/18 03:54:52 schubert Exp $");
JOBMODULE(NtpCluster3DModule, "NtpCluster3DModule",
         "A module for filling Cluster3D ntuple records.");

const Double_t NtpCluster3DModule::kCos45 = 0.70710678;  //used to convert u,v to x,y

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


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

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

  MSG("NtpSR",Msg::kDebug) << 
    "NtpCluster3DModule::DefaultConfig" << endl;

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

  /*
  r.UnLockValues();
  r.LockValues();
  */

  return r;
}

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

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

  MSG("NtpSR",Msg::kDebug) << "NtpCluster3DModule::Config" << endl;

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

  if ( r.Get("PostTrigger",tmpi)) {
    fPostTrigger = fabs((Double_t)(tmpi)*1.e-9);
  }
  if ( r.Get("PostTrigger",tmpd)) {
    fPostTrigger = fabs(tmpd*1.e-9);
  }

  */
}

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

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

  JobCResult result(JobCResult::kPassed);  
  MSG("NtpSR",Msg::kDebug) << "NtpCluster3DModule::Reco" << endl;

  // Reset maps used to associate uid of reconstructed object with array index
  fStripUidMap.clear();
  //fSliceUidMap.clear();
  fClusterUidMap.clear();
  //fTrackUidMap.clear();
  //fEventUidMap.clear();

  // Check that mom exists.
  assert(mom);

  const RawRecord* rawrec = 
   dynamic_cast<const RawRecord*>(mom->GetFragment("RawRecord","","DaqSnarl"));
  if (!rawrec) {
    MSG("NtpSR",Msg::kWarning) << "No DaqSnarl RawRecord in Mom" << endl;
    result.SetWarning().SetFailed();
    return result;
  }

  const CandRecord* cndrec = dynamic_cast<const CandRecord*>
          (mom->GetFragment("CandRecord","PrimaryCandidateRecord"));
  if (!cndrec) {
    MSG("NtpSR",Msg::kWarning) << "No PrimaryCandidateRecord in Mom" << endl;
    result.SetWarning().SetFailed();
    return result;
  }

  // Extract header from CandRecord and use this to create RecCandHeader
  // and NtpCluster3DRecord.
  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());
  NtpCluster3DRecord* ntprec = new NtpCluster3DRecord(ntphdr);
  
  this -> FillNtpStrip(ntprec,cndrec);
  this -> FillNtpCluster(ntprec,cndrec);
  this -> FillNtpClusterSummary(ntprec,cndrec);

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

  return result;

}


void NtpCluster3DModule::FillNtpStrip(NtpCluster3DRecord* /* ntprec */,const CandRecord* cndrec) {
  //
  //  Purpose:  Private method used to fill strip portion of ntuple record.
  //
  //  Arguments: NtpCluster3DRecord and CandRecord
  //  
  //  Return: status.
  // 

  MSG("NtpSR",Msg::kVerbose) << "NtpCluster3DModule::FillNtpStrip" << endl;

  const CandStripListHandle *striplisthandle 
   = dynamic_cast <const CandStripListHandle*> 
     (cndrec->FindCandHandle("CandStripListHandle"));
  if ( !striplisthandle ) return; // no strips => done

  Int_t nstrip = 0;
  TIter stripItr(striplisthandle->GetDaughterIterator());
  while ( CandStripHandle* strip=dynamic_cast<CandStripHandle*>(stripItr())) {
    Int_t uid = strip->GetUidInt();
    fStripUidMap.insert(std::make_pair(uid,nstrip));
    ++nstrip; 

  } // done with all strips

 
  return;

}

void NtpCluster3DModule::FillNtpCluster(NtpCluster3DRecord* ntprec,const CandRecord* cndrec) {
  //
  //  Purpose:  Private method used to fill cluster portion of ntuple record.
  //
  //  Arguments: pointers to NtpCluster3DRecord and CandRecord
  //  
  //  Return: status.
  // 

  MSG("NtpSR",Msg::kVerbose) << "NtpCluster3DModule::FillNtpCluster" << endl;

    const CandCluster3DListHandle *clusterlisthandle 
     = dynamic_cast <const CandCluster3DListHandle*> 
       (cndrec -> FindCandHandle("CandCluster3DListHandle","CandCluster3DList"));

  if ( !clusterlisthandle ) return; // all done


  Int_t ncluster = 0;

  CandCluster3DHandleItr clItr(clusterlisthandle->GetDaughterIterator());

  TClonesArray& clusterarray = *(ntprec->clu3D);

  while ( clItr.IsValid() ) {
     CandCluster3DHandle *cluster = clItr.Ptr();    

    Int_t uid = cluster->GetUidInt();

    fClusterUidMap.insert(std::make_pair(uid,ncluster));
    // Uses new with placement to efficiently create cluster ntp
    NtpCluster3D* ntpcluster 
         = new(clusterarray[ncluster++])NtpCluster3D(cluster->GetNStrip());
    // Fill indices of associated strips in cluster tree
    ntpcluster->index = ncluster - 1;

     CandStripHandleItr stpItr(cluster->GetDaughterIterator());
    Int_t nclusterstrip = 0;
     while ( stpItr.IsValid() ) {
      CandStripHandle *clusterstrip = stpItr.Ptr();
      Int_t uid = clusterstrip->GetUidInt();
      Int_t stripindex = fStripUidMap[uid];
      ntpcluster->AddStripAt(stripindex,nclusterstrip); // add index to strip

    // Set summed charge in cluster
      ntpcluster->ph.raw = cluster->GetCharge(CalStripType::kNone);
      ntpcluster->ph.siglin = cluster->GetCharge(CalStripType::kSigLin);
      ntpcluster->ph.sigcor = cluster->GetCharge(CalStripType::kSigCorr);
      ntpcluster->ph.pe = cluster->GetCharge(CalStripType::kPE);
      //ntpcluster->ph.sigmap = cluster->GetCharge(CalStripType::kSigMapped);
      //ntpcluster->ph.mip = cluster->GetCharge(CalStripType::kMIP);
      //ntpcluster->ph.gev = cluster->GetCharge(CalStripType::kGeV);

      nclusterstrip++;
      stpItr.Next();
     }

    clItr.Next();
  }

  return;
}


void NtpCluster3DModule::FillNtpClusterSummary(NtpCluster3DRecord* ntprec,
                                            const CandRecord* /* cndrec */) {
  //
  //  Purpose:  Private method used to fill event summary portion of ntuple 
  //            record.
  //
  //  Arguments: pointers to NtpCluster3DRecord and CandRecord
  //  
  //  Return: none.
  // 


  MSG("NtpSR",Msg::kVerbose) << "NtpCluster3DModule::FillNtpClusterSummary" << endl;

  NtpCluster3DSummary& ntpclustersummary = ntprec->hdr3D;
  ntpclustersummary.ncluster = fClusterUidMap.size();

  return;

}

---