[u/mrichter/AliRoot.git] / MUON / AliMUONAlignmentTask.cxx

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

//-----------------------------------------------------------------------------
/// \class AliMUONAlignmentTask
/// AliAnalysisTask to align the MUON spectrometer.
/// The Task reads as input ESDs and feeds the MUONTracks to AliMUONAlignment.
/// The alignment itself is performed by AliMillepede.
/// A OCDB entry is written with the alignment parameters.
/// A list of graph are written to monitor the alignment parameters.
///
/// \author Javier Castillo, CEA/Saclay - Irfu/SPhN
//-----------------------------------------------------------------------------

#include <fstream>

#include <TString.h>
#include <TError.h>
#include <TGraphErrors.h>
#include <TTree.h>
#include <TChain.h>
#include <TClonesArray.h>
#include <TGeoGlobalMagField.h>
#include <TGeoManager.h>
#include <Riostream.h>

#include "AliAnalysisTask.h"
#include "AliAnalysisManager.h"
#include "AliESDInputHandler.h"
#include "AliESDEvent.h"
#include "AliESDMuonTrack.h"
#include "AliMagF.h"
#include "AliCDBManager.h"
#include "AliGRPManager.h"
#include "AliCDBMetaData.h"
#include "AliCDBId.h"
#include "AliGeomManager.h"

#include "AliMpCDB.h"
#include "AliMUONAlignment.h"
#include "AliMUONTrack.h"
#include "AliMUONTrackExtrap.h"
#include "AliMUONTrackParam.h"
#include "AliMUONGeometryTransformer.h"
#include "AliMUONESDInterface.h"

#include "AliMUONAlignmentTask.h"

///\cond CLASSIMP  
ClassImp(AliMUONAlignmentTask)
///\endcond

// //________________________________________________________________________
// AliMUONAlignmentTask::AliMUONAlignmentTask(const char *name) 
//   : AliAnalysisTask(name, ""),
//     fESD(0x0),
//     fAlign(0x0),
//     fGeoFilename(""),
//     fTransform(0x0),
//     fTrackTot(0),
//     fTrackOk(0),
//     fMSDEx(0x0), 
//     fMSDEy(0x0), 
//     fMSDEz(0x0),
//     fMSDEp(0x0),
//     fList(0x0)
// {
//   /// Default Constructor
//   // Define input and output slots here
//   // Input slot #0 works with a TChain
//   DefineInput(0, TChain::Class());
//   // Output slot #0 writes NTuple/histos into a TList
//   DefineOutput(0, TList::Class());  

//   // initialize parameters ...
//   for(Int_t k=0;k<4*156;k++) {
//     fParameters[k]=0.;
//     fErrors[k]=0.;
//     fPulls[k]=0.;
//   }

//   fAlign = new AliMUONAlignment();
//   fTransform = new AliMUONGeometryTransformer();  
// }

//________________________________________________________________________
AliMUONAlignmentTask::AliMUONAlignmentTask(const char *name, const char *geofilename, const char *defaultocdb, const char *misalignocdb) 
  : AliAnalysisTask(name, ""),
    fESD(0x0),
    fAlign(0x0),
    fGeoFilename(geofilename),
    fMisAlignOCDB(misalignocdb),
    fDefaultOCDB(defaultocdb),
    fTransform(0x0),
    fTrackTot(0),
    fTrackOk(0),
    fLastRunNumber(-1),
    fMSDEx(0x0), 
    fMSDEy(0x0), 
    fMSDEz(0x0),
    fMSDEp(0x0),
    fList(0x0)
{
  /// Default Constructor
  // Define input and output slots here
  // Input slot #0 works with a TChain
  DefineInput(0, TChain::Class());
  // Output slot #0 writes NTuple/histos into a TList
  DefineOutput(0, TList::Class());  

  // initialize parameters ...
  for(Int_t k=0;k<4*156;k++) {
    fParameters[k]=0.;
    fErrors[k]=0.;
    fPulls[k]=0.;
  }

  fAlign = new AliMUONAlignment();
  fTransform = new AliMUONGeometryTransformer();  
}


