StuDraw3DMuEvent.cxx

// $Id: StuDraw3DMuEvent.cxx,v 1.17 2010/04/02 22:34:09 fine Exp $
// *-- Author :    Valery Fine(fine@bnl.gov)   27/04/2008
#include "StuDraw3DMuEvent.h"
#include "Gtypes.h"
#include "StHelixHelper.h"
#include "StMuDSTMaker/COMMON/StMuTrack.h"
#include "StThreeVector.hh"
#include "StEEmcUtil/EEmcGeom/EEmcGeomSimple.h"
#include "StEmcUtil/geometry/StEmcGeom.h"

      

//___________________________________________________
StuDraw3DMuEvent::StuDraw3DMuEvent( const char *detectorName,TVirtualPad *pad): 
StDraw3D(detectorName,pad),fEndcapGeom(0)
{
   // The detectorName is a comma separated list of the OpenInventor files with no extension
   // For all names on the list one should provide the iv file with the "iv extension:
   //                         <name>.iv
   if (!gMuEventDisplay) gMuEventDisplay = this;
}


//___________________________________________________
StuDraw3DMuEvent::~StuDraw3DMuEvent()
{
   if (gMuEventDisplay == this) gMuEventDisplay = 0;
   delete fEndcapGeom; fEndcapGeom=0;
}


//___________________________________________________
void StuDraw3DMuEvent::Tracks(StTrackType type)
{
   Int_t n_prim=0;
   Int_t n_glob=0;
   TObjArray *globTracks = 0;
   TObjArray *primTracks = 0;
   if (type == global && ( globTracks= StMuDst::globalTracks() ) ){
      n_glob=StMuDst::GetNGlobalTrack();
   } else if ( (primTracks = StMuDst::primaryTracks()) ) {
      n_prim=StMuDst::GetNPrimaryTrack();
   }

   Int_t i_track=0;
   while (i_track < n_prim) {
       StMuTrack &track = *(StMuTrack *)primTracks->UncheckedAt(i_track++);
       double pt =track.pt();
       Style_t sty = Style(kPrimaryTrack).Sty();
       Size_t  siz = Style(kPrimaryTrack).Siz();
       Track(track,StDraw3DStyle::Pt2Color(pt),sty,siz);
       cout << ".";
   }
   i_track=0;
   while (i_track < n_glob) {
       StMuTrack &track = *(StMuTrack *)globTracks->UncheckedAt(i_track++);
       double pt =track.pt();
       Style_t sty = Style(kGlobalTrack).Sty();
       Size_t  siz = Style(kGlobalTrack).Siz();
       Track(track,StDraw3DStyle::Pt2Color(pt),sty,siz);
       cout << "+";
   }
   cout << endl << n_prim << " primary and " << n_glob << " global tracks have been rendered" << endl;

}



//___________________________________________________
TObject *StuDraw3DMuEvent::Track(const StMuTrack &track, Color_t col,Style_t sty,Size_t siz)
{

   StHelixHelper trPnt(track.helix(),track.outerHelix(),track.length());
   Int_t size;
   Float_t *xyz = trPnt.GetPoints(size);
   TObject *l = Line(size,xyz,col,sty,siz);
   SetModel((TObject*)&track);
   return l;
}


//___________________________________________________
TObject *StuDraw3DMuEvent::Track(const StMuTrack &track, EDraw3DStyle sty)
{
   TObject *view = 0;
   const StDraw3DStyle &style =  Style(sty);
   const StMuTrack *thisTrack = 0;
   if (sty == kPrimaryTrack) 
       thisTrack = track.primaryTrack();
   else
      thisTrack = track.globalTrack();
   if (thisTrack) 
       view = Track(*thisTrack, style.Col(),style.Sty(),style.Siz() );
   return view;

}


//___________________________________________________
TObject *StuDraw3DMuEvent::TrackInOut(const StMuTrack &track, Bool_t in
                  ,  Color_t col,  Style_t sty,  Size_t siz)
{
   // to be completed yet
   const StThreeVectorF  &pnt = in ? track.firstPoint() : track.lastPoint();

   return Point(pnt.x(),pnt.y(),pnt.z(), col,sty,siz);
}


//___________________________________________________
TObject *StuDraw3DMuEvent::TrackInOut(const StMuTrack &track, EDraw3DStyle sty,Bool_t in)
{
   const StDraw3DStyle &style =  Style(sty);
   return TrackInOut(track, in, style.Col(),style.Sty(),style.Siz() );
} 

//___________________________________________________
EEmcGeomSimple *StuDraw3DMuEvent::EndcapGeom()
{
   // endcap tower geometry:
   // see http://drupal.star.bnl.gov/STAR/subsys/eemc/endcap-calorimeter/db-usage
   if  (!fEndcapGeom) fEndcapGeom = new EEmcGeomSimple();
   return fEndcapGeom; 
}
//___________________________________________________

//___________________________________________________
void   StuDraw3DMuEvent::Endcaps(Style_t sty)
{
   StMuEmcCollection *emc= StMuDst::muEmcCollection();
   if (emc)  Endcaps(*emc,sty);
}



