AODCollHandleBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
 
//                                                            //
//  Header file for class AODCollHandleBase                   //
//                                                            //
//  Description: Base class for collection handles            //
//                                                            //
//                                                            //
 
#ifndef VP1AODSYSTEMS_AODCOLLHANDLEBASE_H
#define VP1AODSYSTEMS_AODCOLLHANDLEBASE_H
 
//Local
#include "VP1AODSystems/AODSystemController.h"
 
//VP1
#include "VP1Base/VP1Interval.h"
#include "VP1Base/VP1QtUtils.h"
#include "VP1Base/VP1StdCollection.h"
 
//xAOD
//#include "xAODBase/IParticle.h" // for xAOD::Type::ObjectType
#include "xAODBase/ObjectType.h" // for xAOD::Type::ObjectType
 
// Qt
#include <QColor>
#include <QList>
 
//Misc
#include <iostream>
#include <fstream>
 
class AODSysCommonData;
class AODHandleBase;
class SoSeparator;
class SoMaterial;
class VP1ExtraSepLayerHelper;
//class TrackSysCommonData;
 
//namespace Trk {
//  class Track;
//  class Segment;
//  class IExtrapolator;
//}
 
 
class AODCollHandleBase : public VP1StdCollection
{
 
  Q_OBJECT
 
public:
 
  //Each derived class should implement a "static QStringList
  //availableCollections();" method, which the collection widget will
  //use to determine the available collections.
 
  AODCollHandleBase(AODSysCommonData *, const QString& name, xAOD::Type::ObjectType);
//  AODCollHandleBase(AODSysCommonData *, const QString& name);
 
  virtual void init(VP1MaterialButtonBase* matBut=0) =0;//reimplementations must start with a call to VP1StdCollection::init.
  virtual ~AODCollHandleBase();
 
   //Called just after creation. Should set current cut values and
  //connections with controller to monitor future changes. Reimplement
  //the ..Specific method to setup subsystem specific settings.
  void setupSettingsFromController(const AODSystemController*);
protected:
  virtual void setupSettingsFromControllerSpecific(const AODSystemController*) {};
public:
 
  //  For loading the data and resetting after each event: //
 
  //For use by the handles:
  QString name() const;
  const AODSysCommonData * common() const { return m_commonData; }
  AODSysCommonData * common() { return m_commonData; }
  const VP1ExtraSepLayerHelper * sepHelper() const { return m_sephelper; }//For attaching/detaching.
  VP1ExtraSepLayerHelper * sepHelper() { return m_sephelper; }//For attaching/detaching.
  void setSepHelper(VP1ExtraSepLayerHelper * sh) { m_sephelper=sh; }//For attaching/detaching. 
  
  virtual QByteArray persistifiableState() const;
  virtual void setState(const QByteArray&);
  virtual void dumpToJSON( std::ofstream& ) const {}; 
  xAOD::Type::ObjectType xAODType() const {return m_type;} 
    
protected:
 
//  //Must be called from derived classes when filling in new object handles.
  virtual void hintNumberOfHandlesInEvent(unsigned)=0;
  virtual void addHandle(AODHandleBase*)=0;
 
  //For iterating over object handles:
  virtual void handleIterationBegin()=0;
  virtual AODHandleBase* getNextHandle()=0; //Returns 0 when done.
  virtual QList<AODHandleBase*> getHandlesList() const =0;
 
  //Override if special cuts. Remember to call base class implementation also for common cuts.
  virtual bool cut(AODHandleBase*) = 0 ;//Return true if should be shown (based on various cuts), false otherwise.
 
  //Utility (fixme: put in utility class elsewhere).
  template <class T> void cleanupPtrContainer(T&) const;//Delete pointers and calls .clear()
  //   template <class T> void cleanupNodeContainer(T&) const;//unref's pointers and calls .clear()
 
  void recheckCutStatus(AODHandleBase*);//Call in derived classes for handles that might be effected by a change in cuts.
 
