[u/mrichter/AliRoot.git] / MUON / MUONAlignment.C

/**************************************************************************
 * 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.                  *
 **************************************************************************/

/* $Id$ */

/// \ingroup macros
/// \file MUONAlignment.C
/// \brief Macro for MUON alignment using physics tracks. 
///
/// The macro uses the AliMUONAlignment class to calculate the alignment parameters.
/// An array for the alignment parameters is created and can be filled with
/// initial values that will be used as starting values by the alignment
/// algorithm.
///
/// By default the macro run over galice.root in the working directory. If a file list
/// of galice.root is provided as third argument the macro will run over all files.
/// The macro loop over the files, events and tracks. For each track
/// AliMUONAlignment::ProcessTrack(AliMUONTrack * track) and then
/// AliMUONAlignment::LocalFit(Int_t iTrack, Double_t *lTrackParam, Int_t
/// lSingleFit) are called. After all tracks have been procesed and fitted
/// AliMUONAlignment::GlobalFit(Double_t *parameters,Double_t *errors,Double_t *pulls)
/// is called. The array parameters contains the obatained misalignement parameters.
/// A realigned geometry is generated in a local CDB.
///
/// \author: B. Becker and J. Castillo

#if !defined(__CINT__) || defined(__MAKECINT__)

#include "AliMUONAlignment.h"
#include "AliMUONTrack.h"
#include "AliMUONRecoParam.h"
#include "AliMUONTrackParam.h"
#include "AliMUONGeometryTransformer.h"
#include "AliMUONESDInterface.h"
#include "AliMUONCDB.h"

#include "AliESDEvent.h"
#include "AliESDMuonTrack.h"
#include "AliCDBManager.h"
#include "AliCDBMetaData.h"
#include "AliCDBId.h"
#include "AliGeomManager.h"

#include <TString.h>
#include <TError.h>
#include <TH1.h>
#include <TGraphErrors.h>
#include <TFile.h>
#include <TTree.h>
#include <TClonesArray.h>
#include <Riostream.h>

#include <fstream>

#endif

void MUONAlignment(Int_t nEvents = 100000, char* geoFilename = "geometry.root", TString esdFileName = "AliESDs.root", TString fileList = "")
{
 
  // Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
  if ( ! AliGeomManager::GetGeometry() ) {
    AliGeomManager::LoadGeometry(geoFilename);
    if (! AliGeomManager::GetGeometry() ) {
      Error("MUONAlignment", "getting geometry from file %s failed", geoFilename);
      return;
    }
  }
  
  // load necessary data from OCDB
  AliCDBManager* cdbManager = AliCDBManager::Instance();
  cdbManager->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  cdbManager->SetRun(0);
  if (!AliMUONCDB::LoadField()) return;
  AliMUONRecoParam* recoParam = AliMUONCDB::LoadRecoParam();
  if (!recoParam) return;
  
  // reset tracker for restoring initial track parameters at cluster
  AliMUONESDInterface::ResetTracker(recoParam);

  Double_t parameters[4*156];
  Double_t errors[4*156];
  Double_t pulls[4*156];
  for(Int_t k=0;k<4*156;k++) {
    parameters[k]=0.;
    errors[k]=0.;
    pulls[k]=0.;
  }

  Double_t trackParams[8] = {0.,0.,0.,0.,0.,0.,0.,0.};

  // Set initial values here, good guess may help convergence
  // St 1 
  //  Int_t iPar = 0;
  //  parameters[iPar++] =  0.010300 ;  parameters[iPar++] =  0.010600 ;  parameters[iPar++] =  0.000396 ;  

  bool bLoop = kFALSE;
  ifstream sFileList;
  if (fileList.Contains(".list")) {
    cout << "Reading file list: " << fileList.Data() << endl;
    bLoop = kTRUE;

    TString fullListName(".");
    fullListName +="/";
    fullListName +=fileList;
    sFileList.open(fileList.Data());
  }
  
  TH1F *fInvBenMom = new TH1F("fInvBenMom","fInvBenMom",200,-0.1,0.1); 
  TH1F *fBenMom = new TH1F("fBenMom","fBenMom",200,-40,40); 

  AliMUONAlignment* alig = new AliMUONAlignment();
  alig->InitGlobalParameters(parameters);

  AliMUONGeometryTransformer *transform = new AliMUONGeometryTransformer();
  transform->LoadGeometryData();
  alig->SetGeometryTransformer(transform);

  // Do alignment with magnetic field off
  alig->SetBFieldOn(kFALSE);
  
  // Set tracking station to use
  Bool_t bStOnOff[5] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
  Bool_t bChOnOff[10] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};

  // Set degrees of freedom to align (see AliMUONAlignment)
  alig->AllowVariations(bChOnOff);

  // Fix parameters or add constraints here