//________________________________________________________________________
AliMUONAlignmentTask::AliMUONAlignmentTask(const AliMUONAlignmentTask& obj) 
  : AliAnalysisTask(obj),
    fESD(0x0),
    fAlign(0x0),
    fGeoFilename(""),
    fMisAlignOCDB(""),
    fDefaultOCDB(""),
    fTransform(0x0),
    fTrackTot(0),
    fTrackOk(0),
    fLastRunNumber(-1),
    fMSDEx(0x0), 
    fMSDEy(0x0), 
    fMSDEz(0x0),
    fMSDEp(0x0),
    fList(0x0)
{
  /// Copy constructor
  fESD = obj.fESD;
  fAlign = obj.fAlign;
  fGeoFilename = obj.fGeoFilename;
  fTransform = obj.fTransform;
  fTrackTot = obj.fTrackTot;  
  fTrackOk = obj.fTrackOk;  
  fLastRunNumber = obj.fLastRunNumber;
  fMSDEx = obj.fMSDEx; 
  fMSDEy = obj.fMSDEy; 
  fMSDEz = obj.fMSDEz;
  fMSDEp = obj.fMSDEp;
  fList = obj.fList;  
}

//________________________________________________________________________
AliMUONAlignmentTask& AliMUONAlignmentTask::operator=(const AliMUONAlignmentTask& other) 
{
  /// Assignment
  AliAnalysisTask::operator=(other);
  fESD = other.fESD;
  fAlign = other.fAlign;
  fGeoFilename = other.fGeoFilename;
  fMisAlignOCDB = other.fMisAlignOCDB;
  fDefaultOCDB = other.fDefaultOCDB;
  fTransform = other.fTransform;
  fTrackTot = other.fTrackTot;  
  fTrackOk = other.fTrackOk;  
  fLastRunNumber = other.fLastRunNumber;
  fMSDEx = other.fMSDEx; 
  fMSDEy = other.fMSDEy; 
  fMSDEz = other.fMSDEz;
  fMSDEp = other.fMSDEp;
  fList = other.fList;  
  
  return *this;
}

//________________________________________________________________________
AliMUONAlignmentTask::~AliMUONAlignmentTask() 
{ 
  /// Destructor
  if (fAlign) delete fAlign;
  if (fTransform) delete fTransform;
}

//________________________________________________________________________
void AliMUONAlignmentTask::LocalInit() 
{
  /// Local initialization, called once per task on the client machine 
  /// where the analysis train is assembled
  fLastRunNumber = 0; 
  //  Prepare(fGeoFilename.Data(),fDefaultOCDB.Data(),fMisAlignOCDB.Data());
  Prepare(fGeoFilename.Data(),"local://$ALICE_ROOT/OCDB",fMisAlignOCDB.Data());
  fLastRunNumber = -1;

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


  fAlign->InitGlobalParameters(fParameters);


  fTransform->LoadGeometryData();
  fAlign->SetGeometryTransformer(fTransform);

  // Do alignment with magnetic field off
  fAlign->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,kFALSE,kTRUE,kTRUE,kTRUE,kTRUE};

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

  // Fix parameters or add constraints here
  //   for (Int_t iSt=0; iSt<5; iSt++)
  //     if (!bStOnOff[iSt]) fAlign->FixStation(iSt+1);
  for (Int_t iCh=0; iCh<10; iCh++)
    if (!bChOnOff[iCh]) fAlign->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};
  fAlign->FixHalfSpectrometer(bChOnOff,bSpecLROnOff);

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

    // Here we can fix some detection elements
  fAlign->FixDetElem(908);
  fAlign->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};
  //  fAlign->AddConstraints(bChOnOff,bVarXYT,bDetTLBR,bSpecLROnOff);

}

//________________________________________________________________________
void AliMUONAlignmentTask::ConnectInputData(Option_t *) 
{
  /// Connect ESD here. Called on each input data change.

  // Connect ESD here
  TTree* esdTree = dynamic_cast<TTree*> (GetInputData(0));
  if (!esdTree) {
    Printf("ERROR: Could not read chain from input slot 0");
  } 
  else {
    AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());   
    if (!esdH) {
      Printf("ERROR: Could not get ESDInputHandler");
    } 
    else {
      fESD = esdH->GetEvent();
    }
  }
}

//________________________________________________________________________
void AliMUONAlignmentTask::CreateOutputObjects()
{
  /// Executed once on each worker (machine actually running the analysis code)
  //
  // This method has to be called INSIDE the user redefined CreateOutputObjects
  // method, before creating each object corresponding to the output containers
  // that are to be written to a file. This need to be done in general for the big output
  // objects that may not fit memory during processing. 
  //  OpenFile(0); 

  // Creating graphs 
  fMSDEx = new TGraphErrors(156);
  fMSDEy = new TGraphErrors(156);
  fMSDEz = new TGraphErrors(156);
  fMSDEp = new TGraphErrors(156);

  // Add Ntuples to the list
  fList = new TList();
  fList->Add(fMSDEx);
  fList->Add(fMSDEy);
  fList->Add(fMSDEz);
  fList->Add(fMSDEp);
}

