]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - RALICE/AliVertex.cxx
disable doxygen latex output
[u/mrichter/AliRoot.git] / RALICE / AliVertex.cxx
index dde5f0f53a6a5f7a09e206af833b17a4e85b7f39..1d142c7da997bed3e2f46966b891b223b0dabc50 100644 (file)
 //        Float_t pos[3]={7,9,4};
 //        v1.SetPosition(pos,"car");
 //
-// Note : All quantities are in GeV, GeV/c or GeV/c**2
+// Note : By default all quantities are in meter, GeV, GeV/c or GeV/c**2
+//        but the user can indicate the usage of a different scale for
+//        the metric and/or energy-momentum units via the SetUnitScale()
+//        and SetEscale() memberfunctions, respectively.
+//        The actual metric and energy-momentum unit scales in use can be
+//        obtained via the GetUnitScale() and GetEscale() memberfunctions.
 //
 //--- Author: Nick van Eijndhoven 04-apr-1998 UU-SAP Utrecht
 //- Modified: NvE $Date$ UU-SAP Utrecht
@@ -266,7 +271,7 @@ AliVertex::AliVertex(const AliVertex& v) : AliJet(v.fNtinit),AliPosition(v)
  AliTrack* tx=0;
  Int_t jetflag=0,connect=0;
  AliTrack* tx2=0;
- for (Int_t it=1; it<=v.GetNtracks(); it++)
+ for (Int_t it=1; it<=v.fNtrk; it++)
  {
   tx=v.GetTrack(it);
   if (tx)
@@ -556,9 +561,17 @@ void AliVertex::AddVertex(AliVertex& v,Int_t connect)
  }
 }
 ///////////////////////////////////////////////////////////////////////////
-void AliVertex::Data(TString f)
+void AliVertex::Data(TString f,TString u)
 {
 // Provide vertex information within the coordinate frame f
+//
+// The string argument "u" allows to choose between different angular units
+// in case e.g. a spherical frame is selected.
+// u = "rad" : angles provided in radians
+//     "deg" : angles provided in degrees
+//
+// The defaults are f="car" and u="rad".
+
  const char* name=GetName();
  const char* title=GetTitle();
  cout << " *AliVertex::Data*";
@@ -567,19 +580,27 @@ void AliVertex::Data(TString f)
  cout << endl;
  cout << " Id : " << fUserId << " Invmass : " << GetInvmass()
       << " Charge : " << GetCharge() << " Momentum : " << GetMomentum()
-      << " Ntracks : " << GetNtracks() << " Nvertices : " << fNvtx 
-      << " Njets : " << fNjets << endl;
+      << " Ntracks : " << GetNtracks() << endl;
+ cout << " Nvertices : " << fNvtx << " Njets : " << fNjets
+      << " Energy scale : " << fEscale << " GeV" << endl;
  cout << " ";
- Ali4Vector::Data(f);
+ Ali4Vector::Data(f,u);
  cout << "  Position";
- AliPosition::Data(f); 
+ AliPosition::Data(f,u); 
 } 
 ///////////////////////////////////////////////////////////////////////////
