MCTree.cxx

Go to the documentation of this file.

#include "MCTree.h"
#include <Midad/Gui/GuiTree.h>
#include <Midad/Base/Mint.h>
#include <Midad/Base/Jint.h>

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

#include <TParticle.h>
#include <TClonesArray.h>

#include <string>
using namespace std;

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

#include "pdg_kludge.h"



MCTree::MCTree(GuiTree* gtv)
    : fTree(gtv)
{
    pdg_kludge();
}
MCTree::~MCTree()
{
    cerr << "MCTree::~MCTree()\n";
}


void MCTree::Update(Mint* mint)
{
    if (!fTree) return;
    fTree->DeleteAllItems();

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

    SimSnarlRecord* ssr = 
        dynamic_cast<SimSnarlRecord*>(mom->GetFragment("SimSnarlRecord"));
    if (!ssr) return;

    this->AddSimSnarl(ssr);
}

static const char* tool_tip(TParticle* p)
{
    return 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());
}

static void bark(TParticle* p)
{
    cerr << tool_tip(p) << endl;
}

void MCTree::AddSimSnarl(SimSnarlRecord* ssr)
{
    const TClonesArray* ctca = 
        dynamic_cast<const TClonesArray*>
        (ssr->FindComponent("TClonesArray","StdHep"));
    if (!ctca) return;

    vector<GuiTreeEntry*> entries;

    int ind, siz = ctca->GetEntriesFast();
    for (ind=0; ind < siz; ++ind) {
        TParticle* part = dynamic_cast<TParticle*>((*ctca)[ind]);
        if (!part) {
            cerr << "Non particle: " << (*ctca)[ind]->GetName() << endl;
            continue;
        }

        GuiTreeEntry *entry = 0;
        int mother = part->GetFirstMother();
        if (mother != -1) entry = entries[mother];

        cerr << "Mother=" << mother << " ind=" << ind
             << " " << part->GetName() 
             << endl;
        GuiTreeEntry* e=0;
        if (part->GetStatusCode() >= 200) 
            e = fTree->AddEntry(entry,Form("(%s)",part->GetName()));
        else
            e = fTree->AddEntry(entry,part->GetName());
        e->on_mouse_over.connect(bind(slot(bark),part));
        fTree->SetToolTipItem(e,tool_tip(part));

        entries.push_back(e);
        fTree->OpenItem(e);
    }

    
}

---