//________________________________________________________________________
void AliMUONAlignmentTask::Exec(Option_t *) 
{
  /// Main loop, called for each event
  if (!fESD) {
    Printf("ERROR: fESD not available");
    return;
  }

  Double_t trackParams[8] = {0.,0.,0.,0.,0.,0.,0.,0.};
  if (fESD->GetRunNumber()!=fLastRunNumber){
    fLastRunNumber = fESD->GetRunNumber();
    Prepare(fGeoFilename.Data(),fDefaultOCDB.Data(),fMisAlignOCDB.Data());
  }
 
  Int_t nTracks = Int_t(fESD->GetNumberOfMuonTracks());
  //  if (!event%100) cout << " there are " << nTracks << " tracks in event " << event << endl;
  for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
    AliESDMuonTrack* esdTrack = fESD->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);
      fAlign->ProcessTrack(&track);
      fAlign->LocalFit(fTrackOk++,trackParams,0);
    }
    fTrackTot++;
    cout << "Processed " << fTrackTot << " Tracks." << endl;
  }
  
  // Post final data. Write histo list to a file with option "RECREATE"
  PostData(0,fList);
}      

//________________________________________________________________________
void AliMUONAlignmentTask::FinishTaskOutput()
{
  /// Called once per task on the client machine at the end of the analysis.

  cout << "Processed " << fTrackTot << " Tracks." << endl;
  // Perform global fit
  fAlign->GlobalFit(fParameters,fErrors,fPulls);

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

//   // Update pointers reading them from the output slot
//   fList = (TList*)GetOutputData(0);
//   fMSDEx = (TGraphErrors*)fList->At(0);
//   fMSDEy = (TGraphErrors*)fList->At(1);
//   fMSDEz = (TGraphErrors*)fList->At(2);
//   fMSDEp = (TGraphErrors*)fList->At(3);

  // 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]=fParameters[3*iDE+0];
    msdEy[iDE]=fParameters[3*iDE+1];
    msdEz[iDE]=fParameters[3*iDE+3];
    msdEp[iDE]=fParameters[3*iDE+2];
    msdExErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+0);
    msdEyErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+1);
    msdEzErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+3);
    msdEpErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+2);
    fMSDEx->SetPoint(iDE,deId[iDE],fParameters[3*iDE+0]);
    fMSDEx->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+0));
    fMSDEy->SetPoint(iDE,deId[iDE],fParameters[3*iDE+1]);
    fMSDEy->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+1));
    fMSDEp->SetPoint(iDE,deId[iDE],fParameters[3*iDE+2]);
    fMSDEp->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+2));
    fMSDEz->SetPoint(iDE,deId[iDE],fParameters[3*iDE+3]);
    fMSDEz->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+3));
  }

  // Post final data. Write histo list to a file with option "RECREATE"
  PostData(0,fList);

  // Re Align
  AliMUONGeometryTransformer *newTransform = fAlign->ReAlign(fTransform,fParameters,true); 
  newTransform->WriteTransformations("transform2ReAlign.dat");
  
  // Generate realigned data in local cdb
  const TClonesArray* array = newTransform->GetMisAlignmentData();

  // 100 mum residual resolution for chamber misalignments?
  fAlign->SetAlignmentResolution(array,-1,0.01,0.01,0.004,0.003);
   
  // CDB manager
  AliCDBManager* cdbManager = AliCDBManager::Instance();
  cdbManager->SetDefaultStorage(fDefaultOCDB.Data());
  cdbManager->SetSpecificStorage("MUON/Align/Data",fMisAlignOCDB.Data());
  
  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);

}

//________________________________________________________________________
void AliMUONAlignmentTask::Terminate(const Option_t*)
{
  /// Called once per task on the client machine at the end of the analysis.

}

