/*
 *  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 DelphesBranchElement_h
#define DelphesBranchElement_h

#include "TClass.h"
#include "TClonesArray.h"
#include "TColor.h"
#include "TEveElement.h"
#include "TEveTrack.h"
#include "TString.h"
#include "display/DelphesCaloData.h"
#include <exception>
#include <iostream>

// virtual class to represent objects from a Delphes-tree branch
class DelphesBranchBase
{
public:
  DelphesBranchBase(const char *name = "", TClonesArray *branch = NULL, const enum EColor color = kBlack, Float_t maxPt = 50.) :
    name_(name), maxPt_(maxPt), branch_(branch), color_(color) {}
  virtual ~DelphesBranchBase() {}
  const char *GetName() const { return (const char *)name_; }
  const char *GetType() const { return branch_ ? branch_->GetClass()->GetName() : "None"; }
  virtual const char *GetClassName() = 0;
  enum EColor GetColor() const { return color_; }
  virtual void Reset() = 0;
  virtual void SetTrackingVolume(Float_t r, Float_t l, Float_t Bz = 0.)
  {
    tkRadius_ = r;
    tkHalfLength_ = l;
    tk_Bz_ = Bz;
  }
  virtual void ReadBranch() = 0;
  virtual std::vector<TLorentzVector> GetVectors() = 0;

protected:
  TString name_;
  Float_t maxPt_;
  TClonesArray *branch_;
  const enum EColor color_;
  Float_t tkRadius_, tkHalfLength_, tk_Bz_;
};

// concrete implementations. EveContainer can be a TrackList, ElementList or CaloData.
template <typename EveContainer>
class DelphesBranchElement: public DelphesBranchBase
{
public:
  // constructor
  DelphesBranchElement(const char *name = "", TClonesArray *branch = NULL, const enum EColor color = kBlack, Float_t maxPt = 50.) :
    DelphesBranchBase(name, branch, color, maxPt)
  {
    throw std::exception();
  }

  // destructor
  virtual ~DelphesBranchElement() { delete data_; }

  // get the container (ElementList, TrackList, or CaloData)
  EveContainer *GetContainer() { return data_; }

  // tracking volume
  virtual void SetTrackingVolume(Float_t r, Float_t l, Float_t Bz = 0.)
  {
    tkRadius_ = r;
    tkHalfLength_ = l;
    tk_Bz_ = Bz;
  }

  // resets the collection (before moving to the next event)
  virtual void Reset(){};

  // template class name
  virtual const char *GetClassName() { return data_->ClassName(); }

  // read the branch and fill elements for display
  virtual void ReadBranch() {}

  // return the vector for all elements
  virtual std::vector<TLorentzVector> GetVectors()
  {
    std::vector<TLorentzVector> v;
    return v;
  }

private:
  EveContainer *data_;
};

#if !defined(__CINT__) && !defined(__CLING__)

// special case for calo towers
template <>
DelphesBranchElement<DelphesCaloData>::DelphesBranchElement(const char *name, TClonesArray *branch, const enum EColor color, Float_t maxPt);
template <>
void DelphesBranchElement<DelphesCaloData>::Reset();
template <>
void DelphesBranchElement<DelphesCaloData>::ReadBranch();
template <>
std::vector<TLorentzVector> DelphesBranchElement<DelphesCaloData>::GetVectors();

// special case for element lists
template <>
DelphesBranchElement<TEveElementList>::DelphesBranchElement(const char *name, TClonesArray *branch, const enum EColor color, Float_t maxPt);
template <>
void DelphesBranchElement<TEveElementList>::Reset();
template <>
void DelphesBranchElement<TEveElementList>::ReadBranch();
template <>
std::vector<TLorentzVector> DelphesBranchElement<TEveElementList>::GetVectors();

// special case for track lists
template <>
DelphesBranchElement<TEveTrackList>::DelphesBranchElement(const char *name, TClonesArray *branch, const enum EColor color, Float_t maxPt);
template <>
void DelphesBranchElement<TEveTrackList>::SetTrackingVolume(Float_t r, Float_t l, Float_t Bz);
template <>
void DelphesBranchElement<TEveTrackList>::Reset();
template <>
void DelphesBranchElement<TEveTrackList>::ReadBranch();
template <>
std::vector<TLorentzVector> DelphesBranchElement<TEveTrackList>::GetVectors();

#endif // CINT, CLING

#endif //DelphesBranchElement_h