#include "DetSimSD/CaloSensitiveDetector.h"

#include "G4SDManager.hh"

#include <algorithm>

CaloSensitiveDetector::CaloSensitiveDetector(const std::string& name, dd4hep::Detector& description)
    : DDG4SensitiveDetector(name, description),
      m_hc(nullptr) {
    const std::string& coll_name = m_sensitive.hitsCollection();
    collectionName.insert(coll_name);
}

void
CaloSensitiveDetector::Initialize(G4HCofThisEvent* HCE) {

    // the collection name is provided by DD4hep
    const std::string& coll_name = collectionName[0];
    m_hc = new G4THitsCollection<CalorimeterHit>(GetName(), coll_name);

    int HCID = -1;
    if(HCID<0) HCID = G4SDManager::GetSDMpointer()->GetCollectionID(m_hc);
    HCE->AddHitsCollection( HCID, m_hc ); 

}

G4bool
CaloSensitiveDetector::ProcessHits(G4Step* step, G4TouchableHistory*) {

    dd4hep::sim::Geant4StepHandler h(step);
    dd4hep::Position pos = 0.5 * (h.prePos() + h.postPos());
    HitContribution contrib = dd4hep::sim::Geant4Hit::extractContribution(step);
    CalorimeterHit* hit=find(m_hc,dd4hep::sim::HitPositionCompare<CalorimeterHit>(pos));

    //    G4cout << "----------- Geant4GenericSD<Calorimeter>::buildHits : position : " << pos << G4endl;
    if ( !hit ) {
        hit = new CalorimeterHit(pos);
        hit->cellID  = getCellID( step );
        m_hc->insert(hit);
    }
    hit->truth.push_back(contrib);
    hit->energyDeposit += contrib.deposit;


    
    return true;
}

void
CaloSensitiveDetector::EndOfEvent(G4HCofThisEvent* HCE) {

}

CaloSensitiveDetector::CalorimeterHit*
CaloSensitiveDetector::find(const CaloSensitiveDetector::HitCollection* c,
                            const dd4hep::sim::HitCompare<CaloSensitiveDetector::CalorimeterHit>& cmp) {
    typedef std::vector<CalorimeterHit*> _V;
    const _V* v = (const _V*) c->GetVector();
    for (_V::const_iterator i = v->begin(); i != v->end(); ++i) {
        if (cmp(*i)) {
            return *i;
        }
    }
    return 0;
}