//-----------------------------------------------------------------------
void AliMUONAlignmentTask::Prepare(const char* geoFilename, const char* defaultOCDB, const char* misAlignOCDB)
{
  /// Set the geometry, the magnetic field, the mapping and the reconstruction parameters
      
  // Load mapping
  AliCDBManager* man = AliCDBManager::Instance();
  man->SetDefaultStorage(defaultOCDB);
  man->SetSpecificStorage("MUON/Align/Data",misAlignOCDB);
  man->Print();
  man->SetRun(fLastRunNumber);
  if ( ! AliMpCDB::LoadDDLStore() ) {
    Error("MUONRefit","Could not access mapping from OCDB !");
    exit(-1);
  }

  // Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
  if (!gGeoManager) {
    AliGeomManager::LoadGeometry(geoFilename);
    if (!gGeoManager) {
      Error("AliMUONReAlignTask", "getting geometry from file %s failed", "generated/galice.root");
      return;
    }
  }

  // set mag field
  if (!TGeoGlobalMagField::Instance()->GetField()) {
    printf("Loading field map...\n");
    AliGRPManager *grpMan = new AliGRPManager();
    grpMan->ReadGRPEntry();
    grpMan->SetMagField();
    delete grpMan;
  }
  // set the magnetic field for track extrapolations
  AliMUONTrackExtrap::SetField();
  
}
Commit	Line	Data
81f1d3ae	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
4d610fd5	18	//-----------------------------------------------------------------------------
	19	/// \class AliMUONAlignmentTask
	20	/// AliAnalysisTask to align the MUON spectrometer.
	21	/// The Task reads as input ESDs and feeds the MUONTracks to AliMUONAlignment.
	22	/// The alignment itself is performed by AliMillepede.
	23	/// A OCDB entry is written with the alignment parameters.
	24	/// A list of graph are written to monitor the alignment parameters.
	25	///
	26	/// \author Javier Castillo, CEA/Saclay - Irfu/SPhN
	27	//-----------------------------------------------------------------------------
	28
	29	#include <fstream>
	30
	31	#include <TString.h>
2df5c2cf	32	#include <TError.h>
4d610fd5	33	#include <TGraphErrors.h>
	34	#include <TTree.h>
	35	#include <TChain.h>
	36	#include <TClonesArray.h>
	37	#include <TGeoGlobalMagField.h>
cd8521dd	38	#include <TGeoManager.h>
4d610fd5	39	#include <Riostream.h>
	40
	41	#include "AliAnalysisTask.h"
	42	#include "AliAnalysisManager.h"
	43	#include "AliESDInputHandler.h"
	44	#include "AliESDEvent.h"
	45	#include "AliESDMuonTrack.h"
	46	#include "AliMagF.h"
	47	#include "AliCDBManager.h"
cd8521dd	48	#include "AliGRPManager.h"
4d610fd5	49	#include "AliCDBMetaData.h"
	50	#include "AliCDBId.h"
	51	#include "AliGeomManager.h"
	52
	53	#include "AliMpCDB.h"
	54	#include "AliMUONAlignment.h"
	55	#include "AliMUONTrack.h"
	56	#include "AliMUONTrackExtrap.h"
	57	#include "AliMUONTrackParam.h"
	58	#include "AliMUONGeometryTransformer.h"
	59	#include "AliMUONESDInterface.h"
	60
	61	#include "AliMUONAlignmentTask.h"
	62
	63	///\cond CLASSIMP
	64	ClassImp(AliMUONAlignmentTask)
	65	///\endcond
	66
	67	// //________________________________________________________________________
	68	// AliMUONAlignmentTask::AliMUONAlignmentTask(const char *name)
	69	// : AliAnalysisTask(name, ""),
	70	// fESD(0x0),
	71	// fAlign(0x0),
	72	// fGeoFilename(""),
	73	// fTransform(0x0),
	74	// fTrackTot(0),
	75	// fTrackOk(0),
	76	// fMSDEx(0x0),
	77	// fMSDEy(0x0),
	78	// fMSDEz(0x0),
	79	// fMSDEp(0x0),
	80	// fList(0x0)
	81	// {
	82	// /// Default Constructor
	83	// // Define input and output slots here
	84	// // Input slot #0 works with a TChain
	85	// DefineInput(0, TChain::Class());
	86	// // Output slot #0 writes NTuple/histos into a TList
	87	// DefineOutput(0, TList::Class());
	88
	89	// // initialize parameters ...
	90	// for(Int_t k=0;k<4*156;k++) {
	91	// fParameters[k]=0.;
	92	// fErrors[k]=0.;
	93	// fPulls[k]=0.;
	94	// }
	95
	96	// fAlign = new AliMUONAlignment();
	97	// fTransform = new AliMUONGeometryTransformer();
	98	// }
	99
	100	//________________________________________________________________________
