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