EventDisplay/GfxTrackList.cxx

Go to the documentation of this file.

#include "GfxTrackList.h"
#include "GfxTrackListCfg.h"
#include "GfxProxy.h"
#include "ViewState.h"
#include "GfxTrack.h"
#include <Midad/Gui/GuiSlider.h>
#include <Midad/Base/Mint.h>
#include <RecoBase/CandTrackHandle.h>
#include <RecoBase/CandTrackListHandle.h>
#include <RecoBase/CandFitTrackHandle.h>
#include "GfxTrackListMenu.h"
#include <TClonesArray.h>
#include <Record/RecArrayAllocator.h>
#include <DataUtil/CDL2STL.h>
#include <JobControl/JobC.h>
using namespace DataUtil;

#include <sigc++/sigc++.h>
#include <sigc++/class_slot.h>
using namespace SigC;


#include <Buttons.h> // part of ROOT

static GfxProxy<GfxTrackList> 
gsGfxTrackListProxy("TrackList", new GfxTrackListMenu());

GfxTrackList::GfxTrackList()
    :fCfg(0)
      ,fHideTracks(false)
{
    fCfg = new GfxTrackListCfg();
    fCfg->modified_signal.connect
        (SigC::slot_class(*this,&GfxTrackList::ReConfigure));
    RecArrayAllocator& a = RecArrayAllocator::Instance();
    fTracksCA = a.GetArray("GfxTrack");
} 

GfxTrackList::~GfxTrackList()
{
  RecArrayAllocator& a = RecArrayAllocator::Instance();
  if (fTracksCA) { a.ReleaseArray(fTracksCA); fTracksCA = 0; }
}


void GfxTrackList::Init(PageDisplay& pd, EVD& mp)
{

  fEVD= &mp;
  fPageDisplay= &pd;
  ViewState* vs = this->GetViewState();
  vs->spatial_metric.connect(slot_class(*this,&GfxTrackList::ReConfigure));
}

void GfxTrackList::Configure(Mint& mint)
{
  this->ClearTrackList();
  CandRecord* crec = dynamic_cast<CandRecord*>
    (mint.GetJobC().Mom.GetFragment("CandRecord", "PrimaryCandidateRecord"));    
  if (!crec) return;


  
  // const CandTrackListHandle* ctlh = dynamic_cast<CandTrackListHandle*>(crec->FindCandHandle("CandFitTrackCamListHandle","CandFitTrackCamList"));
  CandTrackListHandle * ctlh = mint.GetTracks();
  if (!ctlh) return;
  RangeDouble& time_range = 
    fPageDisplay->GetTimeSlider().GetRangeCtrl().GetRange();
  TIter trackItr(ctlh->GetDaughterIterator());
  while (CandTrackHandle *cth = dynamic_cast<CandTrackHandle*>
         (trackItr())) {
    
    CandTrackHandle * findtrack=0;
    bool addtrack=true;
    if(GetEVD()->GetCandSlice()){
      CandSliceHandle slc1 =  *(GetEVD()->GetCandSlice());
      CandSliceHandle slc2;
      if(cth->GetCandSlice()) slc2=*(cth->GetCandSlice());
      if(slc1!=slc2 && !slc1.IsCloneOf(slc2)){
        addtrack=false;
      }
    }
    else if(GetEVD()->GetCandEvent()){
      addtrack=false;
      for( Int_t itrack=0;itrack<=GetEVD()->GetCandEvent()->GetLastTrack();itrack++){
        const CandFitTrackHandle *evtrack = dynamic_cast<const CandFitTrackHandle *>(GetEVD()->GetCandEvent()->GetTrack(itrack));
        findtrack=evtrack->GetFinderTrack();
        if(findtrack){
          if (*findtrack==*cth){
            addtrack=true;
            break;
          }
        }
        const CandTrackHandle *evtrackbase = dynamic_cast<const CandTrackHandle *>(evtrack);
        if(*evtrackbase==*cth){
          addtrack=true;
          break;
        }
      }
    }
    
    int last = fTracksCA->GetLast()+1;
    if(addtrack && (time_range.InRange(cth->GetVtxT()) || GetEVD()->GetNumSlices()<2) ){
      new ((*fTracksCA)[last]) GfxTrack(*cth,*this);
      GfxTrack* gs = dynamic_cast<GfxTrack*>(fTracksCA->UncheckedAt(last));
      fTracks.push_back(gs);
    }
  }
  this->ReConfigure();

}
void GfxTrackList::ReConfigure()
{
  if (fHideTracks) return;

    for (TrackList_t::iterator it=fTracks.begin(); it != fTracks.end(); ++it) {
        (*it)->Configure();
    }
}

void GfxTrackList::ClearTrackList() 
{
    fTracks.clear();
    fTracksCA->Clear("C");
}

void GfxTrackList::Draw(Option_t* option)
{
//    cerr << "GfxTrackList::Draw("<<option<<")\n";
    for (TrackList_t::iterator it=fTracks.begin(); it != fTracks.end(); ++it) {

      if(GetEVD()->GetDrawAll() || ((*it)->GetColor()==GetEVD()->GetGhostColor() && GetEVD()->GetDrawGhosts()) || ((*it)->GetColor()!=GetEVD()->GetGhostColor() && GetEVD()->GetDrawGhosts())){
        (*it)->Draw(option);
      }
        
    }
}

void GfxTrackList::ExecuteEvent(int event, int /*px*/, int /*py*/, GfxTrack* track)
{
    switch (event) {
    case kMouseEnter: {
        const CandTrackHandle& cth = track->GetTrack();
        text_info.emit(Form("%s: %d strips, P=%.1f",
                            "track", cth.GetNDaughters(),
                            cth.GetMomentum()));
        break;
    }
    default: // nothing
        break;
    }
    
}

void GfxTrackList::HideTracks(bool hide)
{
    fHideTracks = hide;
    this->ReConfigure();
    if(GetGraphicsView()==GraphicsView::kXY)fEVD->UpdateCanvas();

}

---