GfxMCVec.cxx

Go to the documentation of this file.

#include "GfxMCVec.h"
#include "GfxMCVecList.h"
#include "GfxMCVecListCfg.h"
#include "ViewState.h"

#include <UgliGeometry/UgliGeomHandle.h>
#include <UgliGeometry/UgliStripHandle.h>
#include <Plex/PlexSEIdAltL.h>
#include <TEllipse.h>
#include <TBox.h>
#include <TPad.h>
#include <TLine.h>
#include <TMarker.h>
#include <TList.h>
#include <TPolyLine.h>
#include <map>
using namespace std;

ClassImp(GfxMCVec)



GfxMCVec::GfxMCVec(const TParticle& part, const float evntP, GfxMCVecList& parent)
    : fParticle(part)
    , fEventP(evntP)
    , fParent(parent)
    , fImp(0)
{
}

GfxMCVec::GfxMCVec(const GfxMCVec& rhs)
    : TObject()
      ,fParticle(rhs.fParticle), fEventP(rhs.fEventP), fParent(rhs.fParent), fImp(0)
{
}


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

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

void GfxMCVec::Configure(int ipart)
{
    if (fImp) { delete fImp; fImp = 0; }


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

}


bool GfxMCVec::BuildRealImp( int /*ipart*/)
{
   
    float motherlength;

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

    double x1=0;
    double x2=0;
    double y1=0;
    double y2=0;

    int sign=1;
    if((abs(fParticle.GetPdgCode())==12 || 
       abs(fParticle.GetPdgCode())==14 ||
       abs(fParticle.GetPdgCode())==16) && fParticle.GetStatusCode()==0) sign=-1;
    

    if (fParent.GetGraphicsView() == GraphicsView::kUZ) 
      {
        motherlength=4;
        x1 = fParticle.Vz();
        y1 = 0.7071678*(fParticle.Vx()+fParticle.Vy());
        double len=0.2*fParticle.P();
        double pu = 0.7071678*(fParticle.Px()+fParticle.Py());
        double pz = fParticle.Pz();
        double P=fParticle.P();
        if(P>0 && fEventP>0)
          {
            len= motherlength*(fParticle.P()/fEventP);    
            x2=x1+sign*len*pz/P;
            y2=y1+sign*len*pu/P;
          }
      }
    else if(fParent.GetGraphicsView() == GraphicsView::kVZ) 
      {  
        motherlength=4;
        x1  = fParticle.Vz();
        y1 = 0.7071678*(fParticle.Vy()-fParticle.Vx());
        double len=0.2*fParticle.P();
        double pv = 0.7071678*(fParticle.Py()-fParticle.Px());
        double pz = fParticle.Pz();
        double P=fParticle.P();
        if(P>0 && fEventP>0)
          {
            len= motherlength*(fParticle.P()/fEventP);
            x2=x1+sign*len*pz/P;
            y2=y1+sign*len*pv/P;
          }
      }
    else if(fParent.GetGraphicsView() == GraphicsView::kXY) 
      {                
        motherlength=6;
        x1 = fParticle.Vx();
        y1 = fParticle.Vy();
        double px =fParticle.Px(); 
        double py =fParticle.Py();
        double P=fParticle.P();

        double len=0.2*fParticle.P();
        if(P>0 && fEventP>0)
          {
            len= motherlength*(fParticle.P()/fEventP);
            x2=x1+sign*len*px/P;
            y2=y1+sign*len*py/P;
          }
    }
    else if(fParent.GetGraphicsView() == GraphicsView::kUV) 
      {                
        motherlength=6;
        x1 = 0.7071678*(fParticle.Vx()+fParticle.Vy());
        y1 = 0.7071678*(fParticle.Vy()-fParticle.Vx());
        double pu =0.7071678*(fParticle.Px()+fParticle.Py()); 
        double pv =0.7071678*(fParticle.Py()-fParticle.Px());
        double P=fParticle.P();
        double len=0.2*fParticle.P();
        if(P>0 && fEventP>0)
          {
            len= motherlength*(fParticle.P()/fEventP);
            x2=x1+sign*len*pu/P;
            y2=y1+sign*len*pv/P;
          }
      }
    TLine * Vec = new TLine(x1,y1,x2,y2);
    Vec->SetLineColor(15);
    if(abs(fParticle.GetPdgCode())==2212) Vec->SetLineColor(45);
    if(abs(fParticle.GetPdgCode())==12 || 
       abs(fParticle.GetPdgCode())==14 ||
       abs(fParticle.GetPdgCode())==16)  Vec->SetLineColor(1);
    if(abs(fParticle.GetPdgCode())==11)  Vec->SetLineColor(2);  
    if(abs(fParticle.GetPdgCode())==13)  Vec->SetLineColor(3);
    MCVec->Add(Vec);
    
    fImp = MCVec;


    return true;
}

bool GfxMCVec::BuildDiscreteImp(int /*ipart */)
{
  TList* MCVec = new TList;
  MCVec->SetOwner();

  if(fParent.GetGraphicsView() != GraphicsView::kXY && 
     fParent.GetGraphicsView() != GraphicsView::kUV)
      {
      }

    else if(fParent.GetGraphicsView() == GraphicsView::kXY) 
      {                     
     }
    else if(fParent.GetGraphicsView() == GraphicsView::kUV) 
      {                     
  
      }

 
    fImp = MCVec;

    return true;
}

void GfxMCVec::ExecuteEvent(int event, int px, int py)
{
    fParent.ExecuteEvent(event,px,py,this);
}
void GfxMCVec::Paint(Option_t* option)
{
    if (!fImp) return;

    fImp->Paint(option);
}
int GfxMCVec::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;
}

---