source: git/display/DelphesEventDisplay.h@ fafc433

/*
 *  Delphes: a framework for fast simulation of a generic collider experiment
 *  Copyright (C) 2012-2014  Universite catholique de Louvain (UCL), Belgium
 *  
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  
 *  This program 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 General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef DelphesEventDisplay_h
#define DelphesEventDisplay_h

#include <vector>
#include <iostream>
#include "external/ExRootAnalysis/ExRootTreeReader.h"
#include "display/DelphesDisplay.h"
#include "display/Delphes3DGeometry.h"
#include "TChain.h"
#include "TClonesArray.h"

/*
 * assembly.C: sauvegarde as shape-extract -> implement in the geometry class (read/write)
 * histobrowser.C: intégration d'histogrammes dans le display (on pourrait avoir Pt, eta, phi pour les principales collections)
 * also from alice_esd: summary html table
 * 
 */

class DelphesEventDisplay
{
  public:
    DelphesEventDisplay();
    DelphesEventDisplay(const char *configFile, const char *inputFile, Delphes3DGeometry& det3D);
    ~DelphesEventDisplay();

  private:
    void make_gui();
    void load_event();
    void delphes_read();
    void delphes_read_towers(TClonesArray* data, DelphesBranchBase* element);
    void delphes_read_tracks(TClonesArray* data, DelphesBranchBase* element);
    void delphes_read_jets(TClonesArray* data, DelphesBranchBase* element);
    void delphes_read_vectors(TClonesArray* data, DelphesBranchBase* element);
    void readConfig(const char *configFile, Delphes3DGeometry& det3D, std::vector<DelphesBranchBase*>& elements, std::vector<TClonesArray*>& arrays);

    // Configuration and global variables.
    Int_t event_id_;
    ExRootTreeReader *treeReader_;
    Double_t tkRadius_, totRadius_, tkHalfLength_, bz_;
    TChain* chain_;
    std::vector<DelphesBranchBase*> elements_;
    std::vector<TClonesArray*> arrays_;
    DelphesDisplay *delphesDisplay_;

    // gui controls
  public:
     void Fwd() {  
        if (event_id_ < treeReader_->GetEntries() - 1) {
           ++event_id_;
           load_event();
        } else {
           printf("Already at last event.\n");
        }
     }

     void Bck() {
        if (event_id_ > 0) {
           --event_id_;
           load_event();
        } else {
           printf("Already at first event.\n");
        }
     }
};

#endif //DelphesEventDisplay_h