a0b9eacdf5994c8b09223346315254139d4b894e
[u/mrichter/AliRoot.git] / RALICE / AliVertex.h
1 #ifndef ALIVERTEX_H
2 #define ALIVERTEX_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 // $Id$
7
8 #include "Riostream.h"
9 #include <math.h>
10  
11 #include "TObject.h"
12 #include "TObjArray.h"
13 #include "TPolyLine3D.h"
14  
15 #include "AliJet.h"
16 #include "AliPosition.h"
17  
18 class AliVertex : public AliJet,public AliPosition
19 {
20  public:
21   AliVertex();                            // Default constructor
22   AliVertex(Int_t n);                     // Create a vertex to hold initially n tracks
23   ~AliVertex();                           // Default destructor
24   virtual void SetOwner(Bool_t own=kTRUE);// Set ownership of all added objects
25   AliVertex(AliVertex& v);                // Copy constructor
26   void Reset();                           // Reset all values and stored vertex and jet lists
27   void ResetVertices();                   // Reset stored vertex list
28   void AddJet(AliJet& j,Int_t tracks=1);  // Add a jet (and its tracks) to the vertex
29   void AddVertex(AliVertex& v,Int_t connect=1);// Add (and connect) a (sec.) vertex to the current vertex
30   void AddJet(AliJet* j,Int_t tracks=1)    { AddJet(*j,tracks); }
31   void AddVertex(AliVertex* v,Int_t connect=1) { AddVertex(*v,connect); }
32   void Data(TString f="car");             // Print the vertex info within coordinate frame f
33   void List(TString f="car");             // Print vertex prim. track information for coord. frame f
34   void ListAll(TString f="car");          // Print prim. + sec. vertex full track info for coord. frame f
35   Int_t GetNvertices();                   // Return the number of (secondary) vertices
36   AliVertex* GetVertex(Int_t i);          // Provide i-th (secondary) vertex
37   AliVertex* GetIdVertex(Int_t id);       // Provide the vertex with user identifier "id"
38   void SetNvmax(Int_t n=2);               // Set the initial max. number of (secondary) vertices
39   void SetVertexCopy(Int_t j);            // (De)activate creation of private copies in fVertices
40   Int_t GetVertexCopy();                  // Provide VertexCopy flag value      
41   Int_t GetNjets();                       // Return the number of jets
42   AliJet* GetJet(Int_t i);                // Provide i-th jet
43   AliJet* GetIdJet(Int_t id);             // Provide the jet with user identifier "id"
44   void SetNjmax(Int_t n=2);               // Set the initial max. number of jets
45   void SetJetCopy(Int_t j);               // (De)activate creation of private copies in fJets
46   Int_t GetJetCopy();                     // Provide JetCopy flag value      
47   Int_t IsConnectTrack(AliTrack* t);      // Indicate if track is created by vertex connection
48   Int_t IsJetTrack(AliTrack* t);          // Indicate if track is resulting from jet addition
49   void Draw(Int_t secs=1,Int_t cons=1,Int_t jets=0); // Draw the vertex in an event display
50
51  protected:
52   void Init();          // Initialisation of pointers etc... 
53   Int_t fNvmax;         // The maximum number of (secondary) vertices
54   Int_t fNvtx;          // The number of (secondary) vertices
55   TObjArray* fVertices; // Array to hold the pointers to the (secondary) vertices
56   TObjArray* fConnects; // Array to hold the pointers to the auto-generated connecting tracks
57   Int_t fVertexCopy;    // Flag to denote creation of private copies in fVertices
58   Int_t fNjmax;         // The maximum number of jets
59   Int_t fNjets;         // The number of jets
60   TObjArray* fJets;     // Array to hold the pointers to the jets
61   TObjArray* fJetTracks;// Array to hold the pointers to tracks introduced by jet addition
62   Int_t fJetCopy;       // Flag to denote creation of private copies in fJets
63   TObjArray* fLines;    //! Array to (temporarily) store the 3D lines for the event display 
64
65  private:
66   void Dump(AliVertex* v,Int_t n,TString f); // Recursively print all sec. vertices
67  
68  ClassDef(AliVertex,4) // Creation and investigation of an AliVertex.
69 };
70 #endif