AlignmentModule.cxx

Go to the documentation of this file.

#include <iostream>
using std::cerr;
using std::cout;
using std::endl;

#include "Alignment/AlignmentModule.h"
#include "Alignment/AlignHists.h"

#include "Registry/Registry.h"
#include "CandData/CandRecord.h"
#include "RecoBase/CandTrackHandle.h"
#include "RecoBase/CandTrackListHandle.h"
#include "MinosObjectMap/MomNavigator.h"
#include "DataUtil/GetCandidate.h"

#include "JobControl/JobCModuleRegistry.h"
JOBMODULE(AlignmentModule,"Alignment","Aligns detector using straight tracks");

AlignmentModule::AlignmentModule()
{
    this->Init();
}

AlignmentModule::~AlignmentModule()
{
}


void AlignmentModule::Init(void) 
{
    std::string name = this->GetName();
    name += ".config.default";
    fDefConfig.SetName(name.c_str());
    // Paul's by default:
    fDefConfig.Set("TrackName","candfitthrumuonlist");
    fDefConfig.Set("TrackType","CandThruMuonListHandle");
    fDefConfig.Set("HistFileName","alignment-hists.root");
    fDefConfig.LockValues();
    fConfig = fDefConfig;
}


JobCResult AlignmentModule::Ana(const MomNavigator* mom)
{

    CandTrackListHandle* ctlh = DataUtil::GetCandidate<CandTrackListHandle>
        (mom, fConfig.GetCharString("TrackType"),
         fConfig.GetCharString("TrackName"));

    if (!ctlh) {
        cerr << "No track\n";
        return JobCResult::kAOK;
    }

    static bool been_here = false;
    if (!been_here) {           // need mom, o.w. would be in BeginJob.
        UgliGeomHandle ugh(*(ctlh->GetVldContext()));
        fSTA.Init(ugh,fConfig.GetCharString("HistFileName"));
        been_here = true;
    }

    CandTrackHandleItr cth_itr(ctlh->GetDaughterIterator());
    for (;cth_itr.IsValid();cth_itr.Next()) {
        CandTrackHandle* cth = *cth_itr;
        fSTA.AddTrack(*cth);
    }

    return JobCResult::kAOK;
}

void AlignmentModule::EndJob()
{
    cerr << "AlignmentModule::EndJob\n";
    for (int count=0; count<3; ++count) {
        double ret = fSTA.ApplyAllTracks();
        fSTA.ApplyAllOffsets();
        cout << count << " returns: " << ret << endl;
    }
    cerr << "EndJob done\n";
}

// ClassImp(AlignmentModule)

---