New CPV classes: CPVModule, CPVHit, CPVDigit

diff --git a/PHOS/AliPHOSCPV.cxx b/PHOS/AliPHOSCPV.cxx
new file mode 100644 (file)
index 0000000..a50f06a
--- /dev/null
+++ b/PHOS/AliPHOSCPV.cxx
@@ -0,0 +1,190 @@
+/**************************************************************************
+ * 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.                  *
+ **************************************************************************/
+
+/*
+  $Log$
+*/
+
+////////////////////////////////////////////////
+//  Manager and hits classes for set          //
+//  Charged Particle Veto (CPV)               //
+//                                            //
+//  Author: Yuri Kharlov, IHEP, Protvino      //
+//  e-mail: Yuri.Kharlov@cern.ch              //
+//  Last modified: 28 September 2000          //
+////////////////////////////////////////////////
+ 
+// --- ROOT system ---
+#include <TTree.h>
+
+// --- Standard library ---
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <iostream.h>
+
+// --- galice header files ---
+#include "AliPHOSCPV.h"
+#include "AliRun.h"
+
+//==============================================================================
+//                              CPVModule
+//==============================================================================
+
+ClassImp(CPVModule)
+
+//______________________________________________________________________________
+
+CPVModule::CPVModule(void) {
+  //
+  // Allocate an array of hits
+  //
+
+  if ( NULL==(fHits=new TClonesArray("CPVHit",100)) ) {
+    Error("CPV","Can not create array of hits");
+    exit(1);
+  }
+}
+
+//______________________________________________________________________________
+
+CPVModule::~CPVModule(void)
+{
+  Clear();
+}
+
+//______________________________________________________________________________
+
+void CPVModule::Clear(Option_t *opt="")
+{
+// Clear hit information
+
+  fHits  -> Clear(opt);
+}
+
+//______________________________________________________________________________
+
+void CPVModule::AddHit(TLorentzVector p, Float_t *xy, Int_t ipart)
+{
+  //
+  // Add this hit to the hit list in CPV detector.
+  //
+
+  TClonesArray &lhits = *(TClonesArray *)fHits;
+  new(lhits[fHits->GetEntriesFast()]) CPVHit(p,xy,ipart);
+}
+
+//______________________________________________________________________________
+
+void CPVModule::Print(Option_t *opt)
+{
+  //
+  // Print CPVModule information.
+  // options:  'p' - print hits in the module
+  //
+
+  Int_t nhits,hit;
+  if (strcmp(opt,"p")==0) {
+    printf ("CPV module has %d hits\n",nhits=fHits->GetEntriesFast());
+    for (hit=0;hit<nhits;hit++) {
+      CPVHit *cpvHit = (CPVHit*)fHits->UncheckedAt(hit);
+      cpvHit->Print();
+    }
+  }
+}
+
+//______________________________________________________________________________
+
+void CPVModule::MakeBranch(Int_t i)
+{
+  //
+  // Create a new branch for a CPV module #i in the current Root Tree
+  //
+
+  char branchname[10];
+  sprintf(branchname,"CPV%d",i);
+  gAlice->TreeH()->Branch(branchname,&fHits, 1000);
+}
+
+//_____________________________________________________________________________
+void CPVModule::SetTreeAddress(Int_t i)
+{
+  //
+  // Set branch address for the Hits Tree for a CPV module #i
+  //
+
+  TBranch *branch;
+  char branchname[20];
+  TTree *treeH = gAlice->TreeH();
+  if (treeH){
+    sprintf(branchname,"CPV%d",i);
+    branch = treeH->GetBranch(branchname);
+    if (branch) branch->SetAddress(&fHits);
+  }
+}
+
+//==============================================================================
+//                              CPVHit
+//==============================================================================
+
+ClassImp(CPVHit)
+
+//______________________________________________________________________________
+
+CPVHit::CPVHit(TLorentzVector p, Float_t *xy, Int_t ipart)
+{
+  //
+  // Create a CPV hit object
+  //
+
+  fMomentum  = p;
+  fXhit      = xy[0];
+  fYhit      = xy[1];
+  fIpart     = ipart;
+}
+
+//______________________________________________________________________________
+void CPVHit::Print()
+{
+  //
+  // Print CPV hit
+  //
+
+  printf("CPV hit: p  = (% .4f, % .4f, % .4f, % .4f) GeV,\n",
+       GetMomentum().Px(),GetMomentum().Py(),GetMomentum().Pz(),GetMomentum().E());
+  printf("         xy = (%8.4f, %8.4f) cm, ipart = %d\n",
+        fXhit,fYhit,fIpart);
+}
+
+//==============================================================================
+//                              CPVDigit
+//==============================================================================
+
+ClassImp(CPVDigit)
+
+//______________________________________________________________________________
+
+CPVDigit::CPVDigit(Int_t x, Int_t y, Float_t q)
+{
+  //
+  // Create a CPV digit object
+  //
+
+  fXpad = x;
+  fYpad = y;
+  fQpad = q;
+}
+
+//______________________________________________________________________________