//   for (Int_t iSt=0; iSt<5; iSt++)
//     if (!bStOnOff[iSt]) alig->FixStation(iSt+1);
  for (Int_t iCh=0; iCh<10; iCh++)
    if (!bChOnOff[iCh]) alig->FixChamber(iCh+1);

  // Left and right sides of the detector are independent, one can choose to align 
  // only one side
  Bool_t bSpecLROnOff[2] = {kTRUE,kTRUE};
  alig->FixHalfSpectrometer(bChOnOff,bSpecLROnOff);

  alig->SetChOnOff(bChOnOff);
  alig->SetSpecLROnOff(bChOnOff);

  // Here we can fix some detection elements
  alig->FixDetElem(908);
  alig->FixDetElem(1020);

  // Set predifined global constrains: 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};
  //  alig->AddConstraints(bChOnOff,bVarXYT,bDetTLBR,bSpecLROnOff);


  char cFileName[100];  
    
  Int_t lMaxFile = 1000;
  Int_t iFile = 0;
  Int_t iEvent = 0;
  bool bKeepLoop = kTRUE;
  Int_t iTrackTot=0;
  Int_t iTrackOk=0;

  while(bKeepLoop && iFile<lMaxFile){
    iFile++;
    if (bLoop) {
      sFileList.getline(cFileName,100);
      if (sFileList.eof()) bKeepLoop = kFALSE;
    }
    else {
      sprintf(cFileName,esdFileName.Data());
      bKeepLoop = kFALSE;
    }
    if (!strstr(cFileName,"AliESDs")) continue;      
    cout << "Using file: " << cFileName << endl;
    
    // load ESD event
    TFile* esdFile = TFile::Open(cFileName); // open the file
    if (!esdFile || !esdFile->IsOpen()) {
      cout << "opening ESD file " << cFileName << "failed" << endl;
      continue;
    }
    TTree* esdTree = (TTree*) esdFile->Get("esdTree"); // get the tree
    if (!esdTree) {
      cout << "no ESD tree found" << endl;
      esdFile->Close();
      continue;
    }
    AliESDEvent* esdEvent = new AliESDEvent(); // link ESD event to the tree
    esdEvent->ReadFromTree(esdTree);

    Int_t nevents = esdTree->GetEntries();
    cout << "... with " << nevents << endl;
    for(Int_t event = 0; event < nevents; event++) {
      if (iEvent >= nEvents){
	bKeepLoop = kFALSE;
	break;
      }
      iEvent++;

      if (esdTree->GetEvent(event) <= 0) {
	cout << "fails to read ESD object for event " << event << endl;
	continue;
      }

      Int_t nTracks = Int_t(esdEvent->GetNumberOfMuonTracks());
      if (!event%100) cout << " there are " << nTracks << " tracks in event " << event << endl;
      for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
	AliESDMuonTrack* esdTrack = esdEvent->GetMuonTrack(iTrack);
	if (!esdTrack->ClustersStored()) continue;
	Double_t invBenMom = esdTrack->GetInverseBendingMomentum();
	fInvBenMom->Fill(invBenMom);
	fBenMom->Fill(1./invBenMom);
	if (TMath::Abs(invBenMom)<=1.04) {
	  AliMUONTrack track;
	  AliMUONESDInterface::ESDToMUON(*esdTrack, track);
	  alig->ProcessTrack(&track);
	  alig->LocalFit(iTrackOk++,trackParams,0);
	}
	iTrackTot++;
      }
    }
    delete esdEvent;
    esdFile->Close();
    cout << "Processed " << iTrackTot << " Tracks so far." << endl;
  }
  alig->GlobalFit(parameters,errors,pulls);

  cout << "Done with GlobalFit " << endl;

  // Store results
  Double_t DEid[156] = {0};
  Double_t MSDEx[156] = {0};
  Double_t MSDEy[156] = {0};
  Double_t MSDEz[156] = {0};
  Double_t MSDEp[156] = {0};
  Double_t DEidErr[156] = {0};
  Double_t MSDExErr[156] = {0};
  Double_t MSDEyErr[156] = {0};
  Double_t MSDEzErr[156] = {0};
  Double_t MSDEpErr[156] = {0};
  Int_t lNDetElem = 4*2+4*2+18*2+26*2+26*2;
  Int_t lNDetElemCh[10] = {4,4,4,4,18,18,26,26,26,26};
  // Int_t lSNDetElemCh[10] = {4,8,12,16,34,52,78,104,130,156};
  Int_t idOffset = 0; // 400
  Int_t lSDetElemCh = 0;
  for(Int_t iDE=0; iDE<lNDetElem; iDE++){
    DEidErr[iDE] = 0.;
    DEid[iDE] = idOffset+100;
    DEid[iDE] += iDE; 
    lSDetElemCh = 0;
    for(Int_t iCh=0; iCh<9; iCh++){
      lSDetElemCh += lNDetElemCh[iCh];
      if (iDE>=lSDetElemCh) {
	DEid[iDE] += 100;
	DEid[iDE] -= lNDetElemCh[iCh];
      }
    }
    MSDEx[iDE]=parameters[4*iDE+0];
    MSDEy[iDE]=parameters[4*iDE+1];
    MSDEz[iDE]=parameters[4*iDE+3];
    MSDEp[iDE]=parameters[4*iDE+2];
    MSDExErr[iDE]=(Double_t)alig->GetParError(4*iDE+0);
    MSDEyErr[iDE]=(Double_t)alig->GetParError(4*iDE+1);
    MSDEzErr[iDE]=(Double_t)alig->GetParError(4*iDE+3);
    MSDEpErr[iDE]=(Double_t)alig->GetParError(4*iDE+2);
  }

  cout << "Let's create graphs ...  " << endl;

  TGraphErrors *gMSDEx = new TGraphErrors(lNDetElem,DEid,MSDEx,DEidErr,MSDExErr); 
  TGraphErrors *gMSDEy = new TGraphErrors(lNDetElem,DEid,MSDEy,DEidErr,MSDEyErr); 
  TGraphErrors *gMSDEz = new TGraphErrors(lNDetElem,DEid,MSDEz,DEidErr,MSDEzErr); 
  TGraphErrors *gMSDEp = new TGraphErrors(lNDetElem,DEid,MSDEp,DEidErr,MSDEpErr); 

  cout << "... graphs created, open file ...  " << endl;

  TFile *hFile = new TFile("measShifts.root","RECREATE");

  cout << "... file opened ...  " << endl;

  gMSDEx->Write("gMSDEx");
  gMSDEy->Write("gMSDEy");
  gMSDEz->Write("gMSDEz");
  gMSDEp->Write("gMSDEp");
  fInvBenMom->Write();
  fBenMom->Write();
  hFile->Close();
  
  cout << "... and closed!" << endl;
  // Re Align
  AliMUONGeometryTransformer *newTransform = alig->ReAlign(transform,parameters,true); 
  newTransform->WriteTransformations("transform2ReAlign.dat");
  
  // Generate realigned data in local cdb
  const TClonesArray* array = newTransform->GetMisAlignmentData();

  // 100 mum residual resolution for chamber misalignments?
  alig->SetAlignmentResolution(array,-1,0.01,0.01,0.004,0.003);

  cdbManager->SetSpecificStorage("MUON/Align/Data","local://ReAlignCDB");
  
  AliCDBMetaData* cdbData = new AliCDBMetaData();
  cdbData->SetResponsible("Dimuon Offline project");
  cdbData->SetComment("MUON alignment objects with residual misalignment");
  AliCDBId id("MUON/Align/Data", 0, AliCDBRunRange::Infinity()); 
  cdbManager->Put(const_cast<TClonesArray*>(array), id, cdbData);

}
Commit	Line	Data
af9df62e	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
e54bf126	18	/// \ingroup macros
	19	/// \file MUONAlignment.C
	20	/// \brief Macro for MUON alignment using physics tracks.
	21	///
	22	/// The macro uses the AliMUONAlignment class to calculate the alignment parameters.
	23	/// An array for the alignment parameters is created and can be filled with
	24	/// initial values that will be used as starting values by the alignment
	25	/// algorithm.
	26	///
	27	/// By default the macro run over galice.root in the working directory. If a file list
	28	/// of galice.root is provided as third argument the macro will run over all files.
	29	/// The macro loop over the files, events and tracks. For each track
	30	/// AliMUONAlignment::ProcessTrack(AliMUONTrack * track) and then
	31	/// AliMUONAlignment::LocalFit(Int_t iTrack, Double_t *lTrackParam, Int_t
	32	/// lSingleFit) are called. After all tracks have been procesed and fitted
	33	/// AliMUONAlignment::GlobalFit(Double_t parameters,Double_t errors,Double_t *pulls)
	34	/// is called. The array parameters contains the obatained misalignement parameters.
	35	/// A realigned geometry is generated in a local CDB.
	36	///
	37	/// \author: B. Becker and J. Castillo