cd8521dd	101	AliMUONAlignmentTask::AliMUONAlignmentTask(const char name, const char geofilename, const char defaultocdb, const char misalignocdb)
4d610fd5	102	: AliAnalysisTask(name, ""),
	103	fESD(0x0),
	104	fAlign(0x0),
	105	fGeoFilename(geofilename),
cd8521dd	106	fMisAlignOCDB(misalignocdb),
cd8521dd	107	fDefaultOCDB(defaultocdb),
4d610fd5	108	fTransform(0x0),
	109	fTrackTot(0),
	110	fTrackOk(0),
2df5c2cf	111	fLastRunNumber(-1),
4d610fd5	112	fMSDEx(0x0),
	113	fMSDEy(0x0),
	114	fMSDEz(0x0),
	115	fMSDEp(0x0),
	116	fList(0x0)
	117	{
	118	/// Default Constructor
	119	// Define input and output slots here
	120	// Input slot #0 works with a TChain
	121	DefineInput(0, TChain::Class());
	122	// Output slot #0 writes NTuple/histos into a TList
	123	DefineOutput(0, TList::Class());
	124
	125	// initialize parameters ...
	126	for(Int_t k=0;k<4*156;k++) {
	127	fParameters[k]=0.;
	128	fErrors[k]=0.;
	129	fPulls[k]=0.;
	130	}
	131
	132	fAlign = new AliMUONAlignment();
	133	fTransform = new AliMUONGeometryTransformer();
	134	}
	135
	136
	137	//________________________________________________________________________
	138	AliMUONAlignmentTask::AliMUONAlignmentTask(const AliMUONAlignmentTask& obj)
	139	: AliAnalysisTask(obj),
	140	fESD(0x0),
	141	fAlign(0x0),
	142	fGeoFilename(""),
cd8521dd	143	fMisAlignOCDB(""),
cd8521dd	144	fDefaultOCDB(""),
4d610fd5	145	fTransform(0x0),
	146	fTrackTot(0),
	147	fTrackOk(0),
