EventDisplay/GfxTrack.cxx

Go to the documentation of this file.

#include "GfxTrack.h"
#include "GfxTrackList.h"
#include "GfxTrackListCfg.h"
#include "ViewState.h"
#include <RecoBase/CandStripHandle.h>
#include <RecoBase/CandTrackHandle.h>
#include <RecoBase/CandFitTrackHandle.h>
#include <UgliGeometry/UgliGeomHandle.h>
#include <UgliGeometry/UgliStripHandle.h>

#include <CandDigit/CandDigitHandle.h>
#include <Plex/PlexSEIdAltL.h>
#include <TEllipse.h>
#include <TBox.h>
#include <TPad.h>
#include <TMarker.h>
#include <TList.h>
#include <TPolyLine.h>

#include <map>
using namespace std;

ClassImp(GfxTrack)

GfxTrack::GfxTrack(const CandTrackHandle& cth, GfxTrackList& parent)
    : fTrack(cth)
    , fParent(parent)
    , fImp(0)
{
//    cerr << "Creating GfxTrack with " << cth.GetNDaughters() << " strips\n";
}

GfxTrack::GfxTrack(const GfxTrack& rhs)
    : TObject()
      ,fTrack(rhs.fTrack), fParent(rhs.fParent), fImp(0)
{
}


GfxTrack::~GfxTrack()
{
   this->Clear(); 
}

void GfxTrack::Clear(Option_t*)
{
  if (fImp) delete fImp; fImp=0;
}

void GfxTrack::Configure()
{
    if (fImp) { delete fImp; fImp = 0; }


    switch (fParent.GetViewState()->GetSpatialMetric()) {
    case ViewState::metric_is_continuous:
        if (! this->BuildRealImp()) return;
        break;
    case ViewState::metric_is_discreet: default:
        if (! this->BuildDiscreteImp()) return;
        break;
    }

}

int GfxTrack::GetCandTrackData(float *u, float *v, 
                               float *x, float *y, float *z, 
                               float *q, float *t,
                               int *strip_number, int *plane_number)

{

  CandStripHandleItr orderedstripItr(fTrack.GetDaughterIterator());
  CandStripHandleKeyFunc *stripKf = orderedstripItr.CreateKeyFunc();
  stripKf->SetFun(CandStripHandle::KeyFromPlane);
  orderedstripItr.GetSet()->AdoptSortKeyFunc(stripKf);
  stripKf = 0;

  //    TIter titr(fTrack.GetDaughterIterator());
  PlaneView::PlaneView_t view = fParent.GetPlaneView();
  GraphicsView::GraphicsView_t gview = fParent.GetGraphicsView();
  
  int ind=0;
  while (CandStripHandle *csh = dynamic_cast<CandStripHandle*>(orderedstripItr())) {  

      if (csh->GetPlaneView() == view ||
          gview == GraphicsView::kUV || gview == GraphicsView::kXY) 
        {
                
          int plane = csh->GetPlane();
          if (plane_number) {
            plane_number[ind] = plane;
          }
          if (strip_number) {
            strip_number[ind] = csh->GetStrip();
          }
          if (u){
            u[ind] = fTrack.GetU(plane);
          }
          if (v) {
            v[ind] = fTrack.GetV(plane);
          }
          if (x) {
            x[ind] = 0.70710678*(fTrack.GetU(plane)-fTrack.GetV(plane));
          }
          if (y) {
            y[ind] = 0.70710678*(fTrack.GetU(plane)+fTrack.GetV(plane));
          }
          if (z) {
            z[ind] = fTrack.GetZ(plane);
          }
          if (t) {
            t[ind] = fTrack.GetT(plane);
          }
          if (q) {
            q[ind] = csh->GetCharge();
          }     
          ++ind;
        }
    }
    return ind;
}

