DSTViewer.h

#ifndef VIEWERAR_H
#define VIEWERAR_H 1

#include "marlin/Processor.h"
#include "EVENT/MCParticle.h"
#include "lcio.h"
#include <string>
#include <vector>

using namespace lcio ;
using namespace marlin ;


class DSTViewer : public Processor {
  
 public:
  
  virtual Processor*  newProcessor() { return new DSTViewer ; }
  
  DSTViewer() ;
  
  virtual void init() ;
  
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  virtual void processEvent( LCEvent * evt ) ; 
  
  virtual void check( LCEvent * evt ) ; 
  
  virtual void end() ;

  
 protected:

  int _nRun ;
  int _nEvt ;
  // bool dispHelix;
  
//  std::vector<std::string> _caloHitCollections;
//  std::vector<std::string> _simCaloHitCollections;
//  std::vector<std::string> _trackerHitCollections;
//  std::vector<std::string> _simTrackerHitCollections;
//  std::string _trueClustersCollection;
//  std::string _trueTracksCollection;
//  std::string _clustersCollection;
//  std::string _tracksCollection;
  std::string _particleCollection;
  
  StringVec _jetCollections;

  int _layerCaloHit;
  int _layerSimCaloHit;
  int _layerTrackerHit;
  int _layerSimTrackerHit;
  int _layerTrueClusters;
  int _layerTrueTracks;
  int _layerClusters;
  int _layerTracks;
  int _layerMCP;
  int _layerBosons;
  int _layerReco;
  

  int _detModel;

  int _waitForKeyboard;

  std::map<MCParticle *, int > _mcpList;
  
  int returnTrackColor(int type);
  
  int returnClusterColor(float eneCluster, float cutoff_min, float cutoff_max);

    void showLegendSpectrum(const unsigned int color_steps, char scale, int colorMap, float ene_min, float ene_max, unsigned int ticks);

        int returnRGBClusterColor(float eneCluster, float cutoff_min, float cutoff_max, int color_steps, char scale, int colorMap);

        int returnClusterSize(float eneCluster, float cutoff_min, float cutoff_max);
    
        int returnTrackSize(float type);
        
        int returnJetLayer(std::string jetColName);
        
        int returnIpLayer(std::string jetColName);
        
        int returnJetColor(std::string jetColName, int colNumber);
        
        int addAlphaChannelToColor(int color, int alphaChannel);
        
        float * returnConeColor(std::string jetColName);

        float _bField;
    
    void writeLayerDescription(void);

} ;

#endif