EventDisplay.h

/* 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