  //Convenience methods which can be called from derived classes or
  //the trackcollwidget (but specialised procedures might be more
  //optimal)
public:
  void recheckCutStatusOfAllVisibleHandles();
  void recheckCutStatusOfAllNotVisibleHandles();
  void recheckCutStatusOfAllHandles();
  void update3DObjectsOfAllHandles();
  void updateMaterialOfAllHandles();
 
  int nShownHandles() { return m_nshownhandles; }
 
  //For use only by AODHandleBase::setVisible(..):
  void incrementNShownHandles() { ++m_nshownhandles; }
  void decrementNShownHandles() { --m_nshownhandles; }
//
//  virtual bool mayHaveAssociatedObjects() const { return false; }
  
//  enum COLOURBY { COLOUR_PERCOLLECTION, COLOUR_RANDOM, COLOUR_BYPID, COLOUR_MOMENTUM, COLOUR_CHARGE, COLOUR_DISTANCE, COLOUR_VERTEX };
  // static QString toString(const COLOURBY&);
  // 
  // COLOURBY colourBy() const { return m_colourby; }
 
//  virtual COLOURBY defaultColourBy () const { return COLOUR_PERCOLLECTION; }
//  virtual bool     allowColourByPID() const { return false; }
//  virtual bool     allowColourByMomentum() const { return false; }
//  virtual bool     allowColourByCharge() const { return false; }
//  virtual bool     allowColourByVertex() const { return false;  }
//
protected:
 
  qint32          provideCollTypeID() const;
  virtual QString provideSection() const;
  virtual QString provideSectionToolTip() const;
  virtual QString provideText() const { return name(); };//FIXME
  virtual void    assignDefaultMaterial(SoMaterial*) const;
  virtual QColor  defaultColor() const = 0;//Will be used in assignDefaultMaterial
  virtual QString matButtonToolTip() const { return "Edit cuts/properties/material for "+text(); }
  QList<QWidget*> provideExtraWidgetsForGuiRow() const;
  QByteArray      extraWidgetsState() const;
  void            setExtraWidgetsState(const QByteArray&);
 
private Q_SLOTS:
  void colourByComboBoxItemChanged();
  void collVisibilityChanged(bool);
  virtual void resetCachedValuesCuts() = 0; // must be implemented in inherited classes, according to the relevant physics cuts and the related user input form
  
public Q_SLOTS:
//  void rerandomiseRandomTrackColours();
 
  // setters
//  void setCutAllowedPt(const VP1Interval&);
//  void setCutAllowedEta(const VP1Interval&);
//  void setCutAllowedPhi(const QList<VP1Interval>&);
 
  //getters
//  VP1Interval getCutAllowedPt() {return m_cut_allowedPtSq; };
//  VP1Interval getCutAllowedEta() {return m_cut_allowedEta; };
//  QList<VP1Interval> getCutAllowedPhi() {return m_cut_allowedPhi; };
//  bool getPtAllowall() {return m_cut_pt_allowall; };
//  bool getEtaAllowall() {return m_cut_eta_allowall; };
//  bool getPhiAllowall() {return m_cut_phi_allowall; };
 
 
private:
 
  class Imp;
  Imp * m_dbase;
 
  int m_nshownhandles;
  xAOD::Type::ObjectType   m_type;
  AODSysCommonData       * m_commonData;
  VP1ExtraSepLayerHelper * m_sephelper;
 
//  VP1Interval m_cut_allowedPtSq;//We store the allowed interval for pt squared - to avoid sqrt's.
//  VP1Interval m_cut_allowedEta;
//  QList<VP1Interval> m_cut_allowedPhi;
//  bool m_cut_pt_allowall;
//  bool m_cut_eta_allowall;
//  bool m_cut_phi_allowall;
//  bool m_cut_etaptphi_allwillfail;
  void fillObjectBrowser();
  void updateObjectBrowserVisibilityCounts();
};
 
#endif