---

McDisplay.cxx

Go to the documentation of this file.

// $Header: /nfs/slac/g/glast/ground/cvs/McDisplay/src/McDisplay.cxx,v 1.5 2001/08/27 04:24:17 burnett Exp $
// Original Author: T. Burnett

// Include files
#include "GaudiKernel/Algorithm.h"

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/AlgFactory.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/DataSvc.h"

//gui
#include "GuiSvc/IGuiSvc.h"
#include "gui/DisplayControl.h"
#include "gui/GuiMgr.h"


// GlastEvent for creating the McEvent stuff
#include "GlastEvent/TopLevel/EventModel.h"
#include "GlastEvent/MonteCarlo/McParticle.h"
#include "GlastEvent/MonteCarlo/McVertex.h"


class McDisplay : public Algorithm {
    
public:
    McDisplay(const std::string& name, ISvcLocator* pSvcLocator); 
    StatusCode initialize();
    StatusCode execute();
    StatusCode finalize();
    

    class DisplayMcEvent;
 
    class PrintMcEvent;

private:
    bool m_display_enabled;
    bool m_print_enabled;
    const McVertex*   m_root; 
};
//------------------------------------------------------------------------------
//  factory stuff
static const AlgFactory<McDisplay>  Factory;
const IAlgFactory& McDisplayFactory = Factory;

//------------------------------------------------------------------------------
McDisplay::McDisplay(const std::string& name, ISvcLocator* pSvcLocator) 
:Algorithm(name, pSvcLocator) 
{
    declareProperty("display_enabled",  m_display_enabled="true");
    declareProperty("print_enabled",  m_print_enabled="true");

}
//------------------------------------------------------------------------------
class McVertexRep : public gui::DisplayRep
{
    // Represent the particle trajectory and subsequent shower
public:
    McVertexRep( const McVertex * vert);
    void update(){} // do it in the constructor
};
McVertexRep::McVertexRep(const McVertex * vert) {
    if( vert->finalPosition().mag() != 0 ) { 
        // todo: set color based on charge
        bool neutral =  (vert->mcParticle()->particleID())==22; 
        setColor( neutral? "white" : "black");
        moveTo(vert->initialPosition());
        lineTo(vert->finalPosition());
    }
    //Traverse the Tree
    SmartRefVector<McParticle>::const_iterator i;
    for( i = vert->daughterMcParticles().begin(); i != vert->daughterMcParticles().end();i++){
        const McVertex* v = (*i)->mcVertex();
        append(McVertexRep(v));
    }
    
}
class McDisplay::DisplayMcEvent :  public gui::DisplayRep {
public:
    
    DisplayMcEvent(IDataProviderSvc* dps): m_dps(dps){}
    
    void update() {
        McVertexCol* vertList = SmartDataPtr<McVertexCol>(m_dps, EventModel::MC::McVertexCol);//"/Event/MC/McVertexCol");

        if(vertList) {
            setColor("black");
            const McVertex* launch = *vertList->begin();
            markerAt(launch->initialPosition());
            McVertexRep rep(launch );
            append(rep);
        }
    }
private:
    IDataProviderSvc* m_dps;
};
class McDisplay::PrintMcEvent :  public gui::Command {
public:
    
    PrintMcEvent(IDataProviderSvc* dps): m_dps(dps){}
    
    void execute() {
        McVertexList * vertList = SmartDataPtr<McVertexList>(m_dps, "/Event/MC/McVertices");

        if(vertList) {
            std::cout << "print not enabled yet" << std::endl;
//TODO: finish this            PrintMcEvent rep(*vertList->begin() );
        }
    }
private:
    IDataProviderSvc* m_dps;
};

//------------------------------------------------------------------------------
StatusCode McDisplay::initialize() {
    using namespace gui;   
    MsgStream log(msgSvc(), name());
    log << MSG::INFO << "initialize" << endreq;
    //
    // get the  Gui service
    //
    IGuiSvc* guiSvc=0;
    
    if ( service("GuiSvc", guiSvc).isFailure() ){
        log << MSG::WARNING << "GuiSvc not found, so no event display" << endreq;
    } else {
        
        // get the display from the Gui service and add  rep
        DisplayControl& display = guiSvc->guiMgr()->display();
        
        display.add( new DisplayMcEvent(eventSvc()), "McEvent", m_display_enabled);

        // same for interactive print of event 
        PrintControl& printer = guiSvc->guiMgr()->printer();
        printer.add(new PrintMcEvent(eventSvc()), "McEvent", m_print_enabled);
    }
    return StatusCode::SUCCESS;
}
//------------------------------------------------------------------------------
StatusCode McDisplay::execute() {
    return StatusCode::SUCCESS;
}
//------------------------------------------------------------------------------
StatusCode McDisplay::finalize() {
    
    return StatusCode::SUCCESS;
}

---