2df5c2cf	148	fLastRunNumber(-1),
4d610fd5	149	fMSDEx(0x0),
	150	fMSDEy(0x0),
	151	fMSDEz(0x0),
	152	fMSDEp(0x0),
	153	fList(0x0)
	154	{
	155	/// Copy constructor
	156	fESD = obj.fESD;
	157	fAlign = obj.fAlign;
	158	fGeoFilename = obj.fGeoFilename;
	159	fTransform = obj.fTransform;
	160	fTrackTot = obj.fTrackTot;
	161	fTrackOk = obj.fTrackOk;
2df5c2cf	162	fLastRunNumber = obj.fLastRunNumber;
4d610fd5	163	fMSDEx = obj.fMSDEx;
	164	fMSDEy = obj.fMSDEy;
	165	fMSDEz = obj.fMSDEz;
	166	fMSDEp = obj.fMSDEp;
	167	fList = obj.fList;
	168	}
	169
	170	//________________________________________________________________________
	171	AliMUONAlignmentTask& AliMUONAlignmentTask::operator=(const AliMUONAlignmentTask& other)
	172	{
	173	/// Assignment
	174	AliAnalysisTask::operator=(other);
	175	fESD = other.fESD;
	176	fAlign = other.fAlign;
	177	fGeoFilename = other.fGeoFilename;
cd8521dd	178	fMisAlignOCDB = other.fMisAlignOCDB;
cd8521dd	179	fDefaultOCDB = other.fDefaultOCDB;
4d610fd5	180	fTransform = other.fTransform;
	181	fTrackTot = other.fTrackTot;
	182	fTrackOk = other.fTrackOk;
2df5c2cf	183	fLastRunNumber = other.fLastRunNumber;
4d610fd5	184	fMSDEx = other.fMSDEx;
	185	fMSDEy = other.fMSDEy;
	186	fMSDEz = other.fMSDEz;
	187	fMSDEp = other.fMSDEp;
	188	fList = other.fList;
	189
	190	return *this;
	191	}
	192
	193	//________________________________________________________________________
	194	AliMUONAlignmentTask::~AliMUONAlignmentTask()
	195	{
	196	/// Destructor
	197	if (fAlign) delete fAlign;
	198	if (fTransform) delete fTransform;
	199	}
	200
	201	//________________________________________________________________________
	202	void AliMUONAlignmentTask::LocalInit()
	203	{
	204	/// Local initialization, called once per task on the client machine
	205	/// where the analysis train is assembled
2df5c2cf	206	fLastRunNumber = 0;
	207	// Prepare(fGeoFilename.Data(),fDefaultOCDB.Data(),fMisAlignOCDB.Data());
	208	Prepare(fGeoFilename.Data(),"local://$ALICE_ROOT/OCDB",fMisAlignOCDB.Data());
	209	fLastRunNumber = -1;
4d610fd5	210
	211	// Set initial values here, good guess may help convergence
	212	// St 1
	213	// Int_t iPar = 0;
	214	// fParameters[iPar++] = 0.010300 ; fParameters[iPar++] = 0.010600 ; fParameters[iPar++] = 0.000396 ;
	215
	216
	217	fAlign->InitGlobalParameters(fParameters);
	218
	219
	220	fTransform->LoadGeometryData();
	221	fAlign->SetGeometryTransformer(fTransform);
	222
	223	// Do alignment with magnetic field off
	224	fAlign->SetBFieldOn(kFALSE);
	225
	226	// Set tracking station to use
	227	// Bool_t bStOnOff[5] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
	228	Bool_t bChOnOff[10] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kFALSE,kTRUE,kTRUE,kTRUE,kTRUE};
	229
	230	// Set degrees of freedom to align (see AliMUONAlignment)
	231	fAlign->AllowVariations(bChOnOff);
	232
	233	// Fix parameters or add constraints here
	234	// for (Int_t iSt=0; iSt<5; iSt++)
	235	// if (!bStOnOff[iSt]) fAlign->FixStation(iSt+1);
	236	for (Int_t iCh=0; iCh<10; iCh++)
	237	if (!bChOnOff[iCh]) fAlign->FixChamber(iCh+1);
	238
	239	// Left and right sides of the detector are independent, one can choose to align
	240	// only one side
	241	Bool_t bSpecLROnOff[2] = {kTRUE,kTRUE};
	242	fAlign->FixHalfSpectrometer(bChOnOff,bSpecLROnOff);
	243
	244	fAlign->SetChOnOff(bChOnOff);
	245	fAlign->SetSpecLROnOff(bChOnOff);
	246
	247	// Here we can fix some detection elements
	248	fAlign->FixDetElem(908);
	249	fAlign->FixDetElem(1020);
	250
	251	// Set predifined global constrains: X, Y, P, XvsZ, YvsZ, PvsZ, XvsY, YvsY, PvsY
	252	// Bool_t bVarXYT[9] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
	253	// Bool_t bDetTLBR[4] = {kFALSE,kTRUE,kFALSE,kTRUE};
	254	// fAlign->AddConstraints(bChOnOff,bVarXYT,bDetTLBR,bSpecLROnOff);
	255
	256	}
	257
	258	//________________________________________________________________________
	259	void AliMUONAlignmentTask::ConnectInputData(Option_t *)
	260	{
	261	/// Connect ESD here. Called on each input data change.
	262
	263	// Connect ESD here
	264	TTree* esdTree = dynamic_cast<TTree*> (GetInputData(0));
	265	if (!esdTree) {
	266	Printf("ERROR: Could not read chain from input slot 0");
	267	}
	268	else {
	269	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	270	if (!esdH) {
	271	Printf("ERROR: Could not get ESDInputHandler");
	272	}
	273	else {
274	fESD = esdH->GetEvent();
275	}
276	}
277	}
278
279	//________________________________________________________________________
280	void AliMUONAlignmentTask::CreateOutputObjects()
281	{
282	/// Executed once on each worker (machine actually running the analysis code)
283	//
284	// This method has to be called INSIDE the user redefined CreateOutputObjects
285	// method, before creating each object corresponding to the output containers
286	// that are to be written to a file. This need to be done in general for the big output
287	// objects that may not fit memory during processing.
288	// OpenFile(0);
289
290	// Creating graphs
291	fMSDEx = new TGraphErrors(156);
292	fMSDEy = new TGraphErrors(156);
293	fMSDEz = new TGraphErrors(156);
294	fMSDEp = new TGraphErrors(156);
295
296	// Add Ntuples to the list
297	fList = new TList();
298	fList->Add(fMSDEx);
299	fList->Add(fMSDEy);
300	fList->Add(fMSDEz);
301	fList->Add(fMSDEp);
302	}
303
304	//________________________________________________________________________
305	void AliMUONAlignmentTask::Exec(Option_t *)
306	{
307	/// Main loop, called for each event
308	if (!fESD) {
309	Printf("ERROR: fESD not available");
310	return;
311	}
312
313	Double_t trackParams[8] = {0.,0.,0.,0.,0.,0.,0.,0.};
2df5c2cf	314	if (fESD->GetRunNumber()!=fLastRunNumber){
	315	fLastRunNumber = fESD->GetRunNumber();
	316	Prepare(fGeoFilename.Data(),fDefaultOCDB.Data(),fMisAlignOCDB.Data());
	317	}
4d610fd5	318
	319	Int_t nTracks = Int_t(fESD->GetNumberOfMuonTracks());
	320	// if (!event%100) cout << " there are " << nTracks << " tracks in event " << event << endl;
	321	for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
	322	AliESDMuonTrack* esdTrack = fESD->GetMuonTrack(iTrack);
	323	if (!esdTrack->ClustersStored()) continue;
	324	Double_t invBenMom = esdTrack->GetInverseBendingMomentum();
	325	// fInvBenMom->Fill(invBenMom);
	326	// fBenMom->Fill(1./invBenMom);
	327	if (TMath::Abs(invBenMom)<=1.04) {
	328	AliMUONTrack track;
	329	AliMUONESDInterface::ESDToMUON(*esdTrack, track);
	330	fAlign->ProcessTrack(&track);
	331	fAlign->LocalFit(fTrackOk++,trackParams,0);
	332	}
	333	fTrackTot++;
2df5c2cf	334	cout << "Processed " << fTrackTot << " Tracks." << endl;
4d610fd5	335	}
	336
	337	// Post final data. Write histo list to a file with option "RECREATE"
	338	PostData(0,fList);
	339	}
	340
	341	//________________________________________________________________________