af9df62e	38
2688ea9b	39	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
	40
	41	#include "AliMUONAlignment.h"
	42	#include "AliMUONTrack.h"
a99c3449	43	#include "AliMUONRecoParam.h"
2688ea9b	44	#include "AliMUONTrackParam.h"
2688ea9b	45	#include "AliMUONGeometryTransformer.h"
103e6575	46	#include "AliMUONESDInterface.h"
a99c3449	47	#include "AliMUONCDB.h"
103e6575	48
61fed964	49	#include "AliESDEvent.h"
61fed964	50	#include "AliESDMuonTrack.h"
2688ea9b	51	#include "AliCDBManager.h"
	52	#include "AliCDBMetaData.h"
	53	#include "AliCDBId.h"
a02fe82d	54	#include "AliGeomManager.h"
2688ea9b	55
2688ea9b	56	#include <TString.h>
2688ea9b	57	#include <TError.h>
	58	#include <TH1.h>
	59	#include <TGraphErrors.h>
	60	#include <TFile.h>
61fed964	61	#include <TTree.h>
2688ea9b	62	#include <TClonesArray.h>
	63	#include <Riostream.h>
	64
	65	#include <fstream>
	66
	67	#endif
	68
61fed964	69	void MUONAlignment(Int_t nEvents = 100000, char* geoFilename = "geometry.root", TString esdFileName = "AliESDs.root", TString fileList = "")
af9df62e	70	{
af9df62e	71
59830a87	72	// Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
a02fe82d	73	if ( ! AliGeomManager::GetGeometry() ) {
	74	AliGeomManager::LoadGeometry(geoFilename);
	75	if (! AliGeomManager::GetGeometry() ) {
	76	Error("MUONAlignment", "getting geometry from file %s failed", geoFilename);
59830a87	77	return;
	78	}
	79	}
	80
a99c3449	81	// load necessary data from OCDB
cd8521dd	82	AliCDBManager* cdbManager = AliCDBManager::Instance();
	83	cdbManager->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
	84	cdbManager->SetRun(0);
a99c3449	85	if (!AliMUONCDB::LoadField()) return;
	86	AliMUONRecoParam* recoParam = AliMUONCDB::LoadRecoParam();
	87	if (!recoParam) return;
	88
	89	// reset tracker for restoring initial track parameters at cluster
	90	AliMUONESDInterface::ResetTracker(recoParam);
59830a87	91
723b0b5b	92	Double_t parameters[4*156];
	93	Double_t errors[4*156];
	94	Double_t pulls[4*156];
	95	for(Int_t k=0;k<4*156;k++) {
af9df62e	96	parameters[k]=0.;
	97	errors[k]=0.;
	98	pulls[k]=0.;
	99	}
	100
	101	Double_t trackParams[8] = {0.,0.,0.,0.,0.,0.,0.,0.};
	102
af9df62e	103	// Set initial values here, good guess may help convergence
af9df62e	104	// St 1
2688ea9b	105	// Int_t iPar = 0;
2688ea9b	106	// parameters[iPar++] = 0.010300 ; parameters[iPar++] = 0.010600 ; parameters[iPar++] = 0.000396 ;
af9df62e	107
	108	bool bLoop = kFALSE;
	109	ifstream sFileList;
	110	if (fileList.Contains(".list")) {
	111	cout << "Reading file list: " << fileList.Data() << endl;
	112	bLoop = kTRUE;
	113
	114	TString fullListName(".");
	115	fullListName +="/";
	116	fullListName +=fileList;
	117	sFileList.open(fileList.Data());
	118	}
	119
	120	TH1F *fInvBenMom = new TH1F("fInvBenMom","fInvBenMom",200,-0.1,0.1);
	121	TH1F *fBenMom = new TH1F("fBenMom","fBenMom",200,-40,40);
	122
	123	AliMUONAlignment* alig = new AliMUONAlignment();
	124	alig->InitGlobalParameters(parameters);
	125
a02fe82d	126	AliMUONGeometryTransformer *transform = new AliMUONGeometryTransformer();
a02fe82d	127	transform->LoadGeometryData();
af9df62e	128	alig->SetGeometryTransformer(transform);
7f8c1308	129
	130	// Do alignment with magnetic field off
	131	alig->SetBFieldOn(kFALSE);
587965bd	132
	133	// Set tracking station to use
	134	Bool_t bStOnOff[5] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
d8ad38e5	135	Bool_t bChOnOff[10] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
587965bd	136
f3e4edfb	137	// Set degrees of freedom to align (see AliMUONAlignment)
	138	alig->AllowVariations(bChOnOff);
	139
587965bd	140	// Fix parameters or add constraints here
d8ad38e5	141	// for (Int_t iSt=0; iSt<5; iSt++)
	142	// if (!bStOnOff[iSt]) alig->FixStation(iSt+1);
	143	for (Int_t iCh=0; iCh<10; iCh++)
	144	if (!bChOnOff[iCh]) alig->FixChamber(iCh+1);
587965bd	145
	146	// Left and right sides of the detector are independent, one can choose to align
	147	// only one side
	148	Bool_t bSpecLROnOff[2] = {kTRUE,kTRUE};
d8ad38e5	149	alig->FixHalfSpectrometer(bChOnOff,bSpecLROnOff);
	150
	151	alig->SetChOnOff(bChOnOff);
	152	alig->SetSpecLROnOff(bChOnOff);
587965bd	153
723b0b5b	154	// Here we can fix some detection elements
	155	alig->FixDetElem(908);
	156	alig->FixDetElem(1020);
	157
587965bd	158	// Set predifined global constrains: X, Y, P, XvsZ, YvsZ, PvsZ, XvsY, YvsY, PvsY
	159	Bool_t bVarXYT[9] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
	160	Bool_t bDetTLBR[4] = {kFALSE,kTRUE,kFALSE,kTRUE};
723b0b5b	161	// alig->AddConstraints(bChOnOff,bVarXYT,bDetTLBR,bSpecLROnOff);
587965bd	162
af9df62e	163
af9df62e	164	char cFileName[100];
7f8c1308	165
af9df62e	166	Int_t lMaxFile = 1000;
af9df62e	167	Int_t iFile = 0;
587965bd	168	Int_t iEvent = 0;
af9df62e	169	bool bKeepLoop = kTRUE;
587965bd	170	Int_t iTrackTot=0;
	171	Int_t iTrackOk=0;
	172
af9df62e	173	while(bKeepLoop && iFile<lMaxFile){
	174	iFile++;
	175	if (bLoop) {
	176	sFileList.getline(cFileName,100);
	177	if (sFileList.eof()) bKeepLoop = kFALSE;
	178	}
	179	else {
61fed964	180	sprintf(cFileName,esdFileName.Data());
af9df62e	181	bKeepLoop = kFALSE;
af9df62e	182	}
723b0b5b	183	if (!strstr(cFileName,"AliESDs")) continue;
af9df62e	184	cout << "Using file: " << cFileName << endl;
61fed964	185
	186	// load ESD event
	187	TFile* esdFile = TFile::Open(cFileName); // open the file
	188	if (!esdFile \|\| !esdFile->IsOpen()) {
	189	cout << "opening ESD file " << cFileName << "failed" << endl;
	190	continue;
	191	}
	192	TTree* esdTree = (TTree*) esdFile->Get("esdTree"); // get the tree
	193	if (!esdTree) {
	194	cout << "no ESD tree found" << endl;
	195	esdFile->Close();
	196	continue;
	197	}
	198	AliESDEvent* esdEvent = new AliESDEvent(); // link ESD event to the tree
	199	esdEvent->ReadFromTree(esdTree);
7f8c1308	200
61fed964	201	Int_t nevents = esdTree->GetEntries();
587965bd	202	cout << "... with " << nevents << endl;
af9df62e	203	for(Int_t event = 0; event < nevents; event++) {
587965bd	204	if (iEvent >= nEvents){
	205	bKeepLoop = kFALSE;
	206	break;
	207	}
	208	iEvent++;
	209
61fed964	210	if (esdTree->GetEvent(event) <= 0) {
	211	cout << "fails to read ESD object for event " << event << endl;
	212	continue;
	213	}
7f8c1308	214
61fed964	215	Int_t nTracks = Int_t(esdEvent->GetNumberOfMuonTracks());
	216	if (!event%100) cout << " there are " << nTracks << " tracks in event " << event << endl;
	217	for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
	218	AliESDMuonTrack* esdTrack = esdEvent->GetMuonTrack(iTrack);
	219	if (!esdTrack->ClustersStored()) continue;
	220	Double_t invBenMom = esdTrack->GetInverseBendingMomentum();
af9df62e	221	fInvBenMom->Fill(invBenMom);
	222	fBenMom->Fill(1./invBenMom);
	223	if (TMath::Abs(invBenMom)<=1.04) {
103e6575	224	AliMUONTrack track;
ad3c6eda	225	AliMUONESDInterface::ESDToMUON(*esdTrack, track);
61fed964	226	alig->ProcessTrack(&track);
af9df62e	227	alig->LocalFit(iTrackOk++,trackParams,0);
af9df62e	228	}
587965bd	229	iTrackTot++;
af9df62e	230	}
af9df62e	231	}
61fed964	232	delete esdEvent;
61fed964	233	esdFile->Close();
587965bd	234	cout << "Processed " << iTrackTot << " Tracks so far." << endl;
af9df62e	235	}
	236	alig->GlobalFit(parameters,errors,pulls);
	237
	238	cout << "Done with GlobalFit " << endl;
	239
	240	// Store results
	241	Double_t DEid[156] = {0};
	242	Double_t MSDEx[156] = {0};
	243	Double_t MSDEy[156] = {0};
