Merged ITS-working with HEAD. Improved some of the documentation and

[u/mrichter/AliRoot.git] / PHOS / AliPHOSRecPoint.cxx

/**************************************************************************
 * 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.                  *
 **************************************************************************/

//_________________________________________________________________________
// PHOSRecPoint base class deriving from AliRecPoint
//*-- Author : Gines MARTINEZ  SUBATECH 
//////////////////////////////////////////////////////////////////////////////

// --- ROOT system ---
#include "TPad.h"

// --- Standard library ---
#include <iostream>
#include <cstdio>

// --- AliRoot header files ---

#include "AliPHOSGeometry.h"
#include "AliPHOSDigit.h"
#include "AliPHOSRecPoint.h"



ClassImp(AliPHOSRecPoint)


//____________________________________________________________________________
AliPHOSRecPoint::AliPHOSRecPoint()
  : AliRecPoint()
{
  fGeom =   AliPHOSGeometry::GetInstance() ;
  fPHOSMod = 0;
}

//____________________________________________________________________________
AliPHOSRecPoint::~AliPHOSRecPoint()
{
  // dtor
}

//____________________________________________________________________________
Int_t AliPHOSRecPoint::DistancetoPrimitive(Int_t px, Int_t py)
{
  //Compute distance from point px,py to  a AliPHOSRecPoint considered as a Tmarker
  //  Compute the closest distance of approach from point px,py to this marker.
  //  The distance is computed in pixels units.
  //

   TVector3 pos(0.,0.,0.) ;
   GetLocalPosition( pos) ;
   Float_t x =  pos.X() ;
   Float_t y =  pos.Z() ;
   const Int_t kMaxDiff = 10;
   Int_t pxm  = gPad->XtoAbsPixel(x);
   Int_t pym  = gPad->YtoAbsPixel(y);
   Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);

   if (dist > kMaxDiff) return 9999;
   return dist;
}

//___________________________________________________________________________
 void AliPHOSRecPoint::Draw(Option_t *option)
 {
   // Draw this AliPHOSRecPoint with its current attributes
   
   AppendPad(option);
 }

//______________________________________________________________________________
void AliPHOSRecPoint::ExecuteEvent(Int_t event, Int_t px, Int_t py)
{
  // Execute action corresponding to one event
  // This member function is called when a AliPHOSRecPoint is clicked with the locator
  //
  // If Left button is clicked on AliPHOSRecPoint, the digits are switched on    
  // and switched off when the mouse button is released.
  //

  //  static Int_t pxold, pyold;

   static TGraph *  digitgraph = 0 ;
   static TPaveText* clustertext = 0 ;

   if (!gPad->IsEditable()) return;

   switch (event) {


   case kButton1Down:{
     AliPHOSDigit * digit ;
     AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;
     Int_t iDigit;
     Int_t relid[4] ;
     Float_t xi[fMulDigit] ;
     Float_t zi[fMulDigit] ;
 
     for(iDigit=0; iDigit<fMulDigit; iDigit++) {
       digit = (AliPHOSDigit *) fDigitsList[iDigit];
       phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
       phosgeom->RelPosInModule(relid, xi[iDigit], zi[iDigit]) ;
     }

     if (!digitgraph) {
       digitgraph = new TGraph(fMulDigit,xi,zi);
       digitgraph-> SetMarkerStyle(5) ; 
       digitgraph-> SetMarkerSize(1.) ;
       digitgraph-> SetMarkerColor(1) ;
       digitgraph-> Draw("P") ;
     }
     if (!clustertext) {
  
       TVector3 pos(0.,0.,0.) ;
       GetLocalPosition(pos) ;
       clustertext = new TPaveText(pos.X()-10,pos.Z()+10,pos.X()+50,pos.Z()+35,"") ;
       Text_t  line1[40] ;
       Text_t  line2[40] ;
       sprintf(line1,"Energy=%1.2f GeV",GetEnergy()) ;
       sprintf(line2,"%d Digits",GetDigitsMultiplicity()) ;
       clustertext ->AddText(line1) ;
       clustertext ->AddText(line2) ;
       clustertext ->Draw("");
     }
     gPad->Update() ; 
     Print() ;
  }

     break;

   case kButton1Up:
     if (digitgraph) {
       delete digitgraph  ;
       digitgraph = 0 ;
     }
     if (clustertext) {
       delete clustertext ;
       clustertext = 0 ;
     }
     
     break;
     
   }
}

//____________________________________________________________________________
Int_t AliPHOSRecPoint::GetPHOSMod()
{ 
  if(fPHOSMod > 0) 
    return fPHOSMod ;

  Int_t relid[4] ;
  
  AliPHOSDigit * digit   ;
  digit = (AliPHOSDigit *) fDigitsList[0] ;
  AliPHOSGeometry * phosgeom =  (AliPHOSGeometry *) fGeom ;

  phosgeom->AbsToRelNumbering(digit->GetId(), relid) ;
  fPHOSMod = relid[0];
  return fPHOSMod ;
}

//______________________________________________________________________________
void AliPHOSRecPoint::GetPrimaries(Int_t & number, Int_t * list)
{
  AliPHOSDigit * digit ;
  Int_t index ;
  Int_t maxcounter = 3 ;
  Int_t counter    = 0 ;
  Int_t * tempo    = new Int_t[maxcounter] ;
  
  for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
    digit = (AliPHOSDigit *) fDigitsList[index] ; 
    Int_t nprimaries = digit->GetNprimary() ;
    Int_t * newprimaryarray = new Int_t[nprimaries] ;
    Int_t ii ; 
    for ( ii = 0 ; ii < nprimaries ; ii++)
      newprimaryarray[ii] = digit->GetPrimary(ii+1) ; 
    Int_t jndex ;
    for ( jndex = 0 ; jndex < nprimaries ; jndex++ ) { // all primaries in digit
      if ( counter > maxcounter ) {
	number = - 1 ;
	cout << "AliPHOSRecPoint::GetNprimaries ERROR > increase maxcounter " << endl ;
	break ;
      }
      Int_t newprimary = newprimaryarray[jndex] ;
      Int_t kndex ;
      Bool_t already = kFALSE ;
      for ( kndex = 0 ; kndex < counter ; kndex++ ) { //check if not already stored
	if ( newprimary == tempo[kndex] ){
	  already = kTRUE ;
	  break ;
	}
      } // end of check
      if ( !already) { // store it 
	  tempo[counter] = newprimary ; 
	  counter++ ;
      } // store it
    } // all primaries in digit
    delete newprimaryarray ; 
  } // all digits

  number = counter ; 
  for ( index = 0 ; index < number ; index ++ )
    list[index] = tempo[index] ;
  
  delete tempo ; 
}

//______________________________________________________________________________
void AliPHOSRecPoint::Paint(Option_t *)
{
// Paint this ALiRecPoint as a TMarker  with its current attributes

   TVector3 pos(0.,0.,0.)  ;
   GetLocalPosition(pos)   ;
   Coord_t x = pos.X()     ;
   Coord_t y = pos.Z()     ;
   Color_t markercolor = 1 ;
   Size_t  markersize = 1. ;
   Style_t markerstyle = 5 ;

   if (!gPad->IsBatch()) {
     gVirtualX->SetMarkerColor(markercolor) ;
     gVirtualX->SetMarkerSize (markersize)  ;
     gVirtualX->SetMarkerStyle(markerstyle) ;
   }
   gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
   gPad->PaintPolyMarker(1,&x,&y,"") ;
}