CgemAlignAlg.cxx

Go to the documentation of this file.

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/StatusCode.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/PropertyMgr.h"
 
#include "CgemAlignAlg/CgemAlignAlg.h"
#include "EventModel/EventHeader.h"
 
#include "CalibData/EsTime/EsTimeCalibData.h"
#include "CalibData/CalibModel.h"
#include "GaudiKernel/DataSvc.h"
#include "CalibDataSvc/ICalibRootSvc.h"
#include "CalibData/CalibModel.h"
#include "CalibData/CalibBase1.h"
 
#include <iostream>
#include <fstream>
#include <cstring>
 
using namespace Event;
 
CgemAlignAlg::CgemAlignAlg(const std::string& name, ISvcLocator* pSvcLocator) :
    Algorithm(name, pSvcLocator){
 
    m_histname = "NULL";
    m_initAlignParFlg = false;
 
    declareProperty("AlignMethod", m_alignMethod);
    declareProperty("Event", m_evtType);
    declareProperty("HistOutput", m_histname);
    declareProperty("ConfigFile", m_configFile);
    declareProperty("AlignFile", m_alignFile);
    declareProperty("NumEvtDisplay", m_nEvtDisp);
    declareProperty("EventStart", EventStart=0);
    declareProperty("EventEnd", EventEnd=1E10);
    // declareProperty("CombineCosmic", m_combCosmic = false);
    }
 
CgemAlignAlg::~CgemAlignAlg(){
 
    std::ofstream fend("endAlign.out");
    fend << "CgemAlign end." << std::endl;
    fend.close();
 
    std::cout << "CgemAlignAlg End." << std::endl;
}
 