723b0b5b	244	Double_t MSDEz[156] = {0};
723b0b5b	245	Double_t MSDEp[156] = {0};
af9df62e	246	Double_t DEidErr[156] = {0};
	247	Double_t MSDExErr[156] = {0};
	248	Double_t MSDEyErr[156] = {0};
723b0b5b	249	Double_t MSDEzErr[156] = {0};
723b0b5b	250	Double_t MSDEpErr[156] = {0};
af9df62e	251	Int_t lNDetElem = 42+42+182+262+26*2;
af9df62e	252	Int_t lNDetElemCh[10] = {4,4,4,4,18,18,26,26,26,26};
2688ea9b	253	// Int_t lSNDetElemCh[10] = {4,8,12,16,34,52,78,104,130,156};
af9df62e	254	Int_t idOffset = 0; // 400
	255	Int_t lSDetElemCh = 0;
	256	for(Int_t iDE=0; iDE<lNDetElem; iDE++){
	257	DEidErr[iDE] = 0.;
	258	DEid[iDE] = idOffset+100;
	259	DEid[iDE] += iDE;
	260	lSDetElemCh = 0;
	261	for(Int_t iCh=0; iCh<9; iCh++){
	262	lSDetElemCh += lNDetElemCh[iCh];
	263	if (iDE>=lSDetElemCh) {
	264	DEid[iDE] += 100;
	265	DEid[iDE] -= lNDetElemCh[iCh];
	266	}
	267	}
723b0b5b	268	MSDEx[iDE]=parameters[4*iDE+0];
	269	MSDEy[iDE]=parameters[4*iDE+1];
	270	MSDEz[iDE]=parameters[4*iDE+3];
	271	MSDEp[iDE]=parameters[4*iDE+2];
	272	MSDExErr[iDE]=(Double_t)alig->GetParError(4*iDE+0);
	273	MSDEyErr[iDE]=(Double_t)alig->GetParError(4*iDE+1);
	274	MSDEzErr[iDE]=(Double_t)alig->GetParError(4*iDE+3);
	275	MSDEpErr[iDE]=(Double_t)alig->GetParError(4*iDE+2);
af9df62e	276	}
	277
	278	cout << "Let's create graphs ... " << endl;
	279
	280	TGraphErrors *gMSDEx = new TGraphErrors(lNDetElem,DEid,MSDEx,DEidErr,MSDExErr);
	281	TGraphErrors *gMSDEy = new TGraphErrors(lNDetElem,DEid,MSDEy,DEidErr,MSDEyErr);