//___________________________________________________
void   StuDraw3DMuEvent::Endcaps(const StMuEmcCollection &emc,Style_t sty)
{
  Color_t colorResponce = 0;
  int nTowers = emc.getNEndcapTowerADC();
  for (int i=0; i< nTowers; i++) { 
    int adc,  isec, isub, ieta;
    emc.getEndcapTowerADC(i,adc,isec,isub,ieta);
    isec--;    ieta--;    isub--;
    if (adc<=0) continue; // print only non-zero values
    // access geometry info
    float etaCenter     =EndcapGeom()->getEtaMean(ieta);
    if (etaCenter <= 0) continue;
    float phiCenter     =EndcapGeom()->getPhiMean(isec,isub);
    if (phiCenter <= 0) continue;
    static const float dPhi = 2*EndcapGeom()->getPhiHalfWidth(isec,isub);
    static const float deta = 2*EndcapGeom()->getEtaHalfWidth(ieta);
    static const float radius = 270.; // no idea where I should pick it from.
    float energy = adc*60./4096-0.15;
    if ( energy > 0.15 ) {
    
       // If edep less then MIP (~300 MeV), 60GeV <-> 4096 ADC counts
       if (  energy  < 0.3) {         colorResponce = kBlue; 
          // If edep large then MIP but less then 1 GeV 
       } else if (  energy  < 1.0 ) { colorResponce = kGreen;
           // If between 1 GeV and lowest HT threshold (4 GeV for Run7)
       } else if (  energy  < 4.0 ) { colorResponce = kYellow;
           // If above lowest HT thershold
       } else                         colorResponce = kRed;
        
       static const double maxSize =  400.; // (cm)
       static const double scale   =   15; // 120.; // (cm/Gev)
       double size =(energy > 0.3 ? scale : scale/30.)*energy;
       if (size > maxSize)  size = maxSize ;
       Tower( radius
                , StarRoot::StEta(etaCenter,deta)
                , phiCenter, dPhi
                , colorResponce
                , sty
                , size);
        SetComment(Form("Endcap eta=%f, phi=%f, energy=%f",etaCenter,phiCenter,energy));
     }
  }
}
//____________________________________________________________________________________

//____________________________________________________________________________________
TObject *StuDraw3DMuEvent::EmcHit(Int_t emcHitsSoftId, Color_t col,Style_t sty,Size_t siz, const char *detId)
{  
   TObject *model = 0;
   StEmcGeom *emcGeom =StEmcGeom::getEmcGeom(detId);
   if (emcGeom) {
      Int_t softId=emcHitsSoftId;
      Float_t eta;
      Float_t phi;
      emcGeom->getEtaPhi(softId,eta,phi);
      Float_t etaStep = 1.0/emcGeom->NEta();
      Float_t phiStep = TMath::Pi()/60; 
      static int entries = 0;
        //if (entries) return 0;
        // printf(" m=%d, e=%d, s=%d; eta=%e deta=%e phi=%e dphi=%e id %d\n",m, e, s,eta,etaStep ,phi, phiStep, softId);
      entries++;
      model  = Tower(emcGeom->Radius(), StarRoot::StEta(eta,etaStep)
                         , phi, phiStep
                         , col,sty+(strcmp(detId,"bemc")?0:kBarrelStyle),siz);
   } else {
      LOG_ERROR <<  __FILE__ << ":  there is no geometry information for \"" << detId << "\"" << endm;
   }
   return model;
}

//____________________________________________________________________________________
TObject *StuDraw3DMuEvent::EmcHit(Int_t emcHitsSoftId, float energy, const char *detIdt)
{
   return EmcHit<StEmcTowerColor>(emcHitsSoftId, energy, detIdt);
}


//___________________________________________________
StuDraw3DMuEvent *StuDraw3DMuEvent::Display(){ return gMuEventDisplay;}

//___________________________________________________
StEmcTowerColor::StEmcTowerColor(float energy)
{
   Color_t colorResponce=0;
   Size_t  size=0;
   if (energy > 0  && energy < 30) {
       // If edep less then MIP (~300 MeV), 60GeV <-> 4096 ADC counts
      if (  energy  < 0.3)   {   
           colorResponce = kBlue; 
                // style = 4001;                 //wireframe 
               // If edep large then MIP but less then 1 GeV 
      } else if (  energy  < 1.0 ) colorResponce = kGreen;
               // If between 1 GeV and lowest HT threshold (4 GeV for Run7)
      else if (  energy  < 4.0 )   colorResponce = kYellow;
               // If above lowest HT thershold
      else                         colorResponce = kRed;

      static const double maxSize =  400.; // (cm)
      static const double scale   =  200.; // (cm/Gev)
      size =(energy > 0.3 ? scale : scale/30.)*energy;
      if (size > maxSize)  size = maxSize ;
      fStyle.SetCol(colorResponce);
      fStyle.SetSiz(size);
  }
}

StuDraw3DMuEvent *StuDraw3DMuEvent::gMuEventDisplay = new StuDraw3DMuEvent();

ClassImp(StuDraw3DMuEvent)