StatusCode CgemAlignAlg::initialize() {
    MsgStream log( msgSvc(), name() );
 
    std::cout<<"                                                                                                                                                               "<<std::endl;
    std::cout<<"   ██████╗ ███████╗███████╗██╗██╗██╗     ██████╗ ██████╗ ███████╗███╗   ███╗     █████╗ ██╗     ██╗ ██████╗ ███╗   ██╗███╗   ███╗███████╗███╗   ██╗████████╗   "<<std::endl;  
    std::cout<<"   ██╔══██╗██╔════╝██╔════╝██║██║██║    ██╔════╝██╔════╝ ██╔════╝████╗ ████║    ██╔══██╗██║     ██║██╔════╝ ████╗  ██║████╗ ████║██╔════╝████╗  ██║╚══██╔══╝   "<<std::endl;
    std::cout<<"   ██████╔╝█████╗  ███████╗██║██║██║    ██║     ██║  ███╗█████╗  ██╔████╔██║    ███████║██║     ██║██║  ███╗██╔██╗ ██║██╔████╔██║█████╗  ██╔██╗ ██║   ██║      "<<std::endl;
    std::cout<<"   ██╔══██╗██╔══╝  ╚════██║██║██║██║    ██║     ██║   ██║██╔══╝  ██║╚██╔╝██║    ██╔══██║██║     ██║██║   ██║██║╚██╗██║██║╚██╔╝██║██╔══╝  ██║╚██╗██║   ██║      "<<std::endl;
    std::cout<<"   ██████╔╝███████╗███████║██║██║██║    ╚██████╗╚██████╔╝███████╗██║ ╚═╝ ██║    ██║  ██║███████╗██║╚██████╔╝██║ ╚████║██║ ╚═╝ ██║███████╗██║ ╚████║   ██║      "<<std::endl;
    std::cout<<"   ╚═════╝ ╚══════╝╚══════╝╚═╝╚═╝╚═╝     ╚═════╝ ╚═════╝ ╚══════╝╚═╝     ╚═╝    ╚═╝  ╚═╝╚══════╝╚═╝ ╚═════╝ ╚═╝  ╚═══╝╚═╝     ╚═╝╚══════╝╚═╝  ╚═══╝   ╚═╝      "<<std::endl;
    std::cout<<"                                                                                                                                                               "<<std::endl;
 
    log << MSG::INFO << "CgemAlignAlg initialze() ..." << endreq;
 
    if("NULL"==m_histname){
    log << MSG::ERROR << "not defined histogram file." << endreq;
    return StatusCode::FAILURE;
    }
 
    StatusCode sc = service("CgemGeomSvc", m_cgemGeomSvc);
    if(sc != StatusCode::SUCCESS){
    log << MSG::ERROR << "can not use CgemGeomSvc" << endreq;
    }
 
    sc = service("CgemCalibFunSvc", m_cgemFunSvc);
    if( sc != StatusCode::SUCCESS ){
    log << MSG::FATAL << "can not use CgemCalibFunSvc" << endreq;
    }
 
    // initialize m_param
    initParam();
 
    // read cgem config parameters
    int i;
    std::string strconfig;
    std::string strcomment;
    std::string strTes;
    int fgTes[50];
    for(i=0; i<50; i++) fgTes[i] = -999;
    ifstream fconfig( m_configFile.c_str() );
    if( ! fconfig.is_open() ){
    log << MSG::WARNING << "can not open config file " << m_configFile
        << ". Use defalt config parameters" << endreq;
    } else {
    log << MSG::INFO << "Open config file " << m_configFile << endreq;
    while( fconfig >> strconfig ){
        if('#' == strconfig[0]){
        getline(fconfig, strcomment);
        }  else if("TesMin" == strconfig){
        fconfig >> m_param.tesMin;
        }
 
    }
    fconfig.close();
    } 
 
    // set event type
    m_param.particle = m_evtType;
 
    m_alignPar = new CgemAlignPar();
    m_alignPar -> rdAlignPar(m_alignFile);
    cout << std::setprecision(4) << std::fixed;
    cout<<"   The parameters are:"<<endl;
    cout<<"   Layer        Dx(mm)         Dy(mm)         Dz(mm)         Rx(mrad)       Ry(mrad)       Rz(mrad)   "<<endl;
    cout<<"   ===============================================================================================   "<<endl;
    cout<<"   0            "<<std::setw(15)<<m_alignPar->getDx(0)<<std::setw(15)<<m_alignPar->getDy(0)<<std::setw(15)<<m_alignPar->getDz(0)<<std::setw(15)<<m_alignPar->getRx(0)*1000.<<std::setw(15)<<m_alignPar->getRy(0)*1000.<<std::setw(15)<<m_alignPar->getRz(0)*1000.<<endl;
    cout<<"   1            "<<std::setw(15)<<m_alignPar->getDx(1)<<std::setw(15)<<m_alignPar->getDy(1)<<std::setw(15)<<m_alignPar->getDz(1)<<std::setw(15)<<m_alignPar->getRx(1)*1000.<<std::setw(15)<<m_alignPar->getRy(1)*1000.<<std::setw(15)<<m_alignPar->getRz(1)*1000.<<endl;
    cout<<"   2            "<<std::setw(15)<<m_alignPar->getDx(2)<<std::setw(15)<<m_alignPar->getDy(2)<<std::setw(15)<<m_alignPar->getDz(2)<<std::setw(15)<<m_alignPar->getRx(2)*1000.<<std::setw(15)<<m_alignPar->getRy(2)*1000.<<std::setw(15)<<m_alignPar->getRz(2)*1000.<<endl;
    cout<<"   3            "<<std::setw(15)<<m_alignPar->getDx(3)<<std::setw(15)<<m_alignPar->getDy(3)<<std::setw(15)<<m_alignPar->getDz(3)<<std::setw(15)<<m_alignPar->getRx(3)*1000.<<std::setw(15)<<m_alignPar->getRy(3)*1000.<<std::setw(15)<<m_alignPar->getRz(3)*1000.<<endl;
    cout<<"   4            "<<std::setw(15)<<m_alignPar->getDx(4)<<std::setw(15)<<m_alignPar->getDy(4)<<std::setw(15)<<m_alignPar->getDz(4)<<std::setw(15)<<m_alignPar->getRx(4)*1000.<<std::setw(15)<<m_alignPar->getRy(4)*1000.<<std::setw(15)<<m_alignPar->getRz(4)*1000.<<endl;
    cout<<"   5            "<<std::setw(15)<<m_alignPar->getDx(5)<<std::setw(15)<<m_alignPar->getDy(5)<<std::setw(15)<<m_alignPar->getDz(5)<<std::setw(15)<<m_alignPar->getRx(5)*1000.<<std::setw(15)<<m_alignPar->getRy(5)*1000.<<std::setw(15)<<m_alignPar->getRz(5)*1000.<<endl;
    cout<<"   ===============================================================================================   "<<endl;
    cout<<"\n"<<endl;
 
    m_cgemevt = new CgemAliEvent();
    m_cgemevt -> setParam(m_param);
 
    //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
    m_hlist = new TObjArray(0);
 
    if( 0 == m_alignMethod ){
    m_align = new CgemMilleAlign();
    }
    m_align->setParam(m_param);
    m_align->initialize(m_hlist, m_cgemGeomSvc, m_cgemFunSvc);
 
    log << MSG::INFO << "CgemAlignAlg initialze() successfully !!" << endreq;
 
    m_nEvt = 0;
 
    return StatusCode::SUCCESS;
}
 
