VisCaloCluster.h

Go to the documentation of this file.

/******************************************************************************
 *
 *  Copyright (C) 2008 Boris Iven, Max Hermann
 *
 *  This file is part of the "CEDRIC Event Display" application.
 *
 *****************************************************************************/

#ifndef VISCALOCLUSTER_H
#define VISCALOCLUSTER_H

#include "FilterVis.h"
#include "PhastInterface.h"
#include "QPhastInterface.h"
#include "PhastNode.h"
#include "EDException.h"
#include <Phast.h>

#include <iostream>

class VisCaloCluster : public FilterVis
{
public:
        VisCaloCluster( int mode, std::string caloName )
        : FilterVis(mode), m_caloName(caloName) 
        {
                //~ std::cout << "VisCaloCluster: Constructed vis " << this << " for calorimeter \"" << m_caloName << "\"" << std::endl;
        }

        PhastNode* visNode()
        {
                //~ std::cout << "VisCaloCluster: Creating visualization node for calorimeter \"" << m_caloName << "\"" << std::endl;
                
                PhastNode* node = new PhastNode;                
                
                // add all clusters belonging to calorimeter
                const PaEvent& event = PhastInterface::ref().phast()->event;
                try
                {
                        const PaCalorimeter& calo = QPhastInterface::getCalorimeter( m_caloName );
                
                        for( int i=0; i < event.NCaloClus(); ++i )
                        {
                                const PaCaloClus& cluster = event.vCaloClus(i);
                                
                          #if 1 //-- determine corresponding cluster by z-position
                                if( calo.IsMyCluster( cluster ) )
                          #else //-- determine corresponding cluster by name (PROBLEMATIC: sometimes names don't match!)
                                if( cluster.CalorimName().compare( m_caloName ) == 0 )
                          #endif
                                {
                                        node->addChild( new PhastNode(cluster) );
                                }
                        }
                }
                catch(EDException e)
                {
                        std::cout << e.getInfo() << std::endl;
                        return node;
                }
                
                
                return node; 
        }
        
        void filter( PhastScene* scene, std::list<PhastNode*>& nodes )
        {
                nodes.push_back( scene->getDetector( m_caloName ) );
                nodes.push_back( scene->eventNode() );
        }
        
protected:
        std::string m_caloName; 
};

#endif