// -*- C++ -*-
//
// Package:    MultiplicityProducer
// Class:      MultiplicityProducer
//
/**\class MultiplicityProducer MultiplicityProducer.cc DPGAnalysis/SiStripTools/plugins/MultiplicityProducer.cc

 Description: EDProducer of multiplicity maps
 Implementation:

*/
//
// Original Author:  Andrea Venturi
//         Created:  Fri Dec 04 2009
//
//

// system include files
#include <memory>
#include <string>
#include <map>
#include <vector>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/Common/interface/DetSetVector.h"

#include "DataFormats/SiStripCluster/interface/SiStripCluster.h"
#include "DataFormats/SiStripDigi/interface/SiStripDigi.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"

#include "CommonTools/UtilAlgos/interface/DetIdSelector.h"
//
// class decleration
//
template <class T>
class MultiplicityProducer : public edm::global::EDProducer<> {
public:
  explicit MultiplicityProducer(const edm::ParameterSet&);
  ~MultiplicityProducer() override;

private:
  void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
  int multiplicity(typename T::const_iterator det) const;
  int detSetMultiplicity(typename T::const_iterator det) const;

  // ----------member data ---------------------------

  edm::EDGetTokenT<T> m_collectionToken;
  const bool m_clustersize;
  std::map<unsigned int, std::string> m_subdets;
  std::map<unsigned int, DetIdSelector> m_subdetsels;
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
template <class T>
MultiplicityProducer<T>::MultiplicityProducer(const edm::ParameterSet& iConfig)
    : m_collectionToken(consumes<T>(iConfig.getParameter<edm::InputTag>("clusterdigiCollection"))),
      m_clustersize(iConfig.getUntrackedParameter<bool>("withClusterSize", false)),
      m_subdets(),
      m_subdetsels() {
  produces<std::map<unsigned int, int> >();

  //now do what ever other initialization is needed

  std::vector<edm::ParameterSet> wantedsubds(iConfig.getParameter<std::vector<edm::ParameterSet> >("wantedSubDets"));

  for (std::vector<edm::ParameterSet>::iterator ps = wantedsubds.begin(); ps != wantedsubds.end(); ++ps) {
    m_subdets[ps->getParameter<unsigned int>("detSelection")] = ps->getParameter<std::string>("detLabel");
    m_subdetsels[ps->getParameter<unsigned int>("detSelection")] =
        DetIdSelector(ps->getUntrackedParameter<std::vector<std::string> >("selection", std::vector<std::string>()));
  }
}

template <class T>
MultiplicityProducer<T>::~MultiplicityProducer() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

//
// member functions
//

// ------------ method called to produce the data  ------------
template <class T>
void MultiplicityProducer<T>::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  LogDebug("Multiplicity") << " Ready to loop";

  using namespace edm;

  std::unique_ptr<std::map<unsigned int, int> > mults(new std::map<unsigned int, int>);

  Handle<T> digis;
  iEvent.getByToken(m_collectionToken, digis);

  for (std::map<unsigned int, std::string>::const_iterator sdet = m_subdets.begin(); sdet != m_subdets.end(); ++sdet) {
    (*mults)[sdet->first] = 0;
  }

  for (typename T::const_iterator det = digis->begin(); det != digis->end(); ++det) {
    //    if(m_subdets.find(0)!=m_subdets.end()) (*mults)[0]+= det->size();
    if (m_subdets.find(0) != m_subdets.end())
      (*mults)[0] += multiplicity(det);

    DetId detid(det->detId());
    unsigned int subdet = detid.subdetId();

    //    if(m_subdets.find(subdet)!=m_subdets.end() && !m_subdetsels[subdet].isValid() ) (*mults)[subdet] += det->size();
    if (m_subdets.find(subdet) != m_subdets.end()) {
      auto detsel = m_subdetsels.find(subdet);
      if (detsel == m_subdetsels.end() or !detsel->second.isValid())
        (*mults)[subdet] += multiplicity(det);
    }
    for (std::map<unsigned int, DetIdSelector>::const_iterator detsel = m_subdetsels.begin();
         detsel != m_subdetsels.end();
         ++detsel) {
      //      if(detsel->second.isValid() && detsel->second.isSelected(detid)) (*mults)[detsel->first] += det->size();
      if (detsel->second.isValid() && detsel->second.isSelected(detid))
        (*mults)[detsel->first] += multiplicity(det);
    }
  }

  for (std::map<unsigned int, int>::const_iterator it = mults->begin(); it != mults->end(); ++it) {
    LogDebug("Multiplicity") << " Found " << it->second << " digis/clusters in " << it->first << " "
                             << m_subdets.find(it->first)->second;
  }

  iEvent.put(std::move(mults));
}

template <class T>
int MultiplicityProducer<T>::multiplicity(typename T::const_iterator det) const {
  int mult = 0;
  if (m_clustersize) {
    //    edm::LogInfo("multiplicitywithcustersize") << "sono qua: with size";
    mult = detSetMultiplicity(det);

  } else {
    mult = det->size();
    //    edm::LogInfo("multiplicitywithcustersize") << "sono qua senza size";
  }
  return mult;
}

template <class T>
int MultiplicityProducer<T>::detSetMultiplicity(typename T::const_iterator det) const {
  return det->size();
}

template <>
int MultiplicityProducer<edmNew::DetSetVector<SiStripCluster> >::detSetMultiplicity(
    edmNew::DetSetVector<SiStripCluster>::const_iterator det) const {
  int mult = 0;

  for (edmNew::DetSet<SiStripCluster>::const_iterator clus = det->begin(); clus != det->end(); ++clus) {
    //    edm::LogInfo("multiplicitywithcustersize") << "sono qua";
    mult += clus->amplitudes().size();
  }

  return mult;
}

template <>
int MultiplicityProducer<edmNew::DetSetVector<SiPixelCluster> >::detSetMultiplicity(
    edmNew::DetSetVector<SiPixelCluster>::const_iterator det) const {
  int mult = 0;

  for (edmNew::DetSet<SiPixelCluster>::const_iterator clus = det->begin(); clus != det->end(); ++clus) {
    mult += clus->size();
  }

  return mult;
}

//define this as a plug-in
typedef MultiplicityProducer<edmNew::DetSetVector<SiStripCluster> > SiStripClusterMultiplicityProducer;
typedef MultiplicityProducer<edmNew::DetSetVector<SiPixelCluster> > SiPixelClusterMultiplicityProducer;
typedef MultiplicityProducer<edm::DetSetVector<SiStripDigi> > SiStripDigiMultiplicityProducer;

DEFINE_FWK_MODULE(SiStripClusterMultiplicityProducer);
DEFINE_FWK_MODULE(SiPixelClusterMultiplicityProducer);
DEFINE_FWK_MODULE(SiStripDigiMultiplicityProducer);