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.
444 // In case a private copy is made, this is performed via the Clone() memberfunction.
445 // All AliJet and derived classes have the default TObject::Clone() memberfunction.
446 // However, derived classes generally contain an internal data structure which may
447 // include pointers to other objects. Therefore it is recommended to provide
448 // for all derived classes a specific copy constructor and override the default Clone()
449 // memberfunction using this copy constructor.
450 // An example for this may be seen from AliJet.
454 fJets=new TObjArray(fNjmax);
455 if (fJetCopy) fJets->SetOwner();
457 if (fNjets == fNjmax) // Check if maximum jet number is reached
460 fJets->Expand(fNjmax);
463 // Add the jet to the list
465 if (fJetCopy) jx=(AliJet*)j.Clone();
473 // Add the tracks of the jet to this vertex
478 fJetTracks=new TObjArray();
480 Int_t copy=1-(jx->GetTrackCopy());
482 for (Int_t i=1; i<=jx->GetNtracks(); i++)
493 ///////////////////////////////////////////////////////////////////////////
494 void AliVertex::AddVertex(AliVertex& v,Int_t connect)
496 // Add a (secondary) vertex to the current vertex.
497 // In case the maximum number of (secondary) vertices has been reached,
498 // the array space will be extended automatically
500 // Note : By default the 4-momentum and charge of the current (primary) vertex
501 // are updated by automatically creating the track connecting
502 // both vertices. The track parameters are taken from the
503 // 4-momentum and charge of the secondary vertex.
504 // The automatic creation of the connecting track and updating
505 // of the (primary) vertex 4-momentum and charge can be suppressed
506 // by specifying connect=0. In this case, however, the user
507 // has to introduce the connecting track lateron by hand
508 // explicitly in order to match the kinematics and charge.
510 // In case a private copy is made, this is performed via the Clone() memberfunction.
511 // All AliVertex and derived classes have the default TObject::Clone() memberfunction.
512 // However, derived classes generally contain an internal data structure which may
513 // include pointers to other objects. Therefore it is recommended to provide
514 // for all derived classes a specific copy constructor and override the default Clone()
515 // memberfunction using this copy constructor.
516 // An example for this may be seen from AliVertex.
520 fVertices=new TObjArray(fNvmax);
521 if (fVertexCopy) fVertices->SetOwner();
523 if (fNvtx == fNvmax) // Check if maximum vertex number is reached
526 fVertices->Expand(fNvmax);
529 // Add the linked (secondary) vertex to the list
531 if (fVertexCopy) vx=(AliVertex*)v.Clone();
539 // Create connecting track and update 4-momentum and charge for current vertex
542 AliTrack* t=new AliTrack();
543 t->SetBeginPoint(GetPosition());
544 t->SetEndPoint(v.GetPosition());
545 t->SetCharge(v.GetCharge());
546 t->Set4Momentum((Ali4Vector&)v);
552 fConnects=new TObjArray(fNvmax);
553 if (!fTrackCopy) fConnects->SetOwner();
558 ///////////////////////////////////////////////////////////////////////////
559 void AliVertex::Data(TString f)
561 // Provide vertex information within the coordinate frame f
562 const char* name=GetName();
563 const char* title=GetTitle();
564 cout << " *AliVertex::Data*";
565 if (strlen(name)) cout << " Name : " << GetName();
566 if (strlen(title)) cout << " Title : " << GetTitle();
568 cout << " Id : " << fUserId << " Invmass : " << GetInvmass()
569 << " Charge : " << GetCharge() << " Momentum : " << GetMomentum()
570 << " Ntracks : " << GetNtracks() << " Nvertices : " << fNvtx
571 << " Njets : " << fNjets << endl;
575 AliPosition::Data(f);
577 ///////////////////////////////////////////////////////////////////////////
578 void AliVertex::List(TString f)
580 // Provide primary track and sec. vertex information within the coordinate frame f
582 Data(f); // Information of the current vertex
584 // The tracks of this vertex
586 for (Int_t it=1; it<=GetNtracks(); it++)
591 cout << " ---Track no. " << it << endl;
597 cout << " *AliVertex::List* Error : No track present." << endl;
601 // The secondary vertices of this vertex
603 for (Int_t iv=1; iv<=GetNvertices(); iv++)
608 cout << " ---Level 1 sec. vertex no. " << iv << endl;
614 cout << " *AliVertex::List* Error : No sec. vertex present." << endl;
618 ///////////////////////////////////////////////////////////////////////////
619 void AliVertex::ListAll(TString f)
621 // Provide complete (sec) vertex and (decay) track info within the coordinate frame f
623 Data(f); // Information of the current vertex
625 // The tracks of this vertex
627 for (Int_t it=1; it<=GetNtracks(); it++)
632 cout << " ---Track no. " << it << endl;
638 cout << " *AliVertex::ListAll* Error : No track present." << endl;
643 Dumps(v,1,f); // Information of all sec. vertices
645 //////////////////////////////////////////////////////////////////////////
646 void AliVertex::Dumps(AliVertex* v,Int_t n,TString f)
648 // Recursively provide the info of all secondary vertices of this vertex
650 for (Int_t iv=1; iv<=v->GetNvertices(); iv++)
655 cout << " ---Level " << n << " sec. vertex no. " << iv << endl;
659 // The tracks of this vertex
661 for (Int_t it=1; it<=vs->GetNtracks(); it++)
666 cout << " ---Track no. " << it << endl;
672 cout << " *AliVertex::Dumps* Error : No track present." << endl;
676 // Go for next sec. vertex level of this sec. vertex recursively
681 cout << " *AliVertex::Dumps* Error : No sec. vertex present." << endl;
685 //////////////////////////////////////////////////////////////////////////
686 Int_t AliVertex::GetNvertices()
688 // Return the current number of (secondary) vertices
691 ///////////////////////////////////////////////////////////////////////////
692 AliVertex* AliVertex::GetVertex(Int_t i)
694 // Return the i-th (secondary) vertex of the current vertex
697 cout << " *AliVertex*::GetVertex* No (secondary) vertices present." << endl;
704 cout << " *AliVertex*::GetVertex* Invalid argument i : " << i
705 << " Nvtx = " << fNvtx << endl;
710 return (AliVertex*)fVertices->At(i-1);
714 ///////////////////////////////////////////////////////////////////////////
715 AliVertex* AliVertex::GetIdVertex(Int_t id)
717 // Return the (sec.) vertex with user identifier "id"
722 cout << " *AliVertex*::GetIdVertex* No (secondary) vertices present." << endl;
727 for (Int_t i=0; i<fNvtx; i++)
729 vx=(AliVertex*)fVertices->At(i);
730 if (id == vx->GetId()) v=vx;
735 ///////////////////////////////////////////////////////////////////////////
736 void AliVertex::SetVertexCopy(Int_t j)
738 // (De)activate the creation of private copies of the added vertices.
739 // j=0 ==> No private copies are made; pointers of original vertices are stored.
740 // j=1 ==> Private copies of the vertices are made and these pointers are stored.
742 // Note : Once the storage contains pointer(s) to AliVertex objects one cannot
743 // change the VertexCopy mode anymore.
744 // To change the VertexCopy mode for an existing AliVertex containing
745 // vertices one first has to invoke Reset().
754 cout << "*AliVertex::SetVertexCopy* Invalid argument : " << j << endl;
759 cout << "*AliVertex::SetVertexCopy* Storage already contained vertices."
760 << " ==> VertexCopy mode not changed." << endl;
763 ///////////////////////////////////////////////////////////////////////////
764 Int_t AliVertex::GetVertexCopy()
766 // Provide value of the VertexCopy mode.
767 // 0 ==> No private copies are made; pointers of original vertices are stored.
768 // 1 ==> Private copies of the vertices are made and these pointers are stored.
771 ///////////////////////////////////////////////////////////////////////////
772 Int_t AliVertex::GetNjets()
774 // Return the current number of jets
777 ///////////////////////////////////////////////////////////////////////////
778 AliJet* AliVertex::GetJet(Int_t i)
780 // Return the i-th jet of the current vertex
783 cout << " *AliVertex*::GetJet* No jets present." << endl;
788 if (i<=0 || i>fNjets)
790 cout << " *AliVertex*::GetJet* Invalid argument i : " << i
791 << " Njets = " << fNjets << endl;
796 return (AliJet*)fJets->At(i-1);
800 ///////////////////////////////////////////////////////////////////////////
801 AliJet* AliVertex::GetIdJet(Int_t id)
803 // Return the jet with user identifier "id"
808 cout << " *AliVertex*::GetIdJet* No jets present." << endl;
813 for (Int_t i=0; i<fNjets; i++)
815 jx=(AliJet*)fJets->At(i);
816 if (id == jx->GetId()) j=jx;
821 ///////////////////////////////////////////////////////////////////////////
822 void AliVertex::SetJetCopy(Int_t j)
824 // (De)activate the creation of private copies of the added jets.
825 // j=0 ==> No private copies are made; pointers of original jets are stored.
826 // j=1 ==> Private copies of the jets are made and these pointers are stored.
828 // Note : Once the storage contains pointer(s) to AliJet objects one cannot
829 // change the JetCopy mode anymore.
830 // To change the JetCopy mode for an existing AliVertex containing
831 // jets one first has to invoke Reset().
840 cout << "*AliVertex::SetJetCopy* Invalid argument : " << j << endl;
845 cout << "*AliVertex::SetJetCopy* Storage already contained jets."
846 << " ==> JetCopy mode not changed." << endl;
849 ///////////////////////////////////////////////////////////////////////////
850 Int_t AliVertex::GetJetCopy()
852 // Provide value of the JetCopy mode.
853 // 0 ==> No private copies are made; pointers of original jets are stored.
854 // 1 ==> Private copies of the jets are made and these pointers are stored.
857 ///////////////////////////////////////////////////////////////////////////
858 Int_t AliVertex::IsConnectTrack(AliTrack* t)
860 // Indicate whether a track from the tracklist was created via the
861 // connection of a (secondary) vertex or not.
862 // In case the track was the result of (secondary) vertex addition the
863 // return value is 1, otherwise the value 0 will be returned.
867 if (fConnects->FindObject(t)) connect=1;
871 ///////////////////////////////////////////////////////////////////////////
872 Int_t AliVertex::IsJetTrack(AliTrack* t)
874 // Indicate whether a track from the tracklist was created via the
875 // addition of a jet or not.
876 // In case the track was the result of jet addition the return value is 1,
877 // otherwise the value 0 will be returned.
881 if (fJetTracks->FindObject(t)) jetflag=1;
885 ///////////////////////////////////////////////////////////////////////////
886 void AliVertex::Draw(Int_t secs,Int_t cons,Int_t jets)
888 // 3-Dimensional visualisation of an AliVertex with its attributes.
889 // The displayed tracklength is proportional to the momentum of the track.
891 // Color conventions :
892 // -------------------
893 // positive track : red
894 // neutral track : green
895 // negative track : blue
896 // jet-track : magenta (if explicit marking selected)
898 // secs = 1 --> Draw secondary vertices. (Default)
899 // 0 --> Don't draw secondary vertices.
901 // cons = 1 --> Draw (auto generated) connecting tracks. (Default)
902 // 0 --> Don't draw (auto generated) connecting tracks.
904 // jets = 1 --> Mark tracks belonging to jets.
905 // 0 --> Don't mark jet-tracks. (Default)
909 // Auto generated connecting tracks will be drawn as thin lines.
910 // Tracks belonging to jets will be marked as somewhat thinner magenta lines.
911 // This memberfunction is used recursively.
913 Double_t vec[3]={0,0,0};
922 if (fLines) delete fLines;
923 fLines=new TObjArray();
926 Int_t ntk=GetNtracks();
927 for (Int_t jtk=1; jtk<=ntk; jtk++)
933 charge=tx->GetCharge();
935 TPolyLine3D* line=new TPolyLine3D();
938 if (IsConnectTrack(tx))
942 r=tx->GetBeginPoint();
944 r->GetPosition(vec,"car");
945 line->SetNextPoint(vec[0],vec[1],vec[2]);
948 r->GetPosition(vec,"car");
949 line->SetNextPoint(vec[0],vec[1],vec[2]);
950 line->SetLineWidth(1);
955 r=tx->GetClosestPoint();
957 r->GetPosition(vec,"car");
958 line->SetNextPoint(vec[0],vec[1],vec[2]);
959 p=tx->Get3Momentum();
961 p.GetVector(vec,"car");
962 line->SetNextPoint(vec[0],vec[1],vec[2]);
963 line->SetLineWidth(3);
966 line->SetLineColor(kGreen); // Neutral track
967 if (charge>0) line->SetLineColor(kRed); // Positive track
968 if (charge<0) line->SetLineColor(kBlue); // Negative track
970 // Mark tracks belonging to jets
975 line->SetLineWidth(2);
976 line->SetLineColor(kMagenta);
983 // Go for secondary vertices if selected
986 Int_t nvtx=GetNvertices();
987 for (Int_t jvtx=1; jvtx<=nvtx; jvtx++)
990 if (vx) vx->Draw(secs,cons,jets);
994 ///////////////////////////////////////////////////////////////////////////
995 TObject* AliVertex::Clone(const char* name)
997 // Make a deep copy of the current object and provide the pointer to the copy.
998 // This memberfunction enables automatic creation of new objects of the
999 // correct type depending on the object type, a feature which may be very useful
1000 // for containers when adding objects in case the container owns the objects.
1001 // This feature allows e.g. AliEvent to store either AliVertex objects or
1002 // objects derived from AliVertex via the AddVertex memberfunction, provided
1003 // these derived classes also have a proper Clone memberfunction.
1005 AliVertex* vtx=new AliVertex(*this);
1008 if (strlen(name)) vtx->SetName(name);
1012 ///////////////////////////////////////////////////////////////////////////