723b0b5b	282	TGraphErrors *gMSDEz = new TGraphErrors(lNDetElem,DEid,MSDEz,DEidErr,MSDEzErr);
723b0b5b	283	TGraphErrors *gMSDEp = new TGraphErrors(lNDetElem,DEid,MSDEp,DEidErr,MSDEpErr);
af9df62e	284
	285	cout << "... graphs created, open file ... " << endl;
	286
	287	TFile *hFile = new TFile("measShifts.root","RECREATE");
	288
	289	cout << "... file opened ... " << endl;
	290
	291	gMSDEx->Write("gMSDEx");
	292	gMSDEy->Write("gMSDEy");
723b0b5b	293	gMSDEz->Write("gMSDEz");
723b0b5b	294	gMSDEp->Write("gMSDEp");
af9df62e	295	fInvBenMom->Write();
	296	fBenMom->Write();
	297	hFile->Close();
	298
	299	cout << "... and closed!" << endl;
	300	// Re Align
	301	AliMUONGeometryTransformer *newTransform = alig->ReAlign(transform,parameters,true);
	302	newTransform->WriteTransformations("transform2ReAlign.dat");
	303
	304	// Generate realigned data in local cdb
2688ea9b	305	const TClonesArray* array = newTransform->GetMisAlignmentData();
4818a9b7	306
	307	// 100 mum residual resolution for chamber misalignments?
	308	alig->SetAlignmentResolution(array,-1,0.01,0.01,0.004,0.003);
cd8521dd	309
cd8521dd	310	cdbManager->SetSpecificStorage("MUON/Align/Data","local://ReAlignCDB");
af9df62e	311
	312	AliCDBMetaData* cdbData = new AliCDBMetaData();
	313	cdbData->SetResponsible("Dimuon Offline project");
	314	cdbData->SetComment("MUON alignment objects with residual misalignment");
d8ad38e5	315	AliCDBId id("MUON/Align/Data", 0, AliCDBRunRange::Infinity());
2688ea9b	316	cdbManager->Put(const_cast<TClonesArray*>(array), id, cdbData);
af9df62e	317
61fed964	318	}
61fed964	319