1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 ///////////////////////////////////////////////////////////////////////////
20 // Creation and investigation of an AliVertex.
21 // An AliVertex can be constructed by adding AliTracks and/or AliJets.
23 // Note : Also (secondary) vertices can be added to a vertex.
25 // To provide maximal flexibility to the user, two modes of vertex storage
26 // are provided by means of the memberfunction SetVertexCopy().
27 // The same holds for the storage of jets via SetJetCopy().
29 // a) SetVertexCopy(0) (which is the default).
30 // Only the pointers of the 'added' vertices are stored.
31 // This mode is typically used by making vertex studies based on a fixed list
32 // of vertices which stays under user control or is contained for instance
34 // In this way the AliVertex just represents a 'logical structure' for the
35 // physics analysis which can be embedded in e.g. an AliEvent or AliVertex.
38 // Modifications made to the original vertices also affect the AliVertex objects
41 // b) SetVertexCopy(1).
42 // Of every 'added' vertex a private copy will be made of which the pointer
44 // In this way the AliVertex represents an entity on its own and modifications
45 // made to the original vertices do not affect the AliVertex objects which are
47 // This mode will allow 'adding' many different AliVertex objects by
48 // creating only one AliVertex instance in the main programme and using the
49 // AliVertex::Reset, AliVertex::AddTrack and parameter setting memberfunctions.
51 // See also the documentation provided for the memberfunction SetOwner().
53 // Coding example to make 3 vertices v1, v2 and v3.
54 // ------------------------------------------------
55 // v1 contains the tracks 1,2,3 and 4
56 // v2 contains many different tracks
57 // v3 contains the jets 1 and 2
59 // AliTrack t1,t2,t3,t4;
61 // ... // code to fill the track data
66 // ... // code to fill the jet data
70 // v1.SetVertexCopy(1);
77 // Float_t r1[3]={2.4,0.1,-8.5};
78 // v1.SetPosition(r1,"car");
81 // v2.SetTrackCopy(1);
83 // AliTrack* tx=new AliTrack();
84 // for (Int_t i=0; i<10; i++)
87 // ... // code to fill the track data
93 // Float_t r2[3]={1.6,-3.2,5.7};
94 // v2.SetPosition(r2,"car");
101 // Float_t r3[3]={6.2,4.8,1.3};
102 // v3.SetPosition(r3,"car");
108 // Float_t e1=v1.GetEnergy();
109 // Ali3Vector p1=v1.Get3Momentum();
111 // v1.GetPosition(loc,"sph");
112 // AliPosition r=v2.GetPosition();
114 // Int_t nt=v2.GetNtracks();
115 // AliTrack* tv=v2.GetTrack(1); // Access track number 1 of Vertex v2
117 // Specify the vertices v2 and v3 as secondary vertices of v1
124 // Int_t nv=v1.GetNvtx();
125 // AliVertex* vx=v1.GetVertex(1); // Access 1st secondary vertex of v1
126 // Float_t e=vx->GetEnergy();
128 // Float_t M=v1.GetInvmass();
130 // Reconstruct Vertex v1 from scratch
133 // v1.SetNvmax(25); // Increase initial no. of sec. vertices
137 // Float_t pos[3]={7,9,4};
138 // v1.SetPosition(pos,"car");
140 // Note : All quantities are in GeV, GeV/c or GeV/c**2
142 //--- Author: Nick van Eijndhoven 04-apr-1998 UU-SAP Utrecht
143 //- Modified: NvE $Date$ UU-SAP Utrecht
144 ///////////////////////////////////////////////////////////////////////////
146 #include "AliVertex.h"
147 #include "Riostream.h"
149 ClassImp(AliVertex) // Class implementation to enable ROOT I/O
151 AliVertex::AliVertex() : AliJet(),AliPosition()
153 // Default constructor.
154 // All variables initialised to 0.
155 // Initial maximum number of tracks is set to the default value.
156 // Initial maximum number of sec. vertices is set to the default value.
162 ///////////////////////////////////////////////////////////////////////////
163 void AliVertex::Init()
165 // Initialisation of pointers etc...
176 ///////////////////////////////////////////////////////////////////////////
177 AliVertex::AliVertex(Int_t n) : AliJet(n),AliPosition()
179 // Create a vertex to hold initially a maximum of n tracks
180 // All variables initialised to 0
183 cout << " *** This AliJet initialisation was invoked via the AliVertex ctor." << endl;
190 ///////////////////////////////////////////////////////////////////////////
191 AliVertex::~AliVertex()
193 // Default destructor
220 ///////////////////////////////////////////////////////////////////////////
221 void AliVertex::SetOwner(Bool_t own)
223 // Set ownership of all added objects.
224 // The default parameter is own=kTRUE.
226 // Invokation of this memberfunction also sets all the copy modes
227 // (e.g. TrackCopy & co.) according to the value of own.
229 // This function (with own=kTRUE) is particularly useful when reading data
230 // from a tree/file, since Reset() will then actually remove all the
231 // added objects from memory irrespective of the copy mode settings
232 // during the tree/file creation process. In this way it provides a nice way
233 // of preventing possible memory leaks in the reading/analysis process.
235 // In addition this memberfunction can also be used as a shortcut to set all
236 // copy modes in one go during a tree/file creation process.
237 // However, in this case the user has to take care to only set/change the
238 // ownership (and copy mode) for empty objects (e.g. newly created objects
239 // or after invokation of the Reset() memberfunction) otherwise it will
240 // very likely result in inconsistent destructor behaviour.
244 if (fVertices) fVertices->SetOwner(own);
246 if (fJets) fJets->SetOwner(own);
249 AliJet::SetOwner(own);
251 ///////////////////////////////////////////////////////////////////////////
252 AliVertex::AliVertex(AliVertex& v) : AliJet(v.fNtinit),AliPosition(v)
260 SetTrackCopy(v.GetTrackCopy());
261 SetVertexCopy(v.GetVertexCopy());
262 SetJetCopy(v.GetJetCopy());
265 // Copy all tracks except the ones coming from jets
267 Int_t jetflag=0,connect=0;
269 for (Int_t it=1; it<=v.GetNtracks(); it++)
274 jetflag=v.IsJetTrack(tx);
275 connect=v.IsConnectTrack(tx);
277 if (!jetflag && !connect) AddTrack(tx);
283 fConnects=new TObjArray(fNvmax);
284 if (!fTrackCopy) fConnects->SetOwner();
286 tx2=new AliTrack(*tx);
293 // Copy all the (secondary) vertices without re-creating connecting tracks
294 // The connecting tracks have already been copied above
296 for (Int_t iv=1; iv<=v.GetNvertices(); iv++)
299 if (vx) AddVertex(vx,0);
302 // Copy all the jets including the jet tracks for these jets for which
303 // this was also the case in the original vertex
305 for (Int_t ij=1; ij<=v.GetNjets(); ij++)
311 if (jx->GetNtracks())
316 jetflag=v.IsJetTrack(tx);
323 ///////////////////////////////////////////////////////////////////////////
324 void AliVertex::SetNvmax(Int_t n)
326 // Set the initial maximum number of (secondary) vertices
341 ///////////////////////////////////////////////////////////////////////////
342 void AliVertex::SetNjmax(Int_t n)
344 // Set the initial maximum number of jets
359 ///////////////////////////////////////////////////////////////////////////
360 void AliVertex::Reset()
362 // Reset all variables to 0 and reset all stored vertex and jet lists.
363 // The max. number of tracks is set to the initial value again
364 // The max. number of vertices is set to the default value again
365 // The max. number of jets is set to the default value again
369 Double_t a[3]={0,0,0};
370 SetPosition(a,"sph");
371 SetPositionErrors(a,"car");
374 if (fNvmax>0) SetNvmax(fNvmax);
382 if (fNjmax>0) SetNjmax(fNjmax);
395 ///////////////////////////////////////////////////////////////////////////
396 void AliVertex::ResetVertices()
398 // Reset the stored vertex list and delete all connecting tracks which
399 // were generated automatically via connect=1 in AddVertex().
400 // The max. number of vertices is set to the default value again.
401 // All physics quantities are updated according to the removal of the
402 // connecting tracks.
406 for (Int_t i=0; i<=fConnects->GetLast(); i++)
408 t=(AliTrack*)fConnects->At(i);
409 AliTrack* test=(AliTrack*)fTracks->Remove(t);
413 (Ali4Vector&)(*this)-=(Ali4Vector&)(*t);
415 if (fTrackCopy) delete t;
422 if (fNvmax>0) SetNvmax(fNvmax);
429 ///////////////////////////////////////////////////////////////////////////
430 void AliVertex::AddJet(AliJet& j,Int_t tracks)
432 // Add a jet (and its tracks) to the vertex
433 // In case the maximum number of jets has been reached,
434 // the array space will be extended automatically
436 // Note : By default the tracks of the jet are added to the current (primary)
438 // The automatic addition of the tracks of the jet can be suppressed
439 // by specifying tracks=0. In this case only the AliJet object will
440 // be stored according to the mode specified by SetJetCopy().
441 // The latter will enable jet studies based on a fixed list of tracks
442 // as contained e.g. in an AliVertex or AliEvent.
445 fJets=new TObjArray(fNjmax);
446 if (fJetCopy) fJets->SetOwner();
448 if (fNjets == fNjmax) // Check if maximum jet number is reached
451 fJets->Expand(fNjmax);
454 // Add the jet to the list
456 if (fJetCopy) jx=new AliJet(j);
464 // Add the tracks of the jet to this vertex
469 fJetTracks=new TObjArray();
471 Int_t copy=1-(jx->GetTrackCopy());
473 for (Int_t i=1; i<=jx->GetNtracks(); i++)
484 ///////////////////////////////////////////////////////////////////////////
485 void AliVertex::AddVertex(AliVertex& v,Int_t connect)
487 // Add a (secondary) vertex to the current vertex.
488 // In case the maximum number of (secondary) vertices has been reached,
489 // the array space will be extended automatically
491 // Note : By default the 4-momentum and charge of the current (primary) vertex
492 // are updated by automatically creating the track connecting
493 // both vertices. The track parameters are taken from the
494 // 4-momentum and charge of the secondary vertex.
495 // The automatic creation of the connecting track and updating
496 // of the (primary) vertex 4-momentum and charge can be suppressed
497 // by specifying connect=0. In this case, however, the user
498 // has to introduce the connecting track lateron by hand
499 // explicitly in order to match the kinematics and charge.
503 fVertices=new TObjArray(fNvmax);
504 if (fVertexCopy) fVertices->SetOwner();
506 if (fNvtx == fNvmax) // Check if maximum vertex number is reached
509 fVertices->Expand(fNvmax);
512 // Add the linked (secondary) vertex to the list
514 if (fVertexCopy) vx=new AliVertex(v);
522 // Create connecting track and update 4-momentum and charge for current vertex
525 AliTrack* t=new AliTrack();
526 t->SetBeginPoint(GetPosition());
527 t->SetEndPoint(v.GetPosition());
528 t->SetCharge(v.GetCharge());
529 t->Set4Momentum((Ali4Vector&)v);
535 fConnects=new TObjArray(fNvmax);
536 if (!fTrackCopy) fConnects->SetOwner();
541 ///////////////////////////////////////////////////////////////////////////
542 void AliVertex::Data(TString f)
544 // Provide vertex information within the coordinate frame f
545 cout << " *AliVertex::Data* Id : " << fUserId << " Invmass : " << GetInvmass()
546 << " Charge : " << GetCharge() << " Momentum : " << GetMomentum()
547 << " Ntracks : " << GetNtracks() << " Nvertices : " << fNvtx
548 << " Njets : " << fNjets << endl;
552 AliPosition::Data(f);
554 ///////////////////////////////////////////////////////////////////////////
555 void AliVertex::List(TString f)
557 // Provide primary track and sec. vertex information within the coordinate frame f
559 Data(f); // Information of the current vertex
561 // The tracks of this vertex
563 for (Int_t it=1; it<=GetNtracks(); it++)
568 cout << " ---Track no. " << it << endl;
574 cout << " *AliVertex::List* Error : No track present." << endl;
578 // The secondary vertices of this vertex
580 for (Int_t iv=1; iv<=GetNvertices(); iv++)
585 cout << " ---Level 1 sec. vertex no. " << iv << endl;
591 cout << " *AliVertex::List* Error : No sec. vertex present." << endl;
595 ///////////////////////////////////////////////////////////////////////////
596 void AliVertex::ListAll(TString f)
598 // Provide complete (sec) vertex and (decay) track info within the coordinate frame f
600 Data(f); // Information of the current vertex
602 // The tracks of this vertex
604 for (Int_t it=1; it<=GetNtracks(); it++)
609 cout << " ---Track no. " << it << endl;
615 cout << " *AliVertex::ListAll* Error : No track present." << endl;
620 Dump(v,1,f); // Information of all sec. vertices
622 //////////////////////////////////////////////////////////////////////////
623 void AliVertex::Dump(AliVertex* v,Int_t n,TString f)
625 // Recursively provide the info of all secondary vertices of this vertex
627 for (Int_t iv=1; iv<=v->GetNvertices(); iv++)
632 cout << " ---Level " << n << " sec. vertex no. " << iv << endl;
636 // The tracks of this vertex
638 for (Int_t it=1; it<=vs->GetNtracks(); it++)
643 cout << " ---Track no. " << it << endl;
649 cout << " *AliVertex::Dump* Error : No track present." << endl;
653 // Go for next sec. vertex level of this sec. vertex recursively
658 cout << " *AliVertex::Dump* Error : No sec. vertex present." << endl;
662 //////////////////////////////////////////////////////////////////////////
663 Int_t AliVertex::GetNvertices()
665 // Return the current number of (secondary) vertices
668 ///////////////////////////////////////////////////////////////////////////
669 AliVertex* AliVertex::GetVertex(Int_t i)
671 // Return the i-th (secondary) vertex of the current vertex
674 cout << " *AliVertex*::GetVertex* No (secondary) vertices present." << endl;
681 cout << " *AliVertex*::GetVertex* Invalid argument i : " << i
682 << " Nvtx = " << fNvtx << endl;
687 return (AliVertex*)fVertices->At(i-1);
691 ///////////////////////////////////////////////////////////////////////////
692 AliVertex* AliVertex::GetIdVertex(Int_t id)
694 // Return the (sec.) vertex with user identifier "id"
699 cout << " *AliVertex*::GetIdVertex* No (secondary) vertices present." << endl;
704 for (Int_t i=0; i<fNvtx; i++)
706 vx=(AliVertex*)fVertices->At(i);
707 if (id == vx->GetId()) v=vx;
712 ///////////////////////////////////////////////////////////////////////////
713 void AliVertex::SetVertexCopy(Int_t j)
715 // (De)activate the creation of private copies of the added vertices.
716 // j=0 ==> No private copies are made; pointers of original vertices are stored.
717 // j=1 ==> Private copies of the vertices are made and these pointers are stored.
719 // Note : Once the storage contains pointer(s) to AliVertex objects one cannot
720 // change the VertexCopy mode anymore.
721 // To change the VertexCopy mode for an existing AliVertex containing
722 // vertices one first has to invoke Reset().
731 cout << "*AliVertex::SetVertexCopy* Invalid argument : " << j << endl;
736 cout << "*AliVertex::SetVertexCopy* Storage already contained vertices."
737 << " ==> VertexCopy mode not changed." << endl;
740 ///////////////////////////////////////////////////////////////////////////
741 Int_t AliVertex::GetVertexCopy()
743 // Provide value of the VertexCopy mode.
744 // 0 ==> No private copies are made; pointers of original vertices are stored.
745 // 1 ==> Private copies of the vertices are made and these pointers are stored.
748 ///////////////////////////////////////////////////////////////////////////
749 Int_t AliVertex::GetNjets()
751 // Return the current number of jets
754 ///////////////////////////////////////////////////////////////////////////
755 AliJet* AliVertex::GetJet(Int_t i)
757 // Return the i-th jet of the current vertex
760 cout << " *AliVertex*::GetJet* No jets present." << endl;
765 if (i<=0 || i>fNjets)
767 cout << " *AliVertex*::GetJet* Invalid argument i : " << i
768 << " Njets = " << fNjets << endl;
773 return (AliJet*)fJets->At(i-1);
777 ///////////////////////////////////////////////////////////////////////////
778 AliJet* AliVertex::GetIdJet(Int_t id)
780 // Return the jet with user identifier "id"
785 cout << " *AliVertex*::GetIdJet* No jets present." << endl;
790 for (Int_t i=0; i<fNjets; i++)
792 jx=(AliJet*)fJets->At(i);
793 if (id == jx->GetId()) j=jx;
798 ///////////////////////////////////////////////////////////////////////////
799 void AliVertex::SetJetCopy(Int_t j)
801 // (De)activate the creation of private copies of the added jets.
802 // j=0 ==> No private copies are made; pointers of original jets are stored.
803 // j=1 ==> Private copies of the jets are made and these pointers are stored.
805 // Note : Once the storage contains pointer(s) to AliJet objects one cannot
806 // change the JetCopy mode anymore.
807 // To change the JetCopy mode for an existing AliVertex containing
808 // jets one first has to invoke Reset().
817 cout << "*AliVertex::SetJetCopy* Invalid argument : " << j << endl;
822 cout << "*AliVertex::SetJetCopy* Storage already contained jets."
823 << " ==> JetCopy mode not changed." << endl;
826 ///////////////////////////////////////////////////////////////////////////
827 Int_t AliVertex::GetJetCopy()
829 // Provide value of the JetCopy mode.
830 // 0 ==> No private copies are made; pointers of original jets are stored.
831 // 1 ==> Private copies of the jets are made and these pointers are stored.
834 ///////////////////////////////////////////////////////////////////////////
835 Int_t AliVertex::IsConnectTrack(AliTrack* t)
837 // Indicate whether a track from the tracklist was created via the
838 // connection of a (secondary) vertex or not.
839 // In case the track was the result of (secondary) vertex addition the
840 // return value is 1, otherwise the value 0 will be returned.
844 if (fConnects->FindObject(t)) connect=1;
848 ///////////////////////////////////////////////////////////////////////////
849 Int_t AliVertex::IsJetTrack(AliTrack* t)
851 // Indicate whether a track from the tracklist was created via the
852 // addition of a jet or not.
853 // In case the track was the result of jet addition the return value is 1,
854 // otherwise the value 0 will be returned.
858 if (fJetTracks->FindObject(t)) jetflag=1;
862 ///////////////////////////////////////////////////////////////////////////
863 void AliVertex::Draw(Int_t secs,Int_t cons,Int_t jets)
865 // 3-Dimensional visualisation of an AliVertex with its attributes.
866 // The displayed tracklength is proportional to the momentum of the track.
868 // Color conventions :
869 // -------------------
870 // positive track : red
871 // neutral track : green
872 // negative track : blue
873 // jet-track : magenta (if explicit marking selected)
875 // secs = 1 --> Draw secondary vertices.
876 // 0 --> Don't draw secondary vertices.
878 // cons = 1 --> Draw (auto generated) connecting tracks.
879 // 0 --> Don't draw (auto generated) connecting tracks.
881 // jets = 1 --> Mark tracks belonging to jets.
882 // 0 --> Don't mark jet-tracks.
886 // Auto generated connecting tracks will be drawn as thin lines.
887 // Tracks belonging to jets will be marked as somewhat thinner magenta lines.
888 // This memberfunction is used recursively.
890 Double_t vec[3]={0,0,0};
899 if (fLines) delete fLines;
900 fLines=new TObjArray();
903 Int_t ntk=GetNtracks();
904 for (Int_t jtk=1; jtk<=ntk; jtk++)
910 charge=static_cast<int>(tx->GetCharge());
912 TPolyLine3D* line=new TPolyLine3D();
915 if (IsConnectTrack(tx))
919 r=tx->GetBeginPoint();
921 r->GetPosition(vec,"car");
922 line->SetNextPoint(vec[0],vec[1],vec[2]);
925 r->GetPosition(vec,"car");
926 line->SetNextPoint(vec[0],vec[1],vec[2]);
927 line->SetLineWidth(1);
932 r=tx->GetClosestPoint();
934 r->GetPosition(vec,"car");
935 line->SetNextPoint(vec[0],vec[1],vec[2]);
936 p=tx->Get3Momentum();
938 p.GetVector(vec,"car");
939 line->SetNextPoint(vec[0],vec[1],vec[2]);
940 line->SetLineWidth(3);
943 if (charge>0) line->SetLineColor(kRed); // Positive track
944 if (!charge) line->SetLineColor(kGreen); // Neutral track
945 if (charge<0) line->SetLineColor(kBlue); // Negative track
947 // Mark tracks belonging to jets
952 line->SetLineWidth(2);
953 line->SetLineColor(kMagenta);
960 // Go for secondary vertices if selected
963 Int_t nvtx=GetNvertices();
964 for (Int_t jvtx=1; jvtx<=nvtx; jvtx++)
967 if (vx) vx->Draw(secs,cons,jets);
971 ///////////////////////////////////////////////////////////////////////////