Added support macro AddMisalignmentToClusters.C

diff --git a/MFT/AddMisalignmentToClusters.C b/MFT/AddMisalignmentToClusters.C
new file mode 100644 (file)
index 0000000..b8913b0
--- /dev/null
+++ b/MFT/AddMisalignmentToClusters.C
@@ -0,0 +1,89 @@
+#include "AliMFTCluster.h"
+#include "TFile.h"
+#include "AliMFTCluster.h"
+#include "TObjArray.h"
+#include "TClonesArray.h"
+#include "TTree.h"
+#include "AliMFTConstants.h"
+#include "TBranch.h"
+#include "TRandom.h"
+
+TFile *fileIn=0, *fileOut=0;
+
+const Int_t labelMCOffset = 1000000;
+Double_t misalignmentX[AliMFTConstants::fNMaxPlanes] = {0};
+Double_t misalignmentY[AliMFTConstants::fNMaxPlanes] = {0};
+
+//====================================================================================================================================================
+
+void AddMisalignmentToClusters(Char_t *nameDir=".",
+                              Int_t seed = 12345,
+                              Double_t misalignment = 0.0015) {
+
+  gRandom -> SetSeed(seed);
+
+  TClonesArray *fRecPointsPerPlaneIn[AliMFTConstants::fNMaxPlanes] = {0};
+  TClonesArray *fRecPointsPerPlaneOut[AliMFTConstants::fNMaxPlanes] = {0};
+
+  fileIn = new TFile(Form("%s/MFT.RecPoints.root",nameDir));
+  fileOut = new TFile(Form("%s/MFT.RecPoints.Misaligned.root",nameDir), "recreate");
+  
+  Int_t iEv=0;
+  while (fileIn->cd(Form("Event%d",iEv))) {
+
+    for (Int_t iPlane=0; iPlane<AliMFTConstants::fNMaxPlanes; iPlane++) {
+      fRecPointsPerPlaneIn[iPlane] = new TClonesArray("AliMFTCluster");
+      fRecPointsPerPlaneOut[iPlane] = new TClonesArray("AliMFTCluster");
+    }
+
+    printf("Event %d\n",iEv);
+    TTree *treeIn = (TTree*) gDirectory->Get("TreeR");
+    treeIn -> SetName("TreeR_In");
+    fileOut-> cd();
+    TTree *treeOut = new TTree("TreeR", "Reconstructed Points Container");
+
+    Int_t iPlane=0;
+    while (treeIn->GetBranch(Form("Plane_%02d",iPlane))) {
+      //      printf("Plane %02d\n",iPlane);
+      Double_t misalignmentPhi = 2.*TMath::Pi()*gRandom->Rndm();
+      misalignmentX[iPlane] = misalignment*TMath::Cos(misalignmentPhi);
+      misalignmentY[iPlane] = misalignment*TMath::Sin(misalignmentPhi);
+      treeIn ->SetBranchAddress(Form("Plane_%02d",iPlane), &(fRecPointsPerPlaneIn[iPlane]));
+      treeOut->Branch(Form("Plane_%02d",iPlane), &(fRecPointsPerPlaneOut[iPlane])); 
+      //      treeOut->SetBranchAddress(Form("Plane_%02d",iPlane), &(fRecPointsPerPlaneOut[iPlane]));
+      iPlane++;
+    }
+
+    iPlane=0;
+    treeIn -> GetEntry(0);
+    while (treeIn->GetBranch(Form("Plane_%02d",iPlane))) {
+      Int_t nClusters = fRecPointsPerPlaneIn[iPlane]->GetEntries();
+      for (Int_t iCluster=0; iCluster<nClusters; iCluster++) {
+       //      printf("Cluster %4d\n",iCluster);
+       AliMFTCluster *newCluster = (AliMFTCluster*) fRecPointsPerPlaneIn[iPlane]->At(iCluster);
+       newCluster->SetClusterEditable(kTRUE);
+       newCluster->SetX(newCluster->GetX()+misalignmentX[iPlane]);
+       newCluster->SetY(newCluster->GetY()+misalignmentY[iPlane]);
+       new ((*fRecPointsPerPlaneOut[iPlane])[fRecPointsPerPlaneOut[iPlane]->GetEntries()]) AliMFTCluster(*newCluster);
+      }
+      iPlane++;
+    }
+
+    treeOut -> Fill();
+    fileOut -> mkdir(Form("Event%d",iEv));
+    fileOut -> cd(Form("Event%d",iEv));
+    treeOut -> Write();
+    for (Int_t jPlane=0; jPlane<AliMFTConstants::fNMaxPlanes; jPlane++) {
+      delete fRecPointsPerPlaneIn[jPlane];
+      delete fRecPointsPerPlaneOut[jPlane];
+    }
+
+    iEv++;
+
+  }
+
+  fileOut -> Close();
+
+}
+
+//====================================================================================================================================================