MCVectors.cxx

Go to the documentation of this file.

#include "MCVectors.h"
#include "Mint.h"
#include <Midad/Base/Mint.h>
#include <Midad/Base/Jint.h>

#include <MinosObjectMap/MomNavigator.h>
#include <Record/SimSnarlRecord.h>

#include <TParticle.h>
#include <TParticlePDG.h>
#include <TClonesArray.h>
#if ROOT_VERSION_CODE >= ROOT_VERSION(5,15,3)
#include <TView3D.h>
#else
#include <TView.h>
#endif
#include <TList.h>

#include "pdg_kludge.h"

#include <iostream>
using namespace std;

MCVectors::MCVectors(double x1, double y1, double x2, double y2)
{
    pdg_kludge();

    fPad = new TPad("MCVectors","MCVectors",x1,y1,x2,y2);
    fPad->Draw();
    fPad->cd();
#if ROOT_VERSION_CODE >= ROOT_VERSION(5,15,3)
    fView = new TView3D(1,0,0);
#else
    fView = new TView(1);
#endif
    fView->ShowAxis();
    fView->SetAutoRange();
    fView->Centered();

    fParticles = new TList;
}


MCVectors::~MCVectors()
{
}

#if 0
static void spew_particle(TParticle* p)
{
    int mother = p->GetFirstMother();
    cerr << Form("%s: Vtx=(%.2f,%.2f,%.2f,%.2f),  P=(%.2f,%.2f,%.2f,%.2f) ",
                 p->GetName(),
                 p->Vx(),p->Vy(),p->Vz(),p->T(),
                 p->Px(),p->Py(),p->Pz(),p->Energy())
         << "mom = " << mother 
         << endl;
}
#endif

void MCVectors::Update(Mint* mint)
{
    fPad->cd();


//    fPad->Clear();
//    fPad->SetView(fView);
    fParticles->Delete();

    if (!mint) return;
    const MomNavigator* mom = mint->GetJint().GetMom();
    if (!mom) return;

    SimSnarlRecord* ssr = 
        dynamic_cast<SimSnarlRecord*>(mom->GetFragment("SimSnarlRecord"));
    if (!ssr) return;
    
    const TClonesArray* ctca = 
        dynamic_cast<const TClonesArray*>
        (ssr->FindComponent("TClonesArray","StdHep"));
    if (!ctca) return;


    int ind, siz = ctca->GetEntriesFast();
    double x1,y1,z1,x2,y2,z2;
    x1=y1=z1=x2=y2=z2=0.0;
    TParticle* first = 0;
    for (ind=0; ind < siz; ++ind) {
        TParticle* tmp = dynamic_cast<TParticle*>((*ctca)[ind]);
        if (!tmp) {
            cerr << "Non particle: " << (*ctca)[ind]->GetName() << endl;
            continue;
        }

        TParticle* p = new TParticle(*tmp);
        fParticles->Add(p);

        if (!ind) first = p;

        p->SetProductionVertex( 0,0,0,0);

        int mother = p->GetFirstMother();
        if (mother < 0) p->SetLineColor(2);

        p->Draw();
    }
    fPad->Modified();
    fPad->Update();

}


ClassImp(MCVectors)

---