2df5c2cf	342	void AliMUONAlignmentTask::FinishTaskOutput()
4d610fd5	343	{
	344	/// Called once per task on the client machine at the end of the analysis.
	345
	346	cout << "Processed " << fTrackTot << " Tracks." << endl;
	347	// Perform global fit
	348	fAlign->GlobalFit(fParameters,fErrors,fPulls);
	349
	350	cout << "Done with GlobalFit " << endl;
	351
2df5c2cf	352	// // Update pointers reading them from the output slot
	353	// fList = (TList*)GetOutputData(0);
	354	// fMSDEx = (TGraphErrors*)fList->At(0);
	355	// fMSDEy = (TGraphErrors*)fList->At(1);
	356	// fMSDEz = (TGraphErrors*)fList->At(2);
	357	// fMSDEp = (TGraphErrors*)fList->At(3);
4d610fd5	358
4d610fd5	359	// Store results
9ee1d6ff	360	Double_t deId[156] = {0};
	361	Double_t msdEx[156] = {0};
	362	Double_t msdEy[156] = {0};
	363	Double_t msdEz[156] = {0};
	364	Double_t msdEp[156] = {0};
	365	Double_t deIdErr[156] = {0};
	366	Double_t msdExErr[156] = {0};
	367	Double_t msdEyErr[156] = {0};
	368	Double_t msdEzErr[156] = {0};
	369	Double_t msdEpErr[156] = {0};
4d610fd5	370	Int_t lNDetElem = 42+42+182+262+26*2;
	371	Int_t lNDetElemCh[10] = {4,4,4,4,18,18,26,26,26,26};
	372	// Int_t lSNDetElemCh[10] = {4,8,12,16,34,52,78,104,130,156};
	373	Int_t idOffset = 0; // 400
	374	Int_t lSDetElemCh = 0;
	375	for(Int_t iDE=0; iDE<lNDetElem; iDE++){
9ee1d6ff	376	deIdErr[iDE] = 0.;
	377	deId[iDE] = idOffset+100;
	378	deId[iDE] += iDE;
4d610fd5	379	lSDetElemCh = 0;
	380	for(Int_t iCh=0; iCh<9; iCh++){
	381	lSDetElemCh += lNDetElemCh[iCh];
	382	if (iDE>=lSDetElemCh) {
9ee1d6ff	383	deId[iDE] += 100;
9ee1d6ff	384	deId[iDE] -= lNDetElemCh[iCh];
4d610fd5	385	}
4d610fd5	386	}
9ee1d6ff	387	msdEx[iDE]=fParameters[3*iDE+0];
	388	msdEy[iDE]=fParameters[3*iDE+1];
	389	msdEz[iDE]=fParameters[3*iDE+3];
	390	msdEp[iDE]=fParameters[3*iDE+2];
	391	msdExErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+0);
	392	msdEyErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+1);
	393	msdEzErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+3);
	394	msdEpErr[iDE]=(Double_t)fAlign->GetParError(3*iDE+2);
	395	fMSDEx->SetPoint(iDE,deId[iDE],fParameters[3*iDE+0]);
	396	fMSDEx->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+0));
	397	fMSDEy->SetPoint(iDE,deId[iDE],fParameters[3*iDE+1]);
	398	fMSDEy->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+1));
	399	fMSDEp->SetPoint(iDE,deId[iDE],fParameters[3*iDE+2]);
	400	fMSDEp->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+2));
	401	fMSDEz->SetPoint(iDE,deId[iDE],fParameters[3*iDE+3]);
	402	fMSDEz->SetPoint(iDE,deIdErr[iDE],(Double_t)fAlign->GetParError(3*iDE+3));