diff --git a/PHOS/AliPHOSCPV.h b/PHOS/AliPHOSCPV.h
new file mode 100644 (file)
index 0000000..b37ae2e
--- /dev/null
+++ b/PHOS/AliPHOSCPV.h
@@ -0,0 +1,99 @@
+#ifndef ALIPHOSCPV_H
+#define ALIPHOSCPV_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */
+
+////////////////////////////////////////////////
+//  Manager and hits classes for set:CPV      //
+//                                            //
+//  Author: Yuri Kharlov, IHEP, Protvino      //
+//  e-mail: Yuri.Kharlov@cern.ch              //
+//  Last modified: 23 March 2000              //
+////////////////////////////////////////////////
+ 
+// --- ROOT system ---
+#include <TClonesArray.h> 
+#include <TLorentzVector.h>
+
+// --- galice header files ---
+#include "AliHit.h"
+
+//==============================================================================
+//                              CPVModule
+//==============================================================================
+
+class CPVModule : public TObject {
+
+public:
+
+  virtual ~CPVModule(void);
+           CPVModule(void);
+  
+  void     Clear(Option_t *opt="");
+  void     Print(Option_t *opt="");
+  void     AddHit(TLorentzVector p, Float_t *xy, Int_t ipart);
+  void     MakeBranch(Int_t i);
+  void     SetTreeAddress(Int_t i);
+  
+  TClonesArray *Hits    (void) {return fHits;}
+
+private:
+  
+  TClonesArray *fHits;              // List of hits in the Module
+
+  ClassDef(CPVModule,1)             // CPV Module
+};
+
+//==============================================================================
+//                              CPVHit
+//==============================================================================
+
+class CPVHit : public TObject {
+  
+private:
+  TLorentzVector fMomentum;   // 4-momentum of the particle
+  Float_t        fXhit;       // Hit's X-coordinates
+  Float_t        fYhit;       // Hit's Y-coordinates
+  Int_t          fIpart;      // Hit's particle type
+  
+public:
+  virtual ~CPVHit() {}
+           CPVHit() {}
+           CPVHit(TLorentzVector p, Float_t *xy, Int_t ipart);
+  
+  TLorentzVector GetMomentum()  { return  fMomentum; }
+  Float_t        GetX()         { return  fXhit;     }
+  Float_t        GetY()         { return  fYhit;     }
+  Int_t          GetIpart()     { return  fIpart;    }
+  void           Print();
+
+  ClassDef(CPVHit,1)  // Hit object in one CPV module
+};
+ 
+//==============================================================================
+//                              CPVDigit
+//==============================================================================
+
+class CPVDigit : public TObject {
+  
+private:
+  Int_t    fXpad;       // Digit's pad number in Phi
+  Int_t    fYpad;       // Digit's pad number in Z
+  Float_t  fQpad;       // Digit's pad amplitude
+  
+public:
+  virtual ~CPVDigit() {}
+           CPVDigit() {}
+           CPVDigit(Int_t x, Int_t y, Float_t q);
+  
+  void     SetQpad(Float_t q) { fQpad = q;     }
+  Int_t    GetXpad()          { return  fXpad; }
+  Int_t    GetYpad()          { return  fYpad; }
+  Float_t  GetQpad()          { return  fQpad; }
+
+  ClassDef(CPVDigit,1)  // Digit object in one CPV pad
+};
+ 
+#endif