//****************************************************************************
//
// GEANT RICH hit data classes
//
// HGeantRichPhoton, HGeantRichDirect, and HGeantRichMirror
//
// Author: Romain Holzmann (r.holzmann@gsi.de)
// Modified: 1999/10/07 by Witold Przygoda (przygoda@psja1.if.uj.edu.pl)
//
//****************************************************************************

#ifndef HGEANTRICH_H
#define HGEANTRICH_H

using namespace std;
#include <iostream> 
#include <iomanip>
#include "hlinkeddataobject.h"

// ***************************************************************************

class HGeantRichPhoton : public HLinkedDataObject {

// ***************************************************************************

public:

  HGeantRichPhoton();
  HGeantRichPhoton(const HGeantRichPhoton &source);
  ~HGeantRichPhoton();

  void    setTrack(Int_t aTrack) { parentTrack = aTrack; }
  void    setHit(Float_t ax, Float_t ay, Float_t ae);
  void    setAddress(Char_t s) { sector = s; }
  void    setPmtId(Int_t pmtId) {fPmtId = pmtId;}

  virtual Int_t getTrack() { return parentTrack; }
  void    getHit(Float_t &ax, Float_t &ay, Float_t &ae);
  Float_t getEnergy() { return eHit; }
  Float_t getX() { return yHit; /* detector rotation! */ }
  Float_t getY() { return xHit; /* detector rotation! */ }
  Int_t   getSector() { return sector; }
  Int_t   getNLocationIndex() { return 1; }
  Int_t   getLocationIndex(Int_t i); 
  Int_t   getPmtId() {return fPmtId;};
  Bool_t  isNewRich() { return (fPmtId == -1) ? kFALSE:kTRUE; }
private:

  Int_t parentTrack;     // GEANT track number of parent particle
  Float_t xHit;          // x of hit (in mm) in pad plane
  Float_t yHit;          // y of hit (in mm) in pad plane
  Float_t eHit;          // energy of Cherenkov photon (in eV)
  Char_t sector;         // sector number (0...5)
  Int_t fPmtId;          // Id of the PMT which is used for RICH700
  
  ClassDef(HGeantRichPhoton,3) // GEANT RICH data class for photon hits
};

//============================================================================


// ***************************************************************************

class HGeantRichDirect : public HLinkedDataObject {

// ***************************************************************************

public:

  HGeantRichDirect();
  //HGeantRichDirect(const HGeantRichDirect &source);
  ~HGeantRichDirect();

  void    setTrack(Int_t aTrack) { trackNumber = aTrack;}
  void    setTrack(Int_t aTrack, Int_t aID);
  void    setHit(Float_t ax, Float_t ay, Float_t az);
  void    setMomentum(Float_t aMom, Float_t aTheta, Float_t aPhi);
  void    setELoss(Float_t ae, Float_t alen);
  void    setAddress (Char_t s) { sector = s; };
  void    setNumPhot(Int_t num) {numPhot = num;};
  void    setPmtId(Int_t pmtId) {fPmtId = pmtId;}

  void    getTrack(Int_t &aTrack, Int_t &aID);
  virtual Int_t getTrack() {return trackNumber;}
  void    getHit(Float_t &ax, Float_t &ay, Float_t &az);
  void    getMomentum(Float_t &aMom, Float_t &aTheta, Float_t &aPhi);
  void    getELoss(Float_t &ade, Float_t &alen);
  Float_t getX() { return yHit; /* detector rotation! */ }
  Float_t getY() { return xHit; /* detector rotation! */ }
  Float_t getTheta() { return thetaHit; }
  Float_t getPhi() { return phiHit; }
  Float_t getEnergyLoss() { return eLoss; }
  Float_t getTrackLengthInPhotDet() { return trackLength; }
  Int_t   getSector() {return sector;}
  Int_t   getNLocationIndex() { return 1; }
  inline Int_t   getLocationIndex(Int_t i); 
  Int_t   getNumPhot(){return numPhot;}; 
  Int_t   getPmtId() {return fPmtId;};
  Bool_t  isNewRich() { return (fPmtId == -1) ? kFALSE:kTRUE; }

private:

  Int_t trackNumber;     // GEANT track number of particle
  Float_t xHit;          // x of hit (in mm) in pad plane
  Float_t yHit;          // y of hit (in mm) in pad plane
  Float_t zHit;          // z of hit (in mm) in pad plane
  Int_t particleID;      // GEANT particle ID
  Float_t momentum;      // total momentum of particle (in MeV/c)
  Float_t thetaHit;      // theta of incidence (0-180 deg)
  Float_t phiHit;        // azimuthal angle (0-360 deg)
  Float_t eLoss;         // energy loss (in MeV)
  Float_t trackLength;   // track length in gas detector (in mm)
  Char_t sector;         // sector number (0...5)
  Int_t numPhot;         // dummy (not used)
  Int_t fPmtId;          // Id of the PMT which is used for RICH700

  ClassDef(HGeantRichDirect,3) // GEANT RICH hit data class for direct hits
};

//----------------------------------------------------------------------------
inline Int_t HGeantRichDirect::getLocationIndex(Int_t i) {
  switch (i) {
    case 0 : return sector;
  }
  return -1;
}
//============================================================================



//=========================================================================
// ***************************************************************************

class HGeantRichMirror : public HLinkedDataObject {

// ***************************************************************************

public:

  HGeantRichMirror();
//  HGeantRichMirror(const HGeantRichMirror &source);
  ~HGeantRichMirror();

  void    setTrack(Int_t aTrack) { trackNumber = aTrack;}
  void    setTrack(Int_t aTrack, Int_t aID);
  void    setHit(Float_t ax, Float_t ay, Float_t az);
  void    setAddress (Char_t s) { sector = s; }
  void    setNumPhot(Int_t num) { numPhot = num; }
  void    setXRing(Float_t xCoor) {xRing = xCoor;}
  void    setYRing(Float_t yCoor) {yRing = yCoor;}
  void    setXYring(const Float_t ax, const Float_t ay);

  void    getTrack(Int_t &aTrack, Int_t &aID);
  virtual Int_t getTrack() {return trackNumber;}
  void    getHit(Float_t &ax, Float_t &ay, Float_t &az);
  Float_t getX() { return yLep; /* detector rotation! */ }
  Float_t getY() { return xLep; /* detector rotation! */ }

  Float_t getXRing(){return xRing;}
  Float_t getYRing(){return yRing;}      
  Int_t   getSector() {return sector;}
  Int_t   getNLocationIndex() { return 1; }
  inline Int_t   getLocationIndex(Int_t i); 
  Int_t   getNumPhot(){return numPhot;}; 

private:

  Int_t numPhot;       // nb of photons reflected by mirror for lepton hit on mirror
  Int_t trackNumber;   // GEANT track number of particle
  Float_t xRing;       // x position of ring (in mm) in pad plane
  Float_t yRing;       // y position of ring (in mm) in pad plane
  Float_t xLep;        // x of hit (in mm) on mirror
  Float_t yLep;        // y of hit (in mm) on mirror
  Float_t zLep;        // z of hit (in mm) on mirror
  Int_t lepID;         // GEANT particle ID
  Char_t sector;       // sector number (0...5)

  ClassDef(HGeantRichMirror,2) // Geant RICH hit data class for mirror hits
};

//----------------------------------------------------------------------------
inline Int_t HGeantRichMirror::getLocationIndex(Int_t i) {
  switch (i) {
    case 0 : return sector;
  }
  return -1;
}
//============================================================================

//==========================================================================

#endif  // HGEANTRICH_H