StatusCode CgemAlignAlg::execute() {
    MsgStream log(msgSvc(), name());
    log << MSG::INFO << "CgemAlignAlg execute()" << endreq;
 
    // display event number for debug
    SmartDataPtr<Event::EventHeader> eventHeader(eventSvc(),"/Event/EventHeader");
    if (!eventHeader) {
    log << MSG::FATAL << "Could not find Event Header" << endreq;
    return( StatusCode::FAILURE);
    }
    int iRun = eventHeader->runNumber();
    int iEvt = eventHeader->eventNumber();
    if(0 == (m_nEvt % m_nEvtDisp)){
    std::cout << "Run " << iRun << ", Event " << iEvt << ",   Total Event number " << m_nEvt << endl;
    }
 
    if(m_nEvt<EventStart||m_nEvt>EventEnd) 
    {
        cout<<"Skim the event!!"<<endl;
    m_nEvt++;
    return StatusCode::SUCCESS;
    }
    else
    {
 
    if( ! m_initAlignParFlg ){
        m_initAlignParFlg = true;
    }
 
    m_cgemevt -> setRecEvent();
 
    if (!(m_align->fillHist(m_cgemevt, m_alignPar))) {
        m_cgemevt->clear();
        return(StatusCode::SUCCESS);
    }
    m_cgemevt->clear();
 
    m_nEvt++;
    return StatusCode::SUCCESS;
    }
 
}
 
StatusCode CgemAlignAlg::finalize() {
    MsgStream log(msgSvc(), name());
    log << MSG::INFO << "CgemAlignAlg finalize()" << endreq;
 
    m_align->updateAlignPar(m_alignPar);
    m_alignPar->wrtAlignPar();
 
    // write histogram file
    TFile fhist(m_histname.c_str(), "recreate");
    m_hlist -> Write();
    fhist.Close();
    std::cout << m_histname << " was written" << std::endl;
 
    return StatusCode::SUCCESS;
}
 
 
void CgemAlignAlg::initParam(){
    m_param.particle = 0;
    m_param.nEsFlag = 0;
    m_param.tesMin = 0.0;
    m_param.tesMax = 9999.0;
    m_param.resiType = 0;
    m_param.fgAdjacLayerCut = 0;
    m_param.fgBoundLayerCut = 0;
    m_param.hitStatCut = 0;
    m_param.nTrkCut[0] = 2;
    m_param.nTrkCut[1] = 2;
    m_param.nHitLayCut = 35;
    m_param.nHitCut = 70;
    m_param.ptCut[0] = 0.0;
    m_param.ptCut[1] = 100.0;
    m_param.costheCut[0] = -1.0;
    m_param.costheCut[1] = 1.0;
    m_param.drCut = 50.0;       // mm
    m_param.dzCut = 300.0;      // mm
    m_param.maxDocaInner = 8.0;     // mm
    m_param.maxDocaOuter = 12.0;    // mm
    m_param.maxResi = 1.0;      // mm
 
}