[u/mrichter/AliRoot.git] / MUON / ReadRecoCocktail.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 ReadRecoCocktail.C
/// \brief Base macro to read the trees generated with DecodeRecoCocktail.C 
///
/// \author A. De Falco, H. Woehri, INFN Cagliari, July 2006 

#if !defined(__CINT__) || defined(__MAKECINT__)
#include "TFile.h"
#include "TTree.h"
#include "TCanvas.h"
#include "TClonesArray.h"

#include "AliMUONTrackLight.h"
#include "AliMUONPairLight.h"
#endif

void ReadRecoCocktail(char* fname="./MuonLight.root"){ 
  TFile *file = new TFile(fname); 
  TClonesArray *muonArray   = new TClonesArray("AliMUONTrackLight",100); 
  TClonesArray *dimuonArray = new TClonesArray("AliMUONPairLight",100); 
  TTree *tree = (TTree*) file->Get("tree"); 
  tree->SetBranchAddress("muons",&muonArray); 
  tree->SetBranchAddress("dimuons",&dimuonArray); 
  Int_t nev = tree->GetEntriesFast(); 
  printf ("%d events in tree\n",nev); 
  Int_t ndimuUncorr=0; 
  for (Int_t iev=0; iev<nev; iev++) { 
    tree->GetEvent(iev); 
//     Int_t nmu = muonArray->GetEntries(); 
//     for (Int_t imu=0; imu<nmu; imu++) { 
//       AliMUONTrackLight *mu = (AliMUONTrackLight*) muonArray->At(imu); 
//       mu->Dump(); 
//     }
    Int_t ndimu = dimuonArray->GetEntriesFast(); 
    for (Int_t idimu=0; idimu<ndimu; idimu++) { 
      AliMUONPairLight *dimu = (AliMUONPairLight*) dimuonArray->At(idimu); 
      //      dimu->Dump(); 
      if (!dimu->IsCorrelated()) { 
	printf ("Event %d / %d; dimuon %d not correlated\n",iev,nev,idimu); 
	dimu->PrintInfo("H"); 
	ndimuUncorr++; 
      }
    }
  }
  printf ("%d uncorrelated dimuons in file\n",ndimuUncorr);  
  TCanvas *c1=new TCanvas(); 
  c1->Divide(2);
  printf ("plotting the mass and pt spectra of opposite sign dimuons\n");
  c1->cd(1);
  tree->Draw("dimuons.GetPRec().M()","dimuons.GetCharge()==0");
  c1->cd(2);
  tree->Draw("dimuons.GetPRec().Pt()","dimuons.GetCharge()==0");
}
Commit	Line	Data
fa6e7866	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 ReadRecoCocktail.C
	20	/// \brief Base macro to read the trees generated with DecodeRecoCocktail.C
	21	///
	22	/// \author A. De Falco, H. Woehri, INFN Cagliari, July 2006
fa6e7866	23
3c226639	24	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
	25	#include "TFile.h"
	26	#include "TTree.h"
	27	#include "TCanvas.h"
	28	#include "TClonesArray.h"
	29
	30	#include "AliMUONTrackLight.h"
	31	#include "AliMUONPairLight.h"
	32	#endif
	33
fa6e7866	34	void ReadRecoCocktail(char* fname="./MuonLight.root"){
	35	TFile *file = new TFile(fname);
	36	TClonesArray *muonArray = new TClonesArray("AliMUONTrackLight",100);
	37	TClonesArray *dimuonArray = new TClonesArray("AliMUONPairLight",100);
	38	TTree tree = (TTree) file->Get("tree");
	39	tree->SetBranchAddress("muons",&muonArray);
	40	tree->SetBranchAddress("dimuons",&dimuonArray);
	41	Int_t nev = tree->GetEntriesFast();
	42	printf ("%d events in tree\n",nev);
	43	Int_t ndimuUncorr=0;
	44	for (Int_t iev=0; iev<nev; iev++) {
	45	tree->GetEvent(iev);
3c226639	46	// Int_t nmu = muonArray->GetEntries();
	47	// for (Int_t imu=0; imu<nmu; imu++) {
	48	// AliMUONTrackLight mu = (AliMUONTrackLight) muonArray->At(imu);
	49	// mu->Dump();
	50	// }
fa6e7866	51	Int_t ndimu = dimuonArray->GetEntriesFast();
	52	for (Int_t idimu=0; idimu<ndimu; idimu++) {
	53	AliMUONPairLight dimu = (AliMUONPairLight) dimuonArray->At(idimu);
	54	// dimu->Dump();
	55	if (!dimu->IsCorrelated()) {
	56	printf ("Event %d / %d; dimuon %d not correlated\n",iev,nev,idimu);
	57	dimu->PrintInfo("H");
	58	ndimuUncorr++;
	59	}
	60	}
	61	}
	62	printf ("%d uncorrelated dimuons in file\n",ndimuUncorr);
	63	TCanvas *c1=new TCanvas();
	64	c1->Divide(2);
	65	printf ("plotting the mass and pt spectra of opposite sign dimuons\n");
	66	c1->cd(1);
	67	tree->Draw("dimuons.GetPRec().M()","dimuons.GetCharge()==0");
	68	c1->cd(2);
	69	tree->Draw("dimuons.GetPRec().Pt()","dimuons.GetCharge()==0");
	70	}