#ifndef Validation_MuonDTDigis_h
#define Validation_MuonDTDigis_h

/** \class MuonDTDigis
 *  Analyse the the muon-drift-tubes digitizer.
 *
 *  \authors: R. Bellan
 */
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include <FWCore/Framework/interface/Event.h>

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"

#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "Geometry/DTGeometry/interface/DTLayer.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"

#include "DataFormats/DTDigi/interface/DTDigi.h"
#include "DataFormats/DTDigi/interface/DTDigiCollection.h"
#include "DataFormats/MuonDetId/interface/DTLayerId.h"
#include "DataFormats/MuonDetId/interface/DTWireId.h"

#include "SimMuon/DTDigitizer/test/Histograms.h"

#include <vector>

#include <DQMServices/Core/interface/DQMEDAnalyzer.h>
#include <DQMServices/Core/interface/DQMStore.h>

class TH1F;
class TFile;
class PSimHit;
class hDigis;

namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm

class MuonDTDigis : public DQMEDAnalyzer {
public:
  // Constructor
  explicit MuonDTDigis(const edm::ParameterSet &pset);

  // Destructor
  ~MuonDTDigis() override;

protected:
  void bookHistograms(DQMStore::IBooker &, edm::Run const &, edm::EventSetup const &) override;
  // Analysis
  void analyze(const edm::Event &event, const edm::EventSetup &eventSetup) override;

  hDigis *WheelHistos(int wheel);

private:
  typedef std::map<DTWireId, std::vector<const PSimHit *>> DTWireIdMap;

  edm::EDGetTokenT<edm::PSimHitContainer> SimHitToken_;
  edm::EDGetTokenT<DTDigiCollection> DigiToken_;
  //Get DT Geometry
  edm::ESGetToken<DTGeometry, MuonGeometryRecord> muonGeomToken_;
  const DTGeometry *muonGeom;

  // Switch for debug output
  bool verbose_;

  // Monitor elements
  MonitorElement *meDigiTimeBox_;
  MonitorElement *meDigiTimeBox_wheel2m_;
  MonitorElement *meDigiTimeBox_wheel1m_;
  MonitorElement *meDigiTimeBox_wheel0_;
  MonitorElement *meDigiTimeBox_wheel1p_;
  MonitorElement *meDigiTimeBox_wheel2p_;
  MonitorElement *meDigiEfficiency_;
  MonitorElement *meDigiEfficiencyMu_;
  MonitorElement *meDoubleDigi_;
  MonitorElement *meSimvsDigi_;
  MonitorElement *meWire_DoubleDigi_;

  MonitorElement *meMB1_sim_occup_;
  MonitorElement *meMB1_digi_occup_;
  MonitorElement *meMB2_sim_occup_;
  MonitorElement *meMB2_digi_occup_;
  MonitorElement *meMB3_sim_occup_;
  MonitorElement *meMB3_digi_occup_;
  MonitorElement *meMB4_sim_occup_;
  MonitorElement *meMB4_digi_occup_;

  std::vector<MonitorElement *> meDigiTimeBox_SL_;
  MonitorElement *meDigiHisto_;

  //  TH1F *DigiTimeBox;
  //  TFile *file_more_plots;

  std::unique_ptr<hDigis> hDigis_global;
  std::unique_ptr<hDigis> hDigis_W0;
  std::unique_ptr<hDigis> hDigis_W1;
  std::unique_ptr<hDigis> hDigis_W2;
  std::unique_ptr<hHits> hAllHits;
};

#endif