bool GfxTrack::BuildRealImp()
{
    float *x=0, *y=0;
    int *plane_number;
    int npoints=0;
    double cx,cy,dx,dy;

    int vtxColor=12;
    int trkColor=50;
    int trkStripColor=48;
    fColor=trkColor;
    if(fParent.GetEVD()->GetCandEvent()){
      Bool_t found=false;
      CandEventHandle * ev = fParent.GetEVD()->GetCandEvent();
      Int_t nTracks = ev->GetLastTrack()+1;
      for (Int_t it=0;it<nTracks;it++){
        const CandFitTrackHandle * evtrk = dynamic_cast<const CandFitTrackHandle *>(ev->GetTrack(it));  
        const CandTrackHandle * evtrkbase = dynamic_cast<const CandTrackHandle *>(ev->GetTrack(it));
        CandTrackHandle * findtrack=evtrk->GetFinderTrack();
        if(findtrack->IsCloneOf(fTrack))found=true;
        if(evtrkbase->IsCloneOf(fTrack))found=true;
      }
      if(!found){
        vtxColor=fParent.GetEVD()->GetGhostColor();
        trkColor=fParent.GetEVD()->GetGhostColor();
        trkStripColor=fParent.GetEVD()->GetGhostColor();
        fColor=fParent.GetEVD()->GetGhostColor();
      }
    }
    else if(fParent.GetEVD()->GetCandSlice()){
      CandSliceHandle slc1 =  *(fParent.GetEVD()->GetCandSlice());
      CandSliceHandle slc2;
      if(fTrack.GetCandSlice())slc2=*(fTrack.GetCandSlice());
      if(slc1!=slc2 && !slc1.IsCloneOf(slc2)){
        vtxColor=fParent.GetEVD()->GetGhostColor();
        trkColor=fParent.GetEVD()->GetGhostColor();
        trkStripColor=fParent.GetEVD()->GetGhostColor();
        fColor=fParent.GetEVD()->GetGhostColor();
      }
    }

    int n = fTrack.GetNDaughters();
    x = new float[n];
    y = new float[n];
    plane_number = new int[n];

    TList* track = new TList;
    track->SetOwner();
    double vx, vy;
    if (fParent.GetGraphicsView() == GraphicsView::kUZ) {
        npoints = this->GetCandTrackData(y,0,0,0,x,0,0,0,0);
        vx = fTrack.GetVtxZ();
        vy = fTrack.GetVtxU();

        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(1.);
        Vx->SetMarkerColor(vtxColor);
        TPolyLine * Trk = new TPolyLine(npoints,x,y);                   
        Trk->SetLineColor(trkColor);
        Trk->SetLineWidth(1);
        track->Add(Vx); 
        track->Add(Trk);
   }
    else if(fParent.GetGraphicsView() == GraphicsView::kVZ) 
      {                      // kV
        npoints = this->GetCandTrackData(0,y,0,0,x,0,0,0,0);
        vx = fTrack.GetVtxZ();
        vy = fTrack.GetVtxV();
        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(1.);
        Vx->SetMarkerColor(vtxColor);
        TPolyLine * Trk = new TPolyLine(npoints,x,y);                   
        Trk->SetLineColor(trkColor);
        Trk->SetLineWidth(1);
        track->Add(Vx); 
        track->Add(Trk);
    }
    else if(fParent.GetGraphicsView() == GraphicsView::kXY) 
      {                     
        npoints = this->GetCandTrackData(0,0,x,y,0,0,0,0,plane_number);
        vx =  0.70710678*(fTrack.GetVtxU()-fTrack.GetVtxV());
        vy =  0.70710678*(fTrack.GetVtxU()+fTrack.GetVtxV());
        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(1.);
        Vx->SetMarkerColor(vtxColor);
        track->Add(Vx); 
    }
    else if(fParent.GetGraphicsView() == GraphicsView::kUV) 
      {                     
        npoints = this->GetCandTrackData(y,x,0,0,0,0,0,0,plane_number);
        vx = fTrack.GetVtxV();
        vy = fTrack.GetVtxU();
        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(1.);
        Vx->SetMarkerColor(vtxColor);
        track->Add(Vx); 
    }
 
    TIter titr(fTrack.GetDaughterIterator());
    PlaneView::PlaneView_t view = fParent.GetPlaneView();
    GraphicsView::GraphicsView_t gview = fParent.GetGraphicsView();
    while (CandStripHandle* csh = dynamic_cast<CandStripHandle*>(titr())) 
      {
        if (csh->GetPlaneView() == view)
          {
            if(gview !=GraphicsView::kXY && gview !=GraphicsView::kUV)
              { 
                
                TIter digitItr(csh->GetDaughterIterator());
                while (CandDigitHandle *cdh = dynamic_cast<CandDigitHandle*>(digitItr())) {
                  
                  UgliGeomHandle ugh(*cdh->GetVldContext());
                  UgliStripHandle ush;
                  
                  ush = ugh.GetStripHandle(cdh->GetPlexSEIdAltL().GetBestItem().GetSEId());
                  
                  if (!ush.IsValid()) return false;
                  
                  cx = ush.GlobalPos(0).Z();
                  cy = ush.GetTPos();
                  dx = 2.0*ush.GetHalfThickness();
                  dy = 2.0*ush.GetHalfWidth();
                  TEllipse* box = new TEllipse(cx,cy,dx,dy);
                  box->SetFillColor(trkStripColor);
                  box->SetLineColor(trkStripColor);
                  track->Add(box);
                  
                }
              }
            else
              {
                for(int i=0;i<npoints;i++){
       
                  cx = x[i];
                  cy = y[i];
                  dx = 0.02;
                  dy = 0.02;
                  TEllipse* box = new TEllipse(cx,cy,dx,dy);
                  box->SetFillColor(trkStripColor);
                  box->SetLineColor(trkStripColor);
                  track->Add(box);
                }
                
              }
          }
      }

    fImp = track;

    delete [] x;
    delete [] y;
    delete [] plane_number;

    return true;
}

