ROOT logo
/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

//-----------------------------------------------------------------
//   Implementation of the derived class for track residuals
//   based on Chi2 minimization
//
//-----------------------------------------------------------------

#include <TVirtualFitter.h>
#include <TGeoMatrix.h>

#include "AliAlignObj.h"
#include "AliTrackPointArray.h"
#include "AliTrackResidualsChi2.h"



void TrackResidualsChi2Fcn(Int_t &npar, Double_t *gin, Double_t &f, Double_t *x, Int_t iflag);


ClassImp(AliTrackResidualsChi2)


//______________________________________________________________________________
void TrackResidualsChi2Fcn(Int_t &npar, Double_t *gin, Double_t &f, Double_t *par, Int_t iflag)
{
  // This function is called by minuit
  // The corresponding member method is called
  // using SetObjectFit/GetObjectFit methods of TMinuit
  AliTrackResidualsChi2* dummy = (AliTrackResidualsChi2 *)TVirtualFitter::GetFitter()->GetObjectFit();
  dummy->Chi2(npar, gin, f, par, iflag);
}

//______________________________________________________________________________
Bool_t AliTrackResidualsChi2::Minimize()
{
  // Implementation of Chi2 based minimization
  // of track residuala sum
  // 
  // CHOLM: Modified to use the TVirtualFitter interface only.  This
  // makes it possible other fitters than TMinuit (say TMinuit2,
  // TFumili) by simply changing ones configuration file, or setting
  // it up in a script.  Much more robust and flexible. 
  Double_t arglist[10];
  Int_t ierflg = 0;
  TVirtualFitter *fitter = TVirtualFitter::Fitter(this,6);  //initialize TMinuit
  arglist[0] = -1;
  ierflg = fitter->ExecuteCommand("SET PRINT", arglist, 1);

  fitter->SetFCN(TrackResidualsChi2Fcn);

  arglist[0] = 1;
  ierflg = fitter->ExecuteCommand("SET ERR", arglist ,1);

  // Set starting values and step sizes for parameters
  Double_t pars[6] = {0,0,0,0,0,0};
  for(Int_t i=0;i<6;i++) if(fBFixed[i]) pars[i]=fFixed[i];
  Double_t step[6] = {0.0001,0.0001,0.0001,0.0001,0.0001,0.0001};
  ierflg = fitter->SetParameter(0, "dx", pars[0], step[0], 0,0);
  ierflg = fitter->SetParameter(1, "dy", pars[1], step[1], 0,0);
  ierflg = fitter->SetParameter(2, "dz", pars[2], step[2], 0,0);
  ierflg = fitter->SetParameter(3, "psi", pars[3], step[3], 0,0);
  ierflg = fitter->SetParameter(4, "theta", pars[4], step[4], 0,0);
  ierflg = fitter->SetParameter(5, "phi", pars[5], step[5], 0,0);

  // Fix parameters
  if(fBFixed[0]) {printf("Fixing dx=%f\n",pars[0]); fitter->FixParameter(0);}
  if(fBFixed[1]) {printf("Fixing dy=%f\n",pars[1]); fitter->FixParameter(1);}
  if(fBFixed[2]) {printf("Fixing dz=%f\n",pars[2]); fitter->FixParameter(2);}
  if(fBFixed[3]) {printf("Fixing psi=%f\n",pars[3]); fitter->FixParameter(3);}
  if(fBFixed[4]) {printf("Fixing theta=%f\n",pars[4]); fitter->FixParameter(4);}
  if(fBFixed[5]) {printf("Fixing phi=%f\n",pars[5]); fitter->FixParameter(5);}

  // Now ready for minimization step
  arglist[0] = 500;
  arglist[1] = 1.;
  fitter->ExecuteCommand("MIGRAD", arglist ,2);

  // Print results
  Double_t amin,edm,errdef;
  Int_t nvpar,nparx,icstat;
  icstat = fitter->GetStats(amin,edm,errdef,nvpar,nparx);

   //Construct the covariance matrix for AlignObj
  Double_t *cov=fitter->GetCovarianceMatrix(); 
  Int_t unfixedparam=6;
  Int_t fixedparamat[6]={0,0,0,0,0,0};
  if(fBFixed[0]==kTRUE){
    unfixedparam--;
    fixedparamat[0]=1;
  }
  for(Int_t j=1;j<6;j++){
    if(fBFixed[j]==kTRUE){
      unfixedparam--;
      fixedparamat[j]=fixedparamat[j-1]+1;
    }
    else fixedparamat[j]=fixedparamat[j-1];
  }
  
  Double_t cov2[36];
  for(Int_t i=0;i<6;i++){
    for(Int_t j=0;j<6;j++){
      if(fBFixed[i]==kTRUE||fBFixed[j]==kTRUE){
       	cov2[i+6*j]=0.;
      }
      else cov2[i+6*j]=cov[i-fixedparamat[i]+6*(j-fixedparamat[j])];
    
    }
  }
  
  Double_t covmatrarray[21];
  for(Int_t j=0;j<6;j++){
    for(Int_t i=j;i<6;i++){
      covmatrarray[i*(i+1)/2+j]=cov2[i+6*j];
    }
  }
  
  //  printf("covar 2:  %.10f  \n   %.10f ; %.10f ;  \n; %.10f;  %.10f;  %.10f; \n  %.10f; %.10f;  %.10f;  %.10f;  \n  %.10f ; %.10f  ;  %.10f  ;  %.10f  ;  %.10f \n %.10f ;  %.10f ;  %.10f  ;  %.10f  ;  %.10f  ;  %.10f   ;\n",covmatrarray[0],covmatrarray[1],covmatrarray[2],covmatrarray[3],covmatrarray[4],covmatrarray[5],covmatrarray[6],covmatrarray[7],covmatrarray[8],covmatrarray[9],covmatrarray[10],covmatrarray[11],covmatrarray[12],covmatrarray[13],covmatrarray[14],covmatrarray[15],covmatrarray[16],covmatrarray[17],covmatrarray[18],covmatrarray[19],covmatrarray[20]);
  
  
  fAlignObj->SetCorrMatrix(covmatrarray);
  fChi2 = amin; fNdf -= nvpar;
  
  return kTRUE;
}

