/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Mdc/MdcAlignAlg/MdcAlignAlg-00-01-04/share/distAlign/src/mergeAlign.cpp

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <iomanip>
#include <string>
#include <cstring>
#include <vector>

#include "TFile.h"
#include "TTree.h"
#include "TFolder.h"
#include "TProfile.h"
#include "TObjArray.h"
#include "TSpline.h"
#include "TPostScript.h"
#include "TLatex.h"
#include "TCanvas.h"
#include "TStyle.h"

#include "include/MdcAlignPar.h"
#include "include/MdcCosGeom.h"
#include "include/fun.h"
#include "include/ResiAlign.h"

using namespace std;

int main(int argc, char* argv[]){
     char* jobname;
     if(argc>1){
          jobname = argv[1];
     }else{
          cout << "bad argument" << endl;
          return -1;
     }

     string path = "";
     string strJob = jobname;
     cout << "strJob: " << strJob << endl;
     string::size_type ilast = strJob.find_last_of("/");
     if(string::npos != ilast){
          path = strJob.substr(0, ilast);
     }

     int alignMeth;
     string alignFile;
     string confname;
     string str;
     string strtmp;
     ifstream fjob(jobname);
     if( ! fjob.is_open() ){
          cout << "ERROR: can not read jobOption: " << jobname << endl;
          return 0;
     } else{
          cout << "Open jobOption: " << jobname << endl;
          while( getline(fjob, str) ){
               if(str.find("//", 0) != string::npos){
                    continue;
               } else if( str.find("MdcGeomSvc.alignFilePath", 0) != string::npos ){
                    string::size_type i1 = str.find_first_of("\"");
                    string::size_type i2 = str.find_last_of("\"");
                    alignFile = str.substr(i1+1, i2-i1-1);
               } else if(str.find("MdcAlignAlg.ConfigFile", 0) != string::npos){
                    string::size_type i1 = str.find_first_of("\"");
                    string::size_type i2 = str.find_last_of("\"");
                    confname = str.substr(i1+1, i2-i1-1);
               } else if(str.find("MdcAlignAlg.MdcAlignMeth", 0) != string::npos){
                    string::size_type i1 = str.find_first_of("=");
                    string::size_type i2 = str.find_last_of(";");
                    strtmp = str.substr(i1+1, i2-i1-1);
                    sscanf(strtmp.c_str(), "%d", &alignMeth);
               }
          }
     }

     MdcCosGeom* pGeom = 0;
     pGeom = new MdcCosGeom("/home/bes/wulh/document/wireconf.txt", "/home/bes/wulh/calibConst/MdcAlignPar_ini.txt" );
     pGeom -> initialize(0.0);

     TObjArray* hlist = new TObjArray(0);
     AlignBase* pAlign;
     if(0 == alignMeth) pAlign = new ResiAlign();
     else {cout << "Error AlignType" << endl; return 0;}
     pAlign->init(hlist, pGeom);

     vector<string> fhistname = getHistList();
     if(0==fhistname.size()){
          cout << "hist file path: " << path << endl;
          fhistname = getHistList(path);
     }
     for(unsigned nf=0; nf<fhistname.size(); nf++){
          TFile* fin = new TFile(fhistname[nf].c_str());
          if(!fin->IsOpen()){
               continue;
          } else{
               cout << "merge hist file " << nf << ": " << fhistname[nf] << endl;
               pAlign->mergeHist(fin);
               fin->Close();
          }
     }

     // read align par.
     MdcAlignPar* alignPar = new MdcAlignPar();
     alignPar->initAlignPar();
     alignPar->rdAlignPar(alignFile);

     // fit for alignment
     pAlign->align(alignPar);

     TFile fhist("histall.root", "recreate");
     fhist.cd();
     hlist->Write();
     fhist.Close();

     // output new alignment file
     alignPar->wrtAlignPar();

     return 0;
}

---