-void AliVertex::List(TString f)
+void AliVertex::List(TString f,TString u)
 {
 // Provide primary track and sec. vertex information within the coordinate frame f
+//
+// The string argument "u" allows to choose between different angular units
+// in case e.g. a spherical frame is selected.
+// u = "rad" : angles provided in radians
+//     "deg" : angles provided in degrees
+//
+// The defaults are f="car" and u="rad".
 
- Data(f); // Information of the current vertex
+ Data(f,u); // Information of the current vertex
 
  // The tracks of this vertex
  AliTrack* t; 
@@ -590,7 +611,7 @@ void AliVertex::List(TString f)
   {
    cout << "  ---Track no. " << it << endl;
    cout << " ";
-   t->Data(f); 
+   t->Data(f,u); 
   }
   else
   {
@@ -607,7 +628,7 @@ void AliVertex::List(TString f)
   {
    cout << "  ---Level 1 sec. vertex no. " << iv << endl;
    cout << " ";
-   v->Data(f); 
+   v->Data(f,u); 
   }
   else
   {
@@ -616,11 +637,18 @@ void AliVertex::List(TString f)
  }
 } 
 ///////////////////////////////////////////////////////////////////////////
-void AliVertex::ListAll(TString f)
+void AliVertex::ListAll(TString f,TString u)
 {
 // Provide complete (sec) vertex and (decay) track info within the coordinate frame f
+//
+// The string argument "u" allows to choose between different angular units
+// in case e.g. a spherical frame is selected.
+// u = "rad" : angles provided in radians
+//     "deg" : angles provided in degrees
+//
+// The defaults are f="car" and u="rad".
 
- Data(f); // Information of the current vertex
+ Data(f,u); // Information of the current vertex
 
  // The tracks of this vertex
  AliTrack* t; 
@@ -631,7 +659,7 @@ void AliVertex::ListAll(TString f)
   {
    cout << "  ---Track no. " << it << endl;
    cout << " ";
-   t->ListAll(f); 
+   t->ListAll(f,u); 
   }
   else
   {
@@ -640,10 +668,10 @@ void AliVertex::ListAll(TString f)
  }
 
  AliVertex* v=this;
- Dumps(v,1,f); // Information of all sec. vertices
+ Dumps(v,1,f,u); // Information of all sec. vertices
 }
 //////////////////////////////////////////////////////////////////////////
-void AliVertex::Dumps(AliVertex* v,Int_t n,TString f)
+void AliVertex::Dumps(AliVertex* v,Int_t n,TString f,TString u)
 {
 // Recursively provide the info of all secondary vertices of this vertex
  AliVertex* vs; 
@@ -654,7 +682,7 @@ void AliVertex::Dumps(AliVertex* v,Int_t n,TString f)
   {
    cout << "  ---Level " << n << " sec. vertex no. " << iv << endl;
    cout << " ";
-   vs->Data(f); 
+   vs->Data(f,u); 
 
    // The tracks of this vertex
    AliTrack* t; 
@@ -665,7 +693,7 @@ void AliVertex::Dumps(AliVertex* v,Int_t n,TString f)
     {
      cout << "  ---Track no. " << it << endl;
      cout << " ";
-     t->ListAll(f); 
+     t->ListAll(f,u); 
     }
     else
     {
@@ -674,7 +702,7 @@ void AliVertex::Dumps(AliVertex* v,Int_t n,TString f)
    }
 
    // Go for next sec. vertex level of this sec. vertex recursively
-   Dumps(vs,n+1,f);
+   Dumps(vs,n+1,f,u);
   }
   else
   {
@@ -992,6 +1020,161 @@ void AliVertex::Draw(Int_t secs,Int_t cons,Int_t jets)
  }
 }
 ///////////////////////////////////////////////////////////////////////////
+TObjArray* AliVertex::SortJets(Int_t mode,TObjArray* jets)
+{
+// Order the references to an array of jets by looping over the input array "jets"
+// and checking the value of a certain observable.
+// The ordered array is returned as a TObjArray.
+// In case jets=0 (default), the registered jets of the current vertex are used. 
+// Note that the original jet array is not modified.
+// Via the "mode" argument the user can specify the observable to be checked upon
+// and specify whether sorting should be performed in decreasing order (mode<0)
+// or in increasing order (mode>0).
+//
+// The convention for the observable selection is the following :
+// mode : 1 ==> Number of tracks in the jet
+//        2 ==> Jet energy
+//        3 ==> Jet momentum
+//        4 ==> Invariant mass of the jet
+//        5 ==> Transverse momentum of the jet
+//        6 ==> Longitudinal momentum of the jet
+//        7 ==> Transverse energy of the jet
+//        8 ==> Longitudinal energy of the jet
+//        9 ==> Transverse mass of the jet
+//       10 ==> Jet rapidity
+//       11 ==> Pseudo-rapidity of the jet
+//       12 ==> Number of associated signals
+//       13 ==> Total charge of the jet
+//
+// The default is mode=-1.
+//
+// Note : This sorting routine uses a common area in memory, which is used
+//        by various other sorting facilities as well.
+//        This means that the resulting sorted TObjArray may be overwritten
+//        when another sorting is invoked.
+//        To retain the sorted list of pointers, the user is advised to copy
+//        the pointers contained in the returned TObjArray into a private
+//        TObjArray instance.
+
+ if (fSelected)
+ {
+  delete fSelected;
+  fSelected=0;
+ }
+
+ if (!jets) jets=fJets;
+ if (!mode || abs(mode)>13 || !jets) return fSelected;
+
+ Int_t njets=jets->GetEntries();
+ if (!njets)
+ {
+  return fSelected;
+ }
+ else
+ {
+  fSelected=new TObjArray(njets);
+ }
+
+ Double_t val1,val2; // Values of the observable to be tested upon
+ Int_t nord=0;
+ for (Int_t i=0; i<njets; i++) // Loop over all jets of the array
+ {
+  AliJet* jx=(AliJet*)jets->At(i);
+
+  if (!jx) continue;
+  if (nord == 0) // store the first jet at the first ordered position
+  {
+   nord++;
+   fSelected->AddAt(jx,nord-1);
+   continue;
+  }
+  for (Int_t j=0; j<=nord; j++) // put jet in the right ordered position
+  {
+   if (j == nord) // jet has smallest (mode<0) or largest (mode>0) observable value seen so far
+   {
+    nord++;
+    fSelected->AddAt(jx,j); // add jet at the end
+    break; // go for next jet
+   }
+
+   val1=0;
+   val2=0;
+   
+   switch (abs(mode))
+   {
+    case 1:
+     val1=jx->GetNtracks();
+     val2=((AliJet*)fSelected->At(j))->GetNtracks();
+     break;
+    case 2:
+     val1=jx->GetEnergy(1);
+     val2=((AliJet*)fSelected->At(j))->GetEnergy(1);
+     break;
+    case 3:
+     val1=jx->GetMomentum(1);
+     val2=((AliJet*)fSelected->At(j))->GetMomentum(1);
+     break;
+    case 4:
+     val1=jx->GetInvmass(1);
+     val2=((AliJet*)fSelected->At(j))->GetInvmass(1);
+     break;
+    case 5:
+     val1=jx->GetPt(1);
+     val2=((AliJet*)fSelected->At(j))->GetPt(1);
+     break;
+    case 6:
+     val1=jx->GetPl(1);
+     val2=((AliJet*)fSelected->At(j))->GetPl(1);
+     break;
+    case 7:
+     val1=jx->GetEt(1);
+     val2=((AliJet*)fSelected->At(j))->GetEt(1);
+     break;
+    case 8:
+     val1=jx->GetEl(1);
+     val2=((AliJet*)fSelected->At(j))->GetEl(1);
+     break;
+    case 9:
+     val1=jx->GetMt(1);
+     val2=((AliJet*)fSelected->At(j))->GetMt(1);
+     break;
+    case 10:
+     val1=jx->GetRapidity();
+     val2=((AliJet*)fSelected->At(j))->GetRapidity();
+     break;
+    case 11:
+     val1=jx->GetPseudoRapidity();
+     val2=((AliJet*)fSelected->At(j))->GetPseudoRapidity();
+     break;
+    case 12:
+     val1=jx->GetNsignals();
+     val2=((AliJet*)fSelected->At(j))->GetNsignals();
+     break;
+    case 13:
+     val1=jx->GetCharge();
+     val2=((AliJet*)fSelected->At(j))->GetCharge();
+     break;
+   }
+
+   if (mode<0 && val1 <= val2) continue;
+   if (mode>0 && val1 >= val2) continue;
+   nord++;
+   for (Int_t k=nord-1; k>j; k--) // create empty position
+   {
+    fSelected->AddAt(fSelected->At(k-1),k);
+   }
+   fSelected->AddAt(jx,j); // put jet at empty position
+   break; // go for next jet
+  }
+ }
+ return fSelected;
+}
+///////////////////////////////////////////////////////////////////////////
 TObject* AliVertex::Clone(const char* name) const
 {
 // Make a deep copy of the current object and provide the pointer to the copy.