/data/refman/pasoursint/CMSSW_4_1_8_patch13/src/Fireworks/Core/interface/FWGeometryTableManager.h

Go to the documentation of this file.

#ifndef Fireworks_Core_FWGeometryTableManager_h
#define Fireworks_Core_FWGeometryTableManager_h
// -*- C++ -*-
//
// Package:     Core
// Class  :     FWGeometryTableManager
// 
//
// Original Author:  Alja Mrak-Tadel, Matevz Tadel
//         Created:  Thu Jan 27 14:50:40 CET 2011
// $Id: FWGeometryTableManager.h,v 1.6 2011/02/14 20:02:51 amraktad Exp $
//

#include <sigc++/sigc++.h>
#include <boost/tr1/unordered_map.hpp>

#include "Fireworks/TableWidget/interface/FWTableManagerBase.h"
#include "Fireworks/TableWidget/interface/FWTextTreeCellRenderer.h"

#include "Fireworks/TableWidget/interface/FWTextTableCellRenderer.h"
#include "Fireworks/TableWidget/interface/FWTableCellRendererBase.h"

#include "TGeoNode.h"

class FWTableCellRendererBase;
class FWGeometryBrowser;

class TGeoManager;
class TGeoNode;

class FWGeometryTableManager : public FWTableManagerBase
{
private:
   struct NodeInfo
   {
      NodeInfo():m_node(0), m_parent(-1), m_level(-1), 
                 m_imported(false), m_visible(false), m_expanded(false)
      {}  

      TGeoNode*   m_node;
      Int_t       m_parent;
      Short_t     m_level;

      Bool_t      m_imported;
      Bool_t      m_visible;
      Bool_t      m_expanded;

      const char* name() const;
   };
   
   struct Match
   {
      bool m_matches;
      bool m_childMatches;
      
      Match() : m_matches(true), m_childMatches(true) {}
   
      bool accepted() { return m_matches || m_childMatches; }
   };

   
   // AMT: this could be a common base class with FWCollectionSummaryModelCellRenderer ..
   class ColorBoxRenderer : public FWTableCellRendererBase
   { 
   public:
      ColorBoxRenderer();
      virtual ~ColorBoxRenderer();
  
      virtual UInt_t width() const { return m_width; }
      virtual UInt_t height() const { return m_height; }
      void setData(Color_t c, bool);
      virtual void draw(Drawable_t iID, int iX, int iY, unsigned int iWidth, unsigned int iHeight);

      UInt_t  m_width;
      UInt_t  m_height;
      Pixel_t m_color;      
      bool    m_isSelected;
      TGGC*   m_colorContext;

   };

public:
   FWGeometryTableManager(FWGeometryBrowser*);
   virtual ~FWGeometryTableManager();
   
   void firstColumnClicked(int row);

   // virtual functions of FWTableManagerBase
   
   virtual int unsortedRowNumber(int unsorted) const;
   virtual int numberOfRows() const;
   virtual int numberOfColumns() const;
   virtual std::vector<std::string> getTitles() const;
   virtual FWTableCellRendererBase* cellRenderer(int iSortedRowNumber, int iCol) const;

   virtual const std::string title() const;

   int selectedRow() const;
   int selectedColumn() const;
   virtual bool rowIsSelected(int row) const;

   std::vector<int> rowToIndex() { return m_row_to_index; }

   void setSelection(int row, int column, int mask); 
   virtual void implSort(int, bool) {}
   
   void loadGeometry(TGeoManager* geoManager);

private:
   enum   ECol { kName, kColor,  kVisSelf, kVisChild, kMaterial, kPosition, kBBoxSize, kNumCol };

   FWGeometryTableManager(const FWGeometryTableManager&); // stop default
   const FWGeometryTableManager& operator=(const FWGeometryTableManager&); // stop default

   // table mng
   void changeSelection(int iRow, int iColumn);
   void redrawTable();
   void recalculateVisibility();
   
   // geo
   void checkUniqueVolume(TGeoVolume* v); 
   // void checkChildMatches(TGeoVolume* v, bool& res);
   void checkChildMatches(TGeoVolume* v,  std::vector<TGeoVolume*>&);
   int  getNdaughtersLimited(TGeoNode*) const;
   void getNNodesTotal(TGeoNode* geoNode, int level,int& off, bool debug) const;
   void setTableContent();
   void importChildren(int parent_idx, bool recurse);
   void checkHierarchy();

   // signal callbacks
   void updateMode();
   void updateFilter();
   void updateAutoExpand();

   bool filterOff() const;
   // ---------- member data --------------------------------
   typedef std::vector<NodeInfo> Entries_v;
   typedef Entries_v::iterator Entries_i;
   
   typedef boost::unordered_map<TGeoVolume*, Match>  Volumes_t;
   typedef Volumes_t::iterator               Volumes_i;
   
   // table stuff
   mutable FWTextTreeCellRenderer m_renderer;  
   mutable ColorBoxRenderer       m_colorBoxRenderer;  

   std::vector<int>  m_row_to_index;
   int               m_selectedRow;
   int               m_selectedColumn;
   
   // geo stuff
   FWGeometryBrowser*   m_browser;
   TGeoManager*       m_geoManager;
      
   mutable Volumes_t  m_volumes;
   Entries_v          m_entries;
   
   // cached values from browser
   int               m_autoExpand;
   int               m_maxDaughters;
   bool              m_modeVolume;

   
   sigc::signal<void,int,int> indexSelected_;
};

//______________________________________________________________________________
inline int
FWGeometryTableManager::getNdaughtersLimited(TGeoNode* geoNode) const
{
   // used for debugging of table
   return TMath::Min(geoNode->GetNdaughters(), m_maxDaughters);
   //return  geoNode->GetNdaughters();
}

#endif