X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=MUON%2FAliMUONAlignment.cxx;h=f1278d81f1429379334121256ccb3884704ae9bc;hb=f8caeea5c1ff2b08c22161298b18b7e5307708fc;hp=4092759f94820ce3331bf21b972c0844b74a41a5;hpb=c25af45ba08a1599ea69b567595a48db545e0c27;p=u%2Fmrichter%2FAliRoot.git diff --git a/MUON/AliMUONAlignment.cxx b/MUON/AliMUONAlignment.cxx index 4092759f948..f1278d81f14 100644 --- a/MUON/AliMUONAlignment.cxx +++ b/MUON/AliMUONAlignment.cxx @@ -13,16 +13,16 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* $Id$ */ +/* $Id: AliMUONAlignment.cxx 51000 2011-08-08 17:58:17Z ivana $ */ //----------------------------------------------------------------------------- /// \class AliMUONAlignment -/// Alignment class for the ALICE DiMuon spectrometer +/// Alignment class for the ALICE DiMuon spectrometer /// -/// MUON specific alignment class which interface to AliMillepede. +/// MUON specific alignment class which interface to AliMillepede. /// For each track ProcessTrack calculates the local and global derivatives /// at each cluster and fill the corresponding local equations. Provide methods -/// for fixing or constraining detection elements for best results. +/// for fixing or constraining detection elements for best results. /// /// \author Bruce Becker, Javier Castillo //----------------------------------------------------------------------------- @@ -35,1022 +35,1293 @@ #include "AliMUONGeometryModuleTransformer.h" #include "AliMUONGeometryDetElement.h" #include "AliMUONGeometryBuilder.h" -#include "AliMUONConstants.h" -#include "AliMillepede.h" +#include "AliMillePede2.h" #include "AliMpExMap.h" +#include "AliMpExMapIterator.h" +#include "AliAlignObjMatrix.h" #include "AliLog.h" -#include "TMath.h" -#include "TSystem.h" +#include +#include +#include +#include /// \cond CLASSIMP ClassImp(AliMUONAlignment) /// \endcond - Int_t AliMUONAlignment::fgNDetElem = 4*2+4*2+18*2+26*2+26*2; - Int_t AliMUONAlignment::fgNDetElemCh[10] = {4,4,4,4,18,18,26,26,26,26}; - Int_t AliMUONAlignment::fgSNDetElemCh[10] = {4,8,12,16,34,52,78,104,130,156}; - Int_t AliMUONAlignment::fgNParCh = 3; - Int_t AliMUONAlignment::fgNTrkMod = 16; - Int_t AliMUONAlignment::fgNCh = 10; - Int_t AliMUONAlignment::fgNSt = 5; +//_____________________________________________________________________ +// static variables +const Int_t AliMUONAlignment::fgNDetElemCh[AliMUONAlignment::fgNCh] = { 4, 4, 4, 4, 18, 18, 26, 26, 26, 26 }; +const Int_t AliMUONAlignment::fgSNDetElemCh[AliMUONAlignment::fgNCh+1] = { 0, 4, 8, 12, 16, 34, 52, 78, 104, 130, 156 }; -AliMUONAlignment::AliMUONAlignment() +//_____________________________________________________________________ +/// self initialized array, used for adding constraints +class Array +{ + + public: + + /// contructor + Array( void ) + { + for( Int_t i=0; i < AliMUONAlignment::fNGlobal; ++i ) + { values[i] = 0; } + } + + /// array + Double_t values[AliMUONAlignment::fNGlobal]; + + private: + + /// Not implemented + Array(const Array& ); + + /// Not implemented + Array& operator = (const Array& ); + +}; + +//_____________________________________________________________________ +AliMUONAlignment::AliMUONAlignment() : TObject(), - fBFieldOn(kTRUE), - fStartFac(16.), - fResCutInitial(100.), - fResCut(100.), - fMillepede(0), - fTrackParamAtCluster(0), - fTrack(0), - fCluster(0), - fTrackParam(0), - fNGlobal(fgNDetElem*fgNParCh), - fNLocal(4), - fNStdDev(3), - fDetElemId(0), - fDetElemNumber(0), - fPhi(0.), - fCosPhi(1.), - fSinPhi(0.), - fTransform(0) + fInitialized( kFALSE ), + fRunNumber( 0 ), + fBFieldOn( kTRUE ), + fStartFac( 256 ), + fResCutInitial( 100 ), + fResCut( 100 ), + fMillepede( 0L ), + fCluster( 0L ), + fNStdDev( 3 ), + fDetElemNumber( 0 ), + fTrackRecord(), + fTransform( 0 ), + fGeoCombiTransInverse() { - /// Default constructor, setting define alignment parameters - fSigma[0] = 1.0e-1; + /// constructor + fSigma[0] = 1.5e-1; fSigma[1] = 1.0e-2; - fDoF[0] = kTRUE; fDoF[1] = kTRUE; fDoF[2] = kTRUE; - fAllowVar[0] = 0.05; fAllowVar[1] = 0.05; fAllowVar[2] = 0.001; - - AliInfo(Form("fAllowVar[0]: %f\t fAllowVar[1]: %f\t fPhi: %f\t fgNDetElem: %i\t fNGlobal: %i\t fNLocal: %i",fAllowVar[0],fAllowVar[1],fPhi,fgNDetElem,fNGlobal,fNLocal)); + // default allowed variations + fAllowVar[0] = 0.5; // x + fAllowVar[1] = 0.5; // y + fAllowVar[2] = 0.01; // phi_z + fAllowVar[3] = 5; // z + + // initialize millepede + fMillepede = new AliMillePede2(); - fMillepede = new AliMillepede(); + // initialize degrees of freedom + // by default all parameters are free + for( Int_t iPar = 0; iPar < fNGlobal; ++iPar ) + { fGlobalParameterStatus[iPar] = kFreeParId; } - Init(fNGlobal, fNLocal, fNStdDev); + // initialize local equations + for(int i=0; iInitMille(nGlobal,nLocal,nStdDev,fResCut,fResCutInitial); - Bool_t bStOnOff[5] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE}; - Bool_t bSpecLROnOff[2] = {kTRUE,kTRUE}; + /// initialize + /** + initialize millipede + must be called after necessary detectors have been fixed, + but before constrains are added and before global parameters initial value are set + */ + if( fInitialized ) + { AliFatal( "Millepede already initialized" ); } - AllowVariations(bStOnOff); + // assign proper groupID to free parameters + Int_t nGlobal = 0; + for( Int_t iPar = 0; iPar < fNGlobal; ++iPar ) + { - // Fix parameters or add constraints here - for (Int_t iSt=0; iSt<5; iSt++) - if (!bStOnOff[iSt]) FixStation(iSt+1); + if( fGlobalParameterStatus[iPar] == kFixedParId ) + { + // fixed parameters are left unchanged + continue; - FixHalfSpectrometer(bStOnOff,bSpecLROnOff); + } else if( fGlobalParameterStatus[iPar] == kFreeParId || fGlobalParameterStatus[iPar] == kGroupBaseId ) { - ResetConstraints(); - - // Define global constrains to be applied - // X, Y, P, XvsZ, YvsZ, PvsZ, XvsY, YvsY, PvsY - Bool_t bVarXYT[9] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE}; - Bool_t bDetTLBR[4] = {kFALSE,kTRUE,kFALSE,kTRUE}; - // AddConstraints(bStOnOff,bVarXYT,bDetTLBR,bSpecLROnOff); + // free parameters or first element of group are assigned a new group id + fGlobalParameterStatus[iPar] = nGlobal++; + continue; - // Other possible way to add constrains - bVarXYT[0] = kFALSE; bVarXYT[1] = kFALSE; bVarXYT[2] = kTRUE; - bDetTLBR[0] = kFALSE; bDetTLBR[1] = kTRUE; bDetTLBR[2] = kFALSE; bDetTLBR[3] = kFALSE; -// AddConstraints(bStOnOff,bVarXYT,bDetTLBR); + } else if( fGlobalParameterStatus[iPar] < kGroupBaseId ) { - bVarXYT[0] = kTRUE; bVarXYT[1] = kTRUE; bVarXYT[2] = kFALSE; - // AddConstraints(bStOnOff,bVarXYT); - - // Set iterations - if (fStartFac>1) fMillepede->SetIterations(fStartFac); -} + // get detector element id from status, get chamber parameter id + const Int_t iDeBase( kGroupBaseId - 1 - fGlobalParameterStatus[iPar] ); + const Int_t iParBase = iPar%fgNParCh; -void AliMUONAlignment::FixStation(Int_t iSt){ - /// Fix all detection elements of station iSt - Int_t iDetElemFirst = (iSt>1) ? fgSNDetElemCh[2*(iSt-1)-1] : 0; - Int_t iDetElemLast = fgSNDetElemCh[2*(iSt)-1]; - for (Int_t i = iDetElemFirst; i < iDetElemLast; i++){ - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } -} + // check + if( iDeBase < 0 || iDeBase >= iPar/fgNParCh ) + { AliFatal( Form( "Group for parameter index %i has wrong base detector element: %i", iPar, iDeBase ) ); } + + // assign identical group id to current + fGlobalParameterStatus[iPar] = fGlobalParameterStatus[iDeBase*fgNParCh + iParBase]; + AliInfo( Form( "Parameter %i grouped to detector %i (%s)", iPar, iDeBase, GetParameterMaskString( 1<=1 && iCh<=4){ - if ((lDetElemNumber==1 || lDetElemNumber==2) && !lSpecLROnOff[0]){ // From track crossings - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - if ((lDetElemNumber==0 || lDetElemNumber==3) && !lSpecLROnOff[1]){ // From track crossings - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - } - if (iCh>=5 && iCh<=6){ - if ((lDetElemNumber>=5&&lDetElemNumber<=13) && !lSpecLROnOff[0]){ - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - if (((lDetElemNumber>=0&&lDetElemNumber<=4) || - (lDetElemNumber>=14&&lDetElemNumber<=17)) && !lSpecLROnOff[1]){ - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - } - if (iCh>=7 && iCh<=10){ - if ((lDetElemNumber>=7&&lDetElemNumber<=19) && !lSpecLROnOff[0]){ - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - if (((lDetElemNumber>=0&&lDetElemNumber<=6) || - (lDetElemNumber>=20&&lDetElemNumber<=25)) && !lSpecLROnOff[1]){ - FixParameter(i*fgNParCh+0, 0.0); - FixParameter(i*fgNParCh+1, 0.0); - FixParameter(i*fgNParCh+2, 0.0); - } - } - } } -} -void AliMUONAlignment::SetNonLinear(Bool_t *lStOnOff,Bool_t *lVarXYT){ - /// Set non linear parameter flag selected stations and degrees of freedom - for (Int_t i = 0; i < fgNDetElem; i++){ - Int_t iCh=0; - for (iCh=1; iCh<=fgNCh; iCh++){ - if (iInitMille( fNGlobal, fNLocal, fNStdDev, fResCut, fResCutInitial, fGlobalParameterStatus ); + fInitialized = kTRUE; + + // some debug output + for( Int_t iPar = 0; iPar < fgNParCh; ++iPar ) + { AliInfo( Form( "fAllowVar[%i]= %f", iPar, fAllowVar[iPar] ) ); } + + // set allowed variations for all parameters + for( Int_t iDet = 0; iDet < fgNDetElem; ++iDet ) + { + for( Int_t iPar = 0; iPar < fgNParCh; ++iPar ) + { fMillepede->SetParSigma( iDet*fgNParCh + iPar, fAllowVar[iPar] ); } } + + // Set iterations + if (fStartFac>1) fMillepede->SetIterations(fStartFac); + } -void AliMUONAlignment::AddConstraints(Bool_t *lStOnOff,Bool_t *lVarXYT){ - /// Add constraint equations for selected stations and degrees of freedom - for (Int_t i = 0; i < fgNDetElem; i++){ - Int_t iCh=0; - for (iCh=1; iCh<=fgNCh; iCh++){ - if (iGetRecord() ) fMillepede->GetRecord()->Reset(); + + // get number of track parameters + Int_t nTrackParam = track->GetTrackParamAtCluster()->GetEntries(); + + Bool_t first( kTRUE ); + for( Int_t iTrackParam = 0; iTrackParam < nTrackParam; ++iTrackParam ) + { + + // get new pointers + AliMUONTrackParam* trackParam( (AliMUONTrackParam *) track->GetTrackParamAtCluster()->At(iTrackParam) ); + if( !trackParam ) continue; + + AliMUONVCluster* cluster = trackParam->GetClusterPtr(); + if( !cluster ) continue; + + // fill local variables for this position --> one measurement + FillDetElemData( cluster ); + FillRecPointData( cluster ); + FillTrackParamData( trackParam ); + + if( first ) + { + + // for first valid cluster, save track position as "starting" values + first = kFALSE; + + fTrackPos0[0] = fTrackPos[0]; + fTrackPos0[1] = fTrackPos[1]; + fTrackPos0[2] = fTrackPos[2]; + fTrackSlope0[0] = fTrackSlope[0]; + fTrackSlope0[1] = fTrackSlope[1]; + } - Int_t iSt = lStOnOff[(iCh-1)/2] ? (iCh+1)/2 : 0; - if (iSt){ - if (lVarXYT[0]) { // X constraint - fConstraintX[i*fgNParCh+0]=1.0; - } - if (lVarXYT[1]) { // Y constraint - fConstraintY[i*fgNParCh+1]=1.0; - } - if (lVarXYT[2]) { // T constraint - fConstraintP[i*fgNParCh+2]=1.0; - } + + // 'inverse' (GlobalToLocal) rotation matrix + const Double_t* r( fGeoCombiTransInverse.GetRotationMatrix() ); + + // calculate measurements + if( fBFieldOn ) + { + + // use residuals (cluster - track) for measurement + fMeas[0] = r[0]*(fClustPos[0] - fTrackPos[0]) + r[1]*(fClustPos[1] - fTrackPos[1]); + fMeas[1] = r[3]*(fClustPos[0] - fTrackPos[0]) + r[4]*(fClustPos[1] - fTrackPos[1]); + + } else { + + // use cluster position for measurement + fMeas[0] = ( r[0]*fClustPos[0] + r[1]*fClustPos[1] ); + fMeas[1] = ( r[3]*fClustPos[0] + r[4]*fClustPos[1] ); + } + + // Set local equations + LocalEquationX(); + LocalEquationY(); + } - if (lVarXYT[0]) { // X constraint - AddConstraint(fConstraintX,0.0); - } - if (lVarXYT[1]) { // Y constraint - AddConstraint(fConstraintY,0.0); - } - if (lVarXYT[2]) { // T constraint - AddConstraint(fConstraintP,0.0); + + // copy track record + fMillepede->SetRecordRun(fRunNumber); + fMillepede->SetRecordWeight(weight); + fTrackRecord = *fMillepede->GetRecord(); + + // save record data + if( doAlignment ) + { + fMillepede->SaveRecordData(); } + + // return record + return &fTrackRecord; + } -void AliMUONAlignment::AddConstraints(Bool_t *lStOnOff,Bool_t *lVarXYT, Bool_t *lDetTLBR, Bool_t *lSpecLROnOff){ - /// Add constraint equations for selected stations, degrees of freedom and detector half - Double_t lDetElemLocX = 0.; - Double_t lDetElemLocY = 0.; - Double_t lDetElemLocZ = 0.; - Double_t lDetElemGloX = 0.; - Double_t lDetElemGloY = 0.; - Double_t lDetElemGloZ = 0.; - Double_t lMeanY = 0.; - Double_t lSigmaY = 0.; - Double_t lMeanZ = 0.; - Double_t lSigmaZ = 0.; - Int_t lNDetElem = 0; - for (Int_t i = 0; i < fgNDetElem; i++){ - Int_t iCh=0; - for (iCh=1; iCh<=fgNCh; iCh++){ - if (iLocal2Global(lDetElemId,lDetElemLocX,lDetElemLocY,lDetElemLocZ, - lDetElemGloX,lDetElemGloY,lDetElemGloZ); - if (iCh>=1 && iCh<=4){ - if ((lDetElemNumber==1 || lDetElemNumber==2) && lSpecLROnOff[0]){ // From track crossings - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - if ((lDetElemNumber==0 || lDetElemNumber==3) && lSpecLROnOff[1]){ // From track crossings - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - } - if (iCh>=5 && iCh<=6){ - if ((lDetElemNumber>=5&&lDetElemNumber<=13) && lSpecLROnOff[0]){ - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - if (((lDetElemNumber>=0&&lDetElemNumber<=4) || - (lDetElemNumber>=14&&lDetElemNumber<=17)) && lSpecLROnOff[1]){ - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - } - if (iCh>=7 && iCh<=10){ - if ((lDetElemNumber>=7&&lDetElemNumber<=19) && lSpecLROnOff[0]){ - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - if (((lDetElemNumber>=0&&lDetElemNumber<=6) || - (lDetElemNumber>=20&&lDetElemNumber<=25)) && lSpecLROnOff[1]){ - lMeanY += lDetElemGloY; - lSigmaY += lDetElemGloY*lDetElemGloY; - lMeanZ += lDetElemGloZ; - lSigmaZ += lDetElemGloZ*lDetElemGloZ; - lNDetElem++; - } - } - } - } - lMeanY /= lNDetElem; - lSigmaY /= lNDetElem; - lSigmaY = TMath::Sqrt(lSigmaY-lMeanY*lMeanY); - lMeanZ /= lNDetElem; - lSigmaZ /= lNDetElem; - lSigmaZ = TMath::Sqrt(lSigmaZ-lMeanZ*lMeanZ); - AliInfo(Form("Used %i DetElem, MeanZ= %f , SigmaZ= %f", lNDetElem,lMeanZ,lSigmaZ)); +//______________________________________________________________________________ +void AliMUONAlignment::ProcessTrack( AliMillePedeRecord* trackRecord ) +{ + /// process track record + if( !trackRecord ) return; + + // make sure record storage is initialized + if( !fMillepede->GetRecord() ) fMillepede->InitDataRecStorage(); + + // copy content + *fMillepede->GetRecord() = *trackRecord; + + // save record + fMillepede->SaveRecordData(); + + return; - for (Int_t i = 0; i < fgNDetElem; i++){ - Int_t iCh=0; - for (iCh=1; iCh<=fgNCh; iCh++){ - if (iLocal2Global(lDetElemId,lDetElemLocX,lDetElemLocY,lDetElemLocZ, - lDetElemGloX,lDetElemGloY,lDetElemGloZ); - if (lVarXYT[0]) { // X constraint - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintXT,0); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintXL,0); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintXB,0); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintXR,0); // Right half - } - if (lVarXYT[1]) { // Y constraint - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintYT,1); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintYL,1); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintYB,1); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintYR,1); // Right half - } - if (lVarXYT[2]) { // P constraint - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintPT,2); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintPL,2); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintPB,2); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintPR,2); // Right half - } - if (lVarXYT[3]) { // X-Z shearing - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintXZT,0,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintXZL,0,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintXZB,0,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintXZR,0,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Right half - } - if (lVarXYT[4]) { // Y-Z shearing - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintYZT,1,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintYZL,1,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintYZB,1,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintYZR,1,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Right half - } - if (lVarXYT[5]) { // P-Z rotation - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintPZT,2,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintPZL,2,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintPZB,2,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintPZR,2,(lDetElemGloZ-lMeanZ)/lSigmaZ); // Right half - } - if (lVarXYT[6]) { // X-Y shearing - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintXYT,0,(lDetElemGloY-lMeanY)/lSigmaY); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintXYL,0,(lDetElemGloY-lMeanY)/lSigmaY); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintXYB,0,(lDetElemGloY-lMeanY)/lSigmaY); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintXYR,0,(lDetElemGloY-lMeanY)/lSigmaY); // Right half - } - if (lVarXYT[7]) { // Y-Y scaling - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintYYT,1,(lDetElemGloY-lMeanY)/lSigmaY); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintYYL,1,(lDetElemGloY-lMeanY)/lSigmaY); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintYYB,1,(lDetElemGloY-lMeanY)/lSigmaY); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintYYR,1,(lDetElemGloY-lMeanY)/lSigmaY); // Right half - } - if (lVarXYT[8]) { // P-Y rotation - if (lDetTLBR[0]) ConstrainT(i,iCh,fConstraintPYT,2,(lDetElemGloY-lMeanY)/lSigmaY); // Top half - if (lDetTLBR[1]) ConstrainL(i,iCh,fConstraintPYL,2,(lDetElemGloY-lMeanY)/lSigmaY); // Left half - if (lDetTLBR[2]) ConstrainB(i,iCh,fConstraintPYB,2,(lDetElemGloY-lMeanY)/lSigmaY); // Bottom half - if (lDetTLBR[3]) ConstrainR(i,iCh,fConstraintPYR,2,(lDetElemGloY-lMeanY)/lSigmaY); // Right half - } - } - } - if (lVarXYT[0]) { // X constraint - if (lDetTLBR[0]) AddConstraint(fConstraintXT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintXL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintXB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintXR,0.0); // Right half - } - if (lVarXYT[1]) { // Y constraint - if (lDetTLBR[0]) AddConstraint(fConstraintYT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintYL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintYB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintYR,0.0); // Right half - } - if (lVarXYT[2]) { // T constraint - if (lDetTLBR[0]) AddConstraint(fConstraintPT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintPL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintPB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintPR,0.0); // Right half - } - if (lVarXYT[3]) { // X-Z constraint - if (lDetTLBR[0]) AddConstraint(fConstraintXZT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintXZL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintXZB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintXZR,0.0); // Right half - } - if (lVarXYT[4]) { // Y-Z constraint - if (lDetTLBR[0]) AddConstraint(fConstraintYZT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintYZL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintYZB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintYZR,0.0); // Right half - } - if (lVarXYT[5]) { // P-Z constraint - if (lDetTLBR[0]) AddConstraint(fConstraintPZT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintPZL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintPZB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintPZR,0.0); // Right half - } - if (lVarXYT[6]) { // X-Y constraint - if (lDetTLBR[0]) AddConstraint(fConstraintXYT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintXYL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintXYB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintXYR,0.0); // Right half - } - if (lVarXYT[7]) { // Y-Y constraint - if (lDetTLBR[0]) AddConstraint(fConstraintYYT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintYYL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintYYB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintYYR,0.0); // Right half - } - if (lVarXYT[8]) { // P-Y constraint - if (lDetTLBR[0]) AddConstraint(fConstraintPYT,0.0); // Top half - if (lDetTLBR[1]) AddConstraint(fConstraintPYL,0.0); // Left half - if (lDetTLBR[2]) AddConstraint(fConstraintPYB,0.0); // Bottom half - if (lDetTLBR[3]) AddConstraint(fConstraintPYR,0.0); // Right half - } } -void AliMUONAlignment::ConstrainT(Int_t lDetElem, Int_t lCh, Double_t *lConstraintT, Int_t iVar, Double_t /*lWeight*/){ - /// Set constrain equation for top half of spectrometer - Int_t lDetElemNumber = (lCh==1) ? lDetElem : lDetElem-fgSNDetElemCh[lCh-2]; - if (lCh>=1 && lCh<=4){ - if (lDetElemNumber==0 || lDetElemNumber==1){ // From track crossings - lConstraintT[lDetElem*fgNParCh+iVar]=1.0; - } - } - if (lCh>=5 && lCh<=6){ - if (lDetElemNumber>=0&&lDetElemNumber<=9){ - lConstraintT[lDetElem*fgNParCh+iVar]=1.0; - } - } - if (lCh>=7 && lCh<=10){ - if (lDetElemNumber>=0&&lDetElemNumber<=13){ - lConstraintT[lDetElem*fgNParCh+iVar]=1.0; - } +//_____________________________________________________________________ +void AliMUONAlignment::FixAll( UInt_t mask ) +{ + /// fix parameters matching mask, for all chambers + AliInfo( Form( "Fixing %s for all detector elements", GetParameterMaskString( mask ).Data() ) ); + + // fix all stations + for( Int_t i = 0; i < fgNDetElem; ++i ) + { + if( mask & ParX ) FixParameter(i, 0); + if( mask & ParY ) FixParameter(i, 1); + if( mask & ParTZ ) FixParameter(i, 2); + if( mask & ParZ ) FixParameter(i, 3); } + } -void AliMUONAlignment::ConstrainL(Int_t lDetElem, Int_t lCh, Double_t *lConstraintL, Int_t iVar, Double_t lWeight){ - /// Set constrain equation for left half of spectrometer - Int_t lDetElemNumber = (lCh==1) ? lDetElem : lDetElem-fgSNDetElemCh[lCh-2]; - if (lCh>=1 && lCh<=4){ - if (lDetElemNumber==1 || lDetElemNumber==2){ // From track crossings - lConstraintL[lDetElem*fgNParCh+iVar]=lWeight; - } - } - if (lCh>=5 && lCh<=6){ - if (lDetElemNumber>=5&&lDetElemNumber<=13){ - lConstraintL[lDetElem*fgNParCh+iVar]=lWeight; - } - } - if (lCh>=7 && lCh<=10){ - if (lDetElemNumber>=7&&lDetElemNumber<=19){ - lConstraintL[lDetElem*fgNParCh+iVar]=lWeight; - } +//_____________________________________________________________________ +void AliMUONAlignment::FixChamber( Int_t iCh, UInt_t mask ) +{ + /// fix parameters matching mask, for all detector elements in a given chamber, counting from 1 + + // check boundaries + if( iCh < 1 || iCh > 10 ) + { AliFatal( Form( "Invalid chamber index %i", iCh ) ); } + + // get first and last element + const Int_t iDetElemFirst = fgSNDetElemCh[iCh-1]; + const Int_t iDetElemLast = fgSNDetElemCh[iCh]; + for( Int_t i = iDetElemFirst; i < iDetElemLast; ++i ) + { + + AliInfo( Form( "Fixing %s for detector element %i", GetParameterMaskString(mask).Data(), i ) ); + + if( mask & ParX ) FixParameter(i, 0); + if( mask & ParY ) FixParameter(i, 1); + if( mask & ParTZ ) FixParameter(i, 2); + if( mask & ParZ ) FixParameter(i, 3); + } } -void AliMUONAlignment::ConstrainB(Int_t lDetElem, Int_t lCh, Double_t *lConstraintB, Int_t iVar, Double_t /*lWeight*/){ - /// Set constrain equation for bottom half of spectrometer - Int_t lDetElemNumber = (lCh==1) ? lDetElem : lDetElem-fgSNDetElemCh[lCh-2]; - if (lCh>=1 && lCh<=4){ - if (lDetElemNumber==2 || lDetElemNumber==3){ // From track crossings - lConstraintB[lDetElem*fgNParCh+iVar]=1.0; - } - } - if (lCh>=5 && lCh<=6){ - if ((lDetElemNumber>=9&&lDetElemNumber<=17) || - (lDetElemNumber==0)){ - lConstraintB[lDetElem*fgNParCh+iVar]=1.0; - } - } - if (lCh>=7 && lCh<=10){ - if ((lDetElemNumber>=13&&lDetElemNumber<=25) || - (lDetElemNumber==0)){ - lConstraintB[lDetElem*fgNParCh+iVar]=1.0; - } - } +//_____________________________________________________________________ +void AliMUONAlignment::FixDetElem( Int_t iDetElemId, UInt_t mask ) +{ + /// fix parameters matching mask, for a given detector element, counting from 0 + const Int_t iDet( GetDetElemNumber( iDetElemId ) ); + if ( mask & ParX ) FixParameter(iDet, 0); + if ( mask & ParY ) FixParameter(iDet, 1); + if ( mask & ParTZ ) FixParameter(iDet, 2); + if ( mask & ParZ ) FixParameter(iDet, 3); + } -void AliMUONAlignment::ConstrainR(Int_t lDetElem, Int_t lCh, Double_t *lConstraintR, Int_t iVar, Double_t lWeight){ - /// Set constrain equation for right half of spectrometer - Int_t lDetElemNumber = (lCh==1) ? lDetElem : lDetElem-fgSNDetElemCh[lCh-2]; - if (lCh>=1 && lCh<=4){ - if (lDetElemNumber==0 || lDetElemNumber==3){ // From track crossings - lConstraintR[lDetElem*fgNParCh+iVar]=lWeight; +//_____________________________________________________________________ +void AliMUONAlignment::FixHalfSpectrometer( const Bool_t *lChOnOff, UInt_t sidesMask, UInt_t mask ) +{ + + /// Fix parameters matching mask for all detectors in selected chambers and selected sides of the spectrometer + for( Int_t i = 0; i < fgNDetElem; ++i ) + { + + // get chamber matching detector + const Int_t iCh( GetChamberId(i) ); + if( !lChOnOff[iCh-1] ) continue; + + // get detector element in chamber + Int_t lDetElemNumber = i-fgSNDetElemCh[iCh-1]; + + // skip detector if its side is off + // stations 1 and 2 + if( iCh>=1 && iCh<=4 ) + { + if( lDetElemNumber == 0 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber == 1 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber == 2 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber == 3 && !( sidesMask & SideBottomRight ) ) continue; } - } - if (lCh>=5 && lCh<=6){ - if ((lDetElemNumber>=0&&lDetElemNumber<=4) || - (lDetElemNumber>=14&&lDetElemNumber<=17)){ - lConstraintR[lDetElem*fgNParCh+iVar]=lWeight; + + // station 3 + if (iCh>=5 && iCh<=6) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 4 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber >= 5 && lDetElemNumber <= 10 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber >= 11 && lDetElemNumber <= 13 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber >= 14 && lDetElemNumber <= 17 && !( sidesMask & SideBottomRight ) ) continue; } - } - if (lCh>=7 && lCh<=10){ - if ((lDetElemNumber>=0&&lDetElemNumber<=6) || - (lDetElemNumber>=20&&lDetElemNumber<=25)){ - lConstraintR[lDetElem*fgNParCh+iVar]=lWeight; + + // stations 4 and 5 + if (iCh>=7 && iCh<=10) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 6 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber >= 7 && lDetElemNumber <= 13 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber >= 14 && lDetElemNumber <= 19 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber >= 20 && lDetElemNumber <= 25 && !( sidesMask & SideBottomRight ) ) continue; } - } -} -void AliMUONAlignment::ResetConstraints(){ - /// Reset all constraint equations - for (Int_t i = 0; i < fgNDetElem; i++){ - fConstraintX[i*fgNParCh+0]=0.0; - fConstraintX[i*fgNParCh+1]=0.0; - fConstraintX[i*fgNParCh+2]=0.0; - fConstraintY[i*fgNParCh+0]=0.0; - fConstraintY[i*fgNParCh+1]=0.0; - fConstraintY[i*fgNParCh+2]=0.0; - fConstraintP[i*fgNParCh+0]=0.0; - fConstraintP[i*fgNParCh+1]=0.0; - fConstraintP[i*fgNParCh+2]=0.0; - fConstraintXT[i*fgNParCh+0]=0.0; - fConstraintXT[i*fgNParCh+1]=0.0; - fConstraintXT[i*fgNParCh+2]=0.0; - fConstraintYT[i*fgNParCh+0]=0.0; - fConstraintYT[i*fgNParCh+1]=0.0; - fConstraintYT[i*fgNParCh+2]=0.0; - fConstraintPT[i*fgNParCh+0]=0.0; - fConstraintPT[i*fgNParCh+1]=0.0; - fConstraintPT[i*fgNParCh+2]=0.0; - fConstraintXZT[i*fgNParCh+0]=0.0; - fConstraintXZT[i*fgNParCh+1]=0.0; - fConstraintXZT[i*fgNParCh+2]=0.0; - fConstraintYZT[i*fgNParCh+0]=0.0; - fConstraintYZT[i*fgNParCh+1]=0.0; - fConstraintYZT[i*fgNParCh+2]=0.0; - fConstraintPZT[i*fgNParCh+0]=0.0; - fConstraintPZT[i*fgNParCh+1]=0.0; - fConstraintPZT[i*fgNParCh+2]=0.0; - fConstraintXYT[i*fgNParCh+0]=0.0; - fConstraintXYT[i*fgNParCh+1]=0.0; - fConstraintXYT[i*fgNParCh+2]=0.0; - fConstraintYYT[i*fgNParCh+0]=0.0; - fConstraintYYT[i*fgNParCh+1]=0.0; - fConstraintYYT[i*fgNParCh+2]=0.0; - fConstraintPYT[i*fgNParCh+0]=0.0; - fConstraintPYT[i*fgNParCh+1]=0.0; - fConstraintPYT[i*fgNParCh+2]=0.0; - fConstraintXL[i*fgNParCh+0]=0.0; - fConstraintXL[i*fgNParCh+1]=0.0; - fConstraintXL[i*fgNParCh+2]=0.0; - fConstraintYL[i*fgNParCh+0]=0.0; - fConstraintYL[i*fgNParCh+1]=0.0; - fConstraintYL[i*fgNParCh+2]=0.0; - fConstraintPL[i*fgNParCh+0]=0.0; - fConstraintPL[i*fgNParCh+1]=0.0; - fConstraintPL[i*fgNParCh+2]=0.0; - fConstraintXZL[i*fgNParCh+0]=0.0; - fConstraintXZL[i*fgNParCh+1]=0.0; - fConstraintXZL[i*fgNParCh+2]=0.0; - fConstraintYZL[i*fgNParCh+0]=0.0; - fConstraintYZL[i*fgNParCh+1]=0.0; - fConstraintYZL[i*fgNParCh+2]=0.0; - fConstraintPZL[i*fgNParCh+0]=0.0; - fConstraintPZL[i*fgNParCh+1]=0.0; - fConstraintPZL[i*fgNParCh+2]=0.0; - fConstraintXYL[i*fgNParCh+0]=0.0; - fConstraintXYL[i*fgNParCh+1]=0.0; - fConstraintXYL[i*fgNParCh+2]=0.0; - fConstraintYYL[i*fgNParCh+0]=0.0; - fConstraintYYL[i*fgNParCh+1]=0.0; - fConstraintYYL[i*fgNParCh+2]=0.0; - fConstraintPYL[i*fgNParCh+0]=0.0; - fConstraintPYL[i*fgNParCh+1]=0.0; - fConstraintPYL[i*fgNParCh+2]=0.0; - fConstraintXB[i*fgNParCh+0]=0.0; - fConstraintXB[i*fgNParCh+1]=0.0; - fConstraintXB[i*fgNParCh+2]=0.0; - fConstraintYB[i*fgNParCh+0]=0.0; - fConstraintYB[i*fgNParCh+1]=0.0; - fConstraintYB[i*fgNParCh+2]=0.0; - fConstraintPB[i*fgNParCh+0]=0.0; - fConstraintPB[i*fgNParCh+1]=0.0; - fConstraintPB[i*fgNParCh+2]=0.0; - fConstraintXZB[i*fgNParCh+0]=0.0; - fConstraintXZB[i*fgNParCh+1]=0.0; - fConstraintXZB[i*fgNParCh+2]=0.0; - fConstraintYZB[i*fgNParCh+0]=0.0; - fConstraintYZB[i*fgNParCh+1]=0.0; - fConstraintYZB[i*fgNParCh+2]=0.0; - fConstraintPZB[i*fgNParCh+0]=0.0; - fConstraintPZB[i*fgNParCh+1]=0.0; - fConstraintPZB[i*fgNParCh+2]=0.0; - fConstraintXYB[i*fgNParCh+0]=0.0; - fConstraintXYB[i*fgNParCh+1]=0.0; - fConstraintXYB[i*fgNParCh+2]=0.0; - fConstraintYYB[i*fgNParCh+0]=0.0; - fConstraintYYB[i*fgNParCh+1]=0.0; - fConstraintYYB[i*fgNParCh+2]=0.0; - fConstraintPYB[i*fgNParCh+0]=0.0; - fConstraintPYB[i*fgNParCh+1]=0.0; - fConstraintPYB[i*fgNParCh+2]=0.0; - fConstraintXR[i*fgNParCh+0]=0.0; - fConstraintXR[i*fgNParCh+1]=0.0; - fConstraintXR[i*fgNParCh+2]=0.0; - fConstraintYR[i*fgNParCh+0]=0.0; - fConstraintYR[i*fgNParCh+1]=0.0; - fConstraintYR[i*fgNParCh+2]=0.0; - fConstraintPR[i*fgNParCh+0]=0.0; - fConstraintPR[i*fgNParCh+1]=0.0; - fConstraintPR[i*fgNParCh+2]=0.0; - fConstraintXZR[i*fgNParCh+0]=0.0; - fConstraintXZR[i*fgNParCh+1]=0.0; - fConstraintXZR[i*fgNParCh+2]=0.0; - fConstraintYZR[i*fgNParCh+0]=0.0; - fConstraintYZR[i*fgNParCh+1]=0.0; - fConstraintYZR[i*fgNParCh+2]=0.0; - fConstraintPZR[i*fgNParCh+0]=0.0; - fConstraintPZR[i*fgNParCh+1]=0.0; - fConstraintPZR[i*fgNParCh+2]=0.0; - fConstraintPZR[i*fgNParCh+0]=0.0; - fConstraintPZR[i*fgNParCh+1]=0.0; - fConstraintPZR[i*fgNParCh+2]=0.0; - fConstraintXYR[i*fgNParCh+0]=0.0; - fConstraintXYR[i*fgNParCh+1]=0.0; - fConstraintXYR[i*fgNParCh+2]=0.0; - fConstraintYYR[i*fgNParCh+0]=0.0; - fConstraintYYR[i*fgNParCh+1]=0.0; - fConstraintYYR[i*fgNParCh+2]=0.0; - fConstraintPYR[i*fgNParCh+0]=0.0; - fConstraintPYR[i*fgNParCh+1]=0.0; - fConstraintPYR[i*fgNParCh+2]=0.0; + // detector is accepted, fix it + FixDetElem( i, mask ); + } -} -void AliMUONAlignment::AddConstraint(Double_t *par, Double_t value) { - /// Constrain equation defined by par to value - fMillepede->SetGlobalConstraint(par, value); - AliInfo("Adding constraint"); } -void AliMUONAlignment::InitGlobalParameters(Double_t *par) { - /// Initialize global parameters with par array - fMillepede->SetGlobalParameters(par); - AliInfo("Init Global Parameters"); -} - -void AliMUONAlignment::FixParameter(Int_t iPar, Double_t value) { - /// Parameter iPar is encourage to vary in [-value;value]. - /// If value == 0, parameter is fixed - fMillepede->SetParSigma(iPar, value); - if (value==0) AliInfo(Form("Parameter %i Fixed", iPar)); +//______________________________________________________________________ +void AliMUONAlignment::FixParameter( Int_t iPar ) +{ + + /// fix a given parameter, counting from 0 + if( fInitialized ) + { AliFatal( "Millepede already initialized" ); } + + fGlobalParameterStatus[iPar] = kFixedParId; + } -void AliMUONAlignment::ResetLocalEquation() + +//_____________________________________________________________________ +void AliMUONAlignment::ReleaseChamber( Int_t iCh, UInt_t mask ) { - /// Reset the derivative vectors - for(int i=0; i 10 ) + { AliFatal( Form( "Invalid chamber index %i", iCh ) ); } + + // get first and last element + const Int_t iDetElemFirst = fgSNDetElemCh[iCh-1]; + const Int_t iDetElemLast = fgSNDetElemCh[iCh]; + for( Int_t i = iDetElemFirst; i < iDetElemLast; ++i ) + { + + AliInfo( Form( "Releasing %s for detector element %i", GetParameterMaskString(mask).Data(), i ) ); + + if( mask & ParX ) ReleaseParameter(i, 0); + if( mask & ParY ) ReleaseParameter(i, 1); + if( mask & ParTZ ) ReleaseParameter(i, 2); + if( mask & ParZ ) ReleaseParameter(i, 3); + } } -void AliMUONAlignment::AllowVariations(Bool_t *bStOnOff) { - /// Set allowed variation for selected stations based on fDoF and fAllowVar - for (Int_t iSt=1; iSt<=5; iSt++) { - if (bStOnOff[iSt-1]) { - Int_t iDetElemFirst = (iSt>1) ? fgSNDetElemCh[2*(iSt-1)-1] : 0; - Int_t iDetElemLast = fgSNDetElemCh[2*(iSt)-1]; - for (int i=0; iSetNonLinear(iPar); - AliInfo(Form("Parameter %i set to non linear", iPar)); +//______________________________________________________________________ +void AliMUONAlignment::ReleaseParameter( Int_t iPar ) +{ + + /// release a given parameter, counting from 0 + if( fInitialized ) + { AliFatal( "Millepede already initialized" ); } + + fGlobalParameterStatus[iPar] = kFreeParId; + } -void AliMUONAlignment::LocalEquationX() { - /// Define local equation for current track and cluster in x coor. measurement - // set local derivatives - SetLocalDerivative(0, fCosPhi); - SetLocalDerivative(1, fCosPhi * (fTrackPos[2] - fTrackPos0[2])); - SetLocalDerivative(2, fSinPhi); - SetLocalDerivative(3, fSinPhi * (fTrackPos[2] - fTrackPos0[2])); +//_____________________________________________________________________ +void AliMUONAlignment::GroupChamber( Int_t iCh, UInt_t mask ) +{ + /// group parameters matching mask for all detector elements in a given chamber, counting from 1 + if( iCh < 1 || iCh > 10 ) + { AliFatal( Form( "Invalid chamber index %i", iCh ) ); } - // set global derivatives - SetGlobalDerivative(fDetElemNumber*fgNParCh+0, -fCosPhi); - SetGlobalDerivative(fDetElemNumber*fgNParCh+1, -fSinPhi); - if (fBFieldOn){ - SetGlobalDerivative(fDetElemNumber*fgNParCh+2, - -fSinPhi*(fTrackPos[0]-fDetElemPos[0]) - +fCosPhi*(fTrackPos[1]-fDetElemPos[1])); - } - else { - SetGlobalDerivative(fDetElemNumber*fgNParCh+2, - -fSinPhi*(fTrackPos0[0]+fTrackSlope0[0]* - (fTrackPos[2]-fTrackPos0[2])-fDetElemPos[0]) - +fCosPhi*(fTrackPos0[1]+fTrackSlope0[1]* - (fTrackPos[2]-fTrackPos0[2])-fDetElemPos[1])); - } + const Int_t detElemMin = 100*iCh; + const Int_t detElemMax = 100*iCh + fgNDetElemCh[iCh]-1; + GroupDetElems( detElemMin, detElemMax, mask ); - fMillepede->SetLocalEquation(fGlobalDerivatives, fLocalDerivatives, fMeas[0], fSigma[0]); } -void AliMUONAlignment::LocalEquationY() { - /// Define local equation for current track and cluster in y coor. measurement - // set local derivatives - SetLocalDerivative(0,-fSinPhi); - SetLocalDerivative(1,-fSinPhi * (fTrackPos[2] - fTrackPos0[2])); - SetLocalDerivative(2, fCosPhi); - SetLocalDerivative(3, fCosPhi * (fTrackPos[2] - fTrackPos0[2])); +//_____________________________________________________________________ +void AliMUONAlignment::GroupDetElems( Int_t detElemMin, Int_t detElemMax, UInt_t mask ) +{ + /// group parameters matching mask for all detector elements between min and max + // check number of detector elements + const Int_t nDetElem = detElemMax - detElemMin + 1; + if( nDetElem<2 ) + { AliFatal( Form( "Requested group of DEs %d-%d contains less than 2 DE's", detElemMin, detElemMax ) ); } - // set global derivatives - SetGlobalDerivative(fDetElemNumber*fgNParCh+0, fSinPhi); - SetGlobalDerivative(fDetElemNumber*fgNParCh+1, -fCosPhi); - if (fBFieldOn){ - SetGlobalDerivative(fDetElemNumber*fgNParCh+2, - -fCosPhi*(fTrackPos[0]-fDetElemPos[0]) - -fSinPhi*(fTrackPos[1]-fDetElemPos[1])); + // create list + Int_t* detElemList = new int[nDetElem]; + for( Int_t i = 0; i < nDetElem; ++i ) + { detElemList[i] = detElemMin+i; } + + // group + GroupDetElems( detElemList, nDetElem, mask ); + delete[] detElemList; + +} + +//_____________________________________________________________________ +void AliMUONAlignment::GroupDetElems( Int_t* detElemList, Int_t nDetElem, UInt_t mask ) +{ + /// group parameters matching mask for all detector elements in list + if( fInitialized ) + { AliFatal( "Millepede already initialized" ); } + + const Int_t iDeBase( GetDetElemNumber( detElemList[0] ) ); + for( Int_t i = 0; i < nDetElem; ++i ) + { + const Int_t iDeCurrent( GetDetElemNumber( detElemList[i] ) ); + if( mask & ParX ) fGlobalParameterStatus[iDeCurrent*fgNParCh + 0] = (i==0) ? kGroupBaseId : (kGroupBaseId-iDeBase-1); + if( mask & ParY ) fGlobalParameterStatus[iDeCurrent*fgNParCh + 1] = (i==0) ? kGroupBaseId : (kGroupBaseId-iDeBase-1); + if( mask & ParTZ ) fGlobalParameterStatus[iDeCurrent*fgNParCh + 2] = (i==0) ? kGroupBaseId : (kGroupBaseId-iDeBase-1); + if( mask & ParZ ) fGlobalParameterStatus[iDeCurrent*fgNParCh + 3] = (i==0) ? kGroupBaseId : (kGroupBaseId-iDeBase-1); + + if( i== 0 ) AliInfo( Form( "Creating new group for detector %i and variable %s", detElemList[i], GetParameterMaskString( mask ).Data() ) ); + else AliInfo( Form( "Adding detector element %i to current group", detElemList[i] ) ); } - else { - SetGlobalDerivative(fDetElemNumber*fgNParCh+2, - -fCosPhi*(fTrackPos0[0]+fTrackSlope0[0]* - (fTrackPos[2]-fTrackPos0[2])-fDetElemPos[0]) - -fSinPhi*(fTrackPos0[1]+fTrackSlope0[1]* - (fTrackPos[2]-fTrackPos0[2])-fDetElemPos[1])); + +} + +//______________________________________________________________________ +void AliMUONAlignment::SetChamberNonLinear( Int_t iCh, UInt_t mask ) +{ + /// Set parameters matching mask as non linear, for all detector elements in a given chamber, counting from 1 + const Int_t iDetElemFirst = fgSNDetElemCh[iCh-1]; + const Int_t iDetElemLast = fgSNDetElemCh[iCh]; + for( Int_t i = iDetElemFirst; i < iDetElemLast; ++i ) + { + + if( mask & ParX ) SetParameterNonLinear(i, 0); + if( mask & ParY ) SetParameterNonLinear(i, 1); + if( mask & ParTZ ) SetParameterNonLinear(i, 2); + if( mask & ParZ ) SetParameterNonLinear(i, 3); + } - fMillepede->SetLocalEquation(fGlobalDerivatives, fLocalDerivatives, fMeas[1], fSigma[1]); } -void AliMUONAlignment::FillRecPointData() { - /// Get information of current cluster - fClustPos[0] = fCluster->GetX(); - fClustPos[1] = fCluster->GetY(); - fClustPos[2] = fCluster->GetZ(); - fTransform->Global2Local(fDetElemId,fClustPos[0],fClustPos[1],fClustPos[2], - fClustPosLoc[0],fClustPosLoc[1],fClustPosLoc[2]); +//_____________________________________________________________________ +void AliMUONAlignment::SetDetElemNonLinear( Int_t iDetElemId, UInt_t mask ) +{ + /// Set parameters matching mask as non linear, for a given detector element, counting from 0 + const Int_t iDet( GetDetElemNumber( iDetElemId ) ); + if ( mask & ParX ) SetParameterNonLinear(iDet, 0); + if ( mask & ParY ) SetParameterNonLinear(iDet, 1); + if ( mask & ParTZ ) SetParameterNonLinear(iDet, 2); + if ( mask & ParZ ) SetParameterNonLinear(iDet, 3); + } -void AliMUONAlignment::FillTrackParamData() { - /// Get information of current track at current cluster - fTrackPos[0] = fTrackParam->GetNonBendingCoor(); - fTrackPos[1] = fTrackParam->GetBendingCoor(); - fTrackPos[2] = fTrackParam->GetZ(); - fTrackSlope[0] = fTrackParam->GetNonBendingSlope(); - fTrackSlope[1] = fTrackParam->GetBendingSlope(); - fTransform->Global2Local(fDetElemId,fTrackPos[0],fTrackPos[1],fTrackPos[2], - fTrackPosLoc[0],fTrackPosLoc[1],fTrackPosLoc[2]); +//______________________________________________________________________ +void AliMUONAlignment::SetParameterNonLinear( Int_t iPar ) +{ + /// Set nonlinear flag for parameter iPar + if( !fInitialized ) + { AliFatal( "Millepede not initialized" ); } + + fMillepede->SetNonLinear( iPar ); + AliInfo( Form( "Parameter %i set to non linear", iPar ) ); } -void AliMUONAlignment::FillDetElemData() { - /// Get information of current detection element - Double_t lDetElemLocX = 0.; - Double_t lDetElemLocY = 0.; - Double_t lDetElemLocZ = 0.; - fDetElemId = fCluster->GetDetElemId(); - fDetElemNumber = fDetElemId%100; - for (int iCh=0; iChLocal2Global(fDetElemId,lDetElemLocX,lDetElemLocY,lDetElemLocZ, - fDetElemPos[0],fDetElemPos[1],fDetElemPos[2]); + + if( mask & ParX ) AddConstraint(fConstraintX.values,0.0); + if( mask & ParY ) AddConstraint(fConstraintY.values,0.0); + if( mask & ParTZ ) AddConstraint(fConstraintTZ.values,0.0); + if( mask & ParZ ) AddConstraint(fConstraintZ.values,0.0); + } -void AliMUONAlignment::ProcessTrack(AliMUONTrack * track) { - /// Process track; Loop over clusters and set local equations - fTrack = track; - // get tclones arrays. - fTrackParamAtCluster = fTrack->GetTrackParamAtCluster(); - - // get size of arrays - Int_t nTrackParam = fTrackParamAtCluster->GetEntries(); - AliDebug(1,Form("Number of track param entries : %i ", nTrackParam)); - - for(Int_t iCluster=0; iClusterGetTrackParamAtCluster()->At(iCluster); - fCluster = fTrackParam->GetClusterPtr(); - if (!fCluster || !fTrackParam) continue; - // fill local variables for this position --> one measurement - FillDetElemData(); - FillRecPointData(); - FillTrackParamData(); -// if (fDetElemId<500) continue; - fTrackPos0[0] = fTrackPos[0]; - fTrackPos0[1] = fTrackPos[1]; - fTrackPos0[2] = fTrackPos[2]; - fTrackSlope0[0] = fTrackSlope[0]; - fTrackSlope0[1] = fTrackSlope[1]; - break; - } +//______________________________________________________________________ +void AliMUONAlignment::AddConstraints(const Bool_t *lChOnOff, const Bool_t *lVarXYT, UInt_t sidesMask ) +{ + /* + questions: + - is there not redundancy/inconsistency between lDetTLBR and lSpecLROnOff ? shouldn't we use only lDetTLBR ? + - why is weight ignored for ConstrainT and ConstrainB + - why is there no constrain on z + */ + + /// Add constraint equations for selected chambers, degrees of freedom and detector half + Double_t lMeanY = 0.; + Double_t lSigmaY = 0.; + Double_t lMeanZ = 0.; + Double_t lSigmaZ = 0.; + Int_t lNDetElem = 0; - for(Int_t iCluster=0; iClusterGetTrackParamAtCluster()->At(iCluster); - fCluster = fTrackParam->GetClusterPtr(); - if (!fCluster || !fTrackParam) continue; - // fill local variables for this position --> one measurement - FillDetElemData(); - FillRecPointData(); - FillTrackParamData(); -// if (fDetElemId<500) continue; - AliDebug(1,Form("cluster: %i", iCluster)); - AliDebug(1,Form("x: %f\t y: %f\t z: %f\t DetElemID: %i\t ", fClustPos[0], fClustPos[1], fClustPos[2], fDetElemId)); - AliDebug(1,Form("fDetElemPos[0]: %f\t fDetElemPos[1]: %f\t fDetElemPos[2]: %f\t DetElemID: %i\t ", fDetElemPos[0],fDetElemPos[1],fDetElemPos[2], fDetElemId)); - - AliDebug(1,Form("Track Parameter: %i", iCluster)); - AliDebug(1,Form("x: %f\t y: %f\t z: %f\t slopex: %f\t slopey: %f", fTrackPos[0], fTrackPos[1], fTrackPos[2], fTrackSlope[0], fTrackSlope[1])); - AliDebug(1,Form("x0: %f\t y0: %f\t z0: %f\t slopex0: %f\t slopey0: %f", fTrackPos0[0], fTrackPos0[1], fTrackPos0[2], fTrackSlope0[0], fTrackSlope0[1])); - - fCosPhi = TMath::Cos(fPhi); - fSinPhi = TMath::Sin(fPhi); - if (fBFieldOn){ - fMeas[0] = fTrackPos[0] - fClustPos[0]; - fMeas[1] = fTrackPos[1] - fClustPos[1]; + for( Int_t i = 0; i < fgNDetElem; ++i ) + { + + // get chamber matching detector + const Int_t iCh( GetChamberId(i) ); + + // skip detector if chamber is off + if( lChOnOff[iCh-1] ) continue; + + // get detector element id from detector element number + const Int_t lDetElemNumber = i-fgSNDetElemCh[iCh-1]; + const Int_t lDetElemId = iCh*100+lDetElemNumber; + + // skip detector if its side is off + // stations 1 and 2 + if( iCh>=1 && iCh<=4 ) + { + if( lDetElemNumber == 0 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber == 1 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber == 2 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber == 3 && !( sidesMask & SideBottomRight ) ) continue; } - else { - fMeas[0] = - fClustPos[0]; - fMeas[1] = - fClustPos[1]; + + // station 3 + if (iCh>=5 && iCh<=6) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 4 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber >= 5 && lDetElemNumber <= 10 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber >= 11 && lDetElemNumber <= 13 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber >= 14 && lDetElemNumber <= 17 && !( sidesMask & SideBottomRight ) ) continue; } - AliDebug(1,Form("fMeas[0]: %f\t fMeas[1]: %f\t fSigma[0]: %f\t fSigma[1]: %f", fMeas[0], fMeas[1], fSigma[0], fSigma[1])); - // Set local equations - LocalEquationX(); - LocalEquationY(); + + // stations 4 and 5 + if (iCh>=7 && iCh<=10) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 6 && !( sidesMask & SideTopRight ) ) continue; + if( lDetElemNumber >= 7 && lDetElemNumber <= 13 && !( sidesMask & SideTopLeft ) ) continue; + if( lDetElemNumber >= 14 && lDetElemNumber <= 19 && !( sidesMask & SideBottomLeft ) ) continue; + if( lDetElemNumber >= 20 && lDetElemNumber <= 25 && !( sidesMask & SideBottomRight ) ) continue; + } + + // get global x, y and z position + Double_t lDetElemGloX = 0.; + Double_t lDetElemGloY = 0.; + Double_t lDetElemGloZ = 0.; + fTransform->Local2Global( lDetElemId, 0, 0, 0, lDetElemGloX, lDetElemGloY, lDetElemGloZ ); + + // increment mean Y, mean Z, sigmas and number of accepted detectors + lMeanY += lDetElemGloY; + lSigmaY += lDetElemGloY*lDetElemGloY; + lMeanZ += lDetElemGloZ; + lSigmaZ += lDetElemGloZ*lDetElemGloZ; + lNDetElem++; + } -} -void AliMUONAlignment::LocalFit(Int_t iTrack, Double_t *lTrackParam, Int_t lSingleFit) { - /// Call local fit for this tracks - Int_t iRes = fMillepede->LocalFit(iTrack,lTrackParam,lSingleFit); - if (iRes && !lSingleFit) { - fMillepede->SetNLocalEquations(fMillepede->GetNLocalEquations()+1); + // calculate mean values + lMeanY /= lNDetElem; + lSigmaY /= lNDetElem; + lSigmaY = TMath::Sqrt(lSigmaY-lMeanY*lMeanY); + lMeanZ /= lNDetElem; + lSigmaZ /= lNDetElem; + lSigmaZ = TMath::Sqrt(lSigmaZ-lMeanZ*lMeanZ); + AliInfo( Form( "Used %i DetElem, MeanZ= %f , SigmaZ= %f", lNDetElem,lMeanZ,lSigmaZ ) ); + + // create all possible arrays + Array fConstraintX[4]; //Array for constraint equation X + Array fConstraintY[4]; //Array for constraint equation Y + Array fConstraintP[4]; //Array for constraint equation P + Array fConstraintXZ[4]; //Array for constraint equation X vs Z + Array fConstraintYZ[4]; //Array for constraint equation Y vs Z + Array fConstraintPZ[4]; //Array for constraint equation P vs Z + + // do we really need these ? + Array fConstraintXY[4]; //Array for constraint equation X vs Y + Array fConstraintYY[4]; //Array for constraint equation Y vs Y + Array fConstraintPY[4]; //Array for constraint equation P vs Y + + // fill Bool_t sides array based on masks, for convenience + Bool_t lDetTLBR[4]; + lDetTLBR[0] = sidesMask & SideTop; + lDetTLBR[1] = sidesMask & SideLeft; + lDetTLBR[2] = sidesMask & SideBottom; + lDetTLBR[3] = sidesMask & SideRight; + + for( Int_t i = 0; i < fgNDetElem; ++i ) + { + + // get chamber matching detector + const Int_t iCh( GetChamberId(i) ); + + // skip detector if chamber is off + if( !lChOnOff[iCh-1] ) continue; + + // get detector element id from detector element number + const Int_t lDetElemNumber = i-fgSNDetElemCh[iCh-1]; + const Int_t lDetElemId = iCh*100+lDetElemNumber; + + // get global x, y and z position + Double_t lDetElemGloX = 0.; + Double_t lDetElemGloY = 0.; + Double_t lDetElemGloZ = 0.; + fTransform->Local2Global( lDetElemId, 0, 0, 0, lDetElemGloX, lDetElemGloY, lDetElemGloZ ); + + // loop over sides + for( Int_t iSide = 0; iSide < 4; iSide++ ) + { + + // skip if side is not selected + if( !lDetTLBR[iSide] ) continue; + + // skip detector if it is not in the selected side + // stations 1 and 2 + if( iCh>=1 && iCh<=4 ) + { + if( lDetElemNumber == 0 && !(iSide == 0 || iSide == 3) ) continue; // top-right + if( lDetElemNumber == 1 && !(iSide == 0 || iSide == 1) ) continue; // top-left + if( lDetElemNumber == 2 && !(iSide == 2 || iSide == 1) ) continue; // bottom-left + if( lDetElemNumber == 3 && !(iSide == 2 || iSide == 3) ) continue; // bottom-right + } + + // station 3 + if (iCh>=5 && iCh<=6) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 4 && !(iSide == 0 || iSide == 3) ) continue; // top-right + if( lDetElemNumber >= 5 && lDetElemNumber <= 9 && !(iSide == 0 || iSide == 1) ) continue; // top-left + if( lDetElemNumber >= 10 && lDetElemNumber <= 13 && !(iSide == 2 || iSide == 1) ) continue; // bottom-left + if( lDetElemNumber >= 14 && lDetElemNumber <= 17 && !(iSide == 2 || iSide == 3) ) continue; // bottom-right + } + + // stations 4 and 5 + if (iCh>=7 && iCh<=10) + { + if( lDetElemNumber >= 0 && lDetElemNumber <= 6 && !(iSide == 0 || iSide == 3) ) continue; // top-right + if( lDetElemNumber >= 7 && lDetElemNumber <= 13 && !(iSide == 0 || iSide == 1) ) continue; // top-left + if( lDetElemNumber >= 14 && lDetElemNumber <= 19 && !(iSide == 2 || iSide == 1) ) continue; // bottom-left + if( lDetElemNumber >= 20 && lDetElemNumber <= 25 && !(iSide == 2 || iSide == 3) ) continue; // bottom-right + } + + // constrain x + if( lVarXYT[0] ) fConstraintX[iSide].values[i*fgNParCh+0] = 1; + + // constrain y + if( lVarXYT[1] ) fConstraintY[iSide].values[i*fgNParCh+1] = 1; + + // constrain phi (rotation around z) + if( lVarXYT[2] ) fConstraintP[iSide].values[i*fgNParCh+2] = 1; + + // x-z shearing + if( lVarXYT[3] ) fConstraintXZ[iSide].values[i*fgNParCh+0] = (lDetElemGloZ-lMeanZ)/lSigmaZ; + + // y-z shearing + if( lVarXYT[4] ) fConstraintYZ[iSide].values[i*fgNParCh+1] = (lDetElemGloZ-lMeanZ)/lSigmaZ; + + // phi-z shearing + if( lVarXYT[5] ) fConstraintPZ[iSide].values[i*fgNParCh+2] = (lDetElemGloZ-lMeanZ)/lSigmaZ; + + // x-y shearing + if( lVarXYT[6] ) fConstraintXY[iSide].values[i*fgNParCh+0] = (lDetElemGloY-lMeanY)/lSigmaY; + + // y-y shearing + if( lVarXYT[7] ) fConstraintYY[iSide].values[i*fgNParCh+1] = (lDetElemGloY-lMeanY)/lSigmaY; + + // phi-y shearing + if( lVarXYT[8] ) fConstraintPY[iSide].values[i*fgNParCh+2] = (lDetElemGloY-lMeanY)/lSigmaY; + + } + + } + + // pass constraints to millepede + for( Int_t iSide = 0; iSide < 4; iSide++ ) + { + // skip if side is not selected + if( !lDetTLBR[iSide] ) continue; + + if( lVarXYT[0] ) AddConstraint(fConstraintX[iSide].values,0.0); + if( lVarXYT[1] ) AddConstraint(fConstraintY[iSide].values,0.0); + if( lVarXYT[2] ) AddConstraint(fConstraintP[iSide].values,0.0); + if( lVarXYT[3] ) AddConstraint(fConstraintXZ[iSide].values,0.0); + if( lVarXYT[4] ) AddConstraint(fConstraintYZ[iSide].values,0.0); + if( lVarXYT[5] ) AddConstraint(fConstraintPZ[iSide].values,0.0); + if( lVarXYT[6] ) AddConstraint(fConstraintXY[iSide].values,0.0); + if( lVarXYT[7] ) AddConstraint(fConstraintYY[iSide].values,0.0); + if( lVarXYT[8] ) AddConstraint(fConstraintPY[iSide].values,0.0); } + } -void AliMUONAlignment::GlobalFit(Double_t *parameters,Double_t *errors,Double_t *pulls) { - /// Call global fit; Global parameters are stored in parameters - fMillepede->GlobalFit(parameters,errors,pulls); +//______________________________________________________________________ +void AliMUONAlignment::InitGlobalParameters(Double_t *par) +{ + /// Initialize global parameters with par array + if( !fInitialized ) + { AliFatal( "Millepede is not initialized" ); } - AliInfo("Done fitting global parameters!"); - for (int iGlob=0; iGlobSetGlobalParameters(par); } -Double_t AliMUONAlignment::GetParError(Int_t iPar) { - /// Get error of parameter iPar - Double_t lErr = fMillepede->GetParError(iPar); - return lErr; +//______________________________________________________________________ +void AliMUONAlignment::SetAllowedVariation( Int_t iPar, Double_t value ) +{ + /// "Encouraged" variation for degrees of freedom + // check initialization + if( fInitialized ) + { AliFatal( "Millepede already initialized" ); } + + // check initialization + if( !(iPar >= 0 && iPar < fgNParCh ) ) + { AliFatal( Form( "Invalid index: %i", iPar ) ); } + + fAllowVar[iPar] = value; } -void AliMUONAlignment::PrintGlobalParameters() { - /// Print global parameters - fMillepede->PrintGlobalParameters(); +//______________________________________________________________________ +void AliMUONAlignment::SetSigmaXY(Double_t sigmaX, Double_t sigmaY) +{ + + /// Set expected measurement resolution + fSigma[0] = sigmaX; + fSigma[1] = sigmaY; + + // print + for( Int_t i=0; i<2; ++i ) + { AliInfo( Form( "fSigma[%i]=%f", i, fSigma[i] ) ); } + } -//_________________________________________________________________________ -TGeoCombiTrans AliMUONAlignment::ReAlign(const TGeoCombiTrans & transform, double *lMisAlignment) const +//_____________________________________________________ +void AliMUONAlignment::GlobalFit( Double_t *parameters, Double_t *errors, Double_t *pulls ) { - /// Realign given transformation by given misalignment and return the misaligned transformation - - Double_t cartMisAlig[3] = {0,0,0}; - Double_t angMisAlig[3] = {0,0,0}; -// const Double_t *trans = transform.GetTranslation(); -// TGeoRotation *rot; -// // check if the rotation we obtain is not NULL -// if (transform.GetRotation()) { -// rot = transform.GetRotation(); -// } -// else { -// rot = new TGeoRotation("rot"); -// } // default constructor. - - cartMisAlig[0] = -lMisAlignment[0]; - cartMisAlig[1] = -lMisAlignment[1]; - angMisAlig[2] = -lMisAlignment[2]*180./TMath::Pi(); - - TGeoTranslation deltaTrans(cartMisAlig[0], cartMisAlig[1], cartMisAlig[2]); - TGeoRotation deltaRot; - deltaRot.RotateX(angMisAlig[0]); - deltaRot.RotateY(angMisAlig[1]); - deltaRot.RotateZ(angMisAlig[2]); - TGeoCombiTrans deltaTransf(deltaTrans,deltaRot); - TGeoHMatrix newTransfMat = transform * deltaTransf; + /// Call global fit; Global parameters are stored in parameters + fMillepede->GlobalFit( parameters, errors, pulls ); + + AliInfo( "Done fitting global parameters" ); + for( int iDet=0; iDetPrintGlobalParameters(); } + +//_____________________________________________________ +Double_t AliMUONAlignment::GetParError(Int_t iPar) const +{ return fMillepede->GetParError(iPar); } + //______________________________________________________________________ -AliMUONGeometryTransformer * -AliMUONAlignment::ReAlign(const AliMUONGeometryTransformer * transformer, - double *misAlignments, Bool_t verbose) - +AliMUONGeometryTransformer* AliMUONAlignment::ReAlign( + const AliMUONGeometryTransformer * transformer, + const double *misAlignments, Bool_t ) { - ///////////////////////////////////////////////////////////////////// - // Takes the internal geometry module transformers, copies them + + /// Returns a new AliMUONGeometryTransformer with the found misalignments + /// applied. + + // Takes the internal geometry module transformers, copies them // and gets the Detection Elements from them. // Takes misalignment parameters and applies these // to the local transform of the Detection Element // Obtains the global transform by multiplying the module transformer - // transformation with the local transformation + // transformation with the local transformation // Applies the global transform to a new detection element // Adds the new detection element to a new module transformer // Adds the new module transformer to a new geometry transformer // Returns the new geometry transformer - Double_t lModuleMisAlignment[3] = {0.,0.,0.}; - Double_t lDetElemMisAlignment[3] = {0.,0.,0.}; - Int_t iDetElemId = 0; - Int_t iDetElemNumber = 0; - - AliMUONGeometryTransformer *newGeometryTransformer = - new AliMUONGeometryTransformer(); - for (Int_t iMt = 0; iMt < transformer->GetNofModuleTransformers(); iMt++) { - // module transformers - const AliMUONGeometryModuleTransformer *kModuleTransformer = - transformer->GetModuleTransformer(iMt, true); - - AliMUONGeometryModuleTransformer *newModuleTransformer = - new AliMUONGeometryModuleTransformer(iMt); + Double_t lModuleMisAlignment[fgNParCh] = {0}; + Double_t lDetElemMisAlignment[fgNParCh] = {0}; + const TClonesArray* oldMisAlignArray( transformer->GetMisAlignmentData() ); + + AliMUONGeometryTransformer *newGeometryTransformer = new AliMUONGeometryTransformer(); + for( Int_t iMt = 0; iMt < transformer->GetNofModuleTransformers(); ++iMt ) + { + + // module transformers + const AliMUONGeometryModuleTransformer *kModuleTransformer = transformer->GetModuleTransformer(iMt, kTRUE); + + AliMUONGeometryModuleTransformer *newModuleTransformer = new AliMUONGeometryModuleTransformer(iMt); newGeometryTransformer->AddModuleTransformer(newModuleTransformer); - - TGeoCombiTrans moduleTransform = - TGeoCombiTrans(*kModuleTransformer->GetTransformation()); - // New module transformation - TGeoCombiTrans newModuleTransform = ReAlign(moduleTransform,lModuleMisAlignment); + + // get transformation + TGeoHMatrix deltaModuleTransform( DeltaTransform( lModuleMisAlignment ) ); + + // update module + TGeoHMatrix moduleTransform( *kModuleTransformer->GetTransformation() ); + TGeoHMatrix newModuleTransform( AliMUONGeometryBuilder::Multiply( deltaModuleTransform, moduleTransform ) ); newModuleTransformer->SetTransformation(newModuleTransform); - - // Get delta transformation: - // Tdelta = Tnew * Told.inverse - TGeoHMatrix deltaModuleTransform = - AliMUONGeometryBuilder::Multiply(newModuleTransform, - kModuleTransformer->GetTransformation()->Inverse()); + + // Get matching old alignment and update current matrix accordingly + if( oldMisAlignArray ) + { + + const AliAlignObjMatrix* oldAlignObj(0); + const Int_t moduleId( kModuleTransformer->GetModuleId() ); + const Int_t volId = AliGeomManager::LayerToVolUID(AliGeomManager::kMUON, moduleId ); + for( Int_t pos = 0; pos < oldMisAlignArray->GetEntriesFast(); ++pos ) + { + + const AliAlignObjMatrix* localAlignObj( dynamic_cast(oldMisAlignArray->At( pos ) ) ); + if( localAlignObj && localAlignObj->GetVolUID() == volId ) + { + oldAlignObj = localAlignObj; + break; + } + + } + + // multiply + if( oldAlignObj ) + { + + TGeoHMatrix oldMatrix; + oldAlignObj->GetMatrix( oldMatrix ); + deltaModuleTransform.Multiply( &oldMatrix ); + + } + + } + // Create module mis alignment matrix - newGeometryTransformer - ->AddMisAlignModule(kModuleTransformer->GetModuleId(), deltaModuleTransform); - + newGeometryTransformer ->AddMisAlignModule(kModuleTransformer->GetModuleId(), deltaModuleTransform); + AliMpExMap *detElements = kModuleTransformer->GetDetElementStore(); - - if (verbose) - AliInfo(Form("%i DEs in old GeometryStore %i",detElements->GetSize(), iMt)); - - for (Int_t iDe = 0; iDe < detElements->GetSize(); iDe++) { - // detection elements. - AliMUONGeometryDetElement *detElement = - (AliMUONGeometryDetElement *) detElements->GetObject(iDe); - if (!detElement) - AliFatal("Detection element not found."); - - /// make a new detection element - AliMUONGeometryDetElement *newDetElement = - new AliMUONGeometryDetElement(detElement->GetId(), - detElement->GetVolumePath()); + + TIter next(detElements->CreateIterator()); + AliMUONGeometryDetElement* detElement; + Int_t iDe(-1); + while ( ( detElement = static_cast(next()) ) ) + { + ++iDe; + // make a new detection element + AliMUONGeometryDetElement *newDetElement = new AliMUONGeometryDetElement(detElement->GetId(), detElement->GetVolumePath()); TString lDetElemName(detElement->GetDEName()); lDetElemName.ReplaceAll("DE",""); - iDetElemId = lDetElemName.Atoi(); - iDetElemNumber = iDetElemId%100; - for (int iCh=0; iChGetGlobalTransformation() ); + TGeoHMatrix newGlobalTransform( AliMUONGeometryBuilder::Multiply( deltaGlobalTransform, globalTransform ) ); + newDetElement->SetGlobalTransformation( newGlobalTransform ); + newModuleTransformer->GetDetElementStore()->Add(newDetElement->GetId(), newDetElement); + + // Get matching old alignment and update current matrix accordingly + if( oldMisAlignArray ) + { + + const AliAlignObjMatrix* oldAlignObj(0); + const int detElemId( detElement->GetId() ); + const Int_t volId = AliGeomManager::LayerToVolUID(AliGeomManager::kMUON, detElemId ); + for( Int_t pos = 0; pos < oldMisAlignArray->GetEntriesFast(); ++pos ) + { + + const AliAlignObjMatrix* localAlignObj( dynamic_cast(oldMisAlignArray->At( pos ) ) ); + if( localAlignObj && localAlignObj->GetVolUID() == volId ) + { + oldAlignObj = localAlignObj; + break; + } + + } + + // multiply + if( oldAlignObj ) + { + + TGeoHMatrix oldMatrix; + oldAlignObj->GetMatrix( oldMatrix ); + deltaGlobalTransform.Multiply( &oldMatrix ); + + } + } - // local transformation of this detection element. - TGeoCombiTrans localTransform - = TGeoCombiTrans(*detElement->GetLocalTransformation()); - TGeoCombiTrans newLocalTransform = ReAlign(localTransform,lDetElemMisAlignment); - newDetElement->SetLocalTransformation(newLocalTransform); - - // global transformation - TGeoHMatrix newGlobalTransform = - AliMUONGeometryBuilder::Multiply(newModuleTransform, - newLocalTransform); - newDetElement->SetGlobalTransformation(newGlobalTransform); - - // add this det element to module - newModuleTransformer->GetDetElementStore()->Add(newDetElement->GetId(), - newDetElement); - - // In the Alice Alignment Framework misalignment objects store - // global delta transformation - // Get detection "intermediate" global transformation - TGeoHMatrix newOldGlobalTransform = newModuleTransform * localTransform; - // Get detection element global delta transformation: - // Tdelta = Tnew * Told.inverse - TGeoHMatrix deltaGlobalTransform - = AliMUONGeometryBuilder::Multiply(newGlobalTransform, - newOldGlobalTransform.Inverse()); - - // Create mis alignment matrix - newGeometryTransformer - ->AddMisAlignDetElement(detElement->GetId(), deltaGlobalTransform); + + // Create misalignment matrix + newGeometryTransformer->AddMisAlignDetElement(detElement->GetId(), deltaGlobalTransform); + } - - if (verbose) - AliInfo(Form("Added module transformer %i to the transformer", iMt)); + newGeometryTransformer->AddModuleTransformer(newModuleTransformer); } + return newGeometryTransformer; + } +//______________________________________________________________________ +void AliMUONAlignment::SetAlignmentResolution( const TClonesArray* misAlignArray, Int_t rChId, Double_t chResX, Double_t chResY, Double_t deResX, Double_t deResY ) +{ + + /// Set alignment resolution to misalign objects to be stored in CDB + /// if rChId is > 0 set parameters for this chamber only, counting from 1 + TMatrixDSym mChCorrMatrix(6); + mChCorrMatrix[0][0]=chResX*chResX; + mChCorrMatrix[1][1]=chResY*chResY; + + TMatrixDSym mDECorrMatrix(6); + mDECorrMatrix[0][0]=deResX*deResX; + mDECorrMatrix[1][1]=deResY*deResY; + + AliAlignObjMatrix *alignMat = 0x0; + + for( Int_t chId = 0; chId <= 9; ++chId ) + { + + // skip chamber if selection is valid, and does not match + if( rChId > 0 && chId+1 != rChId ) continue; + + TString chName1; + TString chName2; + if (chId<4) + { + + chName1 = Form("GM%d",chId); + chName2 = Form("GM%d",chId); + + } else { + + chName1 = Form("GM%d",4+(chId-4)*2); + chName2 = Form("GM%d",4+(chId-4)*2+1); + + } + + for( int i=0; iGetEntries(); ++i ) + { + + alignMat = (AliAlignObjMatrix*)misAlignArray->At(i); + TString volName(alignMat->GetSymName()); + if((volName.Contains(chName1)&& + ((volName.Last('/')==volName.Index(chName1)+chName1.Length())|| + (volName.Length()==volName.Index(chName1)+chName1.Length())))|| + (volName.Contains(chName2)&& + ((volName.Last('/')==volName.Index(chName2)+chName2.Length())|| + (volName.Length()==volName.Index(chName2)+chName2.Length())))) + { + + volName.Remove(0,volName.Last('/')+1); + if (volName.Contains("GM")) alignMat->SetCorrMatrix(mChCorrMatrix); + else if (volName.Contains("DE")) alignMat->SetCorrMatrix(mDECorrMatrix); + + } + + } + + } + +} + + +//_____________________________________________________ +void AliMUONAlignment::FillDetElemData( AliMUONVCluster* cluster ) +{ + + /// Get information of current detection element + // get detector element number from Alice ID + const Int_t detElemId = cluster->GetDetElemId(); + fDetElemNumber = GetDetElemNumber( detElemId ); + + // get detector element + const AliMUONGeometryDetElement* detElement = fTransform->GetDetElement( detElemId ); + + /* + get the global transformation matrix and store its inverse, in order to manually perform + the global to Local transformations needed to calculate the derivatives + */ + fGeoCombiTransInverse = detElement->GetGlobalTransformation()->Inverse(); + +} + +//______________________________________________________________________ +void AliMUONAlignment::FillRecPointData( AliMUONVCluster* cluster ) +{ + + /// Get information of current cluster + fClustPos[0] = cluster->GetX(); + fClustPos[1] = cluster->GetY(); + fClustPos[2] = cluster->GetZ(); + +} + +//______________________________________________________________________ +void AliMUONAlignment::FillTrackParamData( AliMUONTrackParam* trackParam ) +{ + + /// Get information of current track at current cluster + fTrackPos[0] = trackParam->GetNonBendingCoor(); + fTrackPos[1] = trackParam->GetBendingCoor(); + fTrackPos[2] = trackParam->GetZ(); + fTrackSlope[0] = trackParam->GetNonBendingSlope(); + fTrackSlope[1] = trackParam->GetBendingSlope(); + +} + +//______________________________________________________________________ +void AliMUONAlignment::LocalEquationX( void ) +{ + /// local equation along X + + // 'inverse' (GlobalToLocal) rotation matrix + const Double_t* r( fGeoCombiTransInverse.GetRotationMatrix() ); + + // local derivatives + SetLocalDerivative( 0, r[0] ); + SetLocalDerivative( 1, r[0]*(fTrackPos[2] - fTrackPos0[2]) ); + SetLocalDerivative( 2, r[1] ); + SetLocalDerivative( 3, r[1]*(fTrackPos[2] - fTrackPos0[2]) ); + + // global derivatives + /* + alignment parameters are + 0: delta_x + 1: delta_y + 2: delta_phiz + 3: delta_z + */ + + SetGlobalDerivative( fDetElemNumber*fgNParCh + 0, -r[0] ); + SetGlobalDerivative( fDetElemNumber*fgNParCh + 1, -r[1] ); + + if( fBFieldOn ) + { + + // use local position for derivatives vs 'delta_phi_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 2, -r[1]*fTrackPos[0] + r[0]*fTrackPos[1] ); + + // use local slopes for derivatives vs 'delta_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 3, r[0]*fTrackSlope[0] + r[1]*fTrackSlope[1] ); + + } else { + + // local copy of extrapolated track positions + const Double_t trackPosX = fTrackPos0[0]+fTrackSlope0[0]*( fTrackPos[2]-fTrackPos0[2] ); + const Double_t trackPosY = fTrackPos0[1]+fTrackSlope0[1]*( fTrackPos[2]-fTrackPos0[2] ); + + // use properly extrapolated position for derivatives vs 'delta_phi_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 2, -r[1]*trackPosX + r[0]*trackPosY ); + + // use slopes at origin for derivatives vs 'delta_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 3, r[0]*fTrackSlope0[0] + r[1]*fTrackSlope0[1] ); + + } + + // store local equation + fMillepede->SetLocalEquation( fGlobalDerivatives, fLocalDerivatives, fMeas[0], fSigma[0] ); + +} + +//______________________________________________________________________ +void AliMUONAlignment::LocalEquationY( void ) +{ + /// local equation along Y + + // 'inverse' (GlobalToLocal) rotation matrix + const Double_t* r( fGeoCombiTransInverse.GetRotationMatrix() ); + + // store local derivatives + SetLocalDerivative( 0, r[3] ); + SetLocalDerivative( 1, r[3]*(fTrackPos[2] - fTrackPos0[2] ) ); + SetLocalDerivative( 2, r[4] ); + SetLocalDerivative( 3, r[4]*(fTrackPos[2] - fTrackPos0[2] ) ); + + // set global derivatives + SetGlobalDerivative( fDetElemNumber*fgNParCh + 0, -r[3]); + SetGlobalDerivative( fDetElemNumber*fgNParCh + 1, -r[4]); + + if( fBFieldOn ) + { + + // use local position for derivatives vs 'delta_phi' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 2, -r[4]*fTrackPos[0] + r[3]*fTrackPos[1]); + + // use local slopes for derivatives vs 'delta_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 3, r[3]*fTrackSlope[0]+r[4]*fTrackSlope[1] ); + + } else { + + // local copy of extrapolated track positions + const Double_t trackPosX = fTrackPos0[0]+fTrackSlope0[0]*( fTrackPos[2]-fTrackPos0[2] ); + const Double_t trackPosY = fTrackPos0[1]+fTrackSlope0[1]*( fTrackPos[2]-fTrackPos0[2] ); + + // use properly extrapolated position for derivatives vs 'delta_phi' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 2, -r[4]*trackPosX + r[3]*trackPosY ); + + // use slopes at origin for derivatives vs 'delta_z' + SetGlobalDerivative( fDetElemNumber*fgNParCh + 3, r[3]*fTrackSlope0[0]+r[4]*fTrackSlope0[1] ); + + } + + // store local equation + fMillepede->SetLocalEquation( fGlobalDerivatives, fLocalDerivatives, fMeas[1], fSigma[1] ); + +} + +//_________________________________________________________________________ +TGeoCombiTrans AliMUONAlignment::DeltaTransform( const double *lMisAlignment) const +{ + /// Get Delta Transformation, based on alignment parameters + + // translation + const TGeoTranslation deltaTrans( lMisAlignment[0], lMisAlignment[1], lMisAlignment[3]); + + // rotation + TGeoRotation deltaRot; + deltaRot.RotateZ(lMisAlignment[2]*180./TMath::Pi()); + + // combined rotation and translation. + return TGeoCombiTrans(deltaTrans,deltaRot); + +} + +//______________________________________________________________________ +void AliMUONAlignment::AddConstraint(Double_t *par, Double_t value) +{ + /// Constrain equation defined by par to value + if( !fInitialized ) + { AliFatal( "Millepede is not initialized" ); } + + fMillepede->SetGlobalConstraint(par, value); +} + +//______________________________________________________________________ +Bool_t AliMUONAlignment::DetElemIsValid( Int_t iDetElemId ) const +{ + /// return true if given detector element is valid (and belongs to muon tracker) + const Int_t iCh = iDetElemId/100; + const Int_t iDet = iDetElemId%100; + return ( iCh > 0 && iCh <= fgNCh && iDet < fgNDetElemCh[iCh-1] ); +} + +//______________________________________________________________________ +Int_t AliMUONAlignment::GetDetElemNumber( Int_t iDetElemId ) const +{ + /// get det element number from ID + // get chamber and element number in chamber + const Int_t iCh = iDetElemId/100; + const Int_t iDet = iDetElemId%100; + + // make sure detector index is valid + if( !( iCh > 0 && iCh <= fgNCh && iDet < fgNDetElemCh[iCh-1] ) ) + { AliFatal( Form( "Invalid detector element id: %i", iDetElemId ) ); } + + // add number of detectors up to this chamber + return iDet + fgSNDetElemCh[iCh-1]; + +} + +//______________________________________________________________________ +Int_t AliMUONAlignment::GetChamberId( Int_t iDetElemNumber ) const +{ + /// get chamber (counting from 1) matching a given detector element id + Int_t iCh( 0 ); + for( iCh=0; iCh