4d610fd5	403	}
	404
	405	// Post final data. Write histo list to a file with option "RECREATE"
	406	PostData(0,fList);
	407
4d610fd5	408	// Re Align
	409	AliMUONGeometryTransformer *newTransform = fAlign->ReAlign(fTransform,fParameters,true);
	410	newTransform->WriteTransformations("transform2ReAlign.dat");
	411
	412	// Generate realigned data in local cdb
	413	const TClonesArray* array = newTransform->GetMisAlignmentData();
	414
	415	// 100 mum residual resolution for chamber misalignments?
	416	fAlign->SetAlignmentResolution(array,-1,0.01,0.01,0.004,0.003);
	417
	418	// CDB manager
	419	AliCDBManager* cdbManager = AliCDBManager::Instance();
cd8521dd	420	cdbManager->SetDefaultStorage(fDefaultOCDB.Data());
cd8521dd	421	cdbManager->SetSpecificStorage("MUON/Align/Data",fMisAlignOCDB.Data());
4d610fd5	422
	423	AliCDBMetaData* cdbData = new AliCDBMetaData();
	424	cdbData->SetResponsible("Dimuon Offline project");
	425	cdbData->SetComment("MUON alignment objects with residual misalignment");
	426	AliCDBId id("MUON/Align/Data", 0, AliCDBRunRange::Infinity());
	427	cdbManager->Put(const_cast<TClonesArray*>(array), id, cdbData);
	428
	429	}
	430
2df5c2cf	431	//________________________________________________________________________
	432	void AliMUONAlignmentTask::Terminate(const Option_t*)
	433	{
	434	/// Called once per task on the client machine at the end of the analysis.
	435
	436	}
	437
cd8521dd	438	//-----------------------------------------------------------------------
	439	void AliMUONAlignmentTask::Prepare(const char* geoFilename, const char* defaultOCDB, const char* misAlignOCDB)
	440	{
	441	/// Set the geometry, the magnetic field, the mapping and the reconstruction parameters
2df5c2cf	442
cd8521dd	443	// Load mapping
	444	AliCDBManager* man = AliCDBManager::Instance();
	445	man->SetDefaultStorage(defaultOCDB);
	446	man->SetSpecificStorage("MUON/Align/Data",misAlignOCDB);
	447	man->Print();
2df5c2cf	448	man->SetRun(fLastRunNumber);
cd8521dd	449	if ( ! AliMpCDB::LoadDDLStore() ) {
2df5c2cf	450	Error("MUONRefit","Could not access mapping from OCDB !");
cd8521dd	451	exit(-1);
	452	}
	453
2df5c2cf	454	// Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
	455	if (!gGeoManager) {
	456	AliGeomManager::LoadGeometry(geoFilename);
	457	if (!gGeoManager) {
	458	Error("AliMUONReAlignTask", "getting geometry from file %s failed", "generated/galice.root");
	459	return;
	460	}
	461	}
	462
cd8521dd	463	// set mag field
	464	if (!TGeoGlobalMagField::Instance()->GetField()) {
	465	printf("Loading field map...\n");
	466	AliGRPManager *grpMan = new AliGRPManager();
	467	grpMan->ReadGRPEntry();
	468	grpMan->SetMagField();
	469	delete grpMan;
	470	}
	471	// set the magnetic field for track extrapolations
	472	AliMUONTrackExtrap::SetField();
	473
	474	}