EventDisplay.h

Go to the documentation of this file.

/* Copyright 2011, Technische Universitaet Muenchen,
   Author: Karl Bicker

   This file is part of GENFIT.

   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef genfit_EventDisplay_h
#define genfit_EventDisplay_h

#include "Track.h"
#include "AbsKalmanFitter.h"

#include <TEveBox.h>
#include <TVector3.h>
#include <string>
#include <vector>

#include <TGButton.h>
#include <TGNumberEntry.h>
#include <TGButtonGroup.h>

namespace genfit {

enum eFitterType { SimpleKalman, RefKalman, DafSimple, DafRef };

class EventDisplay : public TNamed {
 private:
  EventDisplay();

 public:
  ~EventDisplay();
  static EventDisplay *getInstance();

  void reset();

  void addEvent(std::vector<genfit::Track *> &tracks);
  void addEvent(std::vector<const genfit::Track *> &tracks);

  void addEvent(const Track *tr);

  void next(unsigned int stp = 1);

  void prev(unsigned int stp = 1);

  void gotoEvent(unsigned int id);

  int getNEvents();

  void setOptions(std::string opts);

  void setErrScale(double errScale = 1.);

  double getErrScale();

  void open();

  void guiGoto();
  void guiGoto2();
  void guiSetDrawParams();
  void guiSelectFitterId(int val);
  void guiSelectMmHandling(int val);

 private:
  void makeGui();

  void drawEvent(unsigned int id, bool resetCam = true);

  TEveBox *boxCreator(TVector3 o, TVector3 u, TVector3 v, float ud, float vd, float depth);

  void makeLines(const StateOnPlane *prevState, const StateOnPlane *state, const AbsTrackRep *rep, const Color_t &color, const Style_t &style, bool drawMarkers, bool drawErrors,
                 double lineWidth = 2, int markerPos = 1);

  static EventDisplay *eventDisplay_;                  
  unsigned int eventId_;                               
  double errorScale_;                                  
  std::vector<std::vector<genfit::Track *> *> events_; 

  TGNumberEntry *guiEvent;
  TGNumberEntry *guiEvent2;

  TGCheckButton *guiDrawGeometry_;
  bool drawGeometry_;
  TGCheckButton *guiDrawDetectors_;
  bool drawDetectors_;
  TGCheckButton *guiDrawHits_;
  bool drawHits_;
  TGCheckButton *guiDrawErrors_;
  bool drawErrors_;

  TGCheckButton *guiDrawPlanes_;
  bool drawPlanes_;
  TGCheckButton *guiDrawTrackMarkers_;
  bool drawTrackMarkers_;

  TGCheckButton *guiDrawTrack_;
  bool drawTrack_;
  TGCheckButton *guiDrawRefTrack_;
  bool drawRefTrack_;
  TGCheckButton *guiDrawForward_;
  bool drawForward_;
  TGCheckButton *guiDrawBackward_;
  bool drawBackward_;

  TGCheckButton *guiDrawAutoScale_;
  bool drawAutoScale_;
  TGCheckButton *guiDrawScaleMan_;
  bool drawScaleMan_;
  TGNumberEntry *guiErrorScale_;

  bool drawSilent_;

  TGCheckButton *guiDrawCardinalRep_;
  bool drawCardinalRep_;
  TGNumberEntry *guiRepId_;
  unsigned int repId_;

  TGCheckButton *guiDrawAllTracks_;
  bool drawAllTracks_;
  TGNumberEntry *guiTrackId_;
  unsigned int trackId_;

  TGCheckButton *guiRefit_;
  bool refit_;
  TGNumberEntry *guiDebugLvl_;
  unsigned int debugLvl_;
  TGButtonGroup *guiFitterId_;
  eFitterType fitterId_;
  TGButtonGroup *guiMmHandling_;
  eMultipleMeasurementHandling mmHandling_;

  TGCheckButton *guiSquareRootFormalism_;
  bool squareRootFormalism_;
  TGNumberEntry *guiDPVal_;
  double dPVal_;
  TGNumberEntry *guiRelChi2_;
  double dRelChi2_;
  TGNumberEntry *guiDChi2Ref_;
  double dChi2Ref_;
  TGNumberEntry *guiNMinIter_;
  unsigned int nMinIter_;
  TGNumberEntry *guiNMaxIter_;
  unsigned int nMaxIter_;
  TGNumberEntry *guiNMaxFailed_;
  int nMaxFailed_;
  TGCheckButton *guiResort_;
  bool resort_;

 public:
  ClassDef(EventDisplay, 1)
};

} /* End of namespace genfit */
#endif // genfit_EventDisplay_h