bool GfxTrack::BuildDiscreteImp()
{
  double cx,cy,dx,dy;
  int      n = fTrack.GetNDaughters();

    // using new [] here causes bizzare segv in unrelated parts of the
    // code, so lets get all FORTRAN about it.
    const int maxstrips = 10000;
    int strip[maxstrips], plane[maxstrips];
    float q[maxstrips];
    float x[maxstrips],y[maxstrips];

    int vtxColor=12;
    int trkColor=50;
    int trkStripColor=48;
    fColor=trkColor;
    if(fParent.GetEVD()->GetCandEvent()){
      Bool_t found=false;
      CandEventHandle * ev = fParent.GetEVD()->GetCandEvent();
      Int_t nTracks = ev->GetLastTrack()+1;
      for (Int_t it=0;it<nTracks;it++){
        const CandFitTrackHandle * evtrk = dynamic_cast<const CandFitTrackHandle*>(ev->GetTrack(it));
        if(evtrk->GetFinderTrack()->IsCloneOf(fTrack))found=true;
      }
      if(!found){
        vtxColor=fParent.GetEVD()->GetGhostColor();
        trkColor=fParent.GetEVD()->GetGhostColor();
        trkStripColor=fParent.GetEVD()->GetGhostColor();
        fColor=fParent.GetEVD()->GetGhostColor();
      }
    }  
    else if(fParent.GetEVD()->GetCandSlice()){
      CandSliceHandle slc1 =  *(fParent.GetEVD()->GetCandSlice());
      CandSliceHandle slc2;
      if(fTrack.GetCandSlice())slc2=*(fTrack.GetCandSlice());
      if(slc1!=slc2 && !slc2.IsCloneOf(slc2)){
        vtxColor=fParent.GetEVD()->GetGhostColor();
        trkColor=fParent.GetEVD()->GetGhostColor();
        trkStripColor=fParent.GetEVD()->GetGhostColor();
        fColor=fParent.GetEVD()->GetGhostColor();
      }
    }


    TList* track = new TList;
    track->SetOwner();

    int ind;

    if(fParent.GetGraphicsView() != GraphicsView::kXY && 
       fParent.GetGraphicsView() != GraphicsView::kUV)
      {
        n = this->GetCandTrackData(0,0,0,0,0,q,0,strip,plane);
        if (n > maxstrips) n = maxstrips;
        
        map<int,float> mean_strip, norm_strip;
        for (ind = 0; ind < n; ++ind) {
          if (mean_strip.find(plane[ind]) == mean_strip.end()) {
            mean_strip[plane[ind]] = 0;
            norm_strip[plane[ind]] = 0;
          }
          mean_strip[plane[ind]] += strip[ind]*q[ind];
          norm_strip[plane[ind]] += q[ind];
        }
        int vtx_plane = fTrack.GetVtxPlane();
        int vtx_ind = -1;
        
        map<int,float>::iterator it, done = mean_strip.end();
        for (ind=0, it = mean_strip.begin(); it != done && n < maxstrips; ++it, ++ind) {
          int plane = it->first;
          float qstrip = it->second;
          float norm = norm_strip[plane];
          x[ind] = plane;
          y[ind] = qstrip/norm;
          if (plane == vtx_plane) {
            vtx_ind = ind;
            //            cerr << Form("vertex: %d: plane=%.0f strip=%.1f\n",ind,x[ind],y[ind]);
          }
        }
        if(vtx_ind>-1 && vtx_ind<=ind){
          TMarker * Vx = new TMarker(x[vtx_ind],y[vtx_ind],29);
          Vx->SetMarkerSize(2);
          Vx->SetMarkerColor(vtxColor);
          track->Add(Vx);
        }       
        TPolyLine * Trk = new TPolyLine(ind,x,y);                       
        Trk->SetLineColor(trkColor);
        track->Add(Trk);
      }

    else if(fParent.GetGraphicsView() == GraphicsView::kXY) 
      {                     
        ind = this->GetCandTrackData(0,0,x,y,0,0,0,0,plane);
        float vx =  0.70710678*(fTrack.GetVtxU()-fTrack.GetVtxV());
        float vy =  0.70710678*(fTrack.GetVtxU()+fTrack.GetVtxV());
        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(2);
        Vx->SetMarkerColor(vtxColor);
        track->Add(Vx);
    }
    else if(fParent.GetGraphicsView() == GraphicsView::kUV) 
      {                     
        ind = this->GetCandTrackData(y,x,0,0,0,0,0,0,plane);
        float vx = fTrack.GetVtxV();
        float vy = fTrack.GetVtxU();
        TMarker * Vx = new TMarker(vx,vy,29);
        Vx->SetMarkerSize(2);
        Vx->SetMarkerColor(vtxColor);
        track->Add(Vx);

    }

 
    TIter titr(fTrack.GetDaughterIterator());
    PlaneView::PlaneView_t view = fParent.GetPlaneView();
    GraphicsView::GraphicsView_t gview = fParent.GetGraphicsView();
    ind=0;
    while (CandStripHandle* csh = dynamic_cast<CandStripHandle*>(titr())) 
      {
        if (csh->GetPlaneView() == view)
          {
            if(gview !=GraphicsView::kXY && gview !=GraphicsView::kUV)
              { 
                
                TIter digitItr(csh->GetDaughterIterator());
                while (CandDigitHandle *cdh = dynamic_cast<CandDigitHandle*>(digitItr())) {
                  

                  cx = cdh->GetPlexSEIdAltL().GetBestItem().GetSEId().GetPlane();
                  cy = cdh->GetPlexSEIdAltL().GetBestItem().GetSEId().GetStrip();
                  dx = 1.0;
                  dy = 1.0;
                  TEllipse* box = new TEllipse(cx,cy,dx,dy);             
                  box->SetFillColor(trkStripColor);
                  box->SetLineColor(trkStripColor);
                  track->Add(box);
                  
                }
              }
            else
              {

                for(int i=0;i<ind;i++){
       
                  cx = x[i];
                  cy = y[i];
                  dx = 1.0;
                  dy = 1.0;
                  TEllipse* box = new TEllipse(cx,cy,dx,dy);
                  box->SetFillColor(trkStripColor);
                  box->SetLineColor(trkStripColor);
                  track->Add(box);
                }
                
              }
          }
      }


    fImp = track;

    return true;
}

void GfxTrack::ExecuteEvent(int event, int px, int py)
{
    fParent.ExecuteEvent(event,px,py,this);
}
void GfxTrack::Paint(Option_t* option)
{
    if (!fImp) return;
    if(!fParent.GetEVD()->GetDrawAll() && ((fColor==fParent.GetEVD()->GetGhostColor() && !fParent.GetEVD()->GetDrawGhosts()) || (fColor!=fParent.GetEVD()->GetGhostColor() && fParent.GetEVD()->GetDrawGhosts())))return;
    fImp->Paint(option);
}
int GfxTrack::DistancetoPrimitive(int px, int py)
{
    int dist = 0xdead;
    if (fImp) {
        TList* l = dynamic_cast<TList*>(fImp);
        if (!l) 
            dist = fImp->DistancetoPrimitive(px,py);
        else {
            TIter next(l);
            TObject* obj;
            while ((obj=next())) {
                int d = obj->DistancetoPrimitive(px,py);
                if (d < dist) dist = d;
            }
        }
    }
    return dist;
}

---