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