X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITSAlignMille2.cxx;h=80c6a439e33a4966b34901e90f6fa78ec57449fd;hb=21b7b81622540b9a21fef5dfe7dd29bed8f9e25f;hp=5be06f9e45582ef97d193c1746d06d070b85dd3b;hpb=22c263366fe1ac37a3577a05e8b9a67a27f51338;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITSAlignMille2.cxx b/ITS/AliITSAlignMille2.cxx index 5be06f9e455..80c6a439e33 100644 --- a/ITS/AliITSAlignMille2.cxx +++ b/ITS/AliITSAlignMille2.cxx @@ -1,3013 +1,5011 @@ -/************************************************************************** - * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. * - * * - * Author: The ALICE Off-line Project. * - * Contributors are mentioned in the code where appropriate. * - * * - * Permission to use, copy, modify and distribute this software and its * - * documentation strictly for non-commercial purposes is hereby granted * - * without fee, provided that the above copyright notice appears in all * - * copies and that both the copyright notice and this permission notice * - * appear in the supporting documentation. The authors make no claims * - * about the suitability of this software for any purpose. It is * - * provided "as is" without express or implied warranty. * - **************************************************************************/ - -/* $Id$ */ - -//----------------------------------------------------------------------------- -// -// Interface to AliMillePede2 alignment class for the ALICE ITS detector -// -// ITS specific alignment class which interface to AliMillepede. -// For each track ProcessTrack calculates the local and global derivatives -// at each hit and fill the corresponding local equations. Provide methods for -// fixing or constraining detection elements for best results. -// -// author M. Lunardon (thanks to J. Castillo), ruben.shahoyan@cern.ch -//----------------------------------------------------------------------------- - -#include -#include -#include -#include -#include -#include -#include -#include "AliITSAlignMille2.h" -#include "AliITSgeomTGeo.h" -#include "AliGeomManager.h" -#include "AliMillePede2.h" -#include "AliTrackPointArray.h" -#include "AliAlignObjParams.h" -#include "AliLog.h" -#include "AliTrackFitterRieman.h" -#include "AliITSAlignMille2Constraint.h" -#include "AliITSAlignMille2ConstrArray.h" -#include "AliITSresponseSDD.h" - -ClassImp(AliITSAlignMille2) - -const Char_t* AliITSAlignMille2::kRecKeys[] = { - "GEOMETRY_FILE", - "SUPERMODULE_FILE", - "CONSTRAINTS_REFERENCE_FILE", - "PREALIGNMENT_FILE", - "PRECALIBSDD_FILE", - "INITCALBSDD_FILE", - "SET_GLOBAL_DELTAS", - "CONSTRAINT_LOCAL", - "MODULE_VOLUID", - "MODULE_INDEX", - "SET_PSEUDO_PARENTS", - "SET_TRACK_FIT_METHOD", - "SET_MINPNT_TRA", - "SET_NSTDDEV", - "SET_RESCUT_INIT", - "SET_RESCUT_OTHER", - "SET_LOCALSIGMAFACTOR", - "SET_STARTFAC", - "SET_B_FIELD", - "SET_SPARSE_MATRIX", - "REQUIRE_POINT", - "CONSTRAINT_ORPHANS", - "CONSTRAINT_SUBUNITS", - "APPLY_CONSTRAINT" -}; - - -//======================================================================================================== - -AliITSAlignMille2* AliITSAlignMille2::fgInstance = 0; -Int_t AliITSAlignMille2::fgInstanceID = 0; - -//________________________________________________________________________________________________________ -AliITSAlignMille2::AliITSAlignMille2(const Char_t *configFilename ) -: TObject(), - fMillepede(0), - fStartFac(16.), - fResCutInitial(100.), - fResCut(100.), - fNGlobal(0), - fNLocal(4), - fNStdDev(3), - fIsMilleInit(kFALSE), - fAllowPseudoParents(kFALSE), - // - fCurrentModule(0), - fTrack(0), - fTrackBuff(0), - fCluster(), - fGlobalDerivatives(0), - // - fMinNPtsPerTrack(3), - fInitTrackParamsMeth(1), - fTotBadLocEqPoints(0), - fRieman(0), - // - fConstraints(0), - // - fUseGlobalDelta(kFALSE), - fRequirePoints(kFALSE), - fTempExcludedModule(-1), - // - fGeometryFileName("geometry.root"), - fPreAlignmentFileName(""), - fConstrRefFileName(""), - fGeoManager(0), - fIsConfigured(kFALSE), - fPreAlignQF(0), -// - fCorrectSDD(0), - fInitialRecSDD(0), - fPrealignment(0), - fConstrRef(0), - fMilleModule(2), - fSuperModule(2), - fNModules(0), - fNSuperModules(0), - fUsePreAlignment(kFALSE), - fBOn(kFALSE), - fBField(0.0), - fBug(0), - fMilleVersion(2) -{ - /// main constructor that takes input from configuration file - for (int i=3;i--;) fSigmaFactor[i] = 1.0; - // - // new RS - for (Int_t i=0; i<6; i++) { - fNReqLayUp[i]=0; - fNReqLayDown[i]=0; - fNReqLay[i]=0; - } - for (Int_t i=0; i<3; i++) { - fNReqDetUp[i]=0; - fNReqDetDown[i]=0; - fNReqDet[i]=0; - } - // - Int_t lc=LoadConfig(configFilename); - if (lc) { - AliError(Form("Error %d loading configuration from %s",lc,configFilename)); - exit(1); - } - // - fMillepede = new AliMillePede2(); - fgInstance = this; - fgInstanceID++; - // -} - -//________________________________________________________________________________________________________ -AliITSAlignMille2::~AliITSAlignMille2() -{ - /// Destructor - if (fMillepede) delete fMillepede; fMillepede = 0; - if (fGlobalDerivatives) delete[] fGlobalDerivatives; fGlobalDerivatives = 0; - if (fRieman) delete fRieman; fRieman = 0; - if (fPrealignment) delete fPrealignment; fPrealignment = 0; - if (fConstrRef) delete fConstrRef; fConstrRef = 0; - if (fCorrectSDD) delete fCorrectSDD; fCorrectSDD = 0; - if (fInitialRecSDD) delete fInitialRecSDD; fInitialRecSDD = 0; - fTrackBuff.Delete(); - fConstraints.Delete(); - fMilleModule.Delete(); - fSuperModule.Delete(); - if (--fgInstanceID==0) fgInstance = 0; -} - -/////////////////////////////////////////////////////////////////////// - - -TObjArray* AliITSAlignMille2::GetConfigRecord(FILE* stream, TString& recTitle, TString& recOpt, Bool_t rew) -{ - // read new record from config file - TString record; - static TObjArray* recElems = 0; - if (recElems) {delete recElems; recElems = 0;} - // - TString keyws = recTitle; - if (!keyws.IsNull()) { - keyws.ToUpper(); - // keyws += " "; - } - while (record.Gets(stream)) { - int cmt=record.Index("#"); - if (cmt>=0) record.Remove(cmt); // skip comment - record.ReplaceAll("\t"," "); - record.ReplaceAll("\r"," "); - record.Remove(TString::kBoth,' '); - if (record.IsNull()) continue; // nothing to decode - if (!keyws.IsNull() && !record.BeginsWith(keyws.Data())) continue; // specific record was requested - // - recElems = record.Tokenize(" "); - recTitle = recElems->At(0)->GetName(); - recTitle.ToUpper(); - recOpt = recElems->GetLast()>0 ? recElems->At(1)->GetName() : ""; - break; - } - if (rew || !recElems) rewind(stream); - return recElems; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::CheckConfigRecords(FILE* stream) -{ - TString record,recTitle; - int lineCnt = 0; - rewind(stream); - while (record.Gets(stream)) { - int cmt=record.Index("#"); - lineCnt++; - if (cmt>=0) record.Remove(cmt); // skip comment - record.ReplaceAll("\t"," "); - record.ReplaceAll("\r"," "); - record.Remove(TString::kBoth,' '); - if (record.IsNull()) continue; // nothing to decode - // extract keyword - int spc = record.Index(" "); - if (spc>0) recTitle = record(0,spc); - else recTitle = record; - recTitle.ToUpper(); - Bool_t strOK = kFALSE; - for (int ik=kNKeyWords;ik--;) if (recTitle == kRecKeys[ik]) {strOK = kTRUE; break;} - if (strOK) continue; - // - AliError(Form("Unknown keyword %s at line %d", - recTitle.Data(),lineCnt)); - return -1; - // - } - // - rewind(stream); - return 0; -} - - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::LoadConfig(const Char_t *cfile) -{ - // return 0 if success - // 1 if error in module index or voluid - // - FILE *pfc=fopen(cfile,"r"); - if (!pfc) return -1; - // - TString record,recTitle,recOpt,recExt; - Int_t nrecElems,irec; - TObjArray *recArr=0; - // - fNModules = 0; - Bool_t stopped = kFALSE; - // - if (CheckConfigRecords(pfc)<0) return -1; - // - while(1) { - // - // ============= 1: we read some obligatory records in predefined order ================ - // - recTitle = kRecKeys[kGeomFile]; - if ( !GetConfigRecord(pfc,recTitle,recOpt,1) || - (fGeometryFileName=recOpt).IsNull() || - gSystem->AccessPathName(recOpt.Data()) || - InitGeometry() ) - { AliError("Failed to find/load Geometry"); stopped = kTRUE; break;} - // - recTitle = kRecKeys[kSuperModileFile]; - if ( !GetConfigRecord(pfc,recTitle,recOpt,1) || - recOpt.IsNull() || - gSystem->AccessPathName(recOpt.Data()) || - LoadSuperModuleFile(recOpt.Data())) - { AliError("Failed to find/load SuperModules"); stopped = kTRUE; break;} - // - recTitle = kRecKeys[kConstrRefFile]; // LOCAL_CONSTRAINTS are defined wrt these deltas - if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) { - if (recOpt.IsNull() || recOpt=="IDEAL") SetConstraintWrtRef( "IDEAL" ); - else if (gSystem->AccessPathName(recOpt.Data()) || SetConstraintWrtRef(recOpt.Data()) ) - { AliError("Failed to load reference deltas for local constraints"); stopped = kTRUE; break;} - } - // - recTitle = kRecKeys[kPrealignFile]; - if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) - if ( (fPreAlignmentFileName=recOpt).IsNull() || - gSystem->AccessPathName(recOpt.Data()) || - ApplyToGeometry()) - { AliError(Form("Failed to load Prealignment file %s",recOpt.Data())); stopped = kTRUE; break;} - // - recTitle = kRecKeys[kPreCalSDDFile]; - if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) { - if ( recOpt.IsNull() || gSystem->AccessPathName(recOpt.Data()) ) {stopped = kTRUE; break;} - AliInfo(Form("Using %s for SDD precalibration",recOpt.Data())); - TFile* precfi = TFile::Open(recOpt.Data()); - if (!precfi->IsOpen()) {stopped = kTRUE; break;} - fCorrectSDD = (AliITSresponseSDD*)precfi->Get("AliITSresponseSDD"); - precfi->Close(); - delete precfi; - if (!fCorrectSDD) {AliError("Precalibration SDD object is not found"); stopped = kTRUE; break;} - } - // - recTitle = kRecKeys[ kInitCalSDDFile ]; - if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) { - if ( recOpt.IsNull() || gSystem->AccessPathName(recOpt.Data()) ) {stopped = kTRUE; break;} - AliInfo(Form("Using %s as SDD calibration used in TrackPoints",recOpt.Data())); - TFile* precf = TFile::Open(recOpt.Data()); - if (!precf->IsOpen()) {stopped = kTRUE; break;} - fInitialRecSDD = (AliITSresponseSDD*)precf->Get("AliITSresponseSDD"); - precf->Close(); - delete precf; - if (!fInitialRecSDD) {AliError("Initial Calibration SDD object is not found"); stopped = kTRUE; break;} - } - // - recTitle = kRecKeys[ kGlobalDeltas ]; - if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) SetUseGlobalDelta(kTRUE); - // - // =========== 2: see if there are local gaussian constraints defined ===================== - // Note that they should be loaded before the modules declaration - // - recTitle = kRecKeys[ kConstrLocal ]; - while( (recArr=GetConfigRecord(pfc,recTitle,recOpt,0)) ) { - nrecElems = recArr->GetLast()+1; - if (recOpt.IsFloat()) {stopped = kTRUE; break;} // wrong name - if (GetConstraint(recOpt.Data())) { - AliError(Form("Existing constraint %s repeated",recOpt.Data())); - stopped = kTRUE; break; - } - recExt = recArr->At(2)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - double val = recExt.Atof(); - recExt = recArr->At(3)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - double err = recExt.Atof(); - int nwgh = nrecElems - 4; - double *wgh = new double[nwgh]; - for (nwgh=0,irec=4;irecAt(irec)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - wgh[nwgh++] = recExt.Atof(); - } - if (stopped) {delete[] wgh; break;} - // - ConstrainLocal(recOpt.Data(),wgh,nwgh,val,err); - delete[] wgh; - // - } // end while for loop over local constraints - if (stopped) break; - // - // =========== 3: now read modules to align =================================== - // - rewind(pfc); - while( (recArr=GetConfigRecord(pfc,recTitle="",recOpt,0)) ) { - if (!(recTitle==kRecKeys[ kModVolID ] || recTitle==kRecKeys[ kModIndex ])) continue; - // Expected format: MODULE id tolX tolY tolZ tolPsi tolTh tolPhi [[sigX sigY sigZ] extra params] - // where tol* is the tolerance (sigma) for given DOF. 0 means fixed - // sig* is the scaling parameters for the errors of the clusters of this module - // extra params are defined for specific modules, e.g. t0 and vdrift corrections of SDD - // - nrecElems = recArr->GetLast()+1; - if (nrecElems<2 || !recOpt.IsDigit()) {stopped = kTRUE; break;} - int idx = recOpt.Atoi(); - UShort_t voluid = (idx<=kMaxITSSensID) ? GetModuleVolumeID(idx) : idx; - AliITSAlignMille2Module* mod = 0; - // - if (voluid>=kMinITSSupeModuleID) { // custom supermodule - for (int j=0; jGetVolumeID()) { - mod = new AliITSAlignMille2Module(*GetSuperModule(j)); - // the matrix might be updated in case some prealignment was applied, check - TGeoHMatrix* mup = AliGeomManager::GetMatrix(mod->GetName()); - if (mup) *(mod->GetMatrix()) = *mup; - fMilleModule.AddAtAndExpand(mod,fNModules); - break; - } - } - } - else if (idx<=kMaxITSSensVID) { - mod = new AliITSAlignMille2Module(voluid); - fMilleModule.AddAtAndExpand(mod,fNModules); - } - if (!mod) {stopped = kTRUE; break;} // bad volid - // - // geometry variation settings - for (int i=0;i= nrecElems) break; - recExt = recArr->At(irec)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - mod->SetFreeDOF(i, recExt.Atof() ); - } - if (stopped) break; - // - // scaling factors for cluster errors - // first set default ones - for (int i=0;i<3;i++) mod->SetSigmaFactor(i, fSigmaFactor[i]); - for (int i=0;i<3;i++) { - irec = i+8; - if (irec >= nrecElems) break; - recExt = recArr->At(irec)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - mod->SetSigmaFactor(i, recExt.Atof() ); - } - if (stopped) break; - // - mod->SetGeomParamsGlobal(fUseGlobalDelta); - // now comes special detectors treatment - if (mod->IsSDD()) { - double vl = 0; - if (nrecElems>11) { - recExt = recArr->At(11)->GetName(); - if (recExt.IsFloat()) vl = recExt.Atof(); - else {stopped = kTRUE; break;} - irec = 11; - } - mod->SetFreeDOF(AliITSAlignMille2Module::kDOFT0,vl); - // - vl = 0; - if (nrecElems>12) { - recExt = recArr->At(12)->GetName(); - if (recExt.IsFloat()) vl = recExt.Atof(); - else {stopped = kTRUE; break;} - irec = 12; - } - mod->SetFreeDOF(AliITSAlignMille2Module::kDOFDV,vl); - } - // - mod->SetUniqueID(fNModules); - mod->EvaluateDOF(); - fNModules++; - // - // now check if there are local constraints on this module - for (++irec;irecAt(irec)->GetName(); - if (recExt.IsFloat()) {stopped=kTRUE;break;} - AliITSAlignMille2ConstrArray* cstr = (AliITSAlignMille2ConstrArray*)GetConstraint(recExt.Data()); - if (!cstr) { - AliInfo(Form("No Local constraint %s was declared",recExt.Data())); - stopped=kTRUE; - break; - } - cstr->AddModule(mod); - } - if (stopped) break; - } // end while for loop over modules - if (stopped) break; - // - if (fNModules==0) {AliError("Failed to find any MODULE"); stopped = kTRUE; break;} - BuildHierarchy(); // preprocess loaded modules - // - // =========== 4: the rest may come in arbitrary order ======================================= - rewind(pfc); - while ( (recArr=GetConfigRecord(pfc,recTitle="",recOpt,0))!=0 ) { - // - nrecElems = recArr->GetLast()+1; - // - // some simple flags ----------------------------------------------------------------------- - // - if (recTitle == kRecKeys[ kPseudoParents ]) SetAllowPseudoParents(kTRUE); - // - // some optional parameters ---------------------------------------------------------------- - else if (recTitle == kRecKeys[ kTrackFitMethod ]) { - if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} - SetInitTrackParamsMeth(recOpt.Atoi()); - } - // - else if (recTitle == kRecKeys[ kMinPntTrack ]) { - if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} - fMinNPtsPerTrack = recOpt.Atoi(); - } - // - else if (recTitle == kRecKeys[ kNStDev ]) { - if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} - fNStdDev = (Int_t)recOpt.Atof(); - } - // - else if (recTitle == kRecKeys[ kResCutInit ]) { - if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} - fResCutInitial = recOpt.Atof(); - } - // - else if (recTitle == kRecKeys[ kResCutOther ]) { - if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} - fResCut = recOpt.Atof(); - } - // - else if (recTitle == kRecKeys[ kLocalSigFactor ]) { //------------------------- - for (irec=0;irec<3;irec++) if (nrecElems>irec+1) { - fSigmaFactor[irec] = ((TObjString*)recArr->At(irec+1))->GetString().Atof(); - if (fSigmaFactor[irec]<=0.) stopped = kTRUE; - } - if (stopped) break; - } - // - else if (recTitle == kRecKeys[ kStartFactor ]) { //------------------------- - if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} - fStartFac = recOpt.Atof(); - } - // - else if (recTitle == kRecKeys[ kBField ]) { //------------------------- - if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} - fBField = recOpt.Atof(); - if (fBField>0) { - fBOn = kTRUE; - fNLocal = 5; // helices - fRieman = new AliTrackFitterRieman(); - } - else { - fBField = 0.0; - fBOn = kFALSE; - fNLocal = 4; - } - } - // - else if (recTitle == kRecKeys[ kSparseMatrix ]) { // matrix solver type - // - AliMillePede2::SetGlobalMatSparse(kTRUE); - if (recOpt.IsNull()) continue; - // solver type and settings - if (recOpt == "MINRES") AliMillePede2::SetIterSolverType( AliMinResSolve::kSolMinRes ); - else if (recOpt == "FGMRES") AliMillePede2::SetIterSolverType( AliMinResSolve::kSolFGMRes ); - else {stopped = kTRUE; break;} - // - if (nrecElems>=3) { // preconditioner type - recExt = recArr->At(2)->GetName(); - if (!recExt.IsDigit()) {stopped = kTRUE; break;} - AliMillePede2::SetMinResPrecondType( recExt.Atoi() ); - } - // - if (nrecElems>=4) { // tolerance - recExt = recArr->At(3)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - AliMillePede2::SetMinResTol( recExt.Atof() ); - } - // - if (nrecElems>=5) { // maxIter - recExt = recArr->At(4)->GetName(); - if (!recExt.IsDigit()) {stopped = kTRUE; break;} - AliMillePede2::SetMinResMaxIter( recExt.Atoi() ); - } - } - // - else if (recTitle == kRecKeys[ kRequirePoint ]) { //------------------------- - // syntax: REQUIRE_POINT where ndet updw nreqpts - // where = LAYER or DETECTOR - // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) - // updw = 1 for Y>0, -1 for Y<0, 0 if not specified - // nreqpts = minimum number of points of that type - if (nrecElems>=5) { - recOpt.ToUpper(); - int lr = ((TObjString*)recArr->At(2))->GetString().Atoi() - 1; - int hb = ((TObjString*)recArr->At(3))->GetString().Atoi(); - int np = ((TObjString*)recArr->At(4))->GetString().Atoi(); - fRequirePoints = kTRUE; - if (recOpt == "LAYER") { - if (lr<0 || lr>5) {stopped = kTRUE; break;} - if (hb>0) fNReqLayUp[lr] = np; - else if (hb<0) fNReqLayDown[lr] = np; - else fNReqLay[lr] = np; - } - else if (recOpt == "DETECTOR") { - if (lr<0 || lr>2) {stopped = kTRUE; break;} - if (hb>0) fNReqDetUp[lr] = np; - else if (hb<0) fNReqDetDown[lr] = np; - else fNReqDet[lr] = np; - } - else {stopped = kTRUE; break;} - } - else {stopped = kTRUE; break;} - } - // - // global constraints on the subunits/orphans - else if (recTitle == kRecKeys[ kConstrOrphans ]) { //------------------------ - // expect CONSTRAINT_ORPHANS MEAN/MEDIAN Value parID0 ... parID1 ... - if (nrecElems<4) {stopped = kTRUE; break;} - recExt = recArr->At(2)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - double val = recExt.Atof(); - UInt_t pattern = 0; - for (irec=3;irecAt(irec)->GetName(); - if (!recExt.IsDigit()) {stopped = kTRUE; break;} - pattern |= 0x1 << recExt.Atoi(); - } - if (stopped) break; - if (recOpt == "MEAN") ConstrainOrphansMean(val,pattern); - else if (recOpt == "MEDIAN") ConstrainOrphansMedian(val,pattern); - else {stopped = kTRUE; break;} - } - // - else if (recTitle == kRecKeys[ kConstrSubunits ]) { //------------------------ - // expect ONSTRAINT_SUBUNITS MEAN/MEDIAN Value parID0 ... parID1 ... VolID1 ... VolIDn - VolIDm - if (nrecElems<5) {stopped = kTRUE; break;} - recExt = recArr->At(2)->GetName(); - if (!recExt.IsFloat()) {stopped = kTRUE; break;} - double val = recExt.Atof(); - UInt_t pattern = 0; - for (irec=3;irecAt(irec)->GetName(); - if (!recExt.IsDigit()) {stopped = kTRUE; break;} - int parid = recExt.Atoi(); - if (paridAt(irec)->GetName(); - if (recExt == "-") {rangeStart = curID; continue;} // range is requested - else if (!recExt.IsDigit()) {stopped = kTRUE; break;} - else curID = recExt.Atoi(); - // - if (curID<=kMaxITSSensID) curID = GetModuleVolumeID(curID); - // this was a range start or single - int start; - if (rangeStart>=0) {start = rangeStart+1; rangeStart=-1;} // continue the range - else start = curID; // create constraint either for single module (or 1st in the range) - for (int id=start;id<=curID;id++) { - int id0 = IsVIDDefined(id); - if (id0<0) {AliDebug(3,Form("Undefined module %d requested in the SubUnits constraint, skipping",id)); continue;} - if (meanC) ConstrainModuleSubUnitsMean(id0,val,pattern); - else ConstrainModuleSubUnitsMedian(id0,val,pattern); - } - } - if (rangeStart>=0) stopped = kTRUE; // unfinished range - if (stopped) break; - } - // - // association of modules with local constraints - else if (recTitle == kRecKeys[ kApplyConstr ]) { //------------------------ - // expect APPLY_CONSTRAINT NAME [NAME1...] [VolID1 ... VolIDn - VolIDm] - if (nrecElems<3) {stopped = kTRUE; break;} - int nmID0=-1,nmID1=-1; - for (irec=1;irecAt(irec)->GetName(); - if (recExt.IsFloat()) break; - // check if such a constraint was declared - if (!GetConstraint(recExt.Data())) { - AliInfo(Form("No Local constraint %s was declared",recExt.Data())); - stopped=kTRUE; - break; - } - if (nmID0<0) nmID0 = irec; - nmID1 = irec; - } - if (stopped) break; - // - if (irec>=nrecElems) {stopped = kTRUE; break;} // no modules provided - // - // now read the list of modules to constrain - int curID = -1; - int rangeStart = -1; - for (;irecAt(irec)->GetName(); - if (recExt == "-") {rangeStart = curID; continue;} // range is requested - else if (!recExt.IsDigit()) {stopped = kTRUE; break;} - else curID = recExt.Atoi(); - // - if (curID<=kMaxITSSensID) curID = GetModuleVolumeID(curID); - // - // this was a range start or single - int start; - if (rangeStart>=0) {start = rangeStart+1; rangeStart=-1;} // continue the range - else start = curID; // create constraint either for single module (or 1st in the range) - for (int id=start;id<=curID;id++) { - AliITSAlignMille2Module *md = GetMilleModuleByVID(id); - if (!md) {AliDebug(3,Form("Undefined module %d requested in the Local constraint, skipping",id)); continue;} - for (int nmid=nmID0;nmid<=nmID1;nmid++) - ((AliITSAlignMille2ConstrArray*)GetConstraint(recArr->At(nmid)->GetName()))->AddModule(md); - } - } - if (rangeStart>=0) stopped = kTRUE; // unfinished range - if (stopped) break; - } - // - else continue; // already processed record - // - } // end of while loop 4 over the various params - // - break; - } // end of while(1) loop - // - fclose(pfc); - if (stopped) { - AliError(Form("Failed on record %s %s ...\n",recTitle.Data(),recOpt.Data())); - return -1; - } - // - fIsConfigured = kTRUE; - return 0; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::BuildHierarchy() -{ - // build the hieararhy of the modules to align - // - if (!GetUseGlobalDelta() && PseudoParentsAllowed()) { - AliInfo("PseudoParents mode is allowed only when the deltas are global\n" - "Since Deltas are local, switching to NoPseudoParents"); - SetAllowPseudoParents(kFALSE); - } - // set parent/child relationship for modules to align - AliInfo("Setting parent/child relationships\n"); - // - // 1) child -> parent reference - for (int ipar=0;iparIsSensor()) continue; // sensor cannot be a parent - // - for (int icld=0;icldBelongsTo(parent)) continue; - // child cannot have more sensors than the parent - if (child->GetNSensitiveVolumes() > parent->GetNSensitiveVolumes()) continue; - // - AliITSAlignMille2Module* parOld = child->GetParent(); - // is this parent candidate closer than the old parent ? - if (parOld && parOld->GetNSensitiveVolumes()GetNSensitiveVolumes()) continue; // parOld is closer - child->SetParent(parent); - } - // - } - // - // add parent -> children reference - for (int icld=0;icldGetParent(); - if (parent) parent->AddChild(child); - } - // - // reorder the modules in such a way that parents come first - for (int icld=0;icldGetParent()) && (parent->GetUniqueID()>child->GetUniqueID()) ) { - // swap - fMilleModule[icld] = parent; - fMilleModule[parent->GetUniqueID()] = child; - child->SetUniqueID(parent->GetUniqueID()); - parent->SetUniqueID(icld); - child = parent; - } - // - } - // - // Go over the child->parent chain and mark modules with explicitly provided sensors. - // If the sensors of the unit are explicitly declared, all undeclared sensors are - // suppresed in this unit. - for (int icld=fNModules;icld--;) { - AliITSAlignMille2Module* child = GetMilleModule(icld); - AliITSAlignMille2Module* parent = child->GetParent(); - if (!parent) continue; - // - // check if this parent was already processed - if (!parent->AreSensorsProvided()) { - parent->DelSensitiveVolumes(); - parent->SetSensorsProvided(kTRUE); - } - // reattach sensors to parent - for (int isc=child->GetNSensitiveVolumes();isc--;) { - UShort_t senVID = child->GetSensVolVolumeID(isc); - if (!parent->IsIn(senVID)) parent->AddSensitiveVolume(senVID); - } - } - // -} - -// pepo -//________________________________________________________________________________________________________ -void AliITSAlignMille2::SetCurrentModule(Int_t id) -{ - // set the current supermodule - // new meaning - if (fMilleVersion>=2) { - fCurrentModule = GetMilleModule(id); - return; - } - // old meaning - if (fMilleVersion<=1) { - Int_t index=id; - /// set as current the SuperModule that contains the 'index' sens.vol. - if (index<0 || index>2197) { - AliInfo("index does not correspond to a sensitive volume!"); - return; - } - UShort_t voluid=AliITSAlignMille2Module::GetVolumeIDFromIndex(index); - Int_t k=IsContained(voluid); - if (k>=0){ - fCluster.SetVolumeID(voluid); - fCluster.SetXYZ(0,0,0); - InitModuleParams(); - } - else - AliInfo(Form("module %d not defined\n",index)); - } -} -// endpepo -//________________________________________________________________________________________________________ -void AliITSAlignMille2::SetRequiredPoint(Char_t* where, Int_t ndet, Int_t updw, Int_t nreqpts) -{ - // set minimum number of points in specific detector or layer - // where = LAYER or DETECTOR - // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) - // updw = 1 for Y>0, -1 for Y<0, 0 if not specified - // nreqpts = minimum number of points of that type - ndet--; - if (strstr(where,"LAYER")) { - if (ndet<0 || ndet>5) return; - if (updw>0) fNReqLayUp[ndet]=nreqpts; - else if (updw<0) fNReqLayDown[ndet]=nreqpts; - else fNReqLay[ndet]=nreqpts; - fRequirePoints=kTRUE; - } - else if (strstr(where,"DETECTOR")) { - if (ndet<0 || ndet>2) return; - if (updw>0) fNReqDetUp[ndet]=nreqpts; - else if (updw<0) fNReqDetDown[ndet]=nreqpts; - else fNReqDet[ndet]=nreqpts; - fRequirePoints=kTRUE; - } -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::GetModuleIndex(const Char_t *symname) -{ - /// index from symname - if (!symname) return -1; - for (Int_t i=0;i<=kMaxITSSensID; i++) { - if (!strcmp(symname,AliITSgeomTGeo::GetSymName(i))) return i; - } - return -1; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::GetModuleIndex(UShort_t voluid) -{ - /// index from volume ID - AliGeomManager::ELayerID lay = AliGeomManager::VolUIDToLayer(voluid); - if (lay<1|| lay>6) return -1; - Int_t idx=Int_t(voluid)-2048*lay; - if (idx>=AliGeomManager::LayerSize(lay)) return -1; - for (Int_t ilay=1; ilay0 && layerId<7 && modId>=0 && modIdGetIndex()==index) return GetSuperModule(i)->GetVolumeID(); - } - } - return 0; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::InitGeometry() -{ - /// initialize geometry - AliInfo("Loading initial geometry"); - AliGeomManager::LoadGeometry(fGeometryFileName.Data()); - fGeoManager = AliGeomManager::GetGeometry(); - if (!fGeoManager) { - AliInfo("Couldn't initialize geometry"); - return -1; - } - return 0; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::SetConstraintWrtRef(const char* reffname) -{ - // Load the global deltas from this file. The local gaussian constraints on some modules - // will be defined with respect to the deltas from this reference file, converted to local - // delta format. Note: conversion to local format requires reloading the geometry! - // - AliInfo(Form("Loading reference deltas for local constraints from %s",reffname)); - if (!fGeoManager) return -1; - fConstrRefFileName = reffname; - if (fConstrRefFileName == "IDEAL") { // the reference is the ideal geometry, just create dummy reference array - fConstrRef = new TClonesArray("AliAlignObjParams",1); - return 0; - } - TFile *pref = TFile::Open(fConstrRefFileName.Data()); - if (!pref->IsOpen()) return -2; - fConstrRef = (TClonesArray*)pref->Get("ITSAlignObjs"); - pref->Close(); - delete pref; - if (!fConstrRef) { - AliError(Form("Did not find reference prealignment deltas in %s",reffname)); - return -1; - } - // - // we need ideal geometry to convert global deltas to local ones - if (fUsePreAlignment) { - AliError("The call of SetConstraintWrtRef must be done before application of the prealignment"); - return -1; - } - // - AliInfo("Converting global reference deltas to local ones"); - Int_t nprea = fConstrRef->GetEntriesFast(); - for (int ix=0; ixAt(ix); - if (!preo->ApplyToGeometry()) return -1; - } - // - // now convert the global reference deltas to local ones - for (int i=fConstrRef->GetEntriesFast();i--;) { - AliAlignObjParams *preo = (AliAlignObjParams*)fConstrRef->At(i); - TGeoHMatrix * mupd = AliGeomManager::GetMatrix(preo->GetSymName()); - if (!mupd) { // this is not alignable entry, need to look in the supermodules - for (int im=fNSuperModules;im--;) { - AliITSAlignMille2Module* mod = GetSuperModule(im); - if ( strcmp(mod->GetName(), preo->GetSymName()) ) continue; - mupd = mod->GetMatrix(); - break; - } - if (!mupd) { - AliError(Form("Failed to find the volume for reference %s",preo->GetSymName())); - return -1; - } - } - TGeoHMatrix preMat; - preo->GetMatrix(preMat); // Delta_Glob - TGeoHMatrix tmpMat = *mupd; // Delta_Glob * Delta_Glob_Par * M - preMat.MultiplyLeft( &tmpMat.Inverse() ); // M^-1 * Delta_Glob_Par^-1 = (Delta_Glob_Par * M)^-1 - tmpMat.MultiplyLeft( &preMat ); // (Delta_Glob_Par * M)^-1 * Delta_Glob * Delta_Glob_Par * M = Delta_loc - preo->SetMatrix(tmpMat); // local corrections - } - // - // we need to reload the geometry spoiled by this reference deltas... - delete fGeoManager; - AliInfo("Reloading initial geometry"); - AliGeomManager::LoadGeometry(fGeometryFileName.Data()); - fGeoManager = AliGeomManager::GetGeometry(); - return 0; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::Init() -{ - // perform global initialization - // - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized!"); - return; - } - // range constraints in such a way that the childs are constrained before their parents - // orphan constraints come last - for (int ic=0;icGetModuleID()GetModuleID()) { - // swap - fConstraints[ic] = cst1; - fConstraints[ic1] = cst0; - } - } - } - // - if (!GetUseGlobalDelta()) { - AliInfo("ATTENTION: The parameters are defined in the local frame, no check for degeneracy will be done"); - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - int npar = mod->GetNParTot(); - // the parameter may have max 1 free instance, otherwise the equations are underdefined - for (int ipar=0;iparIsFreeDOF(ipar)) continue; - mod->SetParOffset(ipar,fNGlobal++); - } - } - } - else { - // init millepede, decide which parameters are to be fitted explicitly - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - int npar = mod->GetNParTot(); - // the parameter may have max 1 free instance, otherwise the equations are underdefined - for (int ipar=0;iparIsFreeDOF(ipar)) continue; // fixed - // - int nFreeInstances = 0; - // - AliITSAlignMille2Module* parent = mod; - Bool_t cstMeanMed=kFALSE,cstGauss=kFALSE; - // - Bool_t addToFit = kFALSE; - // the parameter may be ommitted from explicit fit (if PseudoParentsAllowed is true) if - // 1) it is not explicitly constrained or its does not participate in Gaussian constraint - // 2) the same applies to all of its parents - // 3) it has at least 1 unconstrained direct child - while(parent) { - if (!parent->IsFreeDOF(ipar)) {parent = parent->GetParent(); continue;} - nFreeInstances++; - if (IsParModConstrained(parent,ipar, cstMeanMed, cstGauss)) nFreeInstances--; - if (cstGauss) addToFit = kTRUE; - parent = parent->GetParent(); - } - if (nFreeInstances>1) { - AliError(Form("Parameter#%d of module %s\nhas %d free instances in the " - "unconstrained parents\nSystem is undefined",ipar,mod->GetName(),nFreeInstances)); - exit(1); - } - // - // i) Are PseudoParents allowed? - if (!PseudoParentsAllowed()) addToFit = kTRUE; - // ii) check if this module has no child with such a free parameter. Since the order of this check - // goes from child to parent, by this moment such a parameter must have been already added - else if (!IsParModFamilyVaried(mod,ipar)) addToFit = kTRUE; // no varied children at all - else if (!IsParFamilyFree(mod,ipar,1)) addToFit = kTRUE; // no unconstrained direct children - // otherwise the value of this parameter can be extracted from simple contraint and the values of - // the relevant parameters of its children the fit is done. Hence it is not included - if (!addToFit) continue; - // - // shall add this parameter to explicit fit - // printf("Adding %s %d -> %d\n",mod->GetName(), ipar, fNGlobal); - mod->SetParOffset(ipar,fNGlobal++); - } - } - } - // - AliInfo(Form("Initializing Millepede with %d gpar, %d lpar and %d stddev ...",fNGlobal, fNLocal, fNStdDev)); - fGlobalDerivatives = new Double_t[fNGlobal]; - memset(fGlobalDerivatives,0,fNGlobal*sizeof(Double_t)); - // - fMillepede->InitMille(fNGlobal,fNLocal,fNStdDev,fResCut,fResCutInitial); - fIsMilleInit = kTRUE; - // - ResetLocalEquation(); - AliInfo("Parameters initialized to zero"); - // - /// Fix non free parameters - for (Int_t i=0; iGetNParTot(); j++) { - if (mod->GetParOffset(j)<0) continue; // not varied - FixParameter(mod->GetParOffset(j),mod->GetParConstraint(j)); - fMillepede->SetParamGrID(i, mod->GetParOffset(j)); - } - } - // - // Set iterations - if (fStartFac>1) fMillepede->SetIterations(fStartFac); - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::AddConstraint(Double_t *par, Double_t value, Double_t sigma) -{ - /// Constrain equation defined by par to value - if (!fIsMilleInit) Init(); - fMillepede->SetGlobalConstraint(par, value, sigma); - AliInfo("Adding constraint"); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::InitGlobalParameters(Double_t *par) -{ - /// Initialize global parameters with par array - if (!fIsMilleInit) Init(); - fMillepede->SetGlobalParameters(par); - AliInfo("Init Global Parameters"); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::FixParameter(Int_t iPar, Double_t value) -{ - /// Parameter iPar is encourage to vary in [-value;value]. - /// If value == 0, parameter is fixed - if (!fIsMilleInit) { - AliInfo("Millepede has not been initialized!"); - return; - } - fMillepede->SetParSigma(iPar, value); - if (value==0) AliInfo(Form("Parameter %i Fixed", iPar)); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ResetLocalEquation() -{ - /// Reset the derivative vectors - for(int i=fNLocal;i--;) fLocalDerivatives[i] = 0.0; - memset(fGlobalDerivatives, 0, fNGlobal*sizeof(double) ); -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::ApplyToGeometry() -{ - // apply starting realignment to ideal geometry - AliInfo(Form("Using %s for prealignment",fPreAlignmentFileName.Data())); - if (!fGeoManager) return -1; - TFile *pref = TFile::Open(fPreAlignmentFileName.Data()); - if (!pref->IsOpen()) return -2; - fPrealignment = (TClonesArray*)pref->Get("ITSAlignObjs"); - if (!fPrealignment) return -3; - Int_t nprea = fPrealignment->GetEntriesFast(); - AliInfo(Form("Array of input misalignments with %d entries",nprea)); - // - for (int ix=0; ixAt(ix); - Int_t index=AliITSAlignMille2Module::GetIndexFromVolumeID(preo->GetVolUID()); - if (index>=0) { - if (index>=fPreAlignQF.GetSize()) fPreAlignQF.Set(index+10); - fPreAlignQF[index] = (int) preo->GetUniqueID()+1; - } - //TString nms = preo->GetSymName(); - //if (!nms.Contains("Ladder")) continue; //RRR - //printf("Applying#%4d %s\n",ix,preo->GetSymName()); - if (!preo->ApplyToGeometry()) return -4; - } - // - pref->Close(); - delete pref; - // - fUsePreAlignment = kTRUE; - return 0; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::GetPreAlignmentQualityFactor(Int_t index) const -{ - // quality factors from prealignment - if (!fUsePreAlignment || index<0 || index>=fPreAlignQF.GetSize()) return -1; - return fPreAlignQF[index]-1; -} - -//________________________________________________________________________________________________________ -AliTrackPointArray *AliITSAlignMille2::PrepareTrack(const AliTrackPointArray *atp) -{ - /// create a new AliTrackPointArray keeping only defined modules - /// move points according to a given prealignment, if any - /// sort alitrackpoints w.r.t. global Y direction, if selected - const double kTiny = 1E-12; - // - AliTrackPointArray *atps=NULL; - Int_t idx[20]; - Int_t npts=atp->GetNPoints(); - - /// checks if AliTrackPoints belong to defined modules - Int_t ngoodpts=0; - Int_t intidx[20]; - - for (int j=0; jGetVolumeID()[j]); - if (intidx[j]>=0) ngoodpts++; - } - AliDebug(3,Form("Number of points in defined modules: %d out of %d",ngoodpts,npts)); - - // reject track if not enough points are left - if (ngoodpts> RS - AliTrackPoint p; - // check points in specific places - if (fRequirePoints) { - Int_t nlayup[6],nlaydown[6],nlay[6]; - Int_t ndetup[3],ndetdown[3],ndet[3]; - for (Int_t j=0; j<6; j++) {nlayup[j]=0; nlaydown[j]=0; nlay[j]=0;} - for (Int_t j=0; j<3; j++) {ndetup[j]=0; ndetdown[j]=0; ndet[j]=0;} - - for (int i=0; iGetX()[i]; - Float_t yy=atp->GetY()[i]; - Float_t r=TMath::Sqrt(xx*xx + yy*yy); - int lay=-1; - if (r<5) lay=0; - else if (r>5 && r<10) lay=1; - else if (r>10 && r<18) lay=2; - else if (r>18 && r<30) lay=3; - else if (r>30 && r<40) lay=4; - else if (r>40) lay=5; - if (lay<0) continue; - int det=lay/2; - //printf("Point %d - x=%f y=%f R=%f lay=%d det=%d\n",i,xx,yy,r,lay,det); - - if (yy>=0.0) { // UP point - nlayup[lay]++; - nlay[lay]++; - ndetup[det]++; - ndet[det]++; - } - else { - nlaydown[lay]++; - nlay[lay]++; - ndetdown[det]++; - ndet[det]++; - } - } - - // checks minimum values - Bool_t isok=kTRUE; - for (Int_t j=0; j<6; j++) { - if (nlayup[j]GetY(),idx); // sort descending... - // - Int_t npto=0; - for (int i=0; iGetPoint(p,idx[i]); - - // prealign point if required - // get IDEAL matrix - AliITSAlignMille2Module *mod = GetMilleModule(intidx[idx[i]]); - TGeoHMatrix *svOrigMatrix = mod->GetSensitiveVolumeOrigGlobalMatrix(p.GetVolumeID()); - // get back real local coordinates: use OriginalGlobalMatrix because AliTrackPoints were written - // with idel geometry - Double_t pg[3],pl[3]; - pg[0]=p.GetX(); - pg[1]=p.GetY(); - pg[2]=p.GetZ(); - // printf("Global coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2]); - AliDebug(3,Form("Global coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2])); - svOrigMatrix->MasterToLocal(pg,pl); - - AliDebug(3,Form("Local coordinates of measured point : X=%f Y=%f Z=%f \n",pl[0],pl[1],pl[2])); - // - // this is a temporary code to extract the drift speed used for given point - if (p.GetDriftTime()>0) { // RRR - // calculate the drift speed - int sid = AliITSAlignMille2Module::GetIndexFromVolumeID(p.GetVolumeID());// - kSDDoffsID; - fDriftTime0[npto] = fInitialRecSDD ? fInitialRecSDD->GetTimeZero(sid) : 0.; - /* - AliGeomManager::ELayerID lay = AliGeomManager::VolUIDToLayer(p.GetVolumeID()); - if (lay==3) fDriftTime0[npto] = pg[2]<0 ? 169.5 : 140.1; - else if (lay==4) fDriftTime0[npto] = pg[2]<0 ? 158.3 : 139.0; - else { - AliError(Form("Strange layer %d for moduleID %d",lay,p.GetVolumeID())); - exit(1); - } - */ - double tdif = p.GetDriftTime() - fDriftTime0[npto]; - if (tdif<=0) tdif = 1; - double vdrift = (3.5085-TMath::Abs(pl[0]))/tdif; - if (vdrift<0) vdrift = 0; - // - // TEMPORARY CORRECTION (if provided) -------------->>> - if (fCorrectSDD) { - float t0Upd = fCorrectSDD->GetTimeZero(sid); - vdrift += fCorrectSDD->GetDeltaVDrift(sid); - tdif = p.GetDriftTime() - t0Upd; - // correct Xlocal - pl[0] = TMath::Sign(3.5085 - vdrift*tdif,pl[0]); - fDriftTime0[npto] = t0Upd; - } - // TEMPORARY CORRECTION (if provided) --------------<<< - fDriftSpeed[npto] = TMath::Sign(vdrift,pl[0]); - // - /* - printf("%d %+6.2f %+6.2f %+6.2f %+5.2f %+5.2f %+5.2f %+6.1f %+6.1f %+f %+f\n", - p.GetVolumeID(),pg[0],pg[1],pg[2],pl[0],pl[1],pl[2],p.GetDriftTime(), fDriftTime0[npto], fDriftSpeed[npto],tdif); - */ - } - - // update covariance matrix - TGeoHMatrix hcov; - Double_t hcovel[9]; - hcovel[0]=double(p.GetCov()[0]); - hcovel[1]=double(p.GetCov()[1]); - hcovel[2]=double(p.GetCov()[2]); - hcovel[3]=double(p.GetCov()[1]); - hcovel[4]=double(p.GetCov()[3]); - hcovel[5]=double(p.GetCov()[4]); - hcovel[6]=double(p.GetCov()[2]); - hcovel[7]=double(p.GetCov()[4]); - hcovel[8]=double(p.GetCov()[5]); - hcov.SetRotation(hcovel); - // now rotate in local system - // printf("\nErrMatGlob: before\n"); hcov.Print(""); //RRR - hcov.Multiply(svOrigMatrix); - hcov.MultiplyLeft(&svOrigMatrix->Inverse()); - // now hcov is LOCAL COVARIANCE MATRIX - // apply sigma scaling - // printf("\nErrMatLoc: before\n"); hcov.Print(""); //RRR - Double_t *hcovscl = hcov.GetRotationMatrix(); - // for (int ir=3;ir--;) for (int ic=3;ic--;) hcovscl[ir*3+ic] *= mod->GetSigmaFactor(ir)*mod->GetSigmaFactor(ic); //RRR - // RS TEMPORARY: nullify non-diagonal elements and sigY - hcovscl[5] = 0; - for (int ir=3;ir--;) for (int ic=3;ic--;) { - if (ir==ic) { - if (TMath::Abs(hcovscl[ir*3+ic])GetSigmaFactor(ir)*mod->GetSigmaFactor(ic); //RRR - } - else hcovscl[ir*3+ic] = 0; - } - // - // printf("\nErrMatLoc: after\n"); hcov.Print(""); //RRR - // - if (fBug==1) { - // correzione bug LAYER 5 SSD temporanea.. - int ssdidx=AliITSAlignMille2Module::GetIndexFromVolumeID(p.GetVolumeID()); - if (ssdidx>=500 && ssdidx<1248) { - int ladder=(ssdidx-500)%22; - if (ladder==18) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx+1)); - if (ladder==19) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx-1)); - } - } - /// get (evenctually prealigned) matrix of sens. vol. - TGeoHMatrix *svMatrix = mod->GetSensitiveVolumeMatrix(p.GetVolumeID()); - // modify global coordinates according with pre-aligment - svMatrix->LocalToMaster(pl,pg); - // now rotate in local system - hcov.Multiply(&svMatrix->Inverse()); - hcov.MultiplyLeft(svMatrix); - // hcov is back in GLOBAL RF - // cure once more - for (int ir=3;ir--;) for (int ic=3;ic--;) if (TMath::Abs(hcovscl[ir*3+ic])AddPoint(npto,&p); - AliDebug(2,Form("Adding point[%d] = ( %f , %f , %f ) volid = %d",npto,atps->GetX()[npto], - atps->GetY()[npto],atps->GetZ()[npto],atps->GetVolumeID()[npto] )); - // printf("Adding %d %d %f\n",npto, p.GetVolumeID(), p.GetY()); - npto++; - } - - return atps; -} - -//________________________________________________________________________________________________________ -AliTrackPointArray *AliITSAlignMille2::SortTrack(const AliTrackPointArray *atp) -{ - /// sort alitrackpoints w.r.t. global Y direction - AliTrackPointArray *atps=NULL; - Int_t idx[20]; - Int_t npts=atp->GetNPoints(); - AliTrackPoint p; - atps=new AliTrackPointArray(npts); - - TMath::Sort(npts,atp->GetY(),idx); - - for (int i=0; iGetPoint(p,idx[i]); - atps->AddPoint(i,&p); - AliDebug(2,Form("Point[%d] = ( %f , %f , %f ) volid = %d",i,atps->GetX()[i],atps->GetY()[i],atps->GetZ()[i],atps->GetVolumeID()[i] )); - } - return atps; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::GetCurrentLayer() const -{ - // get current layer id - if (!fGeoManager) { - AliInfo("ITS geometry not initialized!"); - return -1; - } - return (Int_t)AliGeomManager::VolUIDToLayer(fCluster.GetVolumeID()); -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::InitModuleParams() -{ - /// initialize geometry parameters for a given detector - /// for current cluster (fCluster) - /// fGlobalInitParam[] is set as: - /// [tx,ty,tz,psi,theta,phi] - /// (old was [tx,ty,tz,theta,psi,phi] ROOT's angles...) - /// *** At the moment: using Raffalele's angles definition *** - /// - /// return 0 if success - /// If module is found but has no parameters to vary, return 1 - - if (!fGeoManager) { - AliInfo("ITS geometry not initialized!"); - return -1; - } - - // now 'voluid' is the volumeID of a SENSITIVE VOLUME (coming from a cluster) - - // set the internal index (index in module list) - UShort_t voluid=fCluster.GetVolumeID(); - // - // IT IS VERY IMPORTANT: start from the end of the list, where the childs are located !!! - Int_t k=fNModules-1; - fCurrentModule = 0; - // VERY IMPORTANT: if the sensors were explicitly provided, don't look in the supermodules - while (k>=0 && ! (fCurrentModule=GetMilleModule(k))->IsIn(voluid)) k--; - if (k<0) return -3; - // - /* - // Check if the module has free params. If not, go over the parents - AliITSAlignMille2Module* mdtmp = fCurrentModule; - while (mdtmp && mdtmp->GetNParFree()==0) mdtmp = mdtmp->GetParent(); - if (!mdtmp) return 1; // nothing to vary here - fCurrentModule = mdtmp; - */ - // - fModuleInitParam[0] = 0.0; - fModuleInitParam[1] = 0.0; - fModuleInitParam[2] = 0.0; - fModuleInitParam[3] = 0.0; // psi (X) - fModuleInitParam[4] = 0.0; // theta (Y) - fModuleInitParam[5] = 0.0; // phi (Z) - fModuleInitParam[6] = 0.0; - fModuleInitParam[7] = 0.0; - /// get (evenctually prealigned) matrix of sens. vol. - TGeoHMatrix *svMatrix = fCurrentModule->GetSensitiveVolumeMatrix(voluid); - - fMeasGlo[0] = fCluster.GetX(); - fMeasGlo[1] = fCluster.GetY(); - fMeasGlo[2] = fCluster.GetZ(); - svMatrix->MasterToLocal(fMeasGlo,fMeasLoc); - AliDebug(2,Form("Local coordinates of measured point : X=%f Y=%f Z=%f \n",fMeasLoc[0] ,fMeasLoc[1] ,fMeasLoc[2] )); - - // set stdev from cluster - TGeoHMatrix hcov; - Double_t hcovel[9]; - hcovel[0]=double(fCluster.GetCov()[0]); - hcovel[1]=double(fCluster.GetCov()[1]); - hcovel[2]=double(fCluster.GetCov()[2]); - hcovel[3]=double(fCluster.GetCov()[1]); - hcovel[4]=double(fCluster.GetCov()[3]); - hcovel[5]=double(fCluster.GetCov()[4]); - hcovel[6]=double(fCluster.GetCov()[2]); - hcovel[7]=double(fCluster.GetCov()[4]); - hcovel[8]=double(fCluster.GetCov()[5]); - hcov.SetRotation(hcovel); - // now rotate in local system - hcov.Multiply(svMatrix); - hcov.MultiplyLeft(&svMatrix->Inverse()); - // - // set local sigmas - fSigmaLoc[0] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[0])); - fSigmaLoc[1] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[4])); // RS - fSigmaLoc[2] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[8])); - - // set minimum value for SigmaLoc to 10 micron - if (fSigmaLoc[0]<0.0010) fSigmaLoc[0]=0.0010; - if (fSigmaLoc[2]<0.0010) fSigmaLoc[2]=0.0010; - // - /* RRR the rescaling is moved to PrepareTrack - // multiply local sigmas by global and module specific factor - for (int i=3;i--;) fSigmaLoc[i] *= fSigmaFactor[i]*fCurrentModule->GetSigmaFactor(i); - // - */ - AliDebug(2,Form("Setting StDev from CovMat : fSigmaLocX=%g fSigmaLocY=%g fSigmaLocZ=%g \n",fSigmaLoc[0] ,fSigmaLoc[1] ,fSigmaLoc[2] )); - - return 0; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::Print(Option_t*) const -{ - // print current status - printf("*** AliMillepede for ITS ***\n"); - printf(" Number of defined super modules: %d\n",fNModules); - printf(" Obtained parameters refer to %s Deltas\n",fUseGlobalDelta ? "GLOBAL":"LOCAL"); - // - if (fGeoManager) - printf(" geometry loaded from %s\n",fGeometryFileName.Data()); - else - printf(" geometry not loaded\n"); - // - if (fUsePreAlignment) - printf(" using prealignment from %s \n",fPreAlignmentFileName.Data()); - else - printf(" prealignment not used\n"); - // - // - if (fBOn) - printf(" B Field set to %f T - using Riemann's helices\n",fBField); - else - printf(" B Field OFF - using straight lines \n"); - // - if (fRequirePoints) printf(" Required points in tracks:\n"); - for (Int_t i=0; i<6; i++) { - if (fNReqLayUp[i]>0) printf(" Layer %d : %d points with Y>0\n",i+1,fNReqLayUp[i]); - if (fNReqLayDown[i]>0) printf(" Layer %d : %d points with Y<0\n",i+1,fNReqLayDown[i]); - if (fNReqLay[i]>0) printf(" Layer %d : %d points \n",i+1,fNReqLay[i]); - } - for (Int_t i=0; i<3; i++) { - if (fNReqDetUp[i]>0) printf(" Detector %d : %d points with Y>0\n",i+1,fNReqDetUp[i]); - if (fNReqDetDown[i]>0) printf(" Detector %d : %d points with Y<0\n",i+1,fNReqDetDown[i]); - if (fNReqDet[i]>0) printf(" Detector %d : %d points \n",i+1,fNReqDet[i]); - } - // - printf("\n Millepede configuration parameters:\n"); - printf(" init value for chi2 cut : %.4f\n",fStartFac); - printf(" first iteration cut value : %.4f\n",fResCutInitial); - printf(" other iterations cut value : %.4f\n",fResCut); - printf(" number of stddev for chi2 cut : %d\n",fNStdDev); - printf(" def.scaling for local sigmas : %.4f %.4f %.4f\n",fSigmaFactor[0],fSigmaFactor[1],fSigmaFactor[2]); - - printf("List of defined modules:\n"); - printf(" intidx\tindex\tvoluid\tname\n"); - for (int i=0; iGetIndex(),md->GetVolumeID(),md->GetName()); - } -} - -//________________________________________________________________________________________________________ -AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleByVID(UShort_t voluid) const -{ - // return pointer to a defined supermodule - // return NULL if error - Int_t i=IsVIDDefined(voluid); - if (i<0) return NULL; - return GetMilleModule(i); -} - -//________________________________________________________________________________________________________ -AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleBySymName(const Char_t* symname) const -{ - // return pointer to a defined supermodule - // return NULL if error - UShort_t vid = AliITSAlignMille2Module::GetVolumeIDFromSymname(symname); - if (vid>0) return GetMilleModuleByVID(vid); - else { // this is not alignable module, need to look within defined supermodules - int i = IsSymDefined(symname); - if (i>=0) return GetMilleModule(i); - } - return 0; -} - -//________________________________________________________________________________________________________ -AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleIfContained(const Char_t* symname) const -{ - // return pointer to a defined/contained supermodule - // return NULL otherwise - int i = IsSymContained(symname); - return i<0 ? 0 : GetMilleModule(i); -} - -//________________________________________________________________________________________________________ -AliAlignObjParams* AliITSAlignMille2::GetPrealignedObject(const Char_t* symname) const -{ - // get delta from prealignment for given volume - if (!fPrealignment) return 0; - for (int ipre=fPrealignment->GetEntriesFast();ipre--;) { // was the corresponding object prealigned? - AliAlignObjParams* preob = (AliAlignObjParams*)fPrealignment->At(ipre); - if (!strcmp(preob->GetSymName(),symname)) return preob; - } - return 0; -} - -//________________________________________________________________________________________________________ -AliAlignObjParams* AliITSAlignMille2::GetConstrRefObject(const Char_t* symname) const -{ - // get delta with respect to which the constraint is declared - if (!fConstrRef) return 0; - for (int ipre=fConstrRef->GetEntriesFast();ipre--;) { // was the corresponding object prealigned? - AliAlignObjParams* preob = (AliAlignObjParams*)fConstrRef->At(ipre); - if (!strcmp(preob->GetSymName(),symname)) return preob; - } - return 0; -} - -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::InitRiemanFit() -{ - // Initialize Riemann Fitter for current track - // return kFALSE if error - - if (!fBOn) return kFALSE; - - Int_t npts=0; - AliTrackPoint ap; - npts = fTrack->GetNPoints(); - AliDebug(3,Form("Fitting track with %d points",npts)); - - fRieman->Reset(); - fRieman->SetTrackPointArray(fTrack); - - TArrayI ai(npts); - for (Int_t ipt=0; iptGetVolumeID()[ipt]; - - // fit track with 5 params in his own tracking-rotated reference system - // xc = -p[1]/p[0]; - // yc = 1/p[0]; - // R = sqrt( x0*x0 + y0*y0 - y0*p[2]); - if (!fRieman->Fit(&ai,NULL,(AliGeomManager::ELayerID)1,(AliGeomManager::ELayerID)6)) { - return kFALSE; - } - - for (int i=0; i<5; i++) - fLocalInitParam[i] = fRieman->GetParam()[i]; - - return kTRUE; -} - -//________________________________________________________________________________________________________ -void trackFit2D(Int_t &, Double_t *, double &chi2, double *par, int flag) -{ - // local function for minuit - const double kTiny = 1.e-14; - chi2 = 0; - static AliTrackPoint pnt; - static Bool_t fullErr2D; - // - if (flag==1) fullErr2D = kFALSE;//kTRUE; - enum {kAX,kAZ,kBX,kBZ}; - enum {kXX=0,kXY=1,kXZ=2,kYX=kXY,kYY=3,kYZ=4,kZX=kXZ,kZY=kYZ,kZZ=5}; - // - AliITSAlignMille2* alig = AliITSAlignMille2::GetInstance(); - AliTrackPointArray* track = alig->GetCurrentTrack(); - // - int npts = track->GetNPoints(); - for (int ip=0;ipGetPoint(pnt,ip); - const float *cov = pnt.GetCov(); - double y = pnt.GetY(); - double dx = pnt.GetX() - (par[kAX]+y*par[kBX]); - double dz = pnt.GetZ() - (par[kAZ]+y*par[kBZ]); - double xxe = cov[kXX]; - double zze = cov[kZZ]; - double xze = cov[kXZ]; - // - if (fullErr2D) { - xxe += par[kBX]*par[kBX]*cov[kYY]-2.*par[kBX]*cov[kXY]; - zze += par[kBZ]*par[kBZ]*cov[kYY]-2.*par[kBZ]*cov[kZY]; - xze += par[kBX]*par[kBZ]*cov[kYY]-cov[kYZ]*par[kBZ]-cov[kXY]*par[kBX]; - } - // - double det = xxe*zze - xze*xze; - if (det %+e\n",ip,dx,dz,xxeI,zzeI,xzeI, chi2); - } - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::InitTrackParams(int meth) -{ - /// initialize local parameters with different methods - /// for current track (fTrack) - Int_t npts=0; - AliTrackPoint ap; - double sX=0,sXY=0,sZ=0,sZY=0,sY=0,sYY=0,det=0; - // simple linear interpolation - // get local starting parameters (to be substituted by ESD track parms) - // local parms (fLocalInitParam[]) are: - // [0] = global x coord. of straight line intersection at y=0 plane - // [1] = global z coord. of straight line intersection at y=0 plane - // [2] = px/py - // [3] = pz/py - // test #1: linear fit in x(y) and z(y) - npts = fTrack->GetNPoints(); - AliDebug(3,Form("*** initializing track with %d points ***",npts)); - for (int i=npts;i--;) { - sY += fTrack->GetY()[i]; - sYY += fTrack->GetY()[i]*fTrack->GetY()[i]; - sX += fTrack->GetX()[i]; - sXY += fTrack->GetX()[i]*fTrack->GetY()[i]; - sZ += fTrack->GetZ()[i]; - sZY += fTrack->GetZ()[i]*fTrack->GetY()[i]; - } - det = sYY*npts-sY*sY; - if (det==0) det = 1E-20; - fLocalInitParam[0] = (sX*sYY-sY*sXY)/det; - fLocalInitParam[2] = (sXY*npts-sY*sX)/det; - // - fLocalInitParam[1] = (sZ*sYY-sY*sZY)/det; - fLocalInitParam[3] = (sZY*npts-sY*sZ)/det; - // pepo200709 - fLocalInitParam[4] = 0.0; - // endpepo200709 - - AliDebug(2,Form("X = p0gx + ugx*Y : p0gx = %f ugx = %f\n",fLocalInitParam[0],fLocalInitParam[2])); - // - if (meth==1) return; - // - // perform full fit accounting for cov.matrix - static TVirtualFitter *minuit = 0; - static Double_t step[5] = {1E-3,1E-3,1E-4,1E-4,1E-5}; - static Double_t arglist[10]; - // - if (!minuit) { - minuit = TVirtualFitter::Fitter(0,4); - minuit->SetFCN(trackFit2D); - arglist[0] = 1; - minuit->ExecuteCommand("SET ERR",arglist, 1); - // - arglist[0] = -1; - minuit->ExecuteCommand("SET PRINT",arglist,1); - // - } - // - minuit->SetParameter(0, "ax", fLocalInitParam[0], step[0], 0,0); - minuit->SetParameter(1, "az", fLocalInitParam[1], step[1], 0,0); - minuit->SetParameter(2, "bx", fLocalInitParam[2], step[2], 0,0); - minuit->SetParameter(3, "bz", fLocalInitParam[3], step[3], 0,0); - // - arglist[0] = 1000; // number of function calls - arglist[1] = 0.001; // tolerance - minuit->ExecuteCommand("MIGRAD",arglist,2); - // - for (int i=0;i<4;i++) fLocalInitParam[i] = minuit->GetParameter(i); - for (int i=0;i<4;i++) for (int j=0;j<4;j++) fLocalInitParEr[i][j] = minuit->GetCovarianceMatrixElement(i,j); - // -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::IsSymDefined(const Char_t* symname) const -{ - // checks if supermodule with this symname is defined and return the internal index - // return -1 if not. - for (int k=fNModules;k--;) if (!strcmp(symname,GetMilleModule(k)->GetName())) return k; - return -1; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::IsSymContained(const Char_t* symname) const -{ - // checks if module with this symname is defined and return the internal index - // return -1 if not. - UShort_t vid = AliITSAlignMille2Module::GetVolumeIDFromSymname(symname); - if (vid>0) return IsVIDContained(vid); - // only sensors have real vid, but maybe we have a supermodule with fake vid? - // IMPORTANT: always start from the end to start from the sensors - return IsSymDefined(symname); -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::IsVIDDefined(UShort_t voluid) const -{ - // checks if supermodule 'voluid' is defined and return the internal index - // return -1 if not. - for (int k=fNModules;k--;) if (voluid==GetMilleModule(k)->GetVolumeID()) return k; - return -1; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::IsVIDContained(UShort_t voluid) const -{ - // checks if the sensitive module 'voluid' is contained inside a supermodule - // and return the internal index of the last identified supermodule - // return -1 if error - // IMPORTANT: always start from the end to start from the sensors - if (AliITSAlignMille2Module::GetIndexFromVolumeID(voluid)<0) return -1; - for (int k=fNModules;k--;) if (GetMilleModule(k)->IsIn(voluid)) return k; - return -1; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::CheckCurrentTrack() -{ - /// checks if AliTrackPoints belongs to defined modules - /// return number of good poins - /// return 0 if not enough points - - Int_t npts = fTrack->GetNPoints(); - Int_t ngoodpts=0; - // debug points - for (int j=0; jGetVolumeID()[j])>=0) ngoodpts++; - // - if (ngoodptsGetNPoints(); - AliDebug(2,Form("*** Input track with %d points ***",npts)); - - // preprocessing of the input track: keep only points in defined volumes, - // move points if prealignment is set, sort by Yglo if required - - fTrack=PrepareTrack(track); - if (!fTrack) return -1; - - npts = fTrack->GetNPoints(); - if (npts>kMaxPoints) { - AliError(Form("Compiled with kMaxPoints=%d, current track has %d points",kMaxPoints,npts)); - } - AliDebug(2,Form("*** Processing prepared track with %d points ***",npts)); - - if (!fBOn) { // straight lines - // set local starting parameters (to be substituted by ESD track parms) - // local parms (fLocalInitParam[]) are: - // [0] = global x coord. of straight line intersection at y=0 plane - // [1] = global z coord. of straight line intersection at y=0 plane - // [2] = px/py - // [3] = pz/py - InitTrackParams(fInitTrackParamsMeth); - } - else { - // local parms (fLocalInitParam[]) are the Riemann Fitter params - if (!InitRiemanFit()) { - AliInfo("Riemann fit failed! skipping this track..."); - fTrack=NULL; - return -5; - } - } - - Int_t nloceq=0; - Int_t ngloeq=0; - static Mille2Data md[kMaxPoints]; - // - for (Int_t ipt=0; iptGetPoint(fCluster,ipt); - fCluster.SetUniqueID(ipt); - AliDebug(2,Form("\n--- processing point %d --- \n",ipt)); - - // set geometry parameters for the the current module - if (InitModuleParams()) continue; - AliDebug(2,Form(" VolID=%d Index=%d InternalIdx=%d symname=%s\n", - track->GetVolumeID()[ipt], fCurrentModule->GetIndex(), - fCurrentModule->GetUniqueID(), AliGeomManager::SymName(track->GetVolumeID()[ipt]) )); - AliDebug(2,Form(" Preprocessed Point = ( %f , %f , %f ) \n",fCluster.GetX(),fCluster.GetY(),fCluster.GetZ())); - int res = AddLocalEquation(md[nloceq]); - if (res<0) {fTotBadLocEqPoints++; nloceq = 0; break;} - else if (res==0) nloceq++; - else {nloceq++; ngloeq++;} - } // end loop over points - // - fTrack=NULL; - // not enough good points? - if (nloceqSaveRecordData(); // RRR - // - return 0; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::CalcIntersectionPoint(Double_t *lpar, Double_t *gpar) -{ - /// calculate track intersection point in local coordinates - /// according with a given set of parameters (local(4) and global(6)) - /// and fill fPintLoc/Glo - /// local are: pgx0, pgz0, ugx, ugz OR riemann fitters pars - /// global are: tx,ty,tz,psi,theta,phi (Raff's delta angles in deg.) - /// return 0 if success - - AliDebug(3,Form("lpar = %g %g %g %g %g\ngpar= %g %g %g %g %g %g\n",lpar[0],lpar[1],lpar[2],lpar[3],lpar[4],gpar[0],gpar[1],gpar[2],gpar[3],gpar[4],gpar[5])); - AliDebug(3,Form("deltalpar = %g %g %g %g %g\n",lpar[0]-fLocalInitParam[0],lpar[1]-fLocalInitParam[1],lpar[2]-fLocalInitParam[2],lpar[3]-fLocalInitParam[3],lpar[4]-fLocalInitParam[4])); - - - // prepare the TGeoHMatrix - TGeoHMatrix *tempHMat = fCurrentModule->GetSensitiveVolumeModifiedMatrix(fCluster.GetVolumeID(),gpar, - !fUseGlobalDelta); - if (!tempHMat) return -1; - - Double_t v0g[3]; // vector with straight line direction in global coord. - Double_t p0g[3]; // point of the straight line (glo) - - if (fBOn) { // B FIELD! - AliTrackPoint prf; - for (int ip=0; ip<5; ip++) - fRieman->SetParam(ip,lpar[ip]); - - if (!fRieman->GetPCA(fCluster,prf)) { - AliInfo(Form("error in GetPCA for point %d",fCluster.GetVolumeID())); - return -3; - } - // now determine straight line passing tangent to fit curve at prf - // ugx = dX/dY_glo = DeltaX/DeltaY_glo - // mo' P1=(X,Y,Z)_glo_prf - // => (x,y,Z)_trk_prf ruotando di alpha... - Double_t alpha=fRieman->GetAlpha(); - Double_t x1g = prf.GetX(); - Double_t y1g = prf.GetY(); - Double_t z1g = prf.GetZ(); - Double_t x1t = x1g*TMath::Cos(alpha) + y1g*TMath::Sin(alpha); - Double_t y1t = -x1g*TMath::Sin(alpha) + y1g*TMath::Cos(alpha); - Double_t z1t = z1g; - - Double_t x2t = x1t+1.0; - Double_t y2t = y1t+fRieman->GetDYat(x1t); - Double_t z2t = z1t+fRieman->GetDZat(x1t); - Double_t x2g = x2t*TMath::Cos(alpha) - y2t*TMath::Sin(alpha); - Double_t y2g = x2t*TMath::Sin(alpha) + y2t*TMath::Cos(alpha); - Double_t z2g = z2t; - - AliDebug(3,Form("Riemann frame: fAlpha = %f = %f ",alpha,alpha*180./TMath::Pi())); - AliDebug(3,Form(" prf_glo=( %f , %f , %f ) prf_rf=( %f , %f , %f )\n", x1g,y1g,z1g, x1t,y1t,z1t)); - AliDebug(3,Form(" mov_glo=( %f , %f , %f ) rf=( %f , %f , %f )\n",x2g,y2g,z2g, x2t,y2t,z2t)); - - if (TMath::Abs(y2g-y1g)<1e-15) { - AliInfo("DeltaY=0! Cannot proceed..."); - return -1; - } - // ugx,1,ugz - v0g[0] = (x2g-x1g)/(y2g-y1g); - v0g[1]=1.0; - v0g[2] = (z2g-z1g)/(y2g-y1g); - - // point: just keep prf - p0g[0]=x1g; - p0g[1]=y1g; - p0g[2]=z1g; - } - else { // staight line - // vector of initial straight line direction in glob. coord - v0g[0]=lpar[2]; - v0g[1]=1.0; - v0g[2]=lpar[3]; - - // intercept in yg=0 plane in glob coord - p0g[0]=lpar[0]; - p0g[1]=0.0; - p0g[2]=lpar[1]; - } - AliDebug(3,Form("Line vector: ( %f , %f , %f ) point:( %f , %f , %f )\n",v0g[0],v0g[1],v0g[2],p0g[0],p0g[1],p0g[2])); - - // same in local coord. - Double_t p0l[3],v0l[3]; - tempHMat->MasterToLocalVect(v0g,v0l); - tempHMat->MasterToLocal(p0g,p0l); - - if (TMath::Abs(v0l[1])<1e-15) { - AliInfo("Track Y direction in local frame is zero! Cannot proceed..."); - return -1; - } - - // local intersection point - fPintLoc[0] = p0l[0] - (v0l[0]/v0l[1])*p0l[1]; - fPintLoc[1] = 0; - fPintLoc[2] = p0l[2] - (v0l[2]/v0l[1])*p0l[1]; - - // global intersection point - tempHMat->LocalToMaster(fPintLoc,fPintGlo); - AliDebug(3,Form("Intesect. point: L( %f , %f , %f ) G( %f , %f , %f )\n",fPintLoc[0],fPintLoc[1],fPintLoc[2],fPintGlo[0],fPintGlo[1],fPintGlo[2])); - - return 0; -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::CalcDerivatives(Int_t paridx, Bool_t islpar) -{ - /// calculate numerically (ROOT's style) the derivatives for - /// local X intersection and local Z intersection - /// parlist: local (islpar=kTRUE) pgx0, pgz0, ugx0, ugz0 OR riemann's params - /// global (islpar=kFALSE) tx, ty, tz, psi, theta, phi (Raf's angles in deg) - /// return 0 if success - - // copy initial parameters - Double_t lpar[kNLocal]; - Double_t gpar[kNParCh]; - Double_t *derivative; - for (Int_t i=0; iGetNParFree()==0) continue; - nmodTested++; - for (Int_t i=0; iGetParOffset(i)<0) continue; // this parameter is not explicitly fitted - if (!dfDone[i]) { - if (CalcDerivatives(i,kFALSE)) return -1; - else { - dfDone[i] = kTRUE; - if (zeroX) zeroX = fDerivativeGlo[i][0]==0; - if (zeroZ) zeroZ = fDerivativeGlo[i][2]==0; - } - } - // - m.fDerGloX[ifill] = fDerivativeGlo[i][0]; - m.fDerGloZ[ifill] = fDerivativeGlo[i][2]; - m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(i); - } - // - // specific for special sensors - if ( fCurrentModule->IsSDD() && - (fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0)>=0 || - fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFDV)>=0) ) { - // - // assume for sensor local xloc = xloc0 + V0*dT0+dV*(T-T0) - // where V0 and T are the nominal drift velocity, time and time0 - // and the dT0 and dV are the corrections: - // dX/dT0 = dX/dxloc * dxloc/dT0 = dX/dxloc * V0 - // dX/dV = dX/dxloc * dxloc/dV = dX/dxloc * (T-T0) - // IMPORTANT: the geom derivatives are over the SENSOR LOCAL parameters - // - if (!dfDone[AliITSAlignMille2Module::kDOFT0] || !dfDone[AliITSAlignMille2Module::kDOFDV]) { - // - double dXdxlocsens=0., dZdxlocsens=0.; - // - // if the current module is the sensor itself and we work with local params, then - // we can directly take dX/dxloc_sens dZ/dxloc_sens - if (!fUseGlobalDelta && fCurrentModule->GetVolumeID()==fCluster.GetVolumeID()) { - if (dfDone[AliITSAlignMille2Module::kDOFTX]) { - CalcDerivatives(AliITSAlignMille2Module::kDOFTX,kFALSE); - dfDone[AliITSAlignMille2Module::kDOFTX] = kTRUE; - } - dXdxlocsens = fDerivativeGlo[AliITSAlignMille2Module::kDOFTX][0]; - dZdxlocsens = fDerivativeGlo[AliITSAlignMille2Module::kDOFTX][2]; - } - // - else { // need to perform some transformations - // fetch the jacobian of the transformation from the sensors local frame to the frame - // where the parameters are defined: - // Global: dX/dxloc_sens = dX/dxgl*dxgl/dxloc_sens + ...dX/dphigl*dphigl/dxloc_sens - if (fUseGlobalDelta) fCurrentModule->CalcDerivGloLoc(fCluster.GetVolumeID(), - AliITSAlignMille2Module::kDOFTX, jacobian); - // Local: dX/dxloc_sens = dX/dxcurr*dxcurr/dxloc_sens +..+dX/dphicurr * dphicurr/dxloc_sens - else fCurrentModule->CalcDerivCurLoc(fCluster.GetVolumeID(), - AliITSAlignMille2Module::kDOFTX, jacobian); - // - for (int j=0;jGetParOffset(AliITSAlignMille2Module::kDOFT0)>=0) { - m.fDerGloX[ifill] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][0]; - m.fDerGloZ[ifill] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][2]; - m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0); - } - // - if (fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFDV)>=0) { - m.fDerGloX[ifill] = fDerivativeGlo[AliITSAlignMille2Module::kDOFDV][0]; - m.fDerGloZ[ifill] = fDerivativeGlo[AliITSAlignMille2Module::kDOFDV][2]; - m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFDV); - } - } - // - m.fModuleID[m.fNModFilled++] = fCurrentModule->GetUniqueID(); - } while( (fCurrentModule=fCurrentModule->GetParent()) ); - // - if (nmodTested>0 && zeroX) {AliInfo("Skipping: zero global X derivatives!");return -1;} - if (nmodTested>0 && zeroZ) {AliInfo("Skipping: zero global Z derivatives!");return -1;} - // - // ok, can copy to m - AliDebug(2,Form("Adding local equation X with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[0]-fPintLoc0[0]), fSigmaLoc[0])); - m.fMeasX = fMeasLoc[0]-fPintLoc0[0]; - m.fSigmaX = fSigmaLoc[0]; - // - AliDebug(2,Form("Adding local equation Z with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[2]-fPintLoc0[2]), fSigmaLoc[2])); - m.fMeasZ = fMeasLoc[2]-fPintLoc0[2]; - m.fSigmaZ = fSigmaLoc[2]; - // - m.fNGlobFilled = ifill; - fCurrentModule = endModule; - // - return Int_t(!zeroX && !zeroZ); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::SetLocalEquations(const Mille2Data *marr, Int_t neq) -{ - /// Set local equations with data stored in m - /// return 0 if success - // - for (Int_t j=0; jSetLocalEquation(fGlobalDerivatives, fLocalDerivatives, m.fMeasX, m.fSigmaX); - // - // set equation for Zloc coordinate - AliDebug(2,Form("setting local equation Z with fMeas=%.6f and fSigma=%.6f",m.fMeasZ, m.fSigmaZ)); - for (int i=fNLocal; i--;) SetLocalDerivative( i, m.fDerLocZ[i] ); - for (int i=m.fNGlobFilled;i--;) SetGlobalDerivative( m.fParMilleID[i] , m.fDerGloZ[i] ); - fMillepede->SetLocalEquation(fGlobalDerivatives, fLocalDerivatives, m.fMeasZ, m.fSigmaZ); - // - for (int i=m.fNModFilled;i--;) GetMilleModule(m.fModuleID[i])->IncNProcessedPoints(); - // - } -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::GlobalFit() -{ - /// Call global fit; Global parameters are stored in parameters - if (!fIsMilleInit) Init(); - // - ApplyPreConstraints(); - int res = fMillepede->GlobalFit(); - AliInfo(Form("%s fitting global parameters!",res ? "Done":"Failed")); - if (res) { - // fetch the parameters - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - int nprocp = 0; - for (int ip=mod->GetNParTot();ip--;) { - int idp = mod->GetParOffset(ip); - if (idp<0) continue; // was not in the explicit fit - mod->SetParVal(ip,fMillepede->GetFinalParam(idp)); - mod->SetParErr(ip,fMillepede->GetFinalError(idp)); - int np = fMillepede->GetProcessedPoints(idp); - if (TMath::Abs(np)>TMath::Abs(nprocp)) nprocp = np; - } - if (!mod->GetNProcessedPoints()) mod->SetNProcessedPoints(nprocp); - } - - } - ApplyPostConstraints(); - return res; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::PrintGlobalParameters() -{ - /// Print global parameters - if (!fIsMilleInit) { - AliInfo("Millepede has not been initialized!"); - return; - } - fMillepede->PrintGlobalParameters(); -} - -//________________________________________________________________________________________________________ -Int_t AliITSAlignMille2::LoadSuperModuleFile(const Char_t *sfile) -{ - // load definitions of supermodules from a root file - // return 0 if success - - TFile *smf=TFile::Open(sfile); - if (!smf->IsOpen()) { - AliInfo(Form("Cannot open supermodule file %s",sfile)); - return -1; - } - - TClonesArray *sma=(TClonesArray*)smf->Get("ITSMilleSuperModules"); - if (!sma) { - AliInfo(Form("Cannot find ITSMilleSuperModules array in file")); - return -2; - } - Int_t nsma=sma->GetEntriesFast(); - AliInfo(Form("Array of SuperModules with %d entries\n",nsma)); - // - // pepo200709 - Char_t st[2048]; - char symname[250]; - // end pepo200709 - - UShort_t volid; - TGeoHMatrix m; - // - for (Int_t i=0; iUncheckedAt(i); - volid=a->GetVolUID(); - strcpy(st,a->GetSymName()); - a->GetMatrix(m); - // - sscanf(st,"%s",symname); - // - // decode module list - char *stp=strstr(st,"ModuleList:"); - if (!stp) return -3; - stp += 11; - int idx[2200]; - char spp[200]; int jp=0; - char cl[20]; - strcpy(st,stp); - int l=strlen(st); - int j=0; - int n=0; - // - while (j<=l) { - if (st[j]==9 || st[j]==32 || st[j]==10 || st[j]==0) { - spp[jp]=0; - jp=0; - if (strlen(spp)) { - int k=strcspn(spp,"-"); - if (k%d)",idx[j],kMaxITSSensID)); - return -5; - } - } - Int_t smindex=int(2198+volid-14336); // virtual index - // - fSuperModule.AddAtAndExpand(new AliITSAlignMille2Module(smindex,volid,symname,&m,n,volidsv),fNSuperModules); - // - fNSuperModules++; - } - // - smf->Close(); - // - return 0; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainModuleSubUnitsMean(Int_t idm, Double_t val, UInt_t pattern) -{ - // require that sum of modifications for the childs of this module is = val, i.e. - // the internal corrections moves the module as a whole by fixed value (0 by default). - // pattern is the bit pattern for the parameters to constrain - // - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized: no constrain may be added!"); - return; - } - if (!GetMilleModule(idm)->GetNChildren()) return; - TString nm = "cstrSUMean"; - nm += GetNConstraints(); - AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMean, - idm,val,pattern); - cstr->SetConstraintID(GetNConstraints()); - fConstraints.Add(cstr); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainModuleSubUnitsMedian(Int_t idm, Double_t val, UInt_t pattern) -{ - // require that median of the modifications for the childs of this module is = val, i.e. - // the internal corrections moves the module as a whole by fixed value (0 by default) - // module the outliers. - // pattern is the bit pattern for the parameters to constrain - // The difference between the mean and the median will be transfered to the parent - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized: no constrain may be added!"); - return; - } - if (!GetMilleModule(idm)->GetNChildren()) return; - TString nm = "cstrSUMed"; - nm += GetNConstraints(); - AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMedian, - idm,val,pattern); - cstr->SetConstraintID(GetNConstraints()); - fConstraints.Add(cstr); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainOrphansMean(Double_t val, UInt_t pattern) -{ - // require that median of the modifications for the supermodules which have no parents is = val, i.e. - // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. - // pattern is the bit pattern for the parameters to constrain - // - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized: no constrain may be added!"); - return; - } - TString nm = "cstrOMean"; - nm += GetNConstraints(); - AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMean, - -1,val,pattern); - cstr->SetConstraintID(GetNConstraints()); - fConstraints.Add(cstr); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainOrphansMedian(Double_t val, UInt_t pattern) -{ - // require that median of the modifications for the supermodules which have no parents is = val, i.e. - // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. - // pattern is the bit pattern for the parameters to constrain - // - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized: no constrain may be added!"); - return; - } - TString nm = "cstrOMed"; - nm += GetNConstraints(); - AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMedian, - -1,val,pattern); - cstr->SetConstraintID(GetNConstraints()); - fConstraints.Add(cstr); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainLocal(const Char_t* name,Double_t *parcf,Int_t npar,Double_t val,Double_t err) -{ - // apply constraint on parameters in the local frame - if (fIsMilleInit) { - AliInfo("Millepede has been already initialized: no constrain may be added!"); - return; - } - AliITSAlignMille2ConstrArray *cstr = new AliITSAlignMille2ConstrArray(name,parcf,npar,val,err); - cstr->SetConstraintID(GetNConstraints()); - fConstraints.Add(cstr); -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ApplyGaussianConstraint(const AliITSAlignMille2ConstrArray* cstr) -{ - // apply the constraint on the local corrections of a list of modules - int nmod = cstr->GetNModules(); - double jacobian[AliITSAlignMille2Module::kMaxParGeom][AliITSAlignMille2Module::kMaxParGeom]; - // - for (int imd=nmod;imd--;) { - int modID = cstr->GetModuleID(imd); - AliITSAlignMille2Module* mod = GetMilleModule(modID); - ResetLocalEquation(); - int nadded = 0; - double value = cstr->GetValue(); - double sigma = cstr->GetError(); - // - // in case the reference (survey) deltas were imposed for Gaussian constraints - // already accumulated corrections: they must be subtracted from the constraint value. - if (IsConstraintWrtRef()) { - // - Double_t precal[AliITSAlignMille2Module::kMaxParTot]; - Double_t refcal[AliITSAlignMille2Module::kMaxParTot]; - for (int ip=AliITSAlignMille2Module::kMaxParTot;ip--;) {precal[ip]=0; refcal[ip] = 0.;} - // - // check if there was a reference delta provided for this module - AliAlignObjParams* parref = GetConstrRefObject(mod->GetName()); - if (parref) parref->GetPars(refcal, refcal+3); // found reference delta - // - // extract already applied local corrections for this module - if (fPrealignment) { - // - AliAlignObjParams *preo = GetPrealignedObject(mod->GetName()); - if (preo) { - TGeoHMatrix preMat,tmpMat = *mod->GetMatrix(); // Delta_Glob * Delta_Glob_Par * M - preo->GetMatrix(preMat); // Delta_Glob - preMat.MultiplyLeft( &tmpMat.Inverse() ); // M^-1 * Delta_Glob_Par^-1 = (Delta_Glob_Par * M)^-1 - tmpMat.MultiplyLeft( &preMat ); // (Delta_Glob_Par * M)^-1 * Delta_Glob * Delta_Glob_Par * M = Delta_loc - AliAlignObjParams algob; - algob.SetMatrix(tmpMat); - algob.GetPars(precal,precal+3); // local corrections for geometry - } - } - // - // subtract the contribution to constraint from precalibration - for (int ipar=cstr->GetNCoeffs();ipar--;) value += (refcal[ipar]-precal[ipar])*cstr->GetCoeff(ipar); - // - } - // - if (fUseGlobalDelta) mod->CalcDerivLocGlo(&jacobian[0][0]); - // - for (int ipar=cstr->GetNCoeffs();ipar--;) { - double coef = cstr->GetCoeff(ipar); - if (coef==0) continue; - // - if (!fUseGlobalDelta || ipar>= AliITSAlignMille2Module::kMaxParGeom) { // - // we are working with local params or if the given param is not related to geometry, - // apply the constraint directly - int parPos = mod->GetParOffset(ipar); - if (parPos<0) continue; // not in the fit - fGlobalDerivatives[parPos] += coef; - nadded++; - } - else { // we are working with global params, while the constraint is on local ones -> jacobian - for (int jpar=AliITSAlignMille2Module::kMaxParGeom;jpar--;) { - int parPos = mod->GetParOffset(jpar); - if (parPos<0) continue; - fGlobalDerivatives[parPos] += coef*jacobian[ipar][jpar]; - nadded++; - } - } - } - if (nadded) AddConstraint(fGlobalDerivatives, value, sigma); - } - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ApplyPreConstraints() -{ - // apply constriants which cannot be imposed after the fit - int nconstr = GetNConstraints(); - for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) { - ApplyGaussianConstraint( (AliITSAlignMille2ConstrArray*)cstr); - continue; - } - // - if (cstr->GetType() == AliITSAlignMille2Constraint::kTypeMedian) continue; // post type constraint - // - if (!fUseGlobalDelta) continue; // mean/med constraints must be applied to global deltas - // apply constraint on the mean's before the fit - int imd = cstr->GetModuleID(); - if (imd>=0) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - UInt_t pattern = 0; - for (int ipar=mod->GetNParTot();ipar--;) { - if (!cstr->IncludesParam(ipar)) continue; - if (mod->GetParOffset(ipar)<0) continue; // parameter is not in the explicit fit -> post constraint - pattern |= 0x1<SetApplied(ipar); - } - ConstrainModuleSubUnits(imd,cstr->GetValue(),pattern); - // - } - else if (!PseudoParentsAllowed()) { - ConstrainOrphans(cstr->GetValue(),(UInt_t)cstr->GetPattern()); - cstr->SetApplied(-1); - } - } -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ApplyPostConstraints() -{ - // apply constraints which can be imposed after the fit - int nconstr = GetNConstraints(); - Bool_t convGlo = kFALSE; - // check if there is something to do - int ntodo = 0; - for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) continue; - if (cstr->GetRemainingPattern() == 0) continue; - ntodo++; - } - if (!ntodo) return; - // - if (!fUseGlobalDelta) { // need to convert to global params - ConvertParamsToGlobal(); - convGlo = kTRUE; - } - // - for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) continue; - // - int imd = cstr->GetModuleID(); - // - if (imd>=0) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - UInt_t pattern = 0; - for (int ipar=mod->GetNParTot();ipar--;) { - if (cstr->IsApplied(ipar)) continue; - if (!cstr->IncludesParam(ipar)) continue; - if (!mod->IsFreeDOF(ipar)) continue; // parameter is fixed, will not apply constraint - pattern |= 0x1<SetApplied(ipar); - } - if (pattern) PostConstrainModuleSubUnits(cstr->GetType(),cstr->GetModuleID(),cstr->GetValue(),pattern); - // - } - else if (PseudoParentsAllowed()) { - UInt_t pattern = (UInt_t)cstr->GetRemainingPattern(); - PostConstrainOrphans(cstr->GetType(),cstr->GetValue(),pattern); - cstr->SetApplied(-1); - } - } - // if there was a conversion, rewind it - if (convGlo) ConvertParamsToLocal(); - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainModuleSubUnits(Int_t idm, Double_t val, UInt_t pattern) -{ - // require that sum of modifications for the childs of this module is = val, i.e. - // the internal corrections moves the module as a whole by fixed value (0 by default). - // pattern is the bit pattern for the parameters to constrain - // - // - AliITSAlignMille2Module* mod = GetMilleModule(idm); - // - for (int ip=0;ip>ip)&0x1) /*|| !parent->IsFreeDOF(ip)*/) continue; - ResetLocalEquation(); - int nadd = 0; - for (int ich=mod->GetNChildren();ich--;) { - int idpar = ((AliITSAlignMille2Module*)mod->GetChild(ich))->GetParOffset(ip); - if (idpar<0) continue; - fGlobalDerivatives[idpar] = 1.0; - nadd++; - } - // - if (nadd>0) { - AddConstraint(fGlobalDerivatives,val); - AliInfo(Form("Constrained param %d for %d submodules of module #%d: %s",ip,nadd,idm,mod->GetName())); - } - } - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConstrainOrphans(Double_t val, UInt_t pattern) -{ - // require that median of the modifications for the supermodules which have no parents is = val, i.e. - // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. - // pattern is the bit pattern for the parameters to constrain - // - for (int ip=0;ip>ip)&0x1) ) continue; - ResetLocalEquation(); - int nadd = 0; - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - if (mod->GetParent()) continue; // this is not an orphan - int idpar = mod->GetParOffset(ip); - if (idpar<0) continue; - fGlobalDerivatives[idpar] = 1.0; - nadd++; - } - if (nadd>0) { - AddConstraint(fGlobalDerivatives,val); - AliInfo(Form("Constrained param %d for %d orphan modules",ip,nadd)); - } - } - // - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::PostConstrainModuleSubUnits(Int_t type,Int_t idm, Double_t val, UInt_t pattern) -{ - // require that median or mean of the modifications for the childs of this module is = val, i.e. - // the internal corrections moves the module as a whole by fixed value (0 by default) - // module the outliers. - // pattern is the bit pattern for the parameters to constrain - // The difference between the mean and the median will be transfered to the parent - // - AliITSAlignMille2Module* parent = GetMilleModule(idm); - int nc = parent->GetNChildren(); - // - double *tmpArr = new double[nc]; - // - for (int ip=0;ip>ip)&0x1) || !parent->IsFreeDOF(ip)) continue; - // compute the mean and median of the deltas - int nfree = 0; - for (int ich=nc;ich--;) { - AliITSAlignMille2Module* child = parent->GetChild(ich); - // if (!child->IsFreeDOF(ip)) continue; - tmpArr[nfree++] = child->GetParVal(ip); - } - double median=0,mean=0; - for (int ic0=0;ic0tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;} - } - // - int kmed = nfree/2; - median = (tmpArr[kmed]+tmpArr[nfree-kmed-1])/2.; - if (nfree>0) mean /= nfree; - // - double shift = val - (type==AliITSAlignMille2Constraint::kTypeMean ? mean : median); - // - for (int ich=nc;ich--;) { - AliITSAlignMille2Module* child = parent->GetChild(ich); - // if (!child->IsFreeDOF(ip)) continue; - child->SetParVal(ip, child->GetParVal(ip) + shift); - npc++; - } - // - parent->SetParVal(ip, parent->GetParVal(ip) - shift); - AliInfo(Form("%s constraint: added %f shift to param[%d] of %d children of module %d: %s", - type==AliITSAlignMille2Constraint::kTypeMean ? "MEAN" : "MEDIAN",shift, - ip,npc,idm,parent->GetName())); - } - delete[] tmpArr; - // - // -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::PostConstrainOrphans(Int_t type,Double_t val, UInt_t pattern) -{ - // require that median or mean of modifications for the supermodules which have no parents is = val, i.e. - // the corrections moves the whole setup by fixed value (0 by default). - // pattern is the bit pattern for the parameters to constrain - // - int nc = fNModules; - // - int norph = 0; - for (int ich=nc;ich--;) if (!GetMilleModule(ich)->GetParent()) norph ++; - if (!norph) return; - double *tmpArr = new double[norph]; - // - for (int ip=0;ip>ip)&0x1)) continue; - // compute the mean and median of the deltas - int nfree = 0; - for (int ich=nc;ich--;) { - AliITSAlignMille2Module* child = GetMilleModule(ich); - // if (child->GetParent() || !child->IsFreeDOF(ip)) continue; - if (child->GetParent()) continue; - tmpArr[nfree++] = child->GetParVal(ip); - } - double median=0,mean=0; - for (int ic0=0;ic0tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;} - } - // - int kmed = nfree/2; - median = (tmpArr[kmed]+tmpArr[nfree-kmed-1])/2.; - if (nfree>0) mean /= nfree; - // - double shift = val - (type==AliITSAlignMille2Constraint::kTypeMean ? mean : median); - // - for (int ich=nc;ich--;) { - AliITSAlignMille2Module* child = GetMilleModule(ich); - // if (child->GetParent() || !child->IsFreeDOF(ip)) continue; - if (child->GetParent()) continue; - child->SetParVal(ip, child->GetParVal(ip) + shift); - npc++; - } - // - AliInfo(Form("%s constraint: added %f shift to param[%d] of %d orphan modules", - type==AliITSAlignMille2Constraint::kTypeMean ? "MEAN" : "MEDIAN",shift, - ip,npc)); - } - delete[] tmpArr; - // -} - -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::IsParModConstrained(const AliITSAlignMille2Module* mod,Int_t par, Bool_t &meanmed, Bool_t &gaussian) const -{ - // check if par of the module participates in some constraint, and set the flag for their types - meanmed = gaussian = kFALSE; - // - if ( mod->IsParConstrained(par) ) gaussian = kTRUE; // direct constraint on this param - // - for (int icstr=GetNConstraints();icstr--;) { - AliITSAlignMille2Constraint* cstr = GetConstraint(icstr); - // - if (!cstr->IncludesModPar(mod,par)) continue; - if (cstr->GetType()==AliITSAlignMille2ConstrArray::kTypeGaussian) gaussian = kTRUE; - else meanmed = kTRUE; - // - if (meanmed && gaussian) break; // no sense to check further - } - // - return meanmed||gaussian; -} - -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::IsParModFamilyVaried(const AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const -{ - // check if parameter par is varied for this module or its children up to the level depth - if (depth<0) return kFALSE; - if (mod->GetParOffset(par)>=0) return kTRUE; - for (int icld=mod->GetNChildren();icld--;) { - AliITSAlignMille2Module* child = mod->GetChild(icld); - if (IsParModFamilyVaried(child, par, depth-1)) return kTRUE; - } - return kFALSE; - // -} - -/* -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::IsParFamilyFree(AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const -{ - // check if parameter par is varied and is not subjected to gaussian constraint for the children up to the level depth - if (depth<0) return kTRUE; - for (int icld=mod->GetNChildren();icld--;) { - AliITSAlignMille2Module* child = mod->GetChild(icld); - //if (child->GetParOffset(par)<0) continue; // fixed - Bool_t cstMM=kFALSE,cstGS=kFALSE; - // does this child have gaussian constraint ? - if (!IsParModConstrained(child,par,cstMM,cstGS) || !cstGS ) return kTRUE; - // check its children - if (!IsParFamilyFree(child,par,depth-1)) return kTRUE; - } - return kFALSE; - // -} -*/ - -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::IsParFamilyFree(const AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const -{ - // check if parameter par is varied and is not subjected to gaussian constraint for the children up to the level depth - if (depth<0) return kFALSE; - for (int icld=mod->GetNChildren();icld--;) { - AliITSAlignMille2Module* child = mod->GetChild(icld); - //if (child->GetParOffset(par)<0) continue; // fixed - Bool_t cstMM=kFALSE,cstGS=kFALSE; - // does this child have gaussian constraint ? - if (!IsParModConstrained(child,par,cstMM,cstGS) || !cstGS ) return kTRUE; - // check its children - if (IsParFamilyFree(child,par,depth-1)) return kTRUE; - } - return kFALSE; - // -} - -//________________________________________________________________________________________________________ -Double_t AliITSAlignMille2::GetTDriftSDD() const -{ - // obtain drift time corrected for t0 - double t = fCluster.GetDriftTime(); - return t - fDriftTime0[ fCluster.GetUniqueID() ]; -} - -//________________________________________________________________________________________________________ -Double_t AliITSAlignMille2::GetVDriftSDD() const -{ - // obtain corrected drift speed - return fDriftSpeed[ fCluster.GetUniqueID() ]; -} - -//________________________________________________________________________________________________________ -Bool_t AliITSAlignMille2::FixedOrphans() const -{ - // are there fixed modules with no parent (normally in such a case - // the constraints on the orphans should not be applied - if (!IsConfigured()) { - AliInfo("Still not configured"); - return kFALSE; - } - for (int i=0;iGetParent()==0 && md->GetNParFree()==0) return kTRUE; - } - return kFALSE; -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConvertParamsToGlobal() -{ - // convert params in local frame to global one - double pars[AliITSAlignMille2Module::kMaxParGeom]; - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - if (mod->GeomParamsGlobal()) continue; - mod->GetGeomParamsGlo(pars); - mod->SetParVals(pars,AliITSAlignMille2Module::kMaxParGeom); - mod->SetGeomParamsGlobal(kTRUE); - } -} - -//________________________________________________________________________________________________________ -void AliITSAlignMille2::ConvertParamsToLocal() -{ - // convert params in global frame to local one - double pars[AliITSAlignMille2Module::kMaxParGeom]; - for (int imd=fNModules;imd--;) { - AliITSAlignMille2Module* mod = GetMilleModule(imd); - if (!mod->GeomParamsGlobal()) continue; - mod->GetGeomParamsLoc(pars); - mod->SetParVals(pars,AliITSAlignMille2Module::kMaxParGeom); - mod->SetGeomParamsGlobal(kFALSE); - } -} - +/************************************************************************** + * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. * + * * + * Author: The ALICE Off-line Project. * + * Contributors are mentioned in the code where appropriate. * + * * + * Permission to use, copy, modify and distribute this software and its * + * documentation strictly for non-commercial purposes is hereby granted * + * without fee, provided that the above copyright notice appears in all * + * copies and that both the copyright notice and this permission notice * + * appear in the supporting documentation. The authors make no claims * + * about the suitability of this software for any purpose. It is * + * provided "as is" without express or implied warranty. * + **************************************************************************/ + +/* $Id$ */ + +//----------------------------------------------------------------------------- +// +// Interface to AliMillePede2 alignment class for the ALICE ITS detector +// +// ITS specific alignment class which interface to AliMillepede. +// For each track ProcessTrack calculates the local and global derivatives +// at each hit and fill the corresponding local equations. Provide methods for +// fixing or constraning detection elements for best results. +// +// author M. Lunardon (thanks to J. Castillo), ruben.shahoyan@cern.ch +//----------------------------------------------------------------------------- + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "AliITSAlignMille2.h" +#include "AliITSgeomTGeo.h" +#include "AliGeomManager.h" +#include "AliMillePede2.h" +#include "AliTrackPointArray.h" +#include "AliAlignObjParams.h" +#include "AliLog.h" +#include "AliTrackFitterRieman.h" +#include "AliITSAlignMille2Constraint.h" +#include "AliITSAlignMille2ConstrArray.h" +#include "AliITSresponseSDD.h" +#include "AliITSTPArrayFit.h" +#include "AliCDBManager.h" +#include "AliCDBStorage.h" +#include "AliCDBEntry.h" +#include "AliITSsegmentationSDD.h" +#include "AliITSDriftSpeedArraySDD.h" +#include "AliITSCorrectSDDPoints.h" +#include "AliESDVertex.h" + +ClassImp(AliITSAlignMille2) + +const Char_t* AliITSAlignMille2::fgkRecKeys[] = { + "OCDB_PATH", + "OCDB_SPECIFIC", + "GEOMETRY_FILE", + "SUPERMODULE_FILE", + "CONSTRAINTS_REFERENCE_FILE", + "PREALIGNMENT_FILE", + "PRECALIBSDD_FILE", + "PREVDRIFTSDD_FILE", + "PRECORRMAPSDD_FILE", + "INITCORRMAPSDD_FILE", + "INITCALBSDD_FILE", + "INITVDRIFTSDD_FILE", + "INITDELTA_FILE", + "INITGEOM_FILE", + "SET_GLOBAL_DELTAS", + "CONSTRAINT_LOCAL", + "MODULE_VOLUID", + "MODULE_INDEX", + "SET_PSEUDO_PARENTS", + "SET_TRACK_FIT_METHOD", + "SET_MINPNT_TRA", + "SET_NSTDDEV", + "SET_RESCUT_INIT", + "SET_RESCUT_OTHER", + "SET_LOCALSIGMAFACTOR", + "SET_STARTFAC", + "SET_FINALFAC", + "SET_B_FIELD", + "SET_SPARSE_MATRIX", + "REQUIRE_POINT", + "CONSTRAINT_ORPHANS", + "CONSTRAINT_SUBUNITS", + "APPLY_CONSTRAINT", + "SET_EXTRA_CLUSTERS_MODE", + "SET_USE_TPAFITTER", + "SET_USE_LOCAL_YERROR", + "SET_MIN_POINTS_PER_MODULE", + "SET_USE_SDDVDCORRMULT", + "SET_WEIGHT_PT", + "SET_USE_DIAMOND", + "CORRECT_DIAMOND", + "SET_USE_VERTEX", + "SET_SAME_SDDT0" +}; + +const Char_t AliITSAlignMille2::fgkXYZ[] = "XYZ"; + +//======================================================================================================== + +AliITSAlignMille2* AliITSAlignMille2::fgInstance = 0; +Int_t AliITSAlignMille2::fgInstanceID = 0; + +//________________________________________________________________________________________________________ +AliITSAlignMille2::AliITSAlignMille2(const Char_t *configFilename,TList *userInfo ) +: TObject(), + fMillepede(0), + fStartFac(16.), + fFinalFac(1.), + fResCutInitial(100.), + fResCut(100.), + fNGlobal(0), + fNLocal(4), + fNStdDev(3), + fIsMilleInit(kFALSE), + fAllowPseudoParents(kFALSE), + // + fTPAFitter(0), + fCurrentModule(0), + fTrack(0), + fTrackBuff(0), + fCluster(), + fCurrentSensID(-1), + fClusLoc(12*3), + fClusGlo(12*3), + fClusSigLoc(12*3), + fGlobalDerivatives(0), + fMeasLoc(0), + fMeasGlo(0), + fSigmaLoc(0), + fConstrPT(-1), + fConstrPTErr(-1), + fConstrCharge(0), + fRunID(0), + // + fMinNPtsPerTrack(3), + fIniTrackParamsMeth(1), + fTotBadLocEqPoints(0), + fRieman(0), + // + fConstraints(0), + fCacheMatrixOrig(kMaxITSSensID+1), + fCacheMatrixCurr(kMaxITSSensID+1), + // + fUseGlobalDelta(kFALSE), + fTempExcludedModule(-1), + fUserProvided(0), + // + fIniUserInfo(userInfo), + fIniGeomPath(""), + fIniDeltaPath(""), + fIniSDDRespPath(""), + fPreCalSDDRespPath(""), + fIniSDDVDriftPath(""), + fPreSDDVDriftPath(""), + fIniSDDCorrMapPath(""), + fPreSDDCorrMapPath(""), + fConvertPreDeltas(kFALSE), + fGeometryPath(""), + fPreDeltaPath(""), + fConstrRefPath(""), + fDiamondPath(""), + fGeoManager(0), + fIsConfigured(kFALSE), + fPreAlignQF(0), +// + fIniRespSDD(0), + fPreRespSDD(0), + fIniVDriftSDD(0), + fPreVDriftSDD(0), + fIniCorrMapSDD(0), + fPreCorrMapSDD(0), + fSegmentationSDD(0), + fPrealignment(0), + fConstrRef(0), + fMilleModule(2), + fSuperModule(2), + fNModules(0), + fNSuperModules(0), + fUsePreAlignment(kFALSE), + fUseLocalYErr(kFALSE), + fBOn(kFALSE), + fBField(0.0), + fDataType(kCosmics), + fMinPntPerSens(0), + fBug(0), + fMilleVersion(2), + fExtraClustersMode(0), + fTrackWeight(1), + fWeightPt(0.), + fIsSDDVDriftMult(kFALSE), + fDiamond(), + fDiamondI(), + fUseDiamond(kFALSE), + fUseVertex(kFALSE), + fVertexSet(kFALSE), + fDiamondPointID(-1), + fDiamondModID(-1), + fCheckDiamondPoint(kDiamondCheckIfPrompt), + fFixCurvIfConstraned(kTRUE), + fCurvFitWasConstrained(kFALSE), + fConvAlgMatOld(100) +{ + /// main constructor that takes input from configuration file + for (int i=3;i--;) fSigmaFactor[i] = 1.0; + // + // new RS + for (int i=0;i<3;i++) { + fCorrDiamond[i] = 0; + } + for (int itp=0;itp=0) record.Remove(cmt); // skip comment + record.ReplaceAll("\t"," "); + record.ReplaceAll("\r"," "); + record.Remove(TString::kBoth,' '); + if (record.IsNull()) continue; // nothing to decode + if (!keyws.IsNull() && !record.BeginsWith(keyws.Data())) continue; // specific record was requested + // + recElems = record.Tokenize(" "); + recTitle = recElems->At(0)->GetName(); + recTitle.ToUpper(); + recOpt = recElems->GetLast()>0 ? recElems->At(1)->GetName() : ""; + break; + } + if (rew || !recElems) rewind(stream); + return recElems; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CheckConfigRecords(FILE* stream) +{ + // check the correctness of the record + TString record,recTitle; + int lineCnt = 0; + rewind(stream); + while (record.Gets(stream)) { + int cmt=record.Index("#"); + lineCnt++; + if (cmt>=0) record.Remove(cmt); // skip comment + record.ReplaceAll("\t"," "); + record.ReplaceAll("\r"," "); + record.Remove(TString::kBoth,' '); + if (record.IsNull()) continue; // nothing to decode + // extract keyword + int spc = record.Index(" "); + if (spc>0) recTitle = record(0,spc); + else recTitle = record; + recTitle.ToUpper(); + Bool_t strOK = kFALSE; + for (int ik=kNKeyWords;ik--;) if (recTitle == fgkRecKeys[ik]) {strOK = kTRUE; break;} + if (strOK) continue; + // + AliError(Form("Unknown keyword %s at line %d", + recTitle.Data(),lineCnt)); + return -1; + // + } + // + rewind(stream); + return 0; +} + + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadConfig(const Char_t *cfile) +{ + // return 0 if success + // 1 if error in module index or voluid + // + AliInfo(Form("Loading MillePede2 configuration from %s",cfile)); + AliCDBManager::Instance()->SetCacheFlag(kFALSE); + FILE *pfc=fopen(cfile,"r"); + if (!pfc) return -1; + // + TString record,recTitle,recOpt,recExt; + Int_t nrecElems,irec; + TObjArray *recArr=0; + // + fNModules = 0; + Bool_t stopped = kFALSE; + // + if (CheckConfigRecords(pfc)<0) return -1; + // + while(1) { + // + // ============= 1: we read some important records in predefined order ================ + // + recTitle = fgkRecKeys[kOCDBDefaultPath]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) && !recOpt.IsNull() ) { + AliInfo(Form("Configuration sets OCDB default storage to %s",recOpt.Data())); + AliCDBManager::Instance()->SetDefaultStorage( gSystem->ExpandPathName(recOpt.Data()) ); + TObjString* objStr = (TObjString*)AliCDBManager::Instance()->GetStorageMap()->GetValue("default"); + if (!objStr) {stopped = kTRUE; break;} + objStr->SetUniqueID(1); // mark as user set + } + // + if (fIniUserInfo && ProcessUserInfo(fIniUserInfo)) { AliError("Failed to process intial User Info"); stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[kGeomFile]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) fGeometryPath = gSystem->ExpandPathName(recOpt.Data()); + if ( LoadGeometry(fGeometryPath) ) { AliError("Failed to find/load target ideal Geometry"); stopped = kTRUE; break;} + // + // Do we use new TrackPointArray fitter ? + recTitle = fgkRecKeys[kTPAFitter]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) fTPAFitter = new AliITSTPArrayFit(kNLocal); + // + recTitle = fgkRecKeys[kSuperModileFile]; + if ( !GetConfigRecord(pfc,recTitle,recOpt,1) || + recOpt.IsNull() || + gSystem->ExpandPathName(recOpt) || + gSystem->AccessPathName(recOpt.Data()) || + LoadSuperModuleFile(recOpt.Data())) + { AliError("Failed to find/load SuperModules"); stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[kConstrRefFile]; // LOCAL_CONSTRAINTS are defined wrt these deltas + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) ) { + if (recOpt.IsNull() || recOpt=="IDEAL") SetConstraintWrtRef( "IDEAL" ); + else { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + if ( SetConstraintWrtRef(recOpt.Data()) ) + { AliError("Failed to load reference deltas for local constraints"); stopped = kTRUE; break;} + } + } + // + recTitle = fgkRecKeys[kInitGeomFile]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull() ) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fIniGeomPath = recOpt; + gSystem->ExpandPathName(fIniGeomPath); + fUserProvided |= kSameInitGeomBit; + AliInfo(Form("Configuration sets Production Geometry to %s",fIniGeomPath.Data())); + } + if (fIniGeomPath.IsNull()) fIniGeomPath = fGeometryPath; + // + recTitle = fgkRecKeys[kInitDeltaFile]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull() ) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fIniDeltaPath = recOpt; + gSystem->ExpandPathName(fIniDeltaPath); + fUserProvided |= kSameInitDeltasBit; + AliInfo(Form("Configuration sets Production Deltas to %s",fIniDeltaPath.Data())); + } + // + recTitle = fgkRecKeys[kPreDeltaFile]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) ) { + if (!recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fPreDeltaPath = recOpt; + gSystem->ExpandPathName(fPreDeltaPath); + } + else if (!fIniDeltaPath.IsNull()) { + AliInfo("PreAlignment Deltas keyword is present but empty, will set to Init Deltas of the first tree"); + fPreDeltaPath = fIniDeltaPath; + if (fIniGeomPath != fGeometryPath) fConvertPreDeltas = kTRUE; // production and target geometries are different, request conversion + } + AliInfo(Form("Configuration sets PreAlignment Deltas to %s",fPreDeltaPath.Data())); + } + // + // if initial deltas were provided, load them, apply to geometry and store are "original" matrices + if (CacheMatricesOrig()) {stopped = kTRUE; break;} + // + // then load prealignment deltas + if (!fPreDeltaPath.IsNull()) { + if (fConvertPreDeltas) ConvertDeltas(); // Prealignment deltas are the same as production ones, but need conversion to new geometry + else if (LoadDeltas(fPreDeltaPath,fPrealignment)) {stopped = kTRUE; break;} // read deltas from the file + } + if (fPrealignment && ApplyToGeometry()) {stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[ kInitCalSDDFile ]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fIniSDDRespPath = recOpt; + gSystem->ExpandPathName(fIniSDDRespPath); + fUserProvided |= kSameInitSDDRespBit; + AliInfo(Form("Configuration sets Production SDD Response to %s",fIniSDDRespPath.Data())); + } + if (LoadSDDResponse(fIniSDDRespPath, fIniRespSDD) ) {stopped = kTRUE; break;} + // + // + recTitle = fgkRecKeys[ kInitCorrMapSDDFile ]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fIniSDDCorrMapPath = recOpt; + gSystem->ExpandPathName(fIniSDDCorrMapPath); + fUserProvided |= kSameInitSDDCorrMapBit; + AliInfo(Form("Configuration sets Production SDD Correction Map to %s",fIniSDDCorrMapPath.Data())); + } + if (LoadSDDCorrMap(fIniSDDCorrMapPath, fIniCorrMapSDD) ) {stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[kPreCalSDDFile]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) ) { + if (!recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fPreCalSDDRespPath = recOpt; + gSystem->ExpandPathName(fPreCalSDDRespPath); + } + else if (!fIniSDDRespPath.IsNull()) { + AliInfo("PreCalibration SDD response keyword is present but empty, will set to Init SDD repsonse"); + fPreCalSDDRespPath = fIniSDDRespPath; + } + AliInfo(Form("Configuration sets PreCalibration SDD Response to %s",fPreCalSDDRespPath.Data())); + } + // + if (LoadSDDResponse(fPreCalSDDRespPath, fPreRespSDD) ) {stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[kPreCorrMapSDDFile]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) ) { + if (!recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fPreSDDCorrMapPath = recOpt; + gSystem->ExpandPathName(fPreSDDCorrMapPath); + } + else if (!fIniSDDCorrMapPath.IsNull()) { + AliInfo("PreCalibration SDD Correction Map keyword is present but empty, will set to Init SDD Correction Map"); + fPreSDDCorrMapPath = fIniSDDCorrMapPath; + } + AliInfo(Form("Configuration sets PreCalibration SDD Correction Map to %s",fPreSDDCorrMapPath.Data())); + } + // + if (LoadSDDCorrMap(fPreSDDCorrMapPath, fPreCorrMapSDD) ) {stopped = kTRUE; break;} + // // + recTitle = fgkRecKeys[ kInitVDriftSDDFile ]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fIniSDDVDriftPath = recOpt; + gSystem->ExpandPathName(fIniSDDVDriftPath); + fUserProvided |= kSameInitSDDVDriftBit; + AliInfo(Form("Configuration sets Production SDD VDrift to %s",fIniSDDVDriftPath.Data())); + } + if (LoadSDDVDrift(fIniSDDVDriftPath, fIniVDriftSDD) ) {stopped = kTRUE; break;} + // + recTitle = fgkRecKeys[ kPreVDriftSDDFile ]; + if ( (recArr = GetConfigRecord(pfc,recTitle,recOpt,1)) && !recOpt.IsNull()) { + for (int i=2;i<=recArr->GetLast();i++) {recOpt += " "; recOpt += recArr->At(i)->GetName();} // in case of OCDB string + fPreSDDVDriftPath = recOpt; + gSystem->ExpandPathName(fPreSDDVDriftPath); + AliInfo(Form("Configuration sets PreCalibration SDD VDrift to %s",fPreSDDVDriftPath.Data())); + if (LoadSDDVDrift(fPreSDDVDriftPath, fPreVDriftSDD) ) {stopped = kTRUE; break;} + } + // + recTitle = fgkRecKeys[ kGlobalDeltas ]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) SetUseGlobalDelta(kTRUE); + // + recTitle = fgkRecKeys[ kUseDiamond ]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) { + if (!GetUseGlobalDelta()) { + AliError("Diamond constraint is supported only for Global Frame mode"); + stopped = kTRUE; + break; + } + fUseDiamond = kTRUE; + if (!recOpt.IsNull()) { + fDiamondPath = recOpt; + gSystem->ExpandPathName(fDiamondPath); + fUserProvided |= kSameDiamondBit; + AliInfo(Form("Configuration sets Diamond constraint to %s",fDiamondPath.Data())); + } + } + // + recTitle = fgkRecKeys[ kUseVertex ]; + if ( GetConfigRecord(pfc,recTitle,recOpt,1) ) { + if (!GetUseGlobalDelta()) { + AliError("Vertex constraint is supported only for Global Frame mode"); + stopped = kTRUE; + break; + } + fUseVertex = kTRUE; + if (fUseDiamond) { + AliError("Cannot use Vertex and Diamond constraints at the same time"); + stopped = kTRUE; + break; + } + AliInfo("Will use Vertex constraint when available"); + } + // =========== 2: see if there are local gaussian constraints defined ===================== + // Note that they should be loaded before the modules declaration + // + recTitle = fgkRecKeys[ kConstrLocal ]; + while( (recArr=GetConfigRecord(pfc,recTitle,recOpt,0)) ) { + nrecElems = recArr->GetLast()+1; + if (recOpt.IsFloat()) {stopped = kTRUE; break;} // wrong name + if (GetConstraint(recOpt.Data())) { + AliError(Form("Existing constraint %s repeated",recOpt.Data())); + stopped = kTRUE; break; + } + recExt = recArr->At(2)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + double val = recExt.Atof(); + recExt = recArr->At(3)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + double err = recExt.Atof(); + int nwgh = nrecElems - 4; + double *wgh = new double[nwgh]; + for (nwgh=0,irec=4;irecAt(irec)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + wgh[nwgh++] = recExt.Atof(); + } + if (stopped) {delete[] wgh; break;} + // + ConstrainLocal(recOpt.Data(),wgh,nwgh,val,err); + delete[] wgh; + // + } // end while for loop over local constraints + if (stopped) break; + // + // =========== 3: now read modules to align =================================== + // + rewind(pfc); + // create fixed modules + for (int j=0; jIsAlignable()) continue; + AliITSAlignMille2Module* mod = new AliITSAlignMille2Module(*proto); + // the matrix might be updated in case some prealignment was applied, check + TGeoHMatrix* mup = AliGeomManager::GetMatrix(mod->GetName()); + if (mup) *(mod->GetMatrix()) = *mup; + fMilleModule.AddAtAndExpand(mod,fNModules); + mod->SetGeomParamsGlobal(fUseGlobalDelta); + mod->SetUniqueID(fNModules++); + mod->SetNotInConf(kTRUE); + } + CreateVertexModule(); + // + while( (recArr=GetConfigRecord(pfc,recTitle="",recOpt,0)) ) { + if (!(recTitle==fgkRecKeys[ kModVolID ] || recTitle==fgkRecKeys[ kModIndex ])) continue; + // Expected format: MODULE id tolX tolY tolZ tolPsi tolTh tolPhi [[sigX sigY sigZ] extra params] + // where tol* is the tolerance (sigma) for given DOF. 0 means fixed + // sig* is the scaling parameters for the errors of the clusters of this module + // extra params are defined for specific modules, e.g. t0 and vdrift corrections of SDD + // + nrecElems = recArr->GetLast()+1; + if (nrecElems<2 || !recOpt.IsDigit()) {stopped = kTRUE; break;} + int idx = recOpt.Atoi(); + UShort_t voluid = (idx<=kMaxITSSensID) ? GetModuleVolumeID(idx) : idx; + AliITSAlignMille2Module* mod = 0; + // + if (voluid>=kMinITSSupeModuleID) { // custom supermodule + mod = GetMilleModuleByVID(voluid); + if (!mod) { // need to create + for (int j=0; jGetVolumeID()) { + mod = new AliITSAlignMille2Module(*GetSuperModule(j)); + // the matrix might be updated in case some prealignment was applied, check + TGeoHMatrix* mup = AliGeomManager::GetMatrix(mod->GetName()); + if (mup) *(mod->GetMatrix()) = *mup; + fMilleModule.AddAtAndExpand(mod,fNModules); + mod->SetGeomParamsGlobal(fUseGlobalDelta); + mod->SetUniqueID(fNModules++); + break; + } + } + } + mod->SetNotInConf(kFALSE); + } + else if (idx<=kMaxITSSensVID) { + mod = new AliITSAlignMille2Module(voluid); + fMilleModule.AddAtAndExpand(mod,fNModules); + mod->SetGeomParamsGlobal(fUseGlobalDelta); + mod->SetUniqueID(fNModules++); + } + if (!mod) {stopped = kTRUE; break;} // bad volid + // + // geometry variation settings + for (int i=0;i= nrecElems) break; + recExt = recArr->At(irec)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + mod->SetFreeDOF(i, recExt.Atof() ); + } + if (stopped) break; + // + // scaling factors for cluster errors + // first set default ones + for (int i=0;i<3;i++) mod->SetSigmaFactor(i, fSigmaFactor[i]); + for (int i=0;i<3;i++) { + irec = i+8; + if (irec >= nrecElems) break; + recExt = recArr->At(irec)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + mod->SetSigmaFactor(i, recExt.Atof() ); + } + if (stopped) break; + // + // now comes special detectors treatment + if (mod->IsSDD()) { + double vl = 0; + if (nrecElems>11) { + recExt = recArr->At(11)->GetName(); + if (recExt.IsFloat()) vl = recExt.Atof(); + else {stopped = kTRUE; break;} + irec = 11; + } + mod->SetFreeDOF(AliITSAlignMille2Module::kDOFT0,vl); + // + Bool_t cstLR = kFALSE; + for (int lr=0;lr<2;lr++) { // left right side vdrift corrections + vl = 0; + if (nrecElems>12+lr) { + recExt = recArr->At(12+lr)->GetName(); + if (recExt.IsFloat()) vl = recExt.Atof(); + else {stopped = kTRUE; break;} + irec = 12+lr; + } + mod->SetFreeDOF(lr==0 ? AliITSAlignMille2Module::kDOFDVL : AliITSAlignMille2Module::kDOFDVR,vl); + if (lr==1 && vl>=10) cstLR = kTRUE; // the right side should be constrained to left one + } + if (cstLR) mod->SetVDriftLRSame(); + } + // + mod->EvaluateDOF(); + // + // now check if there are local constraints on this module + for (++irec;irecAt(irec)->GetName(); + if (recExt.IsFloat()) {stopped=kTRUE;break;} + AliITSAlignMille2ConstrArray* cstr = (AliITSAlignMille2ConstrArray*)GetConstraint(recExt.Data()); + if (!cstr) { + AliInfo(Form("No Local constraint %s was declared",recExt.Data())); + stopped=kTRUE; + break; + } + cstr->AddModule(mod); + } + if (stopped) break; + } // end while for loop over modules + if (stopped) break; + // + if (fNModules==0) {AliError("Failed to find any MODULE"); stopped = kTRUE; break;} + BuildHierarchy(); // preprocess loaded modules + // + // =========== 4: the rest may come in arbitrary order ======================================= + rewind(pfc); + while ( (recArr=GetConfigRecord(pfc,recTitle="",recOpt,0))!=0 ) { + // + nrecElems = recArr->GetLast()+1; + // + // some simple flags ----------------------------------------------------------------------- + // + if (recTitle == fgkRecKeys[ kPseudoParents ]) SetAllowPseudoParents(kTRUE); + // + // some optional parameters ---------------------------------------------------------------- + else if (recTitle == fgkRecKeys[ kTrackFitMethod ]) { + if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} + SetInitTrackParamsMeth(recOpt.Atoi()); + } + // + else if (recTitle == fgkRecKeys[ kMinPntTrack ]) { + if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} + fMinNPtsPerTrack = recOpt.Atoi(); + } + // + else if (recTitle == fgkRecKeys[ kNStDev ]) { + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + fNStdDev = (Int_t)recOpt.Atof(); + } + // + else if (recTitle == fgkRecKeys[ kResCutInit ]) { + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + fResCutInitial = recOpt.Atof(); + } + // + else if (recTitle == fgkRecKeys[ kResCutOther ]) { + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + fResCut = recOpt.Atof(); + } + // + else if (recTitle == fgkRecKeys[ kLocalSigFactor ]) { //------------------------- + for (irec=0;irec<3;irec++) if (nrecElems>irec+1) { + fSigmaFactor[irec] = ((TObjString*)recArr->At(irec+1))->GetString().Atof(); + if (fSigmaFactor[irec]<=0.) stopped = kTRUE; + } + if (stopped) break; + } + // + else if (recTitle == fgkRecKeys[ kStartFactor ]) { //------------------------- + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + fStartFac = recOpt.Atof(); + } + // + else if (recTitle == fgkRecKeys[ kFinalFactor ]) { //------------------------- + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + fFinalFac = recOpt.Atof(); + } + // + // pepo2708909 + else if (recTitle == fgkRecKeys[ kExtraClustersMode ]) { //------------------------- + if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} + fExtraClustersMode = recOpt.Atoi(); + } + // endpepo270809 + // + else if (recTitle == fgkRecKeys[ kBField ]) { //------------------------- + if (recOpt.IsNull() || !recOpt.IsFloat() ) {stopped = kTRUE; break;} + SetBField( recOpt.Atof() ); + } + // + else if (recTitle == fgkRecKeys[ kSDDVDCorrMult ]) { //------------------------- + SetSDDVDCorrMult( recOpt.IsNull() || (recOpt.IsFloat() && (recOpt.Atof())>-0.5) ); + } + // + else if (recTitle == fgkRecKeys[ kWeightPt ]) { //------------------------- + double wgh = 1; + if (!recOpt.IsNull()) { + if (!recOpt.IsFloat()) {stopped = kTRUE; break;} + else wgh = recOpt.Atof(); + } + SetWeightPt(wgh); + } + // + else if (recTitle == fgkRecKeys[ kSparseMatrix ]) { // matrix solver type + // + AliMillePede2::SetGlobalMatSparse(kTRUE); + if (recOpt.IsNull()) continue; + // solver type and settings + if (recOpt == "MINRES") AliMillePede2::SetIterSolverType( AliMinResSolve::kSolMinRes ); + else if (recOpt == "FGMRES") AliMillePede2::SetIterSolverType( AliMinResSolve::kSolFGMRes ); + else {stopped = kTRUE; break;} + // + if (nrecElems>=3) { // preconditioner type + recExt = recArr->At(2)->GetName(); + if (!recExt.IsDigit()) {stopped = kTRUE; break;} + AliMillePede2::SetMinResPrecondType( recExt.Atoi() ); + } + // + if (nrecElems>=4) { // tolerance + recExt = recArr->At(3)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + AliMillePede2::SetMinResTol( recExt.Atof() ); + } + // + if (nrecElems>=5) { // maxIter + recExt = recArr->At(4)->GetName(); + if (!recExt.IsDigit()) {stopped = kTRUE; break;} + AliMillePede2::SetMinResMaxIter( recExt.Atoi() ); + } + } + // + else if (recTitle == fgkRecKeys[ kRequirePoint ]) { //------------------------- + // syntax: REQUIRE_POINT where ndet updw nreqpts + // where = LAYER or DETECTOR + // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) + // updw = 1 for Y>0, -1 for Y<0, 0 if not specified + // nreqpts = minimum number of points of that type + if (nrecElems>=5) { + recOpt.ToUpper(); + int lr = ((TObjString*)recArr->At(2))->GetString().Atoi() - 1; + int hb = ((TObjString*)recArr->At(3))->GetString().Atoi(); + int np = ((TObjString*)recArr->At(4))->GetString().Atoi(); + // + int rtp = -1; // use for run type + if (nrecElems>5) { + TString tpstr = ((TObjString*)recArr->At(5))->GetString(); + if ( tpstr.Contains("cosmics",TString::kIgnoreCase) ) rtp = kCosmics; + else if ( tpstr.Contains("collision",TString::kIgnoreCase) ) rtp = kCollision; + else {stopped = kTRUE; break;} + } + // + int tpmn= rtp<0 ? 0 : rtp; + int tpmx= rtp<0 ? kNDataType-1 : rtp; + for (int itp=tpmn;itp<=tpmx;itp++) { + fRequirePoints[itp]=kTRUE; + if (recOpt == "LAYER") { + if (lr<0 || lr>5) {stopped = kTRUE; break;} + if (hb>0) fNReqLayUp[itp][lr]=np; + else if (hb<0) fNReqLayDown[itp][lr]=np; + else fNReqLay[itp][lr]=np; + } + else if (recOpt == "DETECTOR") { + if (lr<0 || lr>2) {stopped = kTRUE; break;} + if (hb>0) fNReqDetUp[itp][lr]=np; + else if (hb<0) fNReqDetDown[itp][lr]=np; + else fNReqDet[itp][lr]=np; + } + else {stopped = kTRUE; break;} + } + if (stopped) break; + } + else {stopped = kTRUE; break;} + } + // + // global constraints on the subunits/orphans + else if (recTitle == fgkRecKeys[ kConstrOrphans ]) { //------------------------ + // expect CONSTRAINT_ORPHANS MEAN/MEDIAN Value parID0 ... parID1 ... + if (nrecElems<4) {stopped = kTRUE; break;} + recExt = recArr->At(2)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + double val = recExt.Atof(); + UInt_t pattern = 0; + for (irec=3;irecAt(irec)->GetName(); + if (!recExt.IsDigit()) {stopped = kTRUE; break;} + pattern |= 0x1 << recExt.Atoi(); + } + if (stopped) break; + if (recOpt == "MEAN") ConstrainOrphansMean(val,pattern); + else if (recOpt == "MEDIAN") ConstrainOrphansMedian(val,pattern); + else {stopped = kTRUE; break;} + } + // + else if (recTitle == fgkRecKeys[ kConstrSubunits ]) { //------------------------ + // expect CONSTRAINT_SUBUNITS MEAN/MEDIAN Value parID0 ... parID1 ... VolID1 ... VolIDn - VolIDm + if (nrecElems<5) {stopped = kTRUE; break;} + recExt = recArr->At(2)->GetName(); + if (!recExt.IsFloat()) {stopped = kTRUE; break;} + double val = recExt.Atof(); + UInt_t pattern = 0; + for (irec=3;irecAt(irec)->GetName(); + if (!recExt.IsDigit()) {stopped = kTRUE; break;} + int parid = recExt.Atoi(); + if (paridAt(irec)->GetName(); + if (recExt == "-") {rangeStart = curID; continue;} // range is requested + else if (!recExt.IsDigit()) {stopped = kTRUE; break;} + else curID = recExt.Atoi(); + // + if (curID<=kMaxITSSensID) curID = GetModuleVolumeID(curID); + // this was a range start or single + int start; + if (rangeStart>=0) {start = rangeStart+1; rangeStart=-1;} // continue the range + else start = curID; // create constraint either for single module (or 1st in the range) + for (int id=start;id<=curID;id++) { + int id0 = IsVIDDefined(id); + if (id0<0) {AliDebug(3,Form("Undefined module %d requested in the SubUnits constraint, skipping",id)); continue;} + if (meanC) ConstrainModuleSubUnitsMean(id0,val,pattern); + else ConstrainModuleSubUnitsMedian(id0,val,pattern); + } + } + if (rangeStart>=0) stopped = kTRUE; // unfinished range + if (stopped) break; + } + // + // association of modules with local constraints + else if (recTitle == fgkRecKeys[ kApplyConstr ]) { //------------------------ + // expect APPLY_CONSTRAINT NAME [NAME1...] [VolID1 ... VolIDn - VolIDm] + if (nrecElems<3) {stopped = kTRUE; break;} + int nmID0=-1,nmID1=-1; + for (irec=1;irecAt(irec)->GetName(); + if (recExt.IsFloat()) break; + // check if such a constraint was declared + if (!GetConstraint(recExt.Data())) { + AliInfo(Form("No Local constraint %s was declared",recExt.Data())); + stopped=kTRUE; + break; + } + if (nmID0<0) nmID0 = irec; + nmID1 = irec; + } + if (stopped) break; + // + if (irec>=nrecElems) {stopped = kTRUE; break;} // no modules provided + // + // now read the list of modules to constrain + int curID = -1; + int rangeStart = -1; + for (;irecAt(irec)->GetName(); + if (recExt == "-") {rangeStart = curID; continue;} // range is requested + else if (!recExt.IsDigit()) {stopped = kTRUE; break;} + else curID = recExt.Atoi(); + // + if (curID<=kMaxITSSensID) curID = GetModuleVolumeID(curID); + // + // this was a range start or single + int start; + if (rangeStart>=0) {start = rangeStart+1; rangeStart=-1;} // continue the range + else start = curID; // create constraint either for single module (or 1st in the range) + for (int id=start;id<=curID;id++) { + AliITSAlignMille2Module *md = GetMilleModuleByVID(id); + if (!md) {AliDebug(3,Form("Undefined module %d requested in the Local constraint, skipping",id)); continue;} + for (int nmid=nmID0;nmid<=nmID1;nmid++) + ((AliITSAlignMille2ConstrArray*)GetConstraint(recArr->At(nmid)->GetName()))->AddModule(md); + } + } + if (rangeStart>=0) stopped = kTRUE; // unfinished range + if (stopped) break; + } + // + // request of the same T0 for group of SDD modules + else if (recTitle == fgkRecKeys[ kSameSDDT0 ]) { //------------------------ + // expect SET_SAME_SDDT0 [SensID1 ... SensIDn - SensIDm] + if (nrecElems<3) {stopped = kTRUE; break;} + // + // now read the list of modules to constrain + int curID = -1; + int rangeStart = -1; + AliITSAlignMille2ConstrArray *cstrT0 = new AliITSAlignMille2ConstrArray("SDDT0",0,0,0,0); + int naddM = 0; + cstrT0->SetPattern(BIT(AliITSAlignMille2Module::kDOFT0)); + for (irec=1;irecAt(irec)->GetName(); + if (recExt == "-") {rangeStart = curID; continue;} // range is requested + else if (!recExt.IsDigit()) {stopped = kTRUE; break;} + else curID = recExt.Atoi(); + // + if (curID=kSDDoffsID+kNSDDmod) {stopped = kTRUE; break;} + // + // this was a range start or single + int start; + if (rangeStart>=0) {start = rangeStart+1; rangeStart=-1;} // continue the range + else start = curID; // create constraint either for single module (or 1st in the range) + for (int id=start;id<=curID;id++) { + int vid = AliITSAlignMille2Module::GetVolumeIDFromIndex(id); + if (vid<=1) {AliDebug(3,Form("Undefined module index %d requested in the SAME_SDDT0 constraint, skipping",id)); continue;} + AliITSAlignMille2Module *md = GetMilleModuleByVID(vid); + if (!md) {AliDebug(3,Form("Undefined module %d requested in the Local constraint, skipping",id)); continue;} + cstrT0->AddModule(md,kFALSE); + naddM++; + } + } + if (rangeStart>=0) stopped = kTRUE; // unfinished range + if (stopped) break; + if (naddM<2) delete cstrT0; + else { + cstrT0->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstrT0); + } + } + // + // Do we use new local Y errors? + else if (recTitle == fgkRecKeys[ kUseLocalYErr ]) { + // expect SET_TPAFITTER + fUseLocalYErr = kTRUE; + } + // + else if (recTitle == fgkRecKeys[ kMinPointsSens ]) { //------------------------- + if (recOpt.IsNull() || !recOpt.IsDigit() ) {stopped = kTRUE; break;} + SetMinPointsPerSensor( recOpt.Atoi() ); + } + // + else if (recTitle == fgkRecKeys[ kOCDBSpecificPath ]) { //------------------------- + if (recOpt.IsNull() || nrecElems<3 ) {stopped = kTRUE; break;} + AliCDBManager::Instance()->SetSpecificStorage(recOpt.Data(), gSystem->ExpandPathName(recArr->At(2)->GetName())); + AliInfo(Form("Configuration sets OCDB specific storage %s to %s",recOpt.Data(),recArr->At(2)->GetName())); + TObjString *pths = (TObjString*)AliCDBManager::Instance()->GetStorageMap()->GetValue(recOpt.Data()); + if (!pths) { stopped = kTRUE; break; } + pths->SetUniqueID(1); // mark as set by user + } + // + else if (recTitle == fgkRecKeys[ kCorrectDiamond ] && fUseDiamond) { //------------------------- + if (nrecElems<4) {stopped = kTRUE; break;} + for (int i=0;i<3;i++) fCorrDiamond[i] = ((TObjString*)recArr->At(i+1))->GetString().Atof(); + AliInfo(Form("Correction %+.4f %+.4f %+.4f will be applied to diamond",fCorrDiamond[0],fCorrDiamond[1],fCorrDiamond[2])); + } + // + else continue; // already processed record + // + } // end of while loop 4 over the various params + // + break; + } // end of while(1) loop + // + fclose(pfc); + if (!fDiamondPath.IsNull() && IsDiamondUsed() && LoadDiamond(fDiamondPath) ) stopped = kTRUE; + if (stopped) { + AliError(Form("Failed on record %s %s ...\n",recTitle.Data(),recOpt.Data())); + return -1; + } + // + if (CacheMatricesCurr()) return -1; + SetUseLocalYErrors(fUseLocalYErr); // YErr used only with TPAFitter + fSegmentationSDD = new AliITSsegmentationSDD(); + // + fIsConfigured = kTRUE; + return 0; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::BuildHierarchy() +{ + // build the hieararhy of the modules to align + // + if (!GetUseGlobalDelta() && PseudoParentsAllowed()) { + AliInfo("PseudoParents mode is allowed only when the deltas are global\n" + "Since Deltas are local, switching to NoPseudoParents"); + SetAllowPseudoParents(kFALSE); + } + // set parent/child relationship for modules to align + AliInfo("Setting parent/child relationships\n"); + // + // 1) child -> parent reference + for (int ipar=0;iparIsSensor()) continue; // sensor cannot be a parent + // + for (int icld=0;icldBelongsTo(parent)) continue; + // child cannot have more sensors than the parent + if (child->GetNSensitiveVolumes() > parent->GetNSensitiveVolumes()) continue; + // + AliITSAlignMille2Module* parOld = child->GetParent(); + // is this parent candidate closer than the old parent ? + if (parOld && parOld->GetNSensitiveVolumes()GetNSensitiveVolumes()) continue; // parOld is closer + child->SetParent(parent); + } + // + } + // + // add parent -> children reference + for (int icld=0;icldGetParent(); + if (parent) parent->AddChild(child); + } + // + // reorder the modules in such a way that parents come first + for (int icld=0;icldGetParent()) && (parent->GetUniqueID()>child->GetUniqueID()) ) { + // swap + fMilleModule[icld] = parent; + fMilleModule[parent->GetUniqueID()] = child; + child->SetUniqueID(parent->GetUniqueID()); + parent->SetUniqueID(icld); + child = parent; + } + // + } + // + // Go over the child->parent chain and mark modules with explicitly provided sensors. + // If the sensors of the unit are explicitly declared, all undeclared sensors are + // suppresed in this unit. + for (int icld=fNModules;icld--;) { + AliITSAlignMille2Module* child = GetMilleModule(icld); + AliITSAlignMille2Module* parent = child->GetParent(); + if (!parent) continue; + // + // check if this parent was already processed + if (!parent->AreSensorsProvided()) { + parent->DelSensitiveVolumes(); + parent->SetSensorsProvided(kTRUE); + } + // reattach sensors to parent + for (int isc=child->GetNSensitiveVolumes();isc--;) { + UShort_t senVID = child->GetSensVolVolumeID(isc); + if (!parent->IsIn(senVID)) parent->AddSensitiveVolume(senVID); + } + } + // +} + +// pepo +//________________________________________________________________________________________________________ +void AliITSAlignMille2::SetCurrentModule(Int_t id) +{ + // set the current supermodule + // new meaning + if (fMilleVersion>=2) { + fCurrentModule = GetMilleModule(id); + return; + } + // old meaning + if (fMilleVersion<=1) { + Int_t index=id; + /// set as current the SuperModule that contains the 'index' sens.vol. + if (index<0 || index>2197) { + AliInfo("index does not correspond to a sensitive volume!"); + return; + } + UShort_t voluid=AliITSAlignMille2Module::GetVolumeIDFromIndex(index); + Int_t k=IsContained(voluid); + if (k>=0){ + fCurrentSensID = index; + fCluster.SetVolumeID(voluid); + fCluster.SetXYZ(0,0,0); + InitModuleParams(); + } + else + AliInfo(Form("module %d not defined\n",index)); + } +} +// endpepo +//________________________________________________________________________________________________________ +void AliITSAlignMille2::SetRequiredPoint(Char_t* where, Int_t ndet, Int_t updw, Int_t nreqpts,Int_t runtype) +{ + // set minimum number of points in specific detector or layer + // where = LAYER or DETECTOR + // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) + // updw = 1 for Y>0, -1 for Y<0, 0 if not specified + // nreqpts = minimum number of points of that type + ndet--; + int tpmn= runtype<0 ? 0 : runtype; + int tpmx= runtype<0 ? kNDataType-1 : runtype; + // + for (int itp=tpmn;itp<=tpmx;itp++) { + fRequirePoints[itp]=kTRUE; + if (strstr(where,"LAYER")) { + if (ndet<0 || ndet>5) return; + if (updw>0) fNReqLayUp[itp][ndet]=nreqpts; + else if (updw<0) fNReqLayDown[itp][ndet]=nreqpts; + else fNReqLay[itp][ndet]=nreqpts; + } + else if (strstr(where,"DETECTOR")) { + if (ndet<0 || ndet>2) return; + if (updw>0) fNReqDetUp[itp][ndet]=nreqpts; + else if (updw<0) fNReqDetDown[itp][ndet]=nreqpts; + else fNReqDet[itp][ndet]=nreqpts; + } + } +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetModuleIndex(const Char_t *symname) +{ + /// index from symname + if (!symname) return -1; + for (Int_t i=0;i<=kMaxITSSensID; i++) { + if (!strcmp(symname,AliITSgeomTGeo::GetSymName(i))) return i; + } + return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetModuleIndex(UShort_t voluid) +{ + /// index from volume ID + AliGeomManager::ELayerID lay = AliGeomManager::VolUIDToLayer(voluid); + if (lay<1|| lay>6) return -1; + Int_t idx=Int_t(voluid)-2048*lay; + if (idx>=AliGeomManager::LayerSize(lay)) return -1; + for (Int_t ilay=1; ilay0 && layerId<7 && modId>=0 && modIdGetIndex()==index) return GetSuperModule(i)->GetVolumeID(); + } + } + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadGeometry(TString& path) +{ + // initialize ideal geometry + AliInfo(Form("Loading ideal geometry %s",path.Data())); + if (path.IsNull()) { + AliError("Path to geometry is not provided"); + return -1; + } + // + AliCDBEntry *entry = 0; + TGeoManager *gm = 0; + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + gm = (TGeoManager*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("ALICE")) gm = (TGeoManager*)precf->Get("ALICE"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + gm = (TGeoManager*) entry->GetObject(); + if (gm && gm->InheritsFrom(TGeoManager::Class())) entry->SetObject(NULL); + else gm = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + precf->Close(); + delete precf; + break; + } + // + if (!gm) {AliError(Form("Failed to load geometry from %s",path.Data())); return -1;} + AliGeomManager::SetGeometry(gm); + fGeoManager = AliGeomManager::GetGeometry(); + if (!fGeoManager) { + AliInfo("Couldn't initialize geometry"); + return -1; + } + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::SetConstraintWrtRef(const char* reffname) +{ + // Load the global deltas from this file. The local gaussian constraints on some modules + // will be defined with respect to the deltas from this reference file, converted to local + // delta format. Note: conversion to local format requires reloading the geometry! + // + AliInfo(Form("Loading reference deltas for local constraints from %s",reffname)); + if (!fGeoManager) return -1; + fConstrRefPath = reffname; + if (fConstrRefPath == "IDEAL") { // the reference is the ideal geometry, just create dummy reference array + fConstrRef = new TClonesArray("AliAlignObjParams",1); + return 0; + } + if (LoadDeltas(fConstrRefPath,fConstrRef)) return -1; + // + // we need ideal geometry to convert global deltas to local ones + if (fUsePreAlignment) { + AliError("The call of SetConstraintWrtRef must be done before application of the prealignment"); + return -1; + } + // + AliInfo("Converting global reference deltas to local ones"); + Int_t nprea = fConstrRef->GetEntriesFast(); + for (int ix=0; ixAt(ix); + if (!preo->ApplyToGeometry()) return -1; + } + // + // now convert the global reference deltas to local ones + for (int i=fConstrRef->GetEntriesFast();i--;) { + AliAlignObjParams *preo = (AliAlignObjParams*)fConstrRef->At(i); + TGeoHMatrix * mupd = AliGeomManager::GetMatrix(preo->GetSymName()); + if (!mupd) { // this is not alignable entry, need to look in the supermodules + for (int im=fNSuperModules;im--;) { + AliITSAlignMille2Module* mod = GetSuperModule(im); + if ( strcmp(mod->GetName(), preo->GetSymName()) ) continue; + mupd = mod->GetMatrix(); + break; + } + if (!mupd) { + AliError(Form("Failed to find the volume for reference %s",preo->GetSymName())); + return -1; + } + } + TGeoHMatrix preMat; + preo->GetMatrix(preMat); // Delta_Glob + TGeoHMatrix tmpMat = *mupd; // Delta_Glob * Delta_Glob_Par * M + preMat.MultiplyLeft( &tmpMat.Inverse() ); // M^-1 * Delta_Glob_Par^-1 = (Delta_Glob_Par * M)^-1 + tmpMat.MultiplyLeft( &preMat ); // (Delta_Glob_Par * M)^-1 * Delta_Glob * Delta_Glob_Par * M = Delta_loc + preo->SetMatrix(tmpMat); // local corrections + } + // + // we need to reload the geometry spoiled by this reference deltas... + delete fGeoManager; + AliInfo("Reloading target ideal geometry"); + return LoadGeometry(fGeometryPath); + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::Init() +{ + // perform global initialization + // + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized!"); + return; + } + // range constraints in such a way that the childs are constrained before their parents + // orphan constraints come last + for (int ic=0;icGetModuleID()GetModuleID()) { + // swap + fConstraints[ic] = cst1; + fConstraints[ic1] = cst0; + } + } + } + // + if (!GetUseGlobalDelta()) { + AliInfo("ATTENTION: The parameters are defined in the local frame, no check for degeneracy will be done"); + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + int npar = mod->GetNParTot(); + // the parameter may have max 1 free instance, otherwise the equations are underdefined + for (int ipar=0;iparIsFreeDOF(ipar)) continue; + mod->SetParOffset(ipar,fNGlobal++); + } + } + } + else { + // init millepede, decide which parameters are to be fitted explicitly + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + if (mod->IsNotInConf()) continue; // dummy module + int npar = mod->GetNParTot(); + // the parameter may have max 1 free instance, otherwise the equations are underdefined + for (int ipar=0;iparIsFreeDOF(ipar)) continue; // fixed + // + int nFreeInstances = 0; + // + AliITSAlignMille2Module* parent = mod; + Bool_t cstMeanMed=kFALSE,cstGauss=kFALSE; + // + Bool_t addToFit = kFALSE; + // the parameter may be ommitted from explicit fit (if PseudoParentsAllowed is true) if + // 1) it is not explicitly constrained or its does not participate in Gaussian constraint + // 2) the same applies to all of its parents + // 3) it has at least 1 unconstrained direct child + while(parent) { + if (parent->IsNotInConf()) {parent = parent->GetParent(); continue;} + if (!parent->IsFreeDOF(ipar)) {parent = parent->GetParent(); continue;} + nFreeInstances++; + if (IsParModConstrained(parent,ipar, cstMeanMed, cstGauss)) nFreeInstances--; + if (cstGauss) addToFit = kTRUE; + parent = parent->GetParent(); + } + if (nFreeInstances>1) { + AliError(Form("Parameter#%d of module %s\nhas %d free instances in the " + "unconstrained parents\nSystem is undefined",ipar,mod->GetName(),nFreeInstances)); + exit(1); + } + // + // i) Are PseudoParents allowed? + if (!PseudoParentsAllowed()) addToFit = kTRUE; + // ii) check if this module has no child with such a free parameter. Since the order of this check + // goes from child to parent, by this moment such a parameter must have been already added + else if (!IsParModFamilyVaried(mod,ipar)) addToFit = kTRUE; // no varied children at all + else if (!IsParFamilyFree(mod,ipar,1)) addToFit = kTRUE; // no unconstrained direct children + // otherwise the value of this parameter can be extracted from simple contraint and the values of + // the relevant parameters of its children the fit is done. Hence it is not included + if (!addToFit) continue; + // + // shall add this parameter to explicit fit + // printf("Adding %s %d -> %d\n",mod->GetName(), ipar, fNGlobal); + mod->SetParOffset(ipar,fNGlobal++); + } + } + } + // + AliInfo(Form("Initializing Millepede with %d gpar, %d lpar and %d stddev ...",fNGlobal, kNLocal, fNStdDev)); + fGlobalDerivatives = new Double_t[fNGlobal]; + memset(fGlobalDerivatives,0,fNGlobal*sizeof(Double_t)); + // + fMillepede->InitMille(fNGlobal,kNLocal,fNStdDev,fResCut,fResCutInitial); + fMillepede->SetMinPntValid(fMinPntPerSens); + fIsMilleInit = kTRUE; + // + ResetLocalEquation(); + AliInfo("Parameters initialized to zero"); + // + /// Fix non free parameters + for (Int_t i=0; iGetNParTot(); j++) { + if (mod->GetParOffset(j)<0) continue; // not varied + FixParameter(mod->GetParOffset(j),mod->GetParConstraint(j)); + fMillepede->SetParamGrID(i, mod->GetParOffset(j)); + } + } + // + ResetCovIScale(); + // Set iterations + if (fStartFac>1) fMillepede->SetIterations(fStartFac); + if (fFinalFac>1) fMillepede->SetChi2CutRef(fFinalFac); + fTrackBuff.Expand(24); + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::AddConstraint(Double_t *par, Double_t value, Double_t sigma) +{ + /// Constrain equation defined by par to value + if (!fIsMilleInit) Init(); + fMillepede->SetGlobalConstraint(par, value, sigma); + AliInfo("Adding constraint"); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::InitGlobalParameters(Double_t *par) +{ + /// Initialize global parameters with par array + if (!fIsMilleInit) Init(); + fMillepede->SetGlobalParameters(par); + AliInfo("Init Global Parameters"); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::FixParameter(Int_t iPar, Double_t value) +{ + /// Parameter iPar is encourage to vary in [-value;value]. + /// If value == 0, parameter is fixed + if (!fIsMilleInit) { + AliInfo("Millepede has not been initialized!"); + return; + } + fMillepede->SetParSigma(iPar, value); + if (IsZero(value)) AliInfo(Form("Parameter %i Fixed", iPar)); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ResetLocalEquation() +{ + /// Reset the derivative vectors + for(int i=kNLocal;i--;) fLocalDerivatives[i] = 0.0; + memset(fGlobalDerivatives, 0, fNGlobal*sizeof(double) ); +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::ApplyToGeometry() +{ + // apply prealignment to ideal geometry + Int_t nprea = fPrealignment->GetEntriesFast(); + AliInfo(Form("Array of prealignment deltas: %d entries",nprea)); + // + for (int ix=0; ixAt(ix); + Int_t index=AliITSAlignMille2Module::GetIndexFromVolumeID(preo->GetVolUID()); + if (index>=0) { + if (index>=fPreAlignQF.GetSize()) fPreAlignQF.Set(index+10); + fPreAlignQF[index] = (int) preo->GetUniqueID()+1; + } + if (!preo->ApplyToGeometry()) { + AliError(Form("Failed on ApplyToGeometry at %s",preo->GetSymName())); + return -1; + } + } + // + fUsePreAlignment = kTRUE; + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetPreAlignmentQualityFactor(Int_t index) const +{ + // quality factors from prealignment + if (!fUsePreAlignment || index<0 || index>=fPreAlignQF.GetSize()) return -1; + return fPreAlignQF[index]-1; +} + +//________________________________________________________________________________________________________ +AliTrackPointArray *AliITSAlignMille2::PrepareTrack(const AliTrackPointArray *atp) +{ + /// create a new AliTrackPointArray keeping only defined modules + /// move points according to a given prealignment, if any + /// sort alitrackpoints w.r.t. global Y direction, if cosmics + const Double_t kRad2L[6] = {5*5,10*10,18*18,30*30,40*40,60*60}; + const Float_t kSensSigY2[6] = {200e-4*200e-4/12, 200e-4*200e-4/12, + 300e-4*300e-4/12, 300e-4*300e-4/12, + 300e-4*300e-4/12, 300e-4*300e-4/12}; // thickness^2/12 + // + fTrack = NULL; + Int_t idx[20] = {0}; + Short_t lrID[20] = {0}; + Int_t npts=atp->GetNPoints(); + if (nptsGetVolumeID()[j]); + if (intidx[j]<0) continue; + ngoodpts++; + Float_t xx=atp->GetX()[j]; + Float_t yy=atp->GetY()[j]; + Float_t r=xx*xx + yy*yy; + int lay; + for (lay=0;lay<6;lay++) if (r5) continue; + lrID[j] = lay; + } + // + AliDebug(3,Form("Number of points in defined modules: %d out of %d",ngoodpts,npts)); + + // pepo270809 + Int_t nextra=0; + // extra clusters selection mode + if (fExtraClustersMode) { + // 1 = keep one cluster, remove randomly the extra + // 2 = keep one cluster, remove the internal one + // 10 = keep tracks only if at least one extra is present + + int iextra1[20],iextra2[20],layovl[20]; + // extra clusters mapping + for (Int_t ipt=0; iptGetX()[ipt]; + float p1y=atp->GetY()[ipt]; + float p1z=atp->GetZ()[ipt]; + int lay1=int(AliGeomManager::VolUIDToLayer(atp->GetVolumeID()[ipt])); + float r1 = p1x*p1x + p1y*p1y; + UShort_t volid1=atp->GetVolumeID()[ipt]; + + for (int ik=ipt+1; ikGetVolumeID()[ik])); + // check if same layer + if (lay2 != lay1) continue; + UShort_t volid2=atp->GetVolumeID()[ik]; + // check if different module + if (volid1 == volid2) continue; + + float p2x=atp->GetX()[ik]; + float p2y=atp->GetY()[ik]; + float p2z=atp->GetZ()[ik]; + float r2 = p2x*p2x + p2y*p2y; + float dr= (p1x-p2x)*(p1x-p2x) + (p1y-p2y)*(p1y-p2y) + (p1z-p2z)*(p1z-p2z); + + // looks for pairs with dr<1 cm, same layer but different module + if (dr<1.0) { + // extra1 is the one with smaller radius in rphi plane + if (r1Rndm()<0.5) + intidx[iextra1[ie]]=-1; + else + intidx[iextra2[ie]]=-1; + } + } + + // mode=2: keep only one clusters and remove the other... + if (fExtraClustersMode==2 && nextra) { + for (int ie=0; ie=0) ngoodpts++; + } + } + // endpepo270809 + + // reject track if not enough points are left + if (ngoodpts> RS + AliTrackPoint p; + // check points in specific places + if (fRequirePoints[fDataType]) { + Int_t nlayup[6],nlaydown[6],nlay[6]; + Int_t ndetup[3],ndetdown[3],ndet[3]; + for (Int_t j=0; j<6; j++) {nlayup[j]=0; nlaydown[j]=0; nlay[j]=0;} + for (Int_t j=0; j<3; j++) {ndetup[j]=0; ndetdown[j]=0; ndet[j]=0;} + + for (int i=0; iGetY()[i]; + int lay = lrID[i]; + int det=lay/2; + //printf("Point %d - x=%f y=%f R=%f lay=%d det=%d\n",i,xx,yy,r,lay,det); + + if (yy>=0.0) { // UP point + nlayup[lay]++; + nlay[lay]++; + ndetup[det]++; + ndet[det]++; + } + else { + nlaydown[lay]++; + nlay[lay]++; + ndetdown[det]++; + ndet[det]++; + } + } + // + // checks minimum values + Bool_t isok=kTRUE; + for (Int_t j=0; j<6; j++) { + if (nlayup[j]GetY(),idx); // sort descending... + // + Int_t npto=0; + if (fClusLoc.GetSize()<3*npts) fClusLoc.Set(3*npts); + if (fClusGlo.GetSize()<3*npts) fClusGlo.Set(3*npts); + if (fClusSigLoc.GetSize()<3*npts) fClusSigLoc.Set(3*npts); + // + for (int i=0; iGetPoint(p,idx[i]); + int sid = AliITSAlignMille2Module::GetIndexFromVolumeID(p.GetVolumeID()); + // + // prealign point if required + // get matrix used to produce the digits + AliITSAlignMille2Module *mod = GetMilleModule(intidx[idx[i]]); + TGeoHMatrix *svOrigMatrix = GetSensorOrigMatrixSID(sid); //mod->GetSensitiveVolumeOrigGlobalMatrix(p.GetVolumeID()); + // get back real local coordinate + fMeasLoc = fClusLoc.GetArray() + npto*3; + fMeasGlo = fClusGlo.GetArray() + npto*3; + fSigmaLoc = fClusSigLoc.GetArray() + npto*3; + fMeasGlo[0]=p.GetX(); + fMeasGlo[1]=p.GetY(); + fMeasGlo[2]=p.GetZ(); + AliDebug(3,Form("Global coordinates of measured point : X=%+f Y=%+f Z=%+f \n",fMeasGlo[0],fMeasGlo[1],fMeasGlo[2])); + svOrigMatrix->MasterToLocal(fMeasGlo,fMeasLoc); + AliDebug(3,Form("Local coordinates of measured point : X=%+f Y=%+f Z=%+f \n",fMeasLoc[0],fMeasLoc[1],fMeasLoc[2])); + // + if (p.GetDriftTime()>0) ProcessSDDPointInfo(&p,sid, npto); // for SDD points extract vdrift + // + // update covariance matrix + Double_t hcovel[9]; + hcovel[0]=double(p.GetCov()[0]); + hcovel[1]=double(p.GetCov()[1]); + hcovel[2]=double(p.GetCov()[2]); + hcovel[3]=double(p.GetCov()[1]); + hcovel[4]=double(p.GetCov()[3]); + hcovel[5]=double(p.GetCov()[4]); + hcovel[6]=double(p.GetCov()[2]); + hcovel[7]=double(p.GetCov()[4]); + hcovel[8]=double(p.GetCov()[5]); + hcov.SetRotation(hcovel); + // + if (AliLog::GetGlobalDebugLevel()>=2) { + AliInfo("Original Global Cov Matrix"); + printf("%+.4e %+.4e %+.4e\n%+.4e %+.4e\n%+.4e\n",hcovel[0],hcovel[1],hcovel[2],hcovel[4],hcovel[5],hcovel[8]); + } + // + // now rotate in local system + hcov.Multiply(svOrigMatrix); + hcov.MultiplyLeft(&svOrigMatrix->Inverse()); + // now hcov is LOCAL COVARIANCE MATRIX + // apply sigma scaling + Double_t *hcovscl = hcov.GetRotationMatrix(); + /* + const float *cv = p.GetCov(); + printf("## %d %d %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e %+.3e\n", + sid,p.GetClusterType(), + fMeasGlo[0],fMeasGlo[1],fMeasGlo[2], + fMeasLoc[0],fMeasLoc[1],fMeasLoc[2], + cv[0],cv[1],cv[2],cv[3],cv[4],cv[5], + hcovscl[0],hcovscl[4],hcovscl[8]); + + */ + if (AliLog::GetGlobalDebugLevel()>=2) { + AliInfo("Original Local Cov Matrix"); + printf("%+.4e %+.4e %+.4e\n%+.4e %+.4e\n%+.4e\n",hcovscl[0],hcovscl[1],hcovscl[2],hcovscl[4],hcovscl[5],hcovscl[8]); + } + hcovscl[4] = fUseLocalYErr ? kSensSigY2[lrID[idx[i]]] : 1E-8; // error due to the sensor thickness + // + for (int ir=3;ir--;) for (int ic=3;ic--;) { + if (ir==ic) { + if ( IsZero(hcovscl[ir*3+ic],1e-8) ) hcovscl[ir*3+ic] = 1E-8; + else hcovscl[ir*3+ic] *= mod->GetSigmaFactor(ir)*mod->GetSigmaFactor(ic); //RRR + fSigmaLoc[ir] = TMath::Sqrt(hcovscl[ir*3+ic]); + } + else hcovscl[ir*3+ic] = 0; + } + // + if (AliLog::GetGlobalDebugLevel()>=2) { + AliInfo("Modified Local Cov Matrix"); + printf("%+.4e %+.4e %+.4e\n%+.4e %+.4e\n%+.4e\n",hcovscl[0],hcovscl[1],hcovscl[2],hcovscl[4],hcovscl[5],hcovscl[8]); + } + // + if (fBug==1) { + // correzione bug LAYER 5 SSD temporanea.. + int ssdidx=AliITSAlignMille2Module::GetIndexFromVolumeID(p.GetVolumeID()); + if (ssdidx>=500 && ssdidx<1248) { + int ladder=(ssdidx-500)%22; + if (ladder==18) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx+1)); + if (ladder==19) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx-1)); + } + } + /// get (evenctually prealigned) matrix of sens. vol. + TGeoHMatrix *svMatrix = GetSensorCurrMatrixSID(sid); //mod->GetSensitiveVolumeMatrix(p.GetVolumeID()); + // modify global coordinates according with pre-aligment + svMatrix->LocalToMaster(fMeasLoc,fMeasGlo); + // now rotate in local system + hcov.Multiply(&svMatrix->Inverse()); + hcov.MultiplyLeft(svMatrix); // hcov is back in GLOBAL RF + // cure once more + for (int ir=3;ir--;) for (int ic=3;ic--;) if (IsZero(hcovscl[ir*3+ic])) hcovscl[ir*3+ic] = 0.; + // printf("\nErrMatGlob: after\n"); hcov.Print(""); //RRR + // + if (AliLog::GetGlobalDebugLevel()>=2) { + AliInfo("Modified Global Cov Matrix"); + printf("%+.4e %+.4e %+.4e\n%+.4e %+.4e\n%+.4e\n",hcovscl[0],hcovscl[1],hcovscl[2],hcovscl[4],hcovscl[5],hcovscl[8]); + } + // + Float_t pcov[6]; + pcov[0]=hcovscl[0]; + pcov[1]=hcovscl[1]; + pcov[2]=hcovscl[2]; + pcov[3]=hcovscl[4]; + pcov[4]=hcovscl[5]; + pcov[5]=hcovscl[8]; + // + // make sure the matrix is positive definite + { + enum {kXX=0,kXY=1,kXZ=2,kYX=kXY,kYY=3,kYZ=4,kZX=kXZ,kZY=kYZ,kZZ=5}; + if (pcov[kXX]*pcov[kYY]*0.999AddPoint(npto,&p); + AliDebug(2,Form("Adding point[%d] = ( %+f , %+f , %+f ) volid = %d",npto,fTrack->GetX()[npto], + fTrack->GetY()[npto],fTrack->GetZ()[npto],fTrack->GetVolumeID()[npto] )); + // printf("Adding %d %d %f\n",npto, p.GetVolumeID(), p.GetY()); + npto++; + } + // + fDiamondPointID = -1; + if (addVertex) { + fTrack->AddPoint(npto,&fDiamond); + fMeasLoc = fClusLoc.GetArray() + npto*3; + fMeasGlo = fClusGlo.GetArray() + npto*3; + fSigmaLoc = fClusSigLoc.GetArray() + npto*3; + fMeasLoc[0] = fMeasGlo[0] = fDiamond.GetX(); + fMeasLoc[1] = fMeasGlo[1] = fDiamond.GetY(); + fMeasLoc[2] = fMeasGlo[2] = fDiamond.GetZ(); + fSigmaLoc[0] = TMath::Sqrt(fDiamond.GetCov()[0]); + fSigmaLoc[1] = TMath::Sqrt(fDiamond.GetCov()[3]); + fSigmaLoc[2] = TMath::Sqrt(fDiamond.GetCov()[5]); + fDiamondPointID = npto++; + } + // + return fTrack; +} + +//________________________________________________________________________________________________________ +AliTrackPointArray *AliITSAlignMille2::SortTrack(const AliTrackPointArray *atp) +{ + /// sort alitrackpoints w.r.t. global Y direction + AliTrackPointArray *atps=NULL; + Int_t idx[20]; + Int_t npts=atp->GetNPoints(); + AliTrackPoint p; + atps=new AliTrackPointArray(npts); + + TMath::Sort(npts,atp->GetY(),idx); + + for (int i=0; iGetPoint(p,idx[i]); + atps->AddPoint(i,&p); + AliDebug(2,Form("Point[%d] = ( %+f , %+f , %+f ) volid = %d",i,atps->GetX()[i],atps->GetY()[i],atps->GetZ()[i],atps->GetVolumeID()[i] )); + } + return atps; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetCurrentLayer() const +{ + // get current layer id + if (!fGeoManager) { + AliInfo("ITS geometry not initialized!"); + return -1; + } + return (Int_t)AliGeomManager::VolUIDToLayer(fCluster.GetVolumeID()); +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::InitModuleParams() +{ + /// initialize geometry parameters for a given detector + /// for current cluster (fCluster) + /// fGlobalInitParam[] is set as: + /// [tx,ty,tz,psi,theta,phi] + /// (old was [tx,ty,tz,theta,psi,phi] ROOT's angles...) + /// *** At the moment: using Raffalele's angles definition *** + /// + /// return 0 if success + /// If module is found but has no parameters to vary, return 1 + + if (!fGeoManager) { + AliInfo("ITS geometry not initialized!"); + return -1; + } + + // now 'voluid' is the volumeID of a SENSITIVE VOLUME (coming from a cluster) + + // set the internal index (index in module list) + UShort_t voluid=fCluster.GetVolumeID(); + fCurrentSensID = AliITSAlignMille2Module::GetIndexFromVolumeID(voluid); + // + if (fCurrentSensID==-1) { // this is a special "vertex" module + fCurrentModule = GetMilleModuleByVID(voluid); + fCurrentSensID = fCurrentModule->GetIndex(); + + } + else { + // + // IT IS VERY IMPORTANT: start from the end of the list, where the childs are located !!! + Int_t k=fNModules-1; + fCurrentModule = 0; + // VERY IMPORTANT: if the sensors were explicitly provided, don't look in the supermodules + while (k>=0 && ! (fCurrentModule=GetMilleModule(k))->IsIn(voluid)) k--; + if (k<0) return -3; + } + // + for (int i=AliITSAlignMille2Module::kMaxParTot;i--;) fModuleInitParam[i] = 0.0; + // + int clID = fCluster.GetUniqueID()-1; + if (clID<0) { // external cluster + fMeasGlo = &fExtClusterPar[0]; + fMeasLoc = &fExtClusterPar[3]; + fSigmaLoc = &fExtClusterPar[6]; + fExtClusterPar[0] = fCluster.GetX(); + fExtClusterPar[1] = fCluster.GetY(); + fExtClusterPar[2] = fCluster.GetZ(); + // + TGeoHMatrix *svMatrix = fCurrentModule->GetSensitiveVolumeMatrix(voluid); + svMatrix->MasterToLocal(fMeasGlo,fMeasLoc); + TGeoHMatrix hcov; + Double_t hcovel[9]; + hcovel[0]=double(fCluster.GetCov()[0]); + hcovel[1]=double(fCluster.GetCov()[1]); + hcovel[2]=double(fCluster.GetCov()[2]); + hcovel[3]=double(fCluster.GetCov()[1]); + hcovel[4]=double(fCluster.GetCov()[3]); + hcovel[5]=double(fCluster.GetCov()[4]); + hcovel[6]=double(fCluster.GetCov()[2]); + hcovel[7]=double(fCluster.GetCov()[4]); + hcovel[8]=double(fCluster.GetCov()[5]); + hcov.SetRotation(hcovel); + // now rotate in local system + hcov.Multiply(svMatrix); + hcov.MultiplyLeft(&svMatrix->Inverse()); + if (fSigmaLoc[0]<0.0010) fSigmaLoc[0]=0.0010; + if (fSigmaLoc[2]<0.0010) fSigmaLoc[2]=0.0010; + // + } + else { + int offs = 3*clID; + fMeasGlo = fClusGlo.GetArray() + offs; + fMeasLoc = fClusLoc.GetArray() + offs; + fSigmaLoc = fClusSigLoc.GetArray() + offs; + } + // + // set minimum value for SigmaLoc to 10 micron + if (fSigmaLoc[0]<0.0010) fSigmaLoc[0]=0.0010; + if (fSigmaLoc[2]<0.0010) fSigmaLoc[2]=0.0010; + if (fCurrentSensID==kVtxSensID || fUseLocalYErr) if (fSigmaLoc[1]<0.0010) fSigmaLoc[1]=0.0010; + // + AliDebug(2,Form("Local coordinates of measured point : X=%+f Y=%+f Z=%+f \n",fMeasLoc[0] ,fMeasLoc[1] ,fMeasLoc[2] )); + AliDebug(2,Form("Setting StDev from CovMat : fSigmaLocX=%g fSigmaLocY=%g fSigmaLocZ=%g \n",fSigmaLoc[0] ,fSigmaLoc[1] ,fSigmaLoc[2] )); + // + return 0; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::Print(Option_t*) const +{ + // print current status + printf("*** AliMillepede for ITS ***\n"); + printf(" Number of defined super modules: %d\n",fNModules); + printf(" Obtained parameters refer to %s Deltas\n",fUseGlobalDelta ? "GLOBAL":"LOCAL"); + // + if (fGeoManager) + printf(" geometry loaded from %s\n",fGeometryPath.Data()); + else + printf(" geometry not loaded\n"); + // + if (fUsePreAlignment) + printf(" using prealignment from %s \n",fPreDeltaPath.Data()); + else + printf(" prealignment not used\n"); + // + // + if (fBOn) + printf(" B Field set to %+f T - using helices\n",fBField); + else + printf(" B Field OFF - using straight lines \n"); + // + if (fTPAFitter) + printf(" Using AliITSTPArrayFit class for track fitting\n"); + else + printf(" Using StraightLine/Riemann fitter for track fitting\n"); + // + printf("Using local Y error due to the sensor thickness: %s\n",(fUseLocalYErr && fTPAFitter) ? "ON":"OFF"); + // + for (int itp=0;itp0) printf(" Layer %d : %d points with Y>0\n",i+1,fNReqLayUp[itp][i]); + if (fNReqLayDown[itp][i]>0) printf(" Layer %d : %d points with Y<0\n",i+1,fNReqLayDown[itp][i]); + if (fNReqLay[itp][i]>0) printf(" Layer %d : %d points \n",i+1,fNReqLay[itp][i]); + } + for (Int_t i=0; i<3; i++) { + if (fNReqDetUp[itp][i]>0) printf(" Detector %d : %d points with Y>0\n",i+1,fNReqDetUp[itp][i]); + if (fNReqDetDown[itp][i]>0) printf(" Detector %d : %d points with Y<0\n",i+1,fNReqDetDown[itp][i]); + if (fNReqDet[itp][i]>0) printf(" Detector %d : %d points \n",i+1,fNReqDet[itp][i]); + } + } + printf(" SDD VDrift correction : %s",fIsSDDVDriftMult ? "Mult":"Add"); + printf(" Weight acc. to pT in power : %f",fWeightPt); + // + printf("\n Millepede configuration parameters:\n"); + printf(" init factor for chi2 cut : %.4f\n",fStartFac); + printf(" final factor for chi2 cut : %.4f\n",fFinalFac); + printf(" first iteration cut value : %.4f\n",fResCutInitial); + printf(" other iterations cut value : %.4f\n",fResCut); + printf(" number of stddev for chi2 cut : %d\n",fNStdDev); + printf(" def.scaling for local sigmas : %.4f %.4f %.4f\n",fSigmaFactor[0],fSigmaFactor[1],fSigmaFactor[2]); + printf(" min.tracks per module : %d\n",fMinPntPerSens); + // + printf("List of defined modules:\n"); + printf(" intidx\tindex\tvoluid\tname\n"); + for (int i=0; iGetIndex(),md->GetVolumeID(),md->GetName()); + } +} + +//________________________________________________________________________________________________________ +AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleByVID(UShort_t voluid) const +{ + // return pointer to a defined supermodule + // return NULL if error + Int_t i=IsVIDDefined(voluid); + if (i<0) return NULL; + return GetMilleModule(i); +} + +//________________________________________________________________________________________________________ +AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleBySymName(const Char_t* symname) const +{ + // return pointer to a defined supermodule + // return NULL if error + UShort_t vid = AliITSAlignMille2Module::GetVolumeIDFromSymname(symname); + if (vid>0) return GetMilleModuleByVID(vid); + else { // this is not alignable module, need to look within defined supermodules + int i = IsSymDefined(symname); + if (i>=0) return GetMilleModule(i); + } + return 0; +} + +//________________________________________________________________________________________________________ +AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleIfContained(const Char_t* symname) const +{ + // return pointer to a defined/contained supermodule + // return NULL otherwise + int i = IsSymContained(symname); + return i<0 ? 0 : GetMilleModule(i); +} + +//________________________________________________________________________________________________________ +AliAlignObjParams* AliITSAlignMille2::GetPrealignedObject(const Char_t* symname) const +{ + // get delta from prealignment for given volume + if (!fPrealignment) return 0; + for (int ipre=fPrealignment->GetEntriesFast();ipre--;) { // was the corresponding object prealigned? + AliAlignObjParams* preob = (AliAlignObjParams*)fPrealignment->At(ipre); + if (!strcmp(preob->GetSymName(),symname)) return preob; + } + return 0; +} + +//________________________________________________________________________________________________________ +AliAlignObjParams* AliITSAlignMille2::GetConstrRefObject(const Char_t* symname) const +{ + // get delta with respect to which the constraint is declared + if (!fConstrRef) return 0; + for (int ipre=fConstrRef->GetEntriesFast();ipre--;) { // was the corresponding object prealigned? + AliAlignObjParams* preob = (AliAlignObjParams*)fConstrRef->At(ipre); + if (!strcmp(preob->GetSymName(),symname)) return preob; + } + return 0; +} + +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::InitRiemanFit() +{ + // Initialize Riemann Fitter for current track + // return kFALSE if error + + if (!fBOn) return kFALSE; + + Int_t npts=0; + AliTrackPoint ap; + npts = fTrack->GetNPoints(); + AliDebug(3,Form("Fitting track with %d points",npts)); + if (!fRieman) fRieman = new AliTrackFitterRieman(); + fRieman->Reset(); + fRieman->SetTrackPointArray(fTrack); + + TArrayI ai(npts); + for (Int_t ipt=0; iptGetVolumeID()[ipt]; + + // fit track with 5 params in his own tracking-rotated reference system + // xc = -p[1]/p[0]; + // yc = 1/p[0]; + // R = sqrt( x0*x0 + y0*y0 - y0*p[2]); + if (!fRieman->Fit(&ai,NULL,(AliGeomManager::ELayerID)1,(AliGeomManager::ELayerID)6)) { + return kFALSE; + } + + for (int i=0; i<5; i++) + fLocalInitParam[i] = fRieman->GetParam()[i]; + + return kTRUE; +} + +//________________________________________________________________________________________________________ +void trackFit2D(Int_t &, Double_t *, double &chi2, double *par, int flag) +{ + // local function for minuit + const double kTiny = 1.e-14; + chi2 = 0; + static AliTrackPoint pnt; + static Bool_t fullErr2D; + // + if (flag==1) fullErr2D = kFALSE;//kTRUE; + // fullErr2D = kTRUE; + enum {kAX,kAZ,kBX,kBZ}; + enum {kXX=0,kXY=1,kXZ=2,kYX=kXY,kYY=3,kYZ=4,kZX=kXZ,kZY=kYZ,kZZ=5}; + // + AliITSAlignMille2* alig = AliITSAlignMille2::GetInstance(); + AliTrackPointArray* track = alig->GetCurrentTrack(); + // + int npts = track->GetNPoints(); + for (int ip=0;ipGetPoint(pnt,ip); + const float *cov = pnt.GetCov(); + double y = pnt.GetY(); + double dx = pnt.GetX() - (par[kAX]+y*par[kBX]); + double dz = pnt.GetZ() - (par[kAZ]+y*par[kBZ]); + double xxe = cov[kXX]; + double zze = cov[kZZ]; + double xze = cov[kXZ]; + // + if (fullErr2D) { + xxe += par[kBX]*par[kBX]*cov[kYY]-2.*par[kBX]*cov[kXY]; + zze += par[kBZ]*par[kBZ]*cov[kYY]-2.*par[kBZ]*cov[kZY]; + xze += par[kBX]*par[kBZ]*cov[kYY]-cov[kYZ]*par[kBZ]-cov[kXY]*par[kBX]; + } + // + double det = xxe*zze - xze*xze; + if (det %+e\n",ip,dx,dz,xxeI,zzeI,xzeI, chi2); + } + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::InitTrackParams(int meth) +{ + /// initialize local parameters with different methods + /// for current track (fTrack) + Int_t npts=0; + AliTrackPoint ap; + double sX=0,sXY=0,sZ=0,sZY=0,sY=0,sYY=0,det=0; + // simple linear interpolation + // get local starting parameters (to be substituted by ESD track parms) + // local parms (fLocalInitParam[]) are: + // [0] = global x coord. of straight line intersection at y=0 plane + // [1] = global z coord. of straight line intersection at y=0 plane + // [2] = px/py + // [3] = pz/py + // test #1: linear fit in x(y) and z(y) + npts = fTrack->GetNPoints(); + AliDebug(3,Form("*** initializing track with %d points ***",npts)); + for (int i=npts;i--;) { + sY += fTrack->GetY()[i]; + sYY += fTrack->GetY()[i]*fTrack->GetY()[i]; + sX += fTrack->GetX()[i]; + sXY += fTrack->GetX()[i]*fTrack->GetY()[i]; + sZ += fTrack->GetZ()[i]; + sZY += fTrack->GetZ()[i]*fTrack->GetY()[i]; + } + det = sYY*npts-sY*sY; + if (IsZero(det)) det = 1E-16; + fLocalInitParam[0] = (sX*sYY-sY*sXY)/det; + fLocalInitParam[2] = (sXY*npts-sY*sX)/det; + // + fLocalInitParam[1] = (sZ*sYY-sY*sZY)/det; + fLocalInitParam[3] = (sZY*npts-sY*sZ)/det; + // pepo200709 + fLocalInitParam[4] = 0.0; + // endpepo200709 + + AliDebug(2,Form("X = p0gx + ugx*Y : p0gx = %+f ugx = %+f\n",fLocalInitParam[0],fLocalInitParam[2])); + // + if (meth==1) return; + // + // perform full fit accounting for cov.matrix + static TVirtualFitter *minuit = 0; + static Double_t step[5] = {1E-3,1E-3,1E-4,1E-4,1E-5}; + static Double_t arglist[10]; + // + if (!minuit) { + minuit = TVirtualFitter::Fitter(0,4); + minuit->SetFCN(trackFit2D); + arglist[0] = 1; + minuit->ExecuteCommand("SET ERR",arglist, 1); + // + arglist[0] = -1; + minuit->ExecuteCommand("SET PRINT",arglist,1); + // + } + // + minuit->SetParameter(0, "ax", fLocalInitParam[0], step[0], 0,0); + minuit->SetParameter(1, "az", fLocalInitParam[1], step[1], 0,0); + minuit->SetParameter(2, "bx", fLocalInitParam[2], step[2], 0,0); + minuit->SetParameter(3, "bz", fLocalInitParam[3], step[3], 0,0); + // + arglist[0] = 1000; // number of function calls + arglist[1] = 0.001; // tolerance + minuit->ExecuteCommand("MIGRAD",arglist,2); + // + for (int i=0;i<4;i++) fLocalInitParam[i] = minuit->GetParameter(i); + for (int i=0;i<4;i++) for (int j=0;j<4;j++) fLocalInitParEr[i][j] = minuit->GetCovarianceMatrixElement(i,j); + /* + double amin,edm,errdef; + int nvpar,nparx; + minuit->GetStats(amin,edm,errdef,nvpar,nparx); + amin /= (2*npts - 4); + printf("Mchi2: %+e\n",amin); + */ + // +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::IsSymDefined(const Char_t* symname) const +{ + // checks if supermodule with this symname is defined and return the internal index + // return -1 if not. + for (int k=fNModules;k--;) if (!strcmp(symname,GetMilleModule(k)->GetName())) return k; + return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::IsSymContained(const Char_t* symname) const +{ + // checks if module with this symname is defined and return the internal index + // return -1 if not. + UShort_t vid = AliITSAlignMille2Module::GetVolumeIDFromSymname(symname); + if (vid>0) return IsVIDContained(vid); + // only sensors have real vid, but maybe we have a supermodule with fake vid? + // IMPORTANT: always start from the end to start from the sensors + return IsSymDefined(symname); +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::IsVIDDefined(UShort_t voluid) const +{ + // checks if supermodule 'voluid' is defined and return the internal index + // return -1 if not. + for (int k=fNModules;k--;) if (voluid==GetMilleModule(k)->GetVolumeID()) return k; + return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::IsVIDContained(UShort_t voluid) const +{ + // checks if the sensitive module 'voluid' is contained inside a supermodule + // and return the internal index of the last identified supermodule + // return -1 if error + // IMPORTANT: always start from the end to start from the sensors + if (AliITSAlignMille2Module::GetIndexFromVolumeID(voluid)<0) return -1; + for (int k=fNModules;k--;) if (GetMilleModule(k)->IsIn(voluid)) return k; + return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetRequestedModID(UShort_t voluid) const +{ + // checks if the sensitive module 'voluid' is contained inside a supermodule + // and return the internal index of the last identified supermodule + // return -1 if error + // IMPORTANT: always start from the end to start from the sensors + if (AliITSAlignMille2Module::GetIndexFromVolumeID(voluid)<0) return -1; + int k; + for (k=fNModules;k--;) if (GetMilleModule(k)->IsIn(voluid)) break; + if (k<0) return -1; + AliITSAlignMille2Module* md = GetMilleModule(k); + while (md && md->IsNotInConf()) md = md->GetParent(); + if (md) return int(md->GetUniqueID()); + else return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CheckCurrentTrack() +{ + /// checks if AliTrackPoints belongs to defined modules + /// return number of good poins + /// return 0 if not enough points + + Int_t npts = fTrack->GetNPoints(); + Int_t ngoodpts=0; + // debug points + for (int j=0; jGetVolumeID()[j])>=0) ngoodpts++; + // + if (ngoodptsGetNPoints(); + AliDebug(2,Form("*** Input track with %d points ***",npts)); + + // preprocessing of the input track: keep only points in defined volumes, + // move points if prealignment is set, sort by Yglo if required + fTrackWeight = wgh; + fTrack=PrepareTrack(track); + if (!fTrack) { + RemoveHelixFitConstraint(); + RemoveVertexConstraint(); + return -1; + } + npts = fTrack->GetNPoints(); + if (npts>kMaxPoints) { + AliError(Form("Compiled with kMaxPoints=%d, current track has %d points",kMaxPoints,npts)); + } + AliDebug(2,Form("*** Processing prepared track with %d points ***",npts)); + // + npts = FitTrack(); + if (npts<0) return npts; + // + // printf("Params: "); for (int i=0;iGetPoint(fCluster,ipt); + fCluster.SetUniqueID(ipt+1); + AliDebug(2,Form("\n--- processing point %d --- \n",ipt)); + + // set geometry parameters for the the current module + if (InitModuleParams()) continue; + AliDebug(2,Form(" VolID=%d Index=%d InternalIdx=%d symname=%s\n", + track->GetVolumeID()[ipt], fCurrentModule->GetIndex(), + fCurrentModule->GetUniqueID(), AliGeomManager::SymName(track->GetVolumeID()[ipt]) )); + AliDebug(2,Form(" Preprocessed Point = ( %+f , %+f , %+f ) \n",fCluster.GetX(),fCluster.GetY(),fCluster.GetZ())); + int res = fTPAFitter ? AddLocalEquationTPA(md[nloceq]) : AddLocalEquation(md[nloceq]); + if (res<0) {fTotBadLocEqPoints++; nloceq = 0; break;} + else if (res==0) nloceq++; + else {nloceq++; ngloeq++;} + } // end loop over points + // + fTrack=NULL; + // not enough good points? + if (nloceqSaveRecordData(); // RRR + fCurvFitWasConstrained = kFALSE; // restore default + // + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::FitTrack() +{ + // Fit the track with selected constraints + // + const Double_t kfDiamondTolerance = 0.1; //diamond tolerance on top of the MS error + if (!fTrack) return -1; + int npts = fTrack->GetNPoints(); + // + if (fTPAFitter) { // use dediacted fitter + // + // if the diamond point is attached, for the moment don't include it in the fit + fTPAFitter->AttachPoints(fTrack,0, npts-1); + fTPAFitter->SetBz(fBField); + fTPAFitter->SetTypeCosmics(IsTypeCosmics()); + if (fIniTrackParamsMeth==1) fTPAFitter->SetIgnoreCov(); + // + double chi2; + double chi2f = 0; + double dca2err; + double dca2 = 0.; + Bool_t fitIsDone = kFALSE; + if (fUseDiamond && fDiamondPointID>0 && fCheckDiamondPoint==kDiamondCheckIfPrompt) { // diamond constraint was added, check if the track looks like prompt + fTPAFitter->SetFirstLast(0,fDiamondPointID-1); + if (IsCovIScaleTouched()) for (int i=npts;i--;) fTPAFitter->SetCovIScale(i,GetCovIScale(i)); + // + chi2f = fTPAFitter->Fit(fConstrCharge,fConstrPT,fConstrPTErr); + if ( chi2f<0 || (chi2f>fNStdDev*fStartFac && fTPAFitter->GetNIterations()>=fTPAFitter->GetMaxIterations()) ) { //RRR + AliInfo(Form("Track fit failed on checking if it is prompt! skipping this track... Chi2:%+e",chi2f)); + fTPAFitter->Reset(); + // fTrack = NULL; + return -5; + } + double xyzRes[3]; + fTPAFitter->GetResiduals(xyzRes,&fDiamondI,kTRUE); + dca2 = xyzRes[0]*xyzRes[0] + xyzRes[1]*xyzRes[1]; + double pT = IsFieldON() ? fTPAFitter->GetPt() : 0.45; + if (pT<0.1) pT = 0.1; + dca2err = kfDiamondTolerance + 0.02/pT; + if (dca2>dca2err*dca2err) { // this is secondary + int* clst = (int*) fTrack->GetClusterType(); + clst[fDiamondPointID] = -1;; + fDiamondPointID = -1; + fitIsDone = kTRUE; + npts--; + } + else fTPAFitter->SetFirstLast(0,fDiamondPointID); // fit with diamond + } + // fTPAFitter->SetParAxis(1); + if (!fitIsDone) { + if (IsCovIScaleTouched()) for (int i=npts;i--;) fTPAFitter->SetCovIScale(i,GetCovIScale(i)); + chi2 = fTPAFitter->Fit(fConstrCharge,fConstrPT,fConstrPTErr); + } + // + RemoveHelixFitConstraint(); // suppress eventual constraints to not affect fit of the next track + RemoveVertexConstraint(); // same ... + // + if ( !fitIsDone && (chi2<0 || (chi2>fNStdDev*fStartFac && fTPAFitter->GetNIterations()>=fTPAFitter->GetMaxIterations())) ) { //RRR + AliInfo(Form("Track fit failed! skipping this track... Chi2:%+e",chi2)); + if (fUseDiamond && fDiamondPointID>0 && fCheckDiamondPoint==kDiamondCheckIfPrompt) AliInfo(Form("VertexFree fit gave Chi2:%+e with residual %+e",chi2f,TMath::Sqrt(dca2))); + /* + fTrack->Print(""); + fTPAFitter->FitHelixCrude(); + fTPAFitter->SetFitDone(); + fTPAFitter->Print(); + */ + fTPAFitter->Reset(); + // fTrack = NULL; + return -5; + } + fNLocal = fTPAFitter->IsFieldON() ? 5:4; // Attention: the fitter might have decided to work in line mode + npts = fTPAFitter->GetLast() - fTPAFitter->GetFirst() + 1; // actual number of points + /* + double *pr = fTPAFitter->GetParams(); + printf("FtPar: %+.5e %+.5e %+.5e %+.5e | chi2:%.3e\n",pr[2],pr[0],pr[3],pr[1],chi2); // RRR + */ + } + else { + // + if (!fBOn) { // straight lines + // set local starting parameters (to be substituted by ESD track parms) + // local parms (fLocalInitParam[]) are: + // [0] = global x coord. of straight line intersection at y=0 plane + // [1] = global z coord. of straight line intersection at y=0 plane + // [2] = px/py + // [3] = pz/py + InitTrackParams(fIniTrackParamsMeth); + /* + double *pr = fLocalInitParam; + printf("FtPar: %+.5e %+.5e %+.5e %+.5e |\n",pr[0],pr[1],pr[2],pr[3]); // RRR + */ + } + else { + // local parms (fLocalInitParam[]) are the Riemann Fitter params + if (!InitRiemanFit()) { + AliInfo("Riemann fit failed! skipping this track..."); + fTrack=NULL; + return -5; + } + } + } + return npts; + // +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CalcIntersectionPoint(const Double_t *lpar, const Double_t *gpar) +{ + /// calculate track intersection point in local coordinates + /// according with a given set of parameters (local(4) and global(6)) + /// and fill fPintLoc/Glo + /// local are: pgx0, pgz0, ugx, ugz OR riemann fitters pars + /// global are: tx,ty,tz,psi,theta,phi (Raff's delta angles in deg.) + /// return 0 if success + + AliDebug(3,Form("lpar = %g %g %g %g %g\ngpar= %g %g %g %g %g %g\n",lpar[0],lpar[1],lpar[2],lpar[3],lpar[4],gpar[0],gpar[1],gpar[2],gpar[3],gpar[4],gpar[5])); + AliDebug(3,Form("deltalpar = %g %g %g %g %g\n",lpar[0]-fLocalInitParam[0],lpar[1]-fLocalInitParam[1],lpar[2]-fLocalInitParam[2],lpar[3]-fLocalInitParam[3],lpar[4]-fLocalInitParam[4])); + + + // prepare the TGeoHMatrix + TGeoHMatrix *tempHMat = fCurrentModule->GetSensitiveVolumeModifiedMatrix(fCluster.GetVolumeID(),gpar, + !fUseGlobalDelta); + if (!tempHMat) return -1; + + Double_t v0g[3]; // vector with straight line direction in global coord. + Double_t p0g[3]; // point of the straight line (glo) + + if (fBOn) { // B FIELD! + AliTrackPoint prf; + for (int ip=0; ip<5; ip++) + fRieman->SetParam(ip,lpar[ip]); + + if (!fRieman->GetPCA(fCluster,prf)) { + AliInfo(Form("error in GetPCA for point %d",fCluster.GetVolumeID())); + return -3; + } + // now determine straight line passing tangent to fit curve at prf + // ugx = dX/dY_glo = DeltaX/DeltaY_glo + // mo' P1=(X,Y,Z)_glo_prf + // => (x,y,Z)_trk_prf ruotando di alpha... + Double_t alpha=fRieman->GetAlpha(); + Double_t x1g = prf.GetX(); + Double_t y1g = prf.GetY(); + Double_t z1g = prf.GetZ(); + Double_t x1t = x1g*TMath::Cos(alpha) + y1g*TMath::Sin(alpha); + Double_t y1t = -x1g*TMath::Sin(alpha) + y1g*TMath::Cos(alpha); + Double_t z1t = z1g; + + Double_t x2t = x1t+1.0; + Double_t y2t = y1t+fRieman->GetDYat(x1t); + Double_t z2t = z1t+fRieman->GetDZat(x1t); + Double_t x2g = x2t*TMath::Cos(alpha) - y2t*TMath::Sin(alpha); + Double_t y2g = x2t*TMath::Sin(alpha) + y2t*TMath::Cos(alpha); + Double_t z2g = z2t; + + AliDebug(3,Form("Riemann frame: fAlpha = %+f = %+f ",alpha,alpha*180./TMath::Pi())); + AliDebug(3,Form(" prf_glo=( %+f , %+f , %+f ) prf_rf=( %+f , %+f , %+f )\n", x1g,y1g,z1g, x1t,y1t,z1t)); + AliDebug(3,Form(" mov_glo=( %+f , %+f , %+f ) rf=( %+f , %+f , %+f )\n",x2g,y2g,z2g, x2t,y2t,z2t)); + + if (TMath::Abs(y2g-y1g)<1e-15) { + AliInfo("DeltaY=0! Cannot proceed..."); + return -1; + } + // ugx,1,ugz + v0g[0] = (x2g-x1g)/(y2g-y1g); + v0g[1]=1.0; + v0g[2] = (z2g-z1g)/(y2g-y1g); + + // point: just keep prf + p0g[0]=x1g; + p0g[1]=y1g; + p0g[2]=z1g; + } + else { // staight line + // vector of initial straight line direction in glob. coord + v0g[0]=lpar[2]; + v0g[1]=1.0; + v0g[2]=lpar[3]; + + // intercept in yg=0 plane in glob coord + p0g[0]=lpar[0]; + p0g[1]=0.0; + p0g[2]=lpar[1]; + } + AliDebug(3,Form("Line vector: ( %+f , %+f , %+f ) point:( %+f , %+f , %+f )\n",v0g[0],v0g[1],v0g[2],p0g[0],p0g[1],p0g[2])); + + // same in local coord. + Double_t p0l[3],v0l[3]; + tempHMat->MasterToLocalVect(v0g,v0l); + tempHMat->MasterToLocal(p0g,p0l); + + if (TMath::Abs(v0l[1])<1e-15) { + AliInfo("Track Y direction in local frame is zero! Cannot proceed..."); + return -1; + } + + // local intersection point + fPintLoc[0] = p0l[0] - (v0l[0]/v0l[1])*p0l[1]; + fPintLoc[1] = 0; + fPintLoc[2] = p0l[2] - (v0l[2]/v0l[1])*p0l[1]; + + // global intersection point + tempHMat->LocalToMaster(fPintLoc,fPintGlo); + AliDebug(3,Form("Intesect. point: L( %+f , %+f , %+f ) G( %+f , %+f , %+f )\n",fPintLoc[0],fPintLoc[1],fPintLoc[2],fPintGlo[0],fPintGlo[1],fPintGlo[2])); + + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CalcDerivatives(Int_t paridx, Bool_t islpar) +{ + /// calculate numerically (ROOT's style) the derivatives for + /// local X intersection and local Z intersection + /// parlist: local (islpar=kTRUE) pgx0, pgz0, ugx0, ugz0 OR riemann's params + /// global (islpar=kFALSE) tx, ty, tz, psi, theta, phi (Raf's angles in deg) + /// return 0 if success + + // copy initial parameters + Double_t lpar[kNLocal]; + Double_t gpar[kNParCh]; + Double_t *derivative; + for (Int_t i=0; iGetNParFree()==0) continue; + nmodTested++; + for (Int_t i=0; iGetParOffset(i)<0) continue; // this parameter is not explicitly fitted + if (!dfDone[i]) { + if (CalcDerivatives(i,kFALSE)) return -1; + else { + dfDone[i] = kTRUE; + if (zeroX) zeroX = IsZero(fDerivativeGlo[i][0]); + if (zeroZ) zeroZ = IsZero(fDerivativeGlo[i][2]); + } + } + // + m.fDerGlo[ifill][kX] = fDerivativeGlo[i][0]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[i][2]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(i); + } + // + // specific for special sensors + Int_t sddLR = -1; + if ( fCurrentModule->IsSDD() && + (fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0)>=0 || + // fCurrentModule->GetParOffset(sddLR = fMeasLoc[kX]>0 ? + fCurrentModule->GetParOffset(sddLR = GetVDriftSDD()>0 ? + AliITSAlignMille2Module::kDOFDVL : AliITSAlignMille2Module::kDOFDVR)>=0) + ) { + // + // assume for sensor local xloc = xloc0 + V0*dT0+dV*(T-T0) + // where V0 and T are the nominal drift velocity, time and time0 + // and the dT0 and dV are the corrections: + // dX/dT0 = dX/dxloc * dxloc/dT0 = dX/dxloc * V0 + // dX/dV = dX/dxloc * dxloc/dV = dX/dxloc * (T-T0) + // IMPORTANT: the geom derivatives are over the SENSOR LOCAL parameters + // + if (!dfDone[AliITSAlignMille2Module::kDOFT0] || !dfDone[sddLR]) { + // + double dXdxlocsens=0., dZdxlocsens=0.; + // + // if the current module is the sensor itself and we work with local params, then + // we can directly take dX/dxloc_sens dZ/dxloc_sens + if (!fUseGlobalDelta && fCurrentModule->GetVolumeID()==fCluster.GetVolumeID()) { + if (!dfDone[AliITSAlignMille2Module::kDOFTX]) { + CalcDerivatives(AliITSAlignMille2Module::kDOFTX,kFALSE); + dfDone[AliITSAlignMille2Module::kDOFTX] = kTRUE; + } + dXdxlocsens = fDerivativeGlo[AliITSAlignMille2Module::kDOFTX][0]; + dZdxlocsens = fDerivativeGlo[AliITSAlignMille2Module::kDOFTX][2]; + } + // + else { // need to perform some transformations + // fetch the jacobian of the transformation from the sensors local frame to the frame + // where the parameters are defined: + // Global: dX/dxloc_sens = dX/dxgl*dxgl/dxloc_sens + ...dX/dphigl*dphigl/dxloc_sens + if (fUseGlobalDelta) fCurrentModule->CalcDerivGloLoc(fCluster.GetVolumeID(), + AliITSAlignMille2Module::kDOFTX, jacobian); + // Local: dX/dxloc_sens = dX/dxcurr*dxcurr/dxloc_sens +..+dX/dphicurr * dphicurr/dxloc_sens + else fCurrentModule->CalcDerivCurLoc(fCluster.GetVolumeID(), + AliITSAlignMille2Module::kDOFTX, jacobian); + // + for (int j=0;jGetParOffset(AliITSAlignMille2Module::kDOFT0)>=0) { + m.fDerGlo[ifill][kX] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][0]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][2]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0); + } + // + if (fCurrentModule->GetParOffset(sddLR)>=0) { + m.fDerGlo[ifill][kX] = fDerivativeGlo[sddLR][0]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[sddLR][2]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(sddLR); + } + } + // + m.fModuleID[m.fNModFilled++] = fCurrentModule->GetUniqueID(); + } while( (fCurrentModule=fCurrentModule->GetParent()) ); + // + if (nmodTested>0 && zeroX) {AliInfo("Skipping: zero global X derivatives!");return -1;} + if (nmodTested>0 && zeroZ) {AliInfo("Skipping: zero global Z derivatives!");return -1;} + // + // ok, can copy to m + AliDebug(2,Form("Adding local equation X with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[0]-fPintLoc0[0]), fSigmaLoc[0])); + m.fMeas[kX] = fMeasLoc[0]-fPintLoc0[0]; + m.fSigma[kX] = fSigmaLoc[0]; + // + AliDebug(2,Form("Adding local equation Z with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[2]-fPintLoc0[2]), fSigmaLoc[2])); + m.fMeas[kZ] = fMeasLoc[2]-fPintLoc0[2]; + m.fSigma[kZ] = fSigmaLoc[2]; + // + m.fNGlobFilled = ifill; + fCurrentModule = endModule; + // + status += Int_t(!zeroX && !zeroZ); // 0 - only locals, 1 locals + globals + return status; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::AddLocalEquationTPA(Mille2Data &m) +{ + /// Define local equation for current cluster in X Y and Z coor. + /// and store them to memory + /// return -1 in case of failure to build some equation + /// 0 if no free global parameters were found but local eq is built + /// 1 if both local and global eqs are built + // + static int cnt = 0; + Bool_t dbg = kFALSE;//kTRUE; + if (++cnt>100000) dbg = kFALSE; + + int curpoint = fCluster.GetUniqueID()-1; + TGeoHMatrix *tempHMat = GetSensorCurrMatrixSID(fCurrentSensID);// fCurrentModule->GetSensitiveVolumeMatrix(fCluster.GetVolumeID()); + // + fTPAFitter->GetDResDParams(&fDerivativeLoc[0][0], curpoint); // resid. derivatives over the track parameters + if (fCurvFitWasConstrained && fFixCurvIfConstraned && !IsZero(fBField)) + for (int i=3;i--;) fDerivativeLoc[AliITSTPArrayFit::kR0][i] = 0; //Fix curvarute + // + for (Int_t i=fNLocal; i--;) tempHMat->MasterToLocalVect(fDerivativeLoc[i],m.fDerLoc[i]); + // + int status = 0; + // derivatives over the global parameters ---------------------------------------->>> + Double_t dGL[3]; // derivative of global position vs local X (for SDD) + Double_t dRdP[3][3]; // derivative of local residuals vs local position + Double_t dPdG[AliITSAlignMille2Module::kMaxParGeom][3]; // derivatives of local position vs global params + fTPAFitter->GetDResDPos(&fDerivativeGlo[0][0], curpoint); + if (fCurrentSensID!=kVtxSensID) for (int i=3;i--;) tempHMat->MasterToLocalVect(fDerivativeGlo[i],dRdP[i]); + else for (int i=3;i--;) for (int j=3;j--;) dRdP[i][j] = fDerivativeGlo[i][j]; // vertex constraint is in lab + // + if (dbg) { + printf("\nCurrentMod: %s Sens:%d\n",fCurrentModule->GetName(),fCurrentSensID); //RRR + printf("Module Matrix: "); + fCurrentModule->GetMatrix()->Print(); //RRR + for (int i=0;i<3;i++) { + printf("dRdP[M%d][resI] ",i); for (int j=0;j<3;j++) printf(":[%d] %+.3e ",j,dRdP[i][j]); printf("\n"); + }//RRR + printf("Sensor Matrix: "); tempHMat->Print(); + } + UInt_t ifill=0, dfDone = 0; + m.fNModFilled = 0; + // + AliITSAlignMille2Module* endModule = fCurrentModule; + // + m.fModuleID[0] = fCurrentModule->GetUniqueID(); // always register id of the base module, even if it has no DOF + // + do { + if (fCurrentModule->GetNParFree()==0) continue; + status = 1; + if (!fUseGlobalDelta) dfDone = 0; // for local deltas the derivatives at diff. levels are different + Bool_t jacobOK = kFALSE; + // + for (Int_t i=0; iGetParOffset(i)<0) continue; // this parameter is not explicitly fitted + // + if (!TestWordBit(dfDone,i)) { // need to calculate new derivative + if (!jacobOK) { + if (fCurrentSensID!=kVtxSensID) { + fCurrentModule->CalcDerivDPosDPar(fCluster.GetVolumeID(),fMeasLoc,&dPdG[0][0]); + if (dbg) { + for (int i1=0;i1<3;i1++) { + printf("Jacob:dPdG[gpar%d][Mj]",i1); for (int j1=0;j1<3;j1++) printf(":[%d] %+.3e ",j1,dPdG[i1][j1]); printf("\n");//RRR + } + } + } + else { + // this is a vertex constraint: only lateral shifts are allowed, no rotations + for (int ip=AliITSAlignMille2Module::kMaxParGeom;ip--;) for (int jp=3;jp--;) dPdG[ip][jp] = (ip==jp) ? 1:0; + } + jacobOK = kTRUE; + } + // dRes_j/dGlo_i = \sum_{k=1:3} dRes_j/dPos_k * dPos_k/dGlo_i + fDerivativeGlo[i][kX] = dRdP[kX][kX]*dPdG[i][kX] + dRdP[kY][kX]*dPdG[i][kY] + dRdP[kZ][kX]*dPdG[i][kZ]; + fDerivativeGlo[i][kY] = dRdP[kX][kY]*dPdG[i][kX] + dRdP[kY][kY]*dPdG[i][kY] + dRdP[kZ][kY]*dPdG[i][kZ]; + fDerivativeGlo[i][kZ] = dRdP[kX][kZ]*dPdG[i][kX] + dRdP[kY][kZ]*dPdG[i][kY] + dRdP[kZ][kZ]*dPdG[i][kZ]; + SetWordBit(dfDone,i); + } + // + if (dbg) { + printf("Level %s DGlob[par%d][resJ] ",fCurrentModule->GetName(),i); //RRR + for (int k=0;k<3;k++) printf(":[%d] %+.3e ",k, fDerivativeGlo[i][k]); printf("\n");//RRR + } + m.fDerGlo[ifill][kX] = fDerivativeGlo[i][kX]; + m.fDerGlo[ifill][kY] = fDerivativeGlo[i][kY]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[i][kZ]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(i); + // + } + // + if ( fCurrentModule->IsSDD() ) { // specific for SDD + // + // assume for sensor local xloc = xloc0 + V0*dT0+dV*(T-T0) + // where V0 and T are the nominal drift velocity, time and time0 + // and the dT0 and dV are the corrections: + // drloc_i/dT0 = sum_j drloc_i/dMeasGlo_j * dMeasGlo_j/dT0 = + // = sum_j drloc_i/dMeasGlo_j sum_k dMeasGlo_j/dMeasLoc_k * dMeasLoc_k/dT0 + // = sum_j drloc_i/dMeasGlo_j dMeasGlo_j/dMeasLoc_X * V0 + // + // drloc_i/dV0 = sum_j drloc_i/dMeasGlo_j * dMeasGlo_j/dV0 = + // = sum_j drloc_i/dMeasGlo_j sum_k dMeasGlo_j/dMeasLoc_k * dMeasLoc_k/dV0 + // = sum_j drloc_i/dMeasGlo_j dMeasGlo_j/dMeasLoc_X * T0 + + // IMPORTANT: the geom derivatives are over the SENSOR LOCAL parameters + // + Bool_t jacOK = kFALSE; + //Int_t sddLR = fMeasLoc[kX]>0 ? AliITSAlignMille2Module::kDOFDVL : AliITSAlignMille2Module::kDOFDVR; + Int_t sddLR = GetVDriftSDD()>0 ? AliITSAlignMille2Module::kDOFDVL : AliITSAlignMille2Module::kDOFDVR; + if (fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0)>=0) { + if (!TestWordBit(dfDone, AliITSAlignMille2Module::kDOFT0)) { + double vdrift = GetVDriftSDD(); + JacobianPosGloLoc(kX,dGL); + jacOK = kTRUE; + fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kX] = + vdrift*(dRdP[kX][kX]*dGL[kX] + dRdP[kY][kX]*dGL[kY] + dRdP[kZ][kX]*dGL[kZ]); + fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kY] = + vdrift*(dRdP[kX][kY]*dGL[kX] + dRdP[kY][kY]*dGL[kY] + dRdP[kZ][kY]*dGL[kZ]); + fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kZ] = + vdrift*(dRdP[kX][kZ]*dGL[kX] + dRdP[kY][kZ]*dGL[kY] + dRdP[kZ][kZ]*dGL[kZ]); + // + SetWordBit(dfDone, AliITSAlignMille2Module::kDOFT0); + } + m.fDerGlo[ifill][kX] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kX]; + m.fDerGlo[ifill][kY] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kY]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[AliITSAlignMille2Module::kDOFT0][kZ]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(AliITSAlignMille2Module::kDOFT0); + } + // + if (fCurrentModule->GetParOffset(sddLR)>=0) { + if (!TestWordBit(dfDone, sddLR)) { + double tdrift = TMath::Sign(GetTDriftSDD(), GetVDriftSDD()); + double vdrift = fIsSDDVDriftMult ? TMath::Abs(GetVDriftSDD()) : 1; + if (!jacOK) JacobianPosGloLoc(kX,dGL); + fDerivativeGlo[sddLR][kX] = + -tdrift*vdrift*(dRdP[kX][kX]*dGL[kX] + dRdP[kY][kX]*dGL[kY] + dRdP[kZ][kX]*dGL[kZ]); + fDerivativeGlo[sddLR][kY] = + -tdrift*vdrift*(dRdP[kX][kY]*dGL[kX] + dRdP[kY][kY]*dGL[kY] + dRdP[kZ][kY]*dGL[kZ]); + fDerivativeGlo[sddLR][kZ] = + -tdrift*vdrift*(dRdP[kX][kZ]*dGL[kX] + dRdP[kY][kZ]*dGL[kY] + dRdP[kZ][kZ]*dGL[kZ]); + SetWordBit(dfDone, sddLR); + } + m.fDerGlo[ifill][kX] = fDerivativeGlo[sddLR][kX]; + m.fDerGlo[ifill][kY] = fDerivativeGlo[sddLR][kY]; + m.fDerGlo[ifill][kZ] = fDerivativeGlo[sddLR][kZ]; + m.fParMilleID[ifill++] = fCurrentModule->GetParOffset(sddLR); + } + } + // + m.fModuleID[m.fNModFilled++] = fCurrentModule->GetUniqueID(); + } while( (fCurrentModule=fCurrentModule->GetParent()) ); + // + // store first local residuals + fTPAFitter->GetResiduals(fPintLoc , curpoint); // lab residuals + for (int i=3;i--;) fPintLoc[i] = -fPintLoc[i]; + if (fCurrentSensID!=kVtxSensID) tempHMat->MasterToLocalVect(fPintLoc,m.fMeas); // local residuals + else for (int i=3;i--;) m.fMeas[i] = fPintLoc[i]; + if (dbg) { + printf("res(meas-loc) "); for (int k=0;k<3;k++) printf(":[%d] %+.3e ",k,m.fMeas[k]); printf("\n"); + printf("Fin:%s %+e %+e\n",endModule->GetName(), fDerivativeGlo[kZ][kZ], fPintLoc[kZ]); + }//RRR + m.fSigma[kX] = fSigmaLoc[kX]; + m.fSigma[kY] = fSigmaLoc[kY]; + m.fSigma[kZ] = fSigmaLoc[kZ]; + // + m.fNGlobFilled = ifill; + fCurrentModule = endModule; + // + return status; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::SetLocalEquations(const Mille2Data *marr, Int_t neq) +{ + /// Set local equations with data stored in m + /// return 0 if success + // + Bool_t locPatt[kNLocal] = {0}; // pattern of lacal eq's to account + for (int i=fNLocal; i--;) { + if (locPatt[i]) continue; // already set + for (Int_t j=0; jSetLocalEquation(fGlobalDerivatives, fLocalDerivatives, m.fMeas[ic], m.fSigma[ic]); + filled = kTRUE; + // + } + // + if (filled) for (int i=m.fNModFilled;i--;) GetMilleModule(m.fModuleID[i])->IncNProcessedPoints(); + } + // + double wgh = 1; + if (GetWeightPt() && fTPAFitter) { + wgh = fTPAFitter->GetPt(); + if (wgh>10) wgh = 10.; + if (wgh<0) wgh = fTPAFitter->IsTypeCosmics() ? 7 : 0.5; + if (GetWeightPt()>0) wgh = TMath::Power(wgh,GetWeightPt()); + } + fMillepede->SetRecordWeight(wgh*fTrackWeight); + fMillepede->SetRecordRun(fRunID); + // +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GlobalFit() +{ + /// Call global fit; Global parameters are stored in parameters + if (!fIsMilleInit) Init(); + // + ApplyPreConstraints(); + int res = fMillepede->GlobalFit(); + AliInfo(Form("%s fitting global parameters!",res ? "Done":"Failed")); + if (res) { + // fetch the parameters + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + int nprocp = 0; + for (int ip=mod->GetNParTot();ip--;) { + int idp = mod->GetParOffset(ip); + if (idp<0) continue; // was not in the explicit fit + mod->SetParVal(ip,fMillepede->GetFinalParam(idp)); + mod->SetParErr(ip,fMillepede->GetFinalError(idp)); + int np = fMillepede->GetProcessedPoints(idp); + if (TMath::Abs(np)>TMath::Abs(nprocp)) nprocp = np; + } + if (!mod->GetNProcessedPoints()) mod->SetNProcessedPoints(nprocp); + } + + } + ApplyPostConstraints(); + return res; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::PrintGlobalParameters() +{ + /// Print global parameters + if (!fIsMilleInit) { + AliInfo("Millepede has not been initialized!"); + return; + } + fMillepede->PrintGlobalParameters(); +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadSuperModuleFile(const Char_t *sfile) +{ + // load definitions of supermodules from a root file + // return 0 if success + AliInfo(Form("Loading SuperModule definitions from %s",sfile)); + TFile *smf=TFile::Open(sfile); + if (!smf->IsOpen()) { + AliInfo(Form("Cannot open supermodule file %s",sfile)); + return -1; + } + + TClonesArray *sma=(TClonesArray*)smf->Get("ITSMilleSuperModules"); + if (!sma) { + AliInfo(Form("Cannot find ITSMilleSuperModules array in file")); + return -2; + } + Int_t nsma=sma->GetEntriesFast(); + AliInfo(Form("Array of SuperModules with %d entries\n",nsma)); + // + // pepo200709 + Char_t st[2048]; + char symname[250]; + // end pepo200709 + + UShort_t volid; + TGeoHMatrix m; + // + for (Int_t i=0; iUncheckedAt(i); + volid=a->GetVolUID(); + strncpy(st,a->GetSymName(),TMath::Min(sizeof(st),strlen(a->GetSymName())+1)); + a->GetMatrix(m); + // + memset(symname,0,250*sizeof(char)); + sscanf(st,"%249s",symname); + // + // decode module list + char *stp=strstr(st,"ModuleList:"); + if (!stp) return -3; + stp += 11; + int idx[2200]; + char spp[200]; int jp=0; + char cl[20]; + strncpy(st,stp,TMath::Min(sizeof(st),strlen(stp)+1)); + int l=strlen(st); + int j=0; + int n=0; + // + while (j<=l) { + if (st[j]==9 || st[j]==32 || st[j]==10 || st[j]==0) { + spp[jp]=0; + jp=0; + if (strlen(spp)) { + int k=strcspn(spp,"-"); + if (k%d)",idx[j],kMaxITSSensID)); + return -5; + } + } + Int_t smindex=int(2198+volid-14336); // virtual index + // + fSuperModule.AddAtAndExpand(new AliITSAlignMille2Module(smindex,volid,symname,&m,n,volidsv),fNSuperModules); + // + fNSuperModules++; + } + // + smf->Close(); + // + return 0; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainModuleSubUnitsMean(Int_t idm, Double_t val, UInt_t pattern) +{ + // require that sum of modifications for the childs of this module is = val, i.e. + // the internal corrections moves the module as a whole by fixed value (0 by default). + // pattern is the bit pattern for the parameters to constrain + // + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized: no constrain may be added!"); + return; + } + if (!GetMilleModule(idm)->GetNChildren()) return; + TString nm = "cstrSUMean"; + nm += GetNConstraints(); + AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMean, + idm,val,pattern); + cstr->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstr); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainModuleSubUnitsMedian(Int_t idm, Double_t val, UInt_t pattern) +{ + // require that median of the modifications for the childs of this module is = val, i.e. + // the internal corrections moves the module as a whole by fixed value (0 by default) + // module the outliers. + // pattern is the bit pattern for the parameters to constrain + // The difference between the mean and the median will be transfered to the parent + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized: no constrain may be added!"); + return; + } + if (!GetMilleModule(idm)->GetNChildren()) return; + TString nm = "cstrSUMed"; + nm += GetNConstraints(); + AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMedian, + idm,val,pattern); + cstr->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstr); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainOrphansMean(Double_t val, UInt_t pattern) +{ + // require that median of the modifications for the supermodules which have no parents is = val, i.e. + // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. + // pattern is the bit pattern for the parameters to constrain + // + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized: no constrain may be added!"); + return; + } + TString nm = "cstrOMean"; + nm += GetNConstraints(); + AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMean, + -1,val,pattern); + cstr->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstr); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainOrphansMedian(Double_t val, UInt_t pattern) +{ + // require that median of the modifications for the supermodules which have no parents is = val, i.e. + // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. + // pattern is the bit pattern for the parameters to constrain + // + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized: no constrain may be added!"); + return; + } + TString nm = "cstrOMed"; + nm += GetNConstraints(); + AliITSAlignMille2Constraint *cstr = new AliITSAlignMille2Constraint(nm.Data(),AliITSAlignMille2Constraint::kTypeMedian, + -1,val,pattern); + cstr->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstr); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainLocal(const Char_t* name,Double_t *parcf,Int_t npar,Double_t val,Double_t err) +{ + // apply constraint on parameters in the local frame + if (fIsMilleInit) { + AliInfo("Millepede has been already initialized: no constrain may be added!"); + return; + } + AliITSAlignMille2ConstrArray *cstr = new AliITSAlignMille2ConstrArray(name,parcf,npar,val,err); + cstr->SetConstraintID(GetNConstraints()); + fConstraints.Add(cstr); +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ApplyGaussianConstraint(const AliITSAlignMille2ConstrArray* cstr) +{ + // apply the constraint on the local corrections of a list of modules + int nmod = cstr->GetNModules(); + double jacobian[AliITSAlignMille2Module::kMaxParGeom][AliITSAlignMille2Module::kMaxParGeom]; + // + // check if this not special SDDT0 constraint + if (cstr->GetPattern()==BIT(AliITSAlignMille2Module::kDOFT0)) { + for (int i=0;iGetNModules()-1;i++) { + AliITSAlignMille2Module *mdI = GetMilleModule(cstr->GetModuleID(i)); + if (!mdI->IsFreeDOF(AliITSAlignMille2Module::kDOFT0)) continue; + for (int j=i+1;jGetNModules();j++) { + AliITSAlignMille2Module *mdJ = GetMilleModule(cstr->GetModuleID(j)); + if (!mdJ->IsFreeDOF(AliITSAlignMille2Module::kDOFT0)) continue; + // + ResetLocalEquation(); + fGlobalDerivatives[mdI->GetParOffset(AliITSAlignMille2Module::kDOFT0)] = 1; + fGlobalDerivatives[mdJ->GetParOffset(AliITSAlignMille2Module::kDOFT0)] =-1; + AddConstraint(fGlobalDerivatives, 0, 1.E-6); + } + } + return; + } + + for (int imd=nmod;imd--;) { + int modID = cstr->GetModuleID(imd); + AliITSAlignMille2Module* mod = GetMilleModule(modID); + ResetLocalEquation(); + int nadded = 0; + double value = cstr->GetValue(); + double sigma = cstr->GetError(); + // + // in case the reference (survey) deltas were imposed for Gaussian constraints + // already accumulated corrections: they must be subtracted from the constraint value. + if (IsConstraintWrtRef()) { + // + Double_t precal[AliITSAlignMille2Module::kMaxParTot]; + Double_t refcal[AliITSAlignMille2Module::kMaxParTot]; + for (int ip=AliITSAlignMille2Module::kMaxParTot;ip--;) {precal[ip]=0; refcal[ip] = 0.;} + // + // check if there was a reference delta provided for this module + AliAlignObjParams* parref = GetConstrRefObject(mod->GetName()); + if (parref) parref->GetPars(refcal, refcal+3); // found reference delta + // + // extract already applied local corrections for this module + if (fPrealignment) { + // + AliAlignObjParams *preo = GetPrealignedObject(mod->GetName()); + if (preo) { + TGeoHMatrix preMat,tmpMat = *mod->GetMatrix(); // Delta_Glob * Delta_Glob_Par * M + preo->GetMatrix(preMat); // Delta_Glob + preMat.MultiplyLeft( &tmpMat.Inverse() ); // M^-1 * Delta_Glob_Par^-1 = (Delta_Glob_Par * M)^-1 + tmpMat.MultiplyLeft( &preMat ); // (Delta_Glob_Par * M)^-1 * Delta_Glob * Delta_Glob_Par * M = Delta_loc + AliAlignObjParams algob; + algob.SetMatrix(tmpMat); + algob.GetPars(precal,precal+3); // local corrections for geometry + } + } + // + // subtract the contribution to constraint from precalibration + for (int ipar=cstr->GetNCoeffs();ipar--;) value += (refcal[ipar]-precal[ipar])*cstr->GetCoeff(ipar); + // + } + // + if (fUseGlobalDelta) mod->CalcDerivLocGlo(&jacobian[0][0]); + // + for (int ipar=cstr->GetNCoeffs();ipar--;) { + double coef = cstr->GetCoeff(ipar); + if (IsZero(coef)) continue; + // + if (!fUseGlobalDelta || ipar>= AliITSAlignMille2Module::kMaxParGeom) { // + // we are working with local params or if the given param is not related to geometry, + // apply the constraint directly + int parPos = mod->GetParOffset(ipar); + if (parPos<0) continue; // not in the fit + fGlobalDerivatives[parPos] += coef; + nadded++; + } + else { // we are working with global params, while the constraint is on local ones -> jacobian + for (int jpar=AliITSAlignMille2Module::kMaxParGeom;jpar--;) { + int parPos = mod->GetParOffset(jpar); + if (parPos<0) continue; + fGlobalDerivatives[parPos] += coef*jacobian[ipar][jpar]; + nadded++; + } + } + } + if (nadded) AddConstraint(fGlobalDerivatives, value, sigma); + } + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ApplyPreConstraints() +{ + // apply constriants which cannot be imposed after the fit + int nconstr = GetNConstraints(); + for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) { + ApplyGaussianConstraint( (AliITSAlignMille2ConstrArray*)cstr); + continue; + } + // + if (cstr->GetType() == AliITSAlignMille2Constraint::kTypeMedian) continue; // post type constraint + // + if (!fUseGlobalDelta) continue; // mean/med constraints must be applied to global deltas + // apply constraint on the mean's before the fit + int imd = cstr->GetModuleID(); + if (imd>=0) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + UInt_t pattern = 0; + for (int ipar=mod->GetNParTot();ipar--;) { + if (!cstr->IncludesParam(ipar)) continue; + if (mod->GetParOffset(ipar)<0) continue; // parameter is not in the explicit fit -> post constraint + pattern |= 0x1<SetApplied(ipar); + } + ConstrainModuleSubUnits(imd,cstr->GetValue(),pattern); + // + } + else if (!PseudoParentsAllowed()) { + ConstrainOrphans(cstr->GetValue(),(UInt_t)cstr->GetPattern()); + cstr->SetApplied(-1); + } + } + // + // do we need to tie the SDD left/right VDrift corrections + for (int imd=0;imdIsSDD() && mod->IsVDriftLRSame()) TieSDDVDriftsLR(mod); + } + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ApplyPostConstraints() +{ + // apply constraints which can be imposed after the fit + int nconstr = GetNConstraints(); + Bool_t convGlo = kFALSE; + // check if there is something to do + int ntodo = 0; + for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) continue; + if (cstr->GetRemainingPattern() == 0) continue; + ntodo++; + } + if (!ntodo) return; + // + if (!fUseGlobalDelta) { // need to convert to global params + ConvertParamsToGlobal(); + convGlo = kTRUE; + } + // + for (int i=0;iGetType() == AliITSAlignMille2ConstrArray::kTypeGaussian) continue; + // + int imd = cstr->GetModuleID(); + // + if (imd>=0) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + if (mod->IsNotInConf()) continue; + UInt_t pattern = 0; + for (int ipar=mod->GetNParTot();ipar--;) { + if (cstr->IsApplied(ipar)) continue; + if (!cstr->IncludesParam(ipar)) continue; + if (!mod->IsFreeDOF(ipar)) continue; // parameter is fixed, will not apply constraint + pattern |= 0x1<SetApplied(ipar); + } + if (pattern) PostConstrainModuleSubUnits(cstr->GetType(),cstr->GetModuleID(),cstr->GetValue(),pattern); + // + } + else if (PseudoParentsAllowed()) { + UInt_t pattern = (UInt_t)cstr->GetRemainingPattern(); + PostConstrainOrphans(cstr->GetType(),cstr->GetValue(),pattern); + cstr->SetApplied(-1); + } + } + // if there was a conversion, rewind it + if (convGlo) ConvertParamsToLocal(); + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainModuleSubUnits(Int_t idm, Double_t val, UInt_t pattern) +{ + // require that sum of modifications for the childs of this module is = val, i.e. + // the internal corrections moves the module as a whole by fixed value (0 by default). + // pattern is the bit pattern for the parameters to constrain + // + // + AliITSAlignMille2Module* mod = GetMilleModule(idm); + // + for (int ip=0;ip>ip)&0x1) /*|| !parent->IsFreeDOF(ip)*/) continue; + ResetLocalEquation(); + int nadd = 0; + for (int ich=mod->GetNChildren();ich--;) { + int idpar = ((AliITSAlignMille2Module*)mod->GetChild(ich))->GetParOffset(ip); + if (idpar<0) continue; + fGlobalDerivatives[idpar] = 1.0; + nadd++; + } + // + if (nadd>0) { + AddConstraint(fGlobalDerivatives,val); + AliInfo(Form("Constrained param %d for %d submodules of module #%d: %s",ip,nadd,idm,mod->GetName())); + } + } + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainOrphans(Double_t val, UInt_t pattern) +{ + // require that median of the modifications for the supermodules which have no parents is = val, i.e. + // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. + // pattern is the bit pattern for the parameters to constrain + // + for (int ip=0;ip>ip)&0x1) ) continue; + ResetLocalEquation(); + int nadd = 0; + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + if (mod->IsNotInConf()) continue; // dummy module + AliITSAlignMille2Module* par = mod->GetParent(); + while (par && par->IsNotInConf() ) par = par->GetParent(); // use only decalred parents + if (par) continue; // this is not an orphan + int idpar = mod->GetParOffset(ip); + if (idpar<0) continue; + fGlobalDerivatives[idpar] = 1.0; + nadd++; + } + if (nadd>0) { + AddConstraint(fGlobalDerivatives,val); + AliInfo(Form("Constrained param %d for %d orphan modules",ip,nadd)); + } + } + // + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::PostConstrainModuleSubUnits(Int_t type,Int_t idm, Double_t val, UInt_t pattern) +{ + // require that median or mean of the modifications for the childs of this module is = val, i.e. + // the internal corrections moves the module as a whole by fixed value (0 by default) + // module the outliers. + // pattern is the bit pattern for the parameters to constrain + // The difference between the mean and the median will be transfered to the parent + // + AliITSAlignMille2Module* parent = GetMilleModule(idm); + int nc = parent->GetNChildren(); + // + double *tmpArr = new double[nc]; + // + for (int ip=0;ip>ip)&0x1) || !parent->IsFreeDOF(ip)) continue; + // compute the mean and median of the deltas + int nfree = 0; + for (int ich=nc;ich--;) { + AliITSAlignMille2Module* child = parent->GetChild(ich); + // if (!child->IsFreeDOF(ip)) continue; + tmpArr[nfree++] = child->GetParVal(ip); + } + double median=0,mean=0; + for (int ic0=0;ic0tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;} + } + // + int kmed = nfree/2; + median = (tmpArr[kmed]+tmpArr[nfree-kmed-1])/2.; + if (nfree>0) mean /= nfree; + // + double shift = val - (type==AliITSAlignMille2Constraint::kTypeMean ? mean : median); + // + for (int ich=nc;ich--;) { + AliITSAlignMille2Module* child = parent->GetChild(ich); + // if (!child->IsFreeDOF(ip)) continue; + child->SetParVal(ip, child->GetParVal(ip) + shift); + npc++; + } + // + parent->SetParVal(ip, parent->GetParVal(ip) - shift); + AliInfo(Form("%s constraint: added %+f shift to param[%d] of %d children of module %d: %s", + type==AliITSAlignMille2Constraint::kTypeMean ? "MEAN" : "MEDIAN",shift, + ip,npc,idm,parent->GetName())); + } + delete[] tmpArr; + // + // +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::PostConstrainOrphans(Int_t type,Double_t val, UInt_t pattern) +{ + // require that median or mean of modifications for the supermodules which have no parents is = val, i.e. + // the corrections moves the whole setup by fixed value (0 by default). + // pattern is the bit pattern for the parameters to constrain + // + int nc = fNModules; + // + int norph = 0; + for (int ich=nc;ich--;) { + AliITSAlignMille2Module *par= GetMilleModule(ich)->GetParent(); + while (par && par->IsNotInConf()) par = par->GetParent(); // use only decalred parents + if (!par) norph ++; + } + // + if (!norph) return; + double *tmpArr = new double[norph]; + for (int i=norph;i--;) tmpArr[i] = 0; + // + for (int ip=0;ip>ip)&0x1)) continue; + // compute the mean and median of the deltas + int nfree = 0; + for (int ich=nc;ich--;) { + AliITSAlignMille2Module* child = GetMilleModule(ich); + if (child->IsNotInConf()) continue; // dummy module + // if (child->GetParent() || !child->IsFreeDOF(ip)) continue; + AliITSAlignMille2Module* par = child->GetParent(); + while (par && par->IsNotInConf()) par = par->GetParent(); // count only declared parents + if (par) continue; + tmpArr[nfree++] = child->GetParVal(ip); + } + double median=0,mean=0; + for (int ic0=0;ic0tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;} + } + // + int kmed = nfree/2; + median = (tmpArr[kmed]+tmpArr[nfree-kmed-1])/2.; + if (nfree>0) mean /= nfree; + // + double shift = val - (type==AliITSAlignMille2Constraint::kTypeMean ? mean : median); + // + for (int ich=nc;ich--;) { + AliITSAlignMille2Module* child = GetMilleModule(ich); + if (child->IsNotInConf()) continue; // dummy module + // if (child->GetParent() || !child->IsFreeDOF(ip)) continue; + AliITSAlignMille2Module* par = child->GetParent(); + while (par && par->IsNotInConf()) par = par->GetParent(); // count only declared parents + if (par) continue; + child->SetParVal(ip, child->GetParVal(ip) + shift); + npc++; + } + // + AliInfo(Form("%s constraint: added %+f shift to param[%d] of %d orphan modules", + type==AliITSAlignMille2Constraint::kTypeMean ? "MEAN" : "MEDIAN",shift, + ip,npc)); + } + delete[] tmpArr; + // +} + +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::IsParModConstrained(const AliITSAlignMille2Module* mod,Int_t par, Bool_t &meanmed, Bool_t &gaussian) const +{ + // check if par of the module participates in some constraint, and set the flag for their types + meanmed = gaussian = kFALSE; + // + if ( mod->IsParConstrained(par) ) gaussian = kTRUE; // direct constraint on this param + // + for (int icstr=GetNConstraints();icstr--;) { + AliITSAlignMille2Constraint* cstr = GetConstraint(icstr); + // + if (!cstr->IncludesModPar(mod,par)) continue; + if (cstr->GetType()==AliITSAlignMille2ConstrArray::kTypeGaussian) gaussian = kTRUE; + else meanmed = kTRUE; + // + if (meanmed && gaussian) break; // no sense to check further + } + // + return meanmed||gaussian; +} + +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::IsParModFamilyVaried(const AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const +{ + // check if parameter par is varied for this module or its children up to the level depth + if (depth<0) return kFALSE; + if (mod->GetParOffset(par)>=0) return kTRUE; + for (int icld=mod->GetNChildren();icld--;) { + AliITSAlignMille2Module* child = mod->GetChild(icld); + if (IsParModFamilyVaried(child, par, depth-1)) return kTRUE; + } + return kFALSE; + // +} + +/* +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::IsParFamilyFree(AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const +{ + // check if parameter par is varied and is not subjected to gaussian constraint for the children up to the level depth + if (depth<0) return kTRUE; + for (int icld=mod->GetNChildren();icld--;) { + AliITSAlignMille2Module* child = mod->GetChild(icld); + //if (child->GetParOffset(par)<0) continue; // fixed + Bool_t cstMM=kFALSE,cstGS=kFALSE; + // does this child have gaussian constraint ? + if (!IsParModConstrained(child,par,cstMM,cstGS) || !cstGS ) return kTRUE; + // check its children + if (!IsParFamilyFree(child,par,depth-1)) return kTRUE; + } + return kFALSE; + // +} +*/ + +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::IsParFamilyFree(const AliITSAlignMille2Module* mod,Int_t par,Int_t depth) const +{ + // check if parameter par is varied and is not subjected to gaussian constraint for the children up to the level depth + if (depth<0) return kFALSE; + for (int icld=mod->GetNChildren();icld--;) { + AliITSAlignMille2Module* child = mod->GetChild(icld); + //if (child->GetParOffset(par)<0) continue; // fixed + Bool_t cstMM=kFALSE,cstGS=kFALSE; + // does this child have gaussian constraint ? + if (!IsParModConstrained(child,par,cstMM,cstGS) || !cstGS ) return kTRUE; + // check its children + if (IsParFamilyFree(child,par,depth-1)) return kTRUE; + } + return kFALSE; + // +} + +//________________________________________________________________________________________________________ +Double_t AliITSAlignMille2::GetTDriftSDD() const +{ + // obtain drift time corrected for t0 + double t = fCluster.GetDriftTime(); + return t - fDriftTime0[ fCluster.GetUniqueID()-1 ]; +} + +//________________________________________________________________________________________________________ +Double_t AliITSAlignMille2::GetVDriftSDD() const +{ + // obtain corrected drift speed + return fDriftSpeed[ fCluster.GetUniqueID()-1 ]; +} + +//________________________________________________________________________________________________________ +Bool_t AliITSAlignMille2::FixedOrphans() const +{ + // are there fixed modules with no parent (normally in such a case + // the constraints on the orphans should not be applied + if (!IsConfigured()) { + AliInfo("Still not configured"); + return kFALSE; + } + for (int i=0;iIsNotInConf()) continue; + if (md->GetParent()==0 && md->GetNParFree()==0) return kTRUE; + } + return kFALSE; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConvertParamsToGlobal() const +{ + // convert params in local frame to global one + double pars[AliITSAlignMille2Module::kMaxParGeom]; + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + if (mod->GeomParamsGlobal()) continue; + mod->GetGeomParamsGlo(pars); + mod->SetParVals(pars,AliITSAlignMille2Module::kMaxParGeom); + mod->SetGeomParamsGlobal(kTRUE); + } +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConvertParamsToLocal() const +{ + // convert params in global frame to local one + double pars[AliITSAlignMille2Module::kMaxParGeom]; + for (int imd=fNModules;imd--;) { + AliITSAlignMille2Module* mod = GetMilleModule(imd); + if (!mod->GeomParamsGlobal()) continue; + mod->GetGeomParamsLoc(pars); + mod->SetParVals(pars,AliITSAlignMille2Module::kMaxParGeom); + mod->SetGeomParamsGlobal(kFALSE); + } +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::SetBField(Double_t b) +{ + // set Bz value + if (IsZero(b,1e-5)) { + fBField = 0.0; + fBOn = kFALSE; + fNLocal = 4; + } + else { + fBField = b; + fBOn = kTRUE; + fNLocal = 5; // helices + } +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::ProcessUserInfo(TList* userInfo) +{ + // extract calibration information used for TrackPointArray creation from run info + // + if (!userInfo) { AliInfo("No UserInfo is provided"); return 0;} + // + TMap *cdbMap=0; + TList* cdbList=0; + TObjString *objStr,*objStr1,*keyStr; + TString cdbStr; + AliCDBManager* man = AliCDBManager::Instance(); + man->SetCacheFlag(kFALSE); + // + int run = userInfo->GetUniqueID(); + if (run>0) SetRunID(run); + AliInfo(Form("UserInfo corresponds to run#%d",run)); + cdbMap = (TMap*)userInfo->FindObject("cdbMap"); + const TMap *curMap = man->GetStorageMap(); + if (!cdbMap) {AliInfo("No CDB Map found in UserInfo");} + else { + if ((objStr=(TObjString*)cdbMap->GetValue("default"))) { // first set default CDB path + if ((objStr1=(TObjString*)curMap->GetValue("default")) && objStr1->GetUniqueID()) { + AliInfo(Form("OCDB default path from UserInfo: %s is overriden by user setting %s",objStr->GetName(),objStr1->GetName())); + } + else { + cdbStr = objStr->GetString(); + man->UnsetDefaultStorage(); + if (man->GetRaw()) man->SetRaw(kFALSE); + if (cdbStr.BeginsWith("raw://")) cdbStr = "raw://"; + AliInfo(Form("Default CDB Storage from UserInfo: %s",cdbStr.Data())); + man->SetDefaultStorage( cdbStr.Data() ); // this may be overriden later by configuration file + } + } + if (man->GetRaw() && run>0) man->SetRun(run); + // + // set specific paths relevant for alignment + TIter itMap(cdbMap); + while( (keyStr=(TObjString*)itMap.Next()) ) { + TString keyS = keyStr->GetString(); + if ( keyS == "default" ) continue; + // + TObjString* curPath = (TObjString*)curMap->GetValue(keyStr->GetName()); + if (curPath && curPath->GetUniqueID()) { + AliInfo(Form("Storage for %s from UserInfo\n is overriden by user setting %s",keyS.Data(),curPath->GetName())); + continue; + } + man->SetSpecificStorage( keyS.Data(), cdbMap->GetValue(keyS)->GetName() ); + } + } + // + cdbList = (TList*)userInfo->FindObject("cdbList"); + if (!cdbList) {AliInfo("No CDB List found in UserInfo");} + else { + // Objects used for TrackPointArray production + GetPathFromUserInfo(cdbList,"GRP/Geometry/Data",fIniGeomPath ,kSameInitGeomBit); + GetPathFromUserInfo(cdbList,"ITS/Align/Data" ,fIniDeltaPath,kSameInitDeltasBit); + GetPathFromUserInfo(cdbList,"ITS/Calib/RespSDD",fIniSDDRespPath,kSameInitSDDRespBit); + GetPathFromUserInfo(cdbList,"ITS/Calib/DriftSpeedSDD",fIniSDDVDriftPath,kSameInitSDDVDriftBit); + GetPathFromUserInfo(cdbList,"ITS/Calib/MapsTimeSDD",fIniSDDCorrMapPath,kSameInitSDDCorrMapBit); + GetPathFromUserInfo(cdbList,"GRP/Calib/MeanVertexSPD",fDiamondPath,kSameDiamondBit); + } + // + TList *bzlst = (TList*)userInfo->FindObject("BzkGauss"); + if (bzlst && bzlst->At(0)) { + objStr = (TObjString*)bzlst->At(0); + SetBField( objStr->GetString().Atof() ); + AliInfo(Form("Magnetic field from UserInfo: %+.2e",GetBField())); + } + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::GetPathFromUserInfo(const TList* cdbList,const char* calib,TString& path, Int_t useBit) +{ + // extract the path for specific CDB path from user info. If it is the same as already loaded, set corresponing bit + TIter itList(cdbList); + if (useBit>=0) ResetBit(useBit); + TObjString* objStr; + while( (objStr=(TObjString*)itList.Next()) ) + if (objStr->GetString().Contains(calib)) { + TString newpath = objStr->GetString(); + AliInfo(Form("Found path in UserInfo: %s",newpath.Data())); + if ( useBit>=0 && (fUserProvided&useBit) ) { + AliInfo(Form("Will use the one provided in config: %s",path.Data())); + SetBit(useBit); + } + else if ( useBit>=0 && (newpath == path) ) { + AliInfo(Form("Path %s is the same as already loaded",path.Data())); + SetBit(useBit); + } + else path = newpath; + // + return 0; + } + AliInfo(Form("Did not find path for %s in UserInfo",calib)); + path = ""; + return -1; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadSDDResponse(TString& path, AliITSresponseSDD *&resp) +{ + // load SDD response + if (path.IsNull()) return 0; + AliInfo(Form("Loading SDD response from %s",path.Data())); + // + AliCDBEntry *entry = 0; + delete resp; + resp = 0; + // + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + resp = (AliITSresponseSDD*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("AliITSresponseSDD")) resp = (AliITSresponseSDD*)precf->Get("AliITSresponseSDD"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + resp = (AliITSresponseSDD*) entry->GetObject(); + if (resp && resp->InheritsFrom(AliITSresponseSDD::Class())) entry->SetObject(NULL); + else resp = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + // + precf->Close(); + delete precf; + break; + } + // + if (!resp) {AliError(Form("Failed to load SDD response from %s",path.Data())); return -1;} + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadSDDVDrift(TString& path, TObjArray *&arr) +{ + // load VDrift object + if (path.IsNull()) return 0; + AliInfo(Form("Loading SDD VDrift from %s",path.Data())); + // + AliCDBEntry *entry = 0; + delete arr; + arr = 0; + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + arr = (TObjArray*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("TObjArray")) arr = (TObjArray*)precf->Get("TObjArray"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + arr = (TObjArray*) entry->GetObject(); + if (arr && arr->InheritsFrom(TObjArray::Class())) entry->SetObject(NULL); + else arr = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + // + precf->Close(); + delete precf; + break; + } + // + if (!arr) {AliError(Form("Failed to load SDD vdrift from %s",path.Data())); return -1;} + arr->SetOwner(kTRUE); + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadSDDCorrMap(TString& path, AliITSCorrectSDDPoints *&map) +{ + // Load SDD correction map + // + if (path.IsNull()) return 0; + AliInfo(Form("Loading SDD Correction Maps from %s",path.Data())); + // + AliCDBEntry *entry = 0; + delete map; + map = 0; + TObjArray* arr = 0; + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + arr = (TObjArray*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("TObjArray")) arr = (TObjArray*)precf->Get("TObjArray"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + arr = (TObjArray*) entry->GetObject(); + if (arr && arr->InheritsFrom(TObjArray::Class())) entry->SetObject(NULL); + else arr = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + // + precf->Close(); + delete precf; + break; + } + // + if (!arr) {AliError(Form("Failed to load SDD Correction Map from %s",path.Data())); return -1;} + arr->SetOwner(kTRUE); + map = new AliITSCorrectSDDPoints(arr); + + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadPreSDDCalib() +{ + // Load SDD correction map for prealignment from current CDB + // + AliInfo(Form("Loading SDD Calibration set for run %d",fRunID)); + AliCDBManager* man = AliCDBManager::Instance(); + man->SetRun(fRunID); + AliCDBEntry *entry = man->Get("ITS/Calib/MapsTimeSDD"); + if(!entry){ + AliError("Error accessing OCDB: SDD maps not found"); + return -1; + } + delete fPreCorrMapSDD; + TObjArray* arr = (TObjArray*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + arr->SetOwner(kTRUE); + fPreCorrMapSDD = new AliITSCorrectSDDPoints(arr); + // + entry = man->Get("ITS/Calib/RespSDD"); + if(!entry){ + AliError("Error accessing OCDB: SDD response not found"); + return -1; + } + delete fPreRespSDD; + fPreRespSDD = (AliITSresponseSDD*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // + entry = man->Get("ITS/Calib/DriftSpeedSDD"); + if(!entry){ + AliError("Error accessing OCDB: SDD Drift speed not found"); + return -1; + } + delete fPreVDriftSDD; + fPreVDriftSDD = (TObjArray*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + // + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadDiamond(TString& path) +{ + // load vertex constraint + if (path.IsNull()) return 0; + AliInfo(Form("Loading Diamond Constraint from %s",path.Data())); + // + AliCDBEntry *entry = 0; + AliESDVertex *vtx = 0; + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + vtx = (AliESDVertex*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("AliESDVertex")) vtx = (AliESDVertex*)precf->Get("AliESDVertex"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + vtx = (AliESDVertex*) entry->GetObject(); + if (vtx && vtx->InheritsFrom(AliESDVertex::Class())) entry->SetObject(NULL); + else vtx = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + // + precf->Close(); + delete precf; + break; + } + // + if (!vtx) {AliError(Form("Failed to load Diamond constraint from %s",path.Data())); return -1;} + // + double vtxXYZ[3]; + vtx->GetXYZ(vtxXYZ); + for (int i=3;i--;) vtxXYZ[i] -= fCorrDiamond[i]; + vtx->SetXYZ(vtxXYZ); + SetVertexConstraint(vtx); + AliInfo("Will use following Diamond Constraint (errors inverted):"); + fDiamondI.Print(""); + delete vtx; + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::LoadDeltas(TString& path, TClonesArray *&arr) +{ + // load ITS geom deltas + if (path.IsNull()) return 0; + AliInfo(Form("Loading Alignment Deltas from %s",path.Data())); + // + AliCDBEntry *entry = 0; + delete arr; + arr = 0; + while(1) { + if (path.BeginsWith("path: ")) { // must load from OCDB + entry = GetCDBEntry(path.Data()); + if (!entry) break; + arr = (TClonesArray*) entry->GetObject(); + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + // AliCDBManager::Instance()->UnloadFromCache(path); // don't want cached object, read new copy + // delete cdbId; + // delete entry; + break; + } + // + if (gSystem->AccessPathName(path.Data())) break; + TFile* precf = TFile::Open(path.Data()); + if (precf->FindKey("ITSAlignObjs")) arr = (TClonesArray*)precf->Get("ITSAlignObjs"); + else if (precf->FindKey("AliCDBEntry") && (entry=(AliCDBEntry*)precf->Get("AliCDBEntry"))) { + arr = (TClonesArray*) entry->GetObject(); + if (arr && arr->InheritsFrom(TClonesArray::Class())) entry->SetObject(NULL); + else arr = 0; + entry->SetObject(NULL); + entry->SetOwner(kTRUE); + delete entry; + } + precf->Close(); + delete precf; + break; + } + // + if (!arr) {AliError(Form("Failed to load Deltas from %s",path.Data())); return -1;} + // + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CacheMatricesCurr() +{ + // build arrays for the fast access to sensor matrices from their sensor ID + // + TGeoHMatrix mdel; + AliInfo("Building sensors current matrices cache"); + // + fCacheMatrixCurr.Delete(); + for (int idx=0;idx<=kMaxITSSensID;idx++) { + int volID = AliITSAlignMille2Module::GetVolumeIDFromIndex(idx); + TGeoHMatrix *mcurr = new TGeoHMatrix(); + AliITSAlignMille2Module::SensVolMatrix(volID, mcurr); + fCacheMatrixCurr.AddAtAndExpand(mcurr,idx); + // + } + // + TGeoHMatrix *mcurr = new TGeoHMatrix(); + fCacheMatrixCurr.AddAtAndExpand(mcurr,kVtxSensID); // special unit matrix for diamond constraint + // + fCacheMatrixCurr.SetOwner(kTRUE); + return 0; +} + +//________________________________________________________________________________________________________ +Int_t AliITSAlignMille2::CacheMatricesOrig() +{ + // build arrays for the fast access to sensor original matrices (used for production) + // + TGeoHMatrix mdel; + AliInfo(Form("Building sensors original matrices cache. InitDeltaPath: %s",fIniDeltaPath.Data())); + // + /*if (fIniGeomPath!=fGeometryPath)*/ if (LoadGeometry(fIniGeomPath)) {AliInfo("Failed to re-load ideal geometry");exit(1);} + // + fCacheMatrixOrig.Delete(); + if (!fIniDeltaPath.IsNull()) { + TClonesArray* prealSav = fPrealignment; + fPrealignment = 0; + if (LoadDeltas(fIniDeltaPath,fPrealignment) || ApplyToGeometry()) + { AliError("Failed to load/apply initial deltas used to produce points"); return -1;} + delete fPrealignment; + fPrealignment = prealSav; + } + // + for (int idx=0;idx<=kMaxITSSensID;idx++) { + int volID = AliITSAlignMille2Module::GetVolumeIDFromIndex(idx); + TGeoHMatrix *morig = new TGeoHMatrix(); + AliITSAlignMille2Module::SensVolMatrix(volID,morig); + fCacheMatrixOrig.AddAtAndExpand(morig,idx); + } + // + if (fConvertPreDeltas) { + // in order to convert deltas from old to new geometry we need the final matrices for all alignable objects + int nmat = fGeoManager->GetNAlignable(); + fConvAlgMatOld.Delete(); + int nmatSel = 0; + for (int i=0;iGetAlignableEntry(i)->GetName(); + if (!nm.BeginsWith("ITS")) continue; + TGeoHMatrix *mo = new TGeoHMatrix(); + (*mo) = *(AliGeomManager::GetMatrix(nm)); + fConvAlgMatOld.AddAtAndExpand(mo,nmatSel++); + mo->SetTitle(nm); + mo->SetName(nm); + } + ConvSortHierarchically(fConvAlgMatOld); + } + // + TGeoHMatrix *mcurr = new TGeoHMatrix(); + fCacheMatrixOrig.AddAtAndExpand(mcurr,kVtxSensID); // special unit matrix for diamond constraint + // + fCacheMatrixOrig.SetOwner(kTRUE); + + fUsePreAlignment = 0; + LoadGeometry(fGeometryPath); // reload target geometry + // + return 0; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::RemoveHelixFitConstraint() +{ + // suppress constraint + fConstrCharge = 0; + fConstrPT = fConstrPTErr = -1; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainHelixFitPT(Int_t q,Double_t pt,Double_t pterr) +{ + // constrain q and pT of the helical fit of the track (should be set before process.track) + // + fConstrCharge = q==0 ? q:TMath::Sign(1,q); + fConstrPT = pt; + fConstrPTErr = pterr; + fCurvFitWasConstrained = kTRUE; +} + +//________________________________________________________________________________________________________ +void AliITSAlignMille2::ConstrainHelixFitCurv(Int_t q,Double_t crv,Double_t crverr) +{ + // constrain charge and curvature of the helical fit of the track (should be set before process.track) + // + const double kCQConv = 0.299792458e-3;// R = PT/Bz/fgkCQConv with GeV,kGauss,cm + + fConstrCharge = q==0 ? q:TMath::Sign(1,q); + if (crv<0 || IsZero(crv)) { + fConstrPT = -1; + fConstrPTErr = -1; + fCurvFitWasConstrained = kFALSE; + } + else { + fConstrPT = TMath::Abs(1./crv*fBField*kCQConv); + fConstrPTErr = crverr>1e-10 ? TMath::Abs(fConstrPT/crv*crverr) : 0.; + fCurvFitWasConstrained = kTRUE; + } +} + +//________________________________________________________________________________________________________ +TClonesArray* AliITSAlignMille2::CreateDeltas() +{ + // Create \Deltas for every explicitly or implicitly (via non-alignable volumes) varied + // or prealigned module. + // If the module has inded J in the hierarchy of alignable volumes (0 - the top, most + // coarse level), then its Delta is expressed via MP2 \deltas (in global frame) and + // prealignment \DeltaP's as: + // \Delta_J = Y X Y^-1 + // where X = \delta_J * \DeltaP_J + // Y = Prod_{K=0,J-1} \delta_K + // Note that \delta_L accounts not only for its own correction but also of all non-alignable + // modules in the hierarchy chain from L up to the closest alignable: + // while (parent && !parent->IsAlignable()) { + // \delta_L->MultiplyLeft( \delta_parent ); + // parent = parent->GetParent(); + // } + // + Bool_t convLoc = kFALSE; + if (!GetUseGlobalDelta()) { + ConvertParamsToGlobal(); + convLoc = kTRUE; + } + // + AliAlignObjParams tempAlignObj; + TGeoHMatrix tempMatX,tempMatY,tempMat1; + // + TClonesArray *array = new TClonesArray("AliAlignObjParams",10); + TClonesArray &alobj = *array; + int idx = 0; + // + TGeoManager* geoManager = AliGeomManager::GetGeometry(); + int nalgtot = geoManager->GetNAlignable(); + // + for (int ialg=0;ialgGetAlignableEntry(ialg)->GetName(); + // + AliITSAlignMille2Module* md = GetMilleModuleBySymName(algname); // explicitly varied? + AliITSAlignMille2Module* parent = md ? md->GetParent(): GetMilleModuleIfContained(algname); + if (md && parent) { + TString mdName = md->GetName(); + TString prName = parent->GetName(); + // SPD Sector -> Layer parentship is fake, need special treatment + if ( mdName.CountChar('/')==2 && mdName.BeginsWith("ITS/SPD") && // SPD sector + prName.CountChar('/')==1 && mdName.BeginsWith("ITS/SPD") ) // SPD Layer + parent = parent->GetParent();//: GetMilleModuleIfContained(prName.Data()); + } + // + AliAlignObjParams* preob = GetPrealignedObject(algname); // was it prealigned ? + // + if (!preob && !md && (!parent || parent->IsAlignable())) continue; // noting to do + // + // create matrix X (see comment) ------------------------------------------------->>> + // start from unity matrix + tempMatX.Clear(); + if (preob) { // account prealigngment + preob->GetMatrix(tempMat1); + tempMatX.MultiplyLeft(&tempMat1); + } + // + if (md) { + tempAlignObj.SetTranslation( md->GetParVal(0),md->GetParVal(1),md->GetParVal(2)); + tempAlignObj.SetRotation( md->GetParVal(3),md->GetParVal(4),md->GetParVal(5)); + tempAlignObj.GetMatrix(tempMat1); + tempMatX.MultiplyLeft(&tempMat1); // acount correction to varied module + } + // + // the corrections to all non-alignable modules from current on + // till first alignable should add up to its matrix + while (parent && !parent->IsAlignable()) { + tempAlignObj.SetTranslation( parent->GetParVal(0),parent->GetParVal(1),parent->GetParVal(2)); + tempAlignObj.SetRotation( parent->GetParVal(3),parent->GetParVal(4),parent->GetParVal(5)); + tempAlignObj.GetMatrix(tempMat1); + tempMatX.MultiplyLeft(&tempMat1); // add matrix of non-alignable module + parent = parent->GetParent(); + } + // create matrix X (see comment) ------------------------------------------------<<< + // + // create matrix Y (see comment) ------------------------------------------------>>> + // start from unity matrix + tempMatY.Clear(); + while ( parent ) { + tempAlignObj.SetTranslation( parent->GetParVal(0),parent->GetParVal(1),parent->GetParVal(2)); + tempAlignObj.SetRotation( parent->GetParVal(3),parent->GetParVal(4),parent->GetParVal(5)); + tempAlignObj.GetMatrix(tempMat1); + tempMatY.MultiplyLeft(&tempMat1); + parent = parent->GetParent(); + } + // create matrix Y (see comment) ------------------------------------------------<<< + // + tempMatX.MultiplyLeft(&tempMatY); + tempMatX.Multiply(&tempMatY.Inverse()); + // + if (tempMatX.IsIdentity()) continue; // do not store dummy matrices + UShort_t vid = AliITSAlignMille2Module::GetVolumeIDFromSymname(algname); + new(alobj[idx++]) AliAlignObjParams(algname,vid,tempMatX,kTRUE); + // + } + // + if (convLoc) ConvertParamsToLocal(); + // + return array; + // +} + +//_______________________________________________________________________________________ +AliITSresponseSDD* AliITSAlignMille2::CreateSDDResponse() +{ + // create object with SDD repsonse (t0 and vdrift corrections) accounting for + // eventual precalibration + // + // if there was a precalibration provided, copy it to new arrray + AliITSresponseSDD *precal = GetSDDPrecalResp(); + if (!precal && fIniVDriftSDD) precal = GetSDDInitResp(); // InitResp is used only when IniVDrift is provided + Bool_t isPreCalMult = precal&&precal->IsVDCorrMult() ? kTRUE : kFALSE; + AliITSresponseSDD *calibSDD = new AliITSresponseSDD(); + calibSDD->SetVDCorrMult(fIsSDDVDriftMult); + // + // copy initial values to the new object + if (precal) { + calibSDD->SetTimeOffset(precal->GetTimeOffset()); + calibSDD->SetADC2keV(precal->GetADC2keV()); + calibSDD->SetChargevsTime(precal->GetChargevsTime()); + for (int ind=kSDDoffsID;indSetModuleTimeZero(ind, precal->GetTimeZero(ind)); + calibSDD->SetDeltaVDrift(ind, precal->GetDeltaVDrift(ind,kFALSE),kFALSE); // left + calibSDD->SetDeltaVDrift(ind, precal->GetDeltaVDrift(ind,kTRUE ),kTRUE); // right + calibSDD->SetADCtokeV(ind,precal->GetADCtokeV(ind)); + } + } + else for (int ind=kSDDoffsID;indSetModuleTimeZero(ind,0); + // + Bool_t save = kFALSE; + for (int imd=GetNModules();imd--;) { + AliITSAlignMille2Module* md = GetMilleModule(imd); + if (!md->IsSDD()) continue; + if (md->IsFreeDOF(AliITSAlignMille2Module::kDOFT0) || + md->IsFreeDOF(AliITSAlignMille2Module::kDOFDVL) || + md->IsFreeDOF(AliITSAlignMille2Module::kDOFDVR)) save = kTRUE; + // + for (int is=0;isGetNSensitiveVolumes();is++) { + int ind = md->GetSensVolIndex(is); + float t0 = calibSDD->GetTimeZero(ind) + md->GetParVal(AliITSAlignMille2Module::kDOFT0); + double dvL = md->GetParVal(AliITSAlignMille2Module::kDOFDVL); + double dvR = md->GetParVal(AliITSAlignMille2Module::kDOFDVR); + if (!calibSDD->IsVDCorrMult()) { // save as additive correction + dvL *= 1e4; + dvR *= 1e4; + // + double conv = 1; + if (isPreCalMult) conv = 6.4; // convert multiplicative precal correction to additive + dvL += calibSDD->GetDeltaVDrift(ind,kFALSE)*conv; + dvR += calibSDD->GetDeltaVDrift(ind,kTRUE)*conv; + } + else { // save as multipicative correction + double conv = 1; + if (!isPreCalMult) conv = 1./6.4; // convert additive precal correction to multiplicative + dvL += calibSDD->GetDeltaVDrift(ind,kFALSE)*conv; + dvR += calibSDD->GetDeltaVDrift(ind,kTRUE)*conv; + } + // + calibSDD->SetModuleTimeZero(ind, t0); + calibSDD->SetDeltaVDrift(ind, dvL, kFALSE); // left side correction + calibSDD->SetDeltaVDrift(ind, dvR, kTRUE); // right side correction + } + } + // + if (!save) { + AliInfo("No free parameters for SDD calibration, nothing to save"); + delete calibSDD; + calibSDD = 0; + } + // + return calibSDD; +} + +//_______________________________________________________________________________________ +Int_t AliITSAlignMille2::ReloadInitCalib(TList *userInfo) +{ + // Use provided UserInfo to + // load the initial calib parameters (geometry, SDD response...) + // Can be used if set of data was processed with different calibration + // + if (!userInfo) { + AliInfo("Reloading of the Calibration parameters was called with empty userInfo"); + return 1; + } + if (ProcessUserInfo(userInfo)) { + AliInfo("Error in processing user info"); + userInfo->Print(); + exit(1); + } + return ReloadInitCalib(); +} + +//_______________________________________________________________________________________ +Int_t AliITSAlignMille2::ReloadInitCalib() +{ + // Load the initial calib parameters (geometry, SDD response...) + // Can be used if set of data was processed with different calibration + // + AliInfo(Form("SameInitDelta: %d | SameInitGeom: %d",TestBit(kSameInitDeltasBit), TestBit(kSameInitGeomBit))); + // 1st cache original matrices + if (!(TestBit(kSameInitDeltasBit) && TestBit(kSameInitGeomBit))) { // need to reload geometry + // + if (CacheMatricesOrig()) { + AliInfo("Failed to cache new initial geometry"); + exit(1); + } + // RS : commented because we don't need to reload prealignment deltas, they are already loaded + // then reload the prealignment geometry + // if (LoadDeltas(fPreDeltaPath,fPrealignment)) { + // AliInfo(Form("Failed to reload the prealigned geometry %s",fPreDeltaPath.Data())); + // exit(1); + // } + // + if (fPrealignment && ApplyToGeometry()) { + AliInfo(Form("Failed re-apply prealigned geometry %s",fPreDeltaPath.Data())); + exit(1); + } + // + // usually no need to re-cache the prealignment geometry, it was not changed + if (fCacheMatrixCurr.GetEntriesFast() != fCacheMatrixOrig.GetEntriesFast()) { + // CacheMatricesCurr(); + AliInfo(Form("Failed to cache the prealigned geometry %s",fPreDeltaPath.Data())); + exit(1); + } + } + else ResetBit(kSameInitDeltasBit); + // + // reload initial SDD response + if (!TestBit(kSameInitSDDRespBit)) { + if (LoadSDDResponse(fIniSDDRespPath, fIniRespSDD) ) { + AliInfo(Form("Failed to load new SDD response %s",fIniSDDRespPath.Data())); + exit(1); + } + } + else ResetBit(kSameInitSDDRespBit); + // + // reload initial SDD vdrift + if (!TestBit(kSameInitSDDVDriftBit)) { + if (LoadSDDVDrift(fIniSDDVDriftPath, fIniVDriftSDD) ) { + AliInfo(Form("Failed to load new SDD VDrift %s",fIniSDDVDriftPath.Data())); + exit(1); + } + } + else ResetBit(kSameInitSDDRespBit); + // + // reload SDD corr.map + if (!TestBit(kSameInitSDDCorrMapBit)) { + if (LoadSDDCorrMap(fIniSDDCorrMapPath, fIniCorrMapSDD) ) { + AliInfo(Form("Failed to load new SDD Correction Map %s",fIniSDDCorrMapPath.Data())); + exit(1); + } + } + else ResetBit(kSameInitSDDRespBit); + // + // reload diamond info + if (!TestBit(kSameDiamondBit)) { + if (LoadDiamond(fDiamondPath) ) { + AliInfo(Form("Failed to load new Diamond constraint %s",fDiamondPath.Data())); + exit(1); + } + } + else ResetBit(kSameInitSDDRespBit); + // + return 0; +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::JacobianPosGloLoc(int locid,double* jacobian) +{ + // calculate the locid row of the jacobian for transformation of the local coordinate to global at current point + TGeoHMatrix* mat = GetSensorCurrMatrixSID(fCurrentSensID); + const Double_t dpar = 1e-2; + double sav = fMeasLoc[locid]; + fMeasLoc[locid] += dpar; + mat->LocalToMaster(fMeasLoc,jacobian); + fMeasLoc[locid] = sav; // recover original value + for (int i=3;i--;) jacobian[i] = (jacobian[i]-fMeasGlo[i])/dpar; // the transformation is linear!!! +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::TieSDDVDriftsLR(AliITSAlignMille2Module* mod) +{ + // impose equality of Left/Right sides VDrift correction for SDD + ResetLocalEquation(); + if ( (mod->IsFreeDOF(AliITSAlignMille2Module::kDOFDVL) + mod->IsFreeDOF(AliITSAlignMille2Module::kDOFDVR))==1) { + AliError("Left/Right VDrift equality is requested for SDD module with only one side VDrift free"); + mod->Print(); + return; + } + if (mod->GetParOffset(AliITSAlignMille2Module::kDOFDVL)>=0) SetGlobalDerivative(mod->GetParOffset(AliITSAlignMille2Module::kDOFDVL), 1.); + if (mod->GetParOffset(AliITSAlignMille2Module::kDOFDVR)>=0) SetGlobalDerivative(mod->GetParOffset(AliITSAlignMille2Module::kDOFDVR), -1.); + AddConstraint(fGlobalDerivatives, 0, 1e-12); + // +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::ProcessSDDPointInfo(const AliTrackPoint* pnt,Int_t sID, Int_t pntID) +{ + // extract the drift information from SDD track point + // + fDriftTime0[pntID] = fIniRespSDD ? fIniRespSDD->GetTimeZero(sID) : 0.; + double tdif = pnt->GetDriftTime() - fDriftTime0[pntID]; + if (tdif<0) tdif = 1; + // + // VDrift extraction + double vdrift=0,vdrift0=0; + Bool_t sddSide = kFALSE; + int sID0 = 2*(sID-kSDDoffsID); + double zanode = -999; + // + if (fIniVDriftSDD) { // SDD VDrift object is provided, use the vdrift from it + AliITSDriftSpeedArraySDD* drarr; + double vdR,vdL,xlR,xlL; + // sometimes xlocal on right side is negative due to the wrong calibration, need to test both hypothesis + double xlabs = TMath::Abs(fMeasLoc[kX]); + drarr = (AliITSDriftSpeedArraySDD*)fIniVDriftSDD->At(sID0); // left side, xloc>0 + zanode = fSegmentationSDD->GetAnodeFromLocal(xlabs,fMeasLoc[kZ]); + vdL = drarr->GetDriftSpeed(0, zanode); + if (fIniRespSDD) { + double corr = fIniRespSDD->GetDeltaVDrift(sID, kFALSE); + if (fIniRespSDD->IsVDCorrMult()) vdL *= (1+corr); + else vdL += corr; + } + xlL = (fSegmentationSDD->Dx() - vdL*tdif)*1e-4; + // + drarr = (AliITSDriftSpeedArraySDD*)fIniVDriftSDD->At(sID0+1); // right side, xloc<0 + zanode = fSegmentationSDD->GetAnodeFromLocal(-xlabs,fMeasLoc[kZ]) - 256; + vdR = drarr->GetDriftSpeed(0, zanode); + if (fIniRespSDD) { + double corr = fIniRespSDD->GetDeltaVDrift(sID, kTRUE); + if (fIniRespSDD->IsVDCorrMult()) vdR *= (1+corr); + else vdR += corr; + } + xlR = -(fSegmentationSDD->Dx() - vdR*tdif)*1e-4; + // + if (TMath::Abs(xlL-fMeasLoc[kX])Dx()*1e-4 - TMath::Abs(fMeasLoc[kX]))/tdif; + sddSide = fMeasLoc[kX]<0; // 0 = left (xloc>0) ; 1 = right (xloc<1) + } + // + if (fPreVDriftSDD) { // use imposed vdrift as a starting point + zanode = fSegmentationSDD->GetAnodeFromLocal(0.5-sddSide,fMeasLoc[kZ]); + if (sddSide) zanode -= 256; + vdrift = ((AliITSDriftSpeedArraySDD*)fPreVDriftSDD->At(sID0+sddSide))->GetDriftSpeed(0, zanode)*1e-4; + } + // + if (vdrift<0) vdrift = 0; + vdrift0 = vdrift; + // at this point we have vdrift and t0 used to create the original point. + // see if precalibration was provided + if (fPreRespSDD) { + float t0Upd = fPreRespSDD->GetTimeZero(sID); + double corr = fPreRespSDD->GetDeltaVDrift(sID, sddSide); + if (fPreRespSDD->IsVDCorrMult()) vdrift *= 1+corr; // right side (xloc<0) may have different correction + else vdrift += corr*1e-4; + // + // if IniRespSDD was used, it should be subtracted back, since it is accounted in the PreResp + if (fIniVDriftSDD&&fIniRespSDD && (fPreVDriftSDD==0)) { + double corr1 = fIniRespSDD->GetDeltaVDrift(sID, sddSide); + if (fIniRespSDD->IsVDCorrMult()) vdrift *= (1-corr1); + else vdrift -= corr1*1e-4; + } + tdif = pnt->GetDriftTime() - t0Upd; + // correct Xlocal + fMeasLoc[0] = fSegmentationSDD->Dx()*1e-4 - vdrift*tdif; + if (sddSide) fMeasLoc[0] = -fMeasLoc[0]; + fDriftTime0[pntID] = t0Upd; + } + // + if (fPreCorrMapSDD) { // apply correction map + fMeasLoc[0] += fPreCorrMapSDD->GetCorrection(sID,fMeasLoc[2],fMeasLoc[0]); + } + + // TEMPORARY CORRECTION (if provided) --------------<<< + fDriftSpeed[pntID] = sddSide ? -vdrift : vdrift; + fDriftSpeed0[pntID] = sddSide ? -vdrift0 : vdrift0; + // + // printf("#%d: t:%+e x:%+e v:%+e: side:%d\n",pntID,fDriftTime0[pntID],fMeasLoc[0],fDriftSpeed[pntID],sddSide); +} + +//_______________________________________________________________________________________ +AliITSAlignMille2Module* AliITSAlignMille2::CreateVertexModule() +{ + // creates dummy module for vertex constraint + TGeoHMatrix mt; + AliITSAlignMille2Module* mod = new AliITSAlignMille2Module(kVtxSensID,kVtxSensVID,"VTX",&mt,0,0); + fMilleModule.AddAtAndExpand(mod,fNModules); + mod->SetGeomParamsGlobal(fUseGlobalDelta); + fDiamondModID = fNModules; + mod->SetUniqueID(fNModules++); + mod->SetNotInConf(kTRUE); + return mod; + // +} + +//_______________________________________________________________________________________ +AliCDBEntry* AliITSAlignMille2::GetCDBEntry(const char* path) +{ + // return object from the OCDB + AliCDBEntry *entry = 0; + AliInfo(Form("Loading object %s",path)); + AliCDBManager* man = AliCDBManager::Instance(); + AliCDBId* cdbId = AliCDBId::MakeFromString(path); + if (!cdbId) { + AliError("Failed to create cdbId"); + return 0; + } + // + AliCDBStorage* stor = man->GetDefaultStorage(); + if (!stor && !man->GetRaw()) man->SetDefaultStorage("raw://"); + if (man->GetRaw()) man->SetRun(fRunID>0 ? fRunID : cdbId->GetFirstRun()); + if (stor) { + TString tp = stor->GetType(); + if (tp.Contains("alien",TString::kIgnoreCase) && !gGrid) TGrid::Connect("alien:"); + } + entry = man->Get(cdbId->GetPath(),cdbId->GetFirstRun(),cdbId->GetVersion(),cdbId->GetSubVersion()); + // entry = man->Get( *cdbId ); + man->ClearCache(); + // + delete cdbId; + return entry; + // +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::SetVertexConstraint(const AliESDVertex* vtx) +{ + // set vertex for constraint + if (!vtx) return; + // + double cmat[6]; + float cmatF[6]; + vtx->GetCovMatrix(cmat); + AliITSAlignMille2Module* diamMod = GetMilleModuleByVID(kVtxSensVID); + if (diamMod) { + cmat[0] *= diamMod->GetSigmaXFactor()*diamMod->GetSigmaXFactor(); + cmat[2] *= diamMod->GetSigmaYFactor()*diamMod->GetSigmaYFactor(); + cmat[5] *= diamMod->GetSigmaZFactor()*diamMod->GetSigmaZFactor(); + cmat[1] *= diamMod->GetSigmaXFactor()*diamMod->GetSigmaYFactor(); + cmat[3] *= diamMod->GetSigmaXFactor()*diamMod->GetSigmaZFactor(); + cmat[4] *= diamMod->GetSigmaYFactor()*diamMod->GetSigmaZFactor(); + } + cmatF[0] = cmat[0]; // xx + cmatF[1] = cmat[1]; // xy + cmatF[2] = cmat[3]; // xz + cmatF[3] = cmat[2]; // yy + cmatF[4] = cmat[4]; // yz + cmatF[5] = cmat[5]; // zz + + fDiamond.SetXYZ(vtx->GetX(),vtx->GetY(),vtx->GetZ(), cmatF); + // + Double_t t0 = cmat[2]*cmat[5] - cmat[4]*cmat[4]; + Double_t t1 = cmat[1]*cmat[5] - cmat[3]*cmat[4]; + Double_t t2 = cmat[1]*cmat[4] - cmat[2]*cmat[3]; + Double_t det = cmat[0]*t0 - cmat[1]*t1 + cmat[3]*t2; + if (TMath::Abs(det)<1e-36) { + vtx->Print(); + AliFatal("Vertex constraint cov.matrix is singular"); + } + cmatF[0] = t0/det; + cmatF[1] = -t1/det; + cmatF[2] = t2/det; + cmatF[3] = (cmat[0]*cmat[5] - cmat[3]*cmat[3])/det; + cmatF[4] = (cmat[1]*cmat[3] - cmat[0]*cmat[4])/det; + cmatF[5] = (cmat[0]*cmat[2] - cmat[1]*cmat[1])/det; + fDiamondI.SetXYZ(vtx->GetX(),vtx->GetY(),vtx->GetZ(), cmatF); + fVertexSet = kTRUE; + // +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::ConvertDeltas() +{ + // convert prealignment deltas from old geometry to new one + // NOTE: the target geometry must be loaded at time this method is called + // + // NOTE: This method can be ONLY used when as a prealignment deltas those used for the production + // of trackpoints (e.g. extracted from the UserInfo). + // The prealignment deltas provided by user via config file must be already converted to target geometry: + // this can be done externally using the macro ConvertDeltas.C + // + // delta_j_new = delta_j_old * Xj_old * Xj_new^-1 + // where X = Prod{delta_i,i=j-1:0} M_j + // with j - the level of the alignable volume in the hierarchy, M - corresponding ideal matrix + // Note that delta_j * Xj is equal to final (misaligned) matrix of corresponding geometry, G_j. + // Since this method is used ONLY in the case where the prealignment deltas are equal to production deltas, + // we have already loaded G_j_old in the fConvAlgMatOld (filled in the CacheMatricesOrig) + // Hence, delta_j_new = G_j_old * Xj_new^-1 + // + AliInfo("Converting deltas from initial to target geometry"); + int nMatOld = fConvAlgMatOld.GetEntriesFast(); // number of alignable matrices + TClonesArray* deltArrNew = new TClonesArray("AliAlignObjParams",10); + // + TGeoHMatrix dmPar; + int nDelNew = 0; + // + for (int im=0;imGetTitle(); + UShort_t vID = AliITSAlignMille2Module::GetVolumeIDFromSymname(algname.Data()); + // + // build X_new >>> + TGeoHMatrix* parent = mtGjold; + TGeoHMatrix xNew; + int parID; + while ( (parID=parent->GetUniqueID()-1)>=0 ) { + parent = (TGeoHMatrix*)fConvAlgMatOld[parID]; + AliAlignObjParams* deltaPar = ConvFindDelta(deltArrNew,parent->GetTitle()); + if (deltaPar) deltaPar->GetMatrix(dmPar); xNew *= dmPar; + } + AliGeomManager::GetOrigGlobalMatrix(algname,dmPar); // ideal matrix of new geometry + xNew *= dmPar; + // build X_new <<< + // + dmPar = *mtGjold; + dmPar *= xNew.Inverse(); + new((*deltArrNew)[nDelNew++]) AliAlignObjParams(algname.Data(),vID,dmPar,kTRUE); + // + } + delete fPrealignment; + fPrealignment = deltArrNew; + // + // we don't need anymore old matrices + fConvAlgMatOld.Delete(); + // +} + +//_______________________________________________________________________________________ +void AliITSAlignMille2::ConvSortHierarchically(TObjArray& matArr) +{ + // Used only for the deltas conversion from one geometry to another + // Sort the matrices according to hiearachy (coarse -> fine) + // + int nmat = matArr.GetEntriesFast(); + // + for (int i=0;iSetUniqueID(0); + for (int j=i;j--;) { + TGeoHMatrix* matJ = (TGeoHMatrix*) matArr[j]; + if (ConvIsJParentOfI(matI,matJ)) { matI->SetUniqueID(j+1); break; } + } + } +} + +//_______________________________________________________________________________________ +Bool_t AliITSAlignMille2::ConvIsJParentOfI(const TGeoHMatrix* matI,const TGeoHMatrix* matJ) const +{ + // Used only for the deltas conversion from one geometry to another + // True if matJ is higher in hierarchy than + // + TString nmI = matI->GetTitle(); + TString nmJ = matJ->GetTitle(); + // + int nlrI = nmI.CountChar('/'); + int nlrJ = nmJ.CountChar('/'); + if (nlrJ>=nlrI) return kFALSE; + // + // special case of SPD sectors + if (nmI.BeginsWith("ITS/SPD1") && nmJ.BeginsWith("ITS/SPD0") && nlrJ==2) nmJ.ReplaceAll("SPD0","SPD1"); + return (nmI.BeginsWith(nmJ)) ? kTRUE:kFALSE; + // +} + +//_______________________________________________________________________________________ +AliAlignObjParams* AliITSAlignMille2::ConvFindDelta(const TClonesArray* arrDelta,const TString& algname) const +{ + // find the delta for given module + if (!arrDelta) return 0; + AliAlignObjParams* delta = 0; + int nDeltas = arrDelta->GetEntries(); + for (int id=0;idAt(id); + if (algname==delta->GetSymName()) break; + delta = 0; + } + return delta; +}