//______________________________________________________________________________
void AliTrackResidualsChi2::Chi2(Int_t & /* npar */, Double_t * /* gin */, Double_t &f, Double_t *par, Int_t /* iflag */)
{
  // Chi2 function to be minimized
  // Sums all the track residuals
  Double_t chi2 = 0;

  fAlignObj->SetPars(par[0],par[1],par[2],par[3],par[4],par[5]);

  AliTrackPoint p1,p2;

  Bool_t count = kFALSE;
  if (fNdf == 0) count = kTRUE;

  for (Int_t itrack = 0; itrack < fLast; itrack++) {
    if (!fVolArray[itrack] || !fTrackArray[itrack]) continue;
    for (Int_t ipoint = 0; ipoint < fVolArray[itrack]->GetNPoints(); ipoint++) {
      fVolArray[itrack]->GetPoint(p1,ipoint);
      fAlignObj->Transform(p1);
      fTrackArray[itrack]->GetPoint(p2,ipoint);
      Float_t residual = p2.GetResidual(p1,kTRUE);
      chi2 += residual;
      if (count) fNdf += 3;
    }
  }
  f = chi2;
}


 AliTrackResidualsChi2.cxx:1
 AliTrackResidualsChi2.cxx:2
 AliTrackResidualsChi2.cxx:3
 AliTrackResidualsChi2.cxx:4
 AliTrackResidualsChi2.cxx:5
 AliTrackResidualsChi2.cxx:6
 AliTrackResidualsChi2.cxx:7
 AliTrackResidualsChi2.cxx:8
 AliTrackResidualsChi2.cxx:9
 AliTrackResidualsChi2.cxx:10
 AliTrackResidualsChi2.cxx:11
 AliTrackResidualsChi2.cxx:12
 AliTrackResidualsChi2.cxx:13
 AliTrackResidualsChi2.cxx:14
 AliTrackResidualsChi2.cxx:15
 AliTrackResidualsChi2.cxx:16
 AliTrackResidualsChi2.cxx:17
 AliTrackResidualsChi2.cxx:18
 AliTrackResidualsChi2.cxx:19
 AliTrackResidualsChi2.cxx:20
 AliTrackResidualsChi2.cxx:21
 AliTrackResidualsChi2.cxx:22
 AliTrackResidualsChi2.cxx:23
 AliTrackResidualsChi2.cxx:24
 AliTrackResidualsChi2.cxx:25
 AliTrackResidualsChi2.cxx:26
 AliTrackResidualsChi2.cxx:27
 AliTrackResidualsChi2.cxx:28
 AliTrackResidualsChi2.cxx:29
 AliTrackResidualsChi2.cxx:30
 AliTrackResidualsChi2.cxx:31
 AliTrackResidualsChi2.cxx:32
 AliTrackResidualsChi2.cxx:33
 AliTrackResidualsChi2.cxx:34
 AliTrackResidualsChi2.cxx:35
 AliTrackResidualsChi2.cxx:36
 AliTrackResidualsChi2.cxx:37
 AliTrackResidualsChi2.cxx:38
 AliTrackResidualsChi2.cxx:39
 AliTrackResidualsChi2.cxx:40
 AliTrackResidualsChi2.cxx:41
 AliTrackResidualsChi2.cxx:42
 AliTrackResidualsChi2.cxx:43
 AliTrackResidualsChi2.cxx:44
 AliTrackResidualsChi2.cxx:45
 AliTrackResidualsChi2.cxx:46
 AliTrackResidualsChi2.cxx:47
 AliTrackResidualsChi2.cxx:48
 AliTrackResidualsChi2.cxx:49
 AliTrackResidualsChi2.cxx:50
 AliTrackResidualsChi2.cxx:51
 AliTrackResidualsChi2.cxx:52
 AliTrackResidualsChi2.cxx:53
 AliTrackResidualsChi2.cxx:54
 AliTrackResidualsChi2.cxx:55
 AliTrackResidualsChi2.cxx:56
 AliTrackResidualsChi2.cxx:57
 AliTrackResidualsChi2.cxx:58
 AliTrackResidualsChi2.cxx:59
 AliTrackResidualsChi2.cxx:60
 AliTrackResidualsChi2.cxx:61
 AliTrackResidualsChi2.cxx:62
 AliTrackResidualsChi2.cxx:63
 AliTrackResidualsChi2.cxx:64
 AliTrackResidualsChi2.cxx:65
 AliTrackResidualsChi2.cxx:66
 AliTrackResidualsChi2.cxx:67
 AliTrackResidualsChi2.cxx:68
 AliTrackResidualsChi2.cxx:69
 AliTrackResidualsChi2.cxx:70
 AliTrackResidualsChi2.cxx:71
 AliTrackResidualsChi2.cxx:72
 AliTrackResidualsChi2.cxx:73
 AliTrackResidualsChi2.cxx:74
 AliTrackResidualsChi2.cxx:75
 AliTrackResidualsChi2.cxx:76
 AliTrackResidualsChi2.cxx:77
 AliTrackResidualsChi2.cxx:78
 AliTrackResidualsChi2.cxx:79
 AliTrackResidualsChi2.cxx:80
 AliTrackResidualsChi2.cxx:81
 AliTrackResidualsChi2.cxx:82
 AliTrackResidualsChi2.cxx:83
 AliTrackResidualsChi2.cxx:84
 AliTrackResidualsChi2.cxx:85
 AliTrackResidualsChi2.cxx:86
 AliTrackResidualsChi2.cxx:87
 AliTrackResidualsChi2.cxx:88
 AliTrackResidualsChi2.cxx:89
 AliTrackResidualsChi2.cxx:90
 AliTrackResidualsChi2.cxx:91
 AliTrackResidualsChi2.cxx:92
 AliTrackResidualsChi2.cxx:93
 AliTrackResidualsChi2.cxx:94
 AliTrackResidualsChi2.cxx:95
 AliTrackResidualsChi2.cxx:96
 AliTrackResidualsChi2.cxx:97
 AliTrackResidualsChi2.cxx:98
 AliTrackResidualsChi2.cxx:99
 AliTrackResidualsChi2.cxx:100
 AliTrackResidualsChi2.cxx:101
 AliTrackResidualsChi2.cxx:102
 AliTrackResidualsChi2.cxx:103
 AliTrackResidualsChi2.cxx:104
 AliTrackResidualsChi2.cxx:105
 AliTrackResidualsChi2.cxx:106
 AliTrackResidualsChi2.cxx:107
 AliTrackResidualsChi2.cxx:108
 AliTrackResidualsChi2.cxx:109
 AliTrackResidualsChi2.cxx:110
 AliTrackResidualsChi2.cxx:111
 AliTrackResidualsChi2.cxx:112
 AliTrackResidualsChi2.cxx:113
 AliTrackResidualsChi2.cxx:114
 AliTrackResidualsChi2.cxx:115
 AliTrackResidualsChi2.cxx:116
 AliTrackResidualsChi2.cxx:117
 AliTrackResidualsChi2.cxx:118
 AliTrackResidualsChi2.cxx:119
 AliTrackResidualsChi2.cxx:120
 AliTrackResidualsChi2.cxx:121
 AliTrackResidualsChi2.cxx:122
 AliTrackResidualsChi2.cxx:123
 AliTrackResidualsChi2.cxx:124
 AliTrackResidualsChi2.cxx:125
 AliTrackResidualsChi2.cxx:126
 AliTrackResidualsChi2.cxx:127
 AliTrackResidualsChi2.cxx:128
 AliTrackResidualsChi2.cxx:129
 AliTrackResidualsChi2.cxx:130
 AliTrackResidualsChi2.cxx:131
 AliTrackResidualsChi2.cxx:132
 AliTrackResidualsChi2.cxx:133
 AliTrackResidualsChi2.cxx:134
 AliTrackResidualsChi2.cxx:135
 AliTrackResidualsChi2.cxx:136
 AliTrackResidualsChi2.cxx:137
 AliTrackResidualsChi2.cxx:138
 AliTrackResidualsChi2.cxx:139
 AliTrackResidualsChi2.cxx:140
 AliTrackResidualsChi2.cxx:141
 AliTrackResidualsChi2.cxx:142
 AliTrackResidualsChi2.cxx:143
 AliTrackResidualsChi2.cxx:144
 AliTrackResidualsChi2.cxx:145
 AliTrackResidualsChi2.cxx:146
 AliTrackResidualsChi2.cxx:147
 AliTrackResidualsChi2.cxx:148
 AliTrackResidualsChi2.cxx:149
 AliTrackResidualsChi2.cxx:150
 AliTrackResidualsChi2.cxx:151
 AliTrackResidualsChi2.cxx:152
 AliTrackResidualsChi2.cxx:153
 AliTrackResidualsChi2.cxx:154
 AliTrackResidualsChi2.cxx:155
 AliTrackResidualsChi2.cxx:156
 AliTrackResidualsChi2.cxx:157
 AliTrackResidualsChi2.cxx:158
 AliTrackResidualsChi2.cxx:159
 AliTrackResidualsChi2.cxx:160
 AliTrackResidualsChi2.cxx:161
 AliTrackResidualsChi2.cxx:162
 AliTrackResidualsChi2.cxx:163
 AliTrackResidualsChi2.cxx:164
 AliTrackResidualsChi2.cxx:165
 AliTrackResidualsChi2.cxx:166
 AliTrackResidualsChi2.cxx:167
 AliTrackResidualsChi2.cxx:168