#include <cmath>
#include <algorithm>
#include <iostream>
#include <set>

#include <tut.h>
#include <TGeoManager.h>

#include "tutRunSimulation.hxx"
#include "tutTestHits.hxx"

#include "HEPUnits.hxx"
#include "ICOMETEvent.hxx"
#include "IG4HitMonitor.hxx"
#include "IG4Trajectory.hxx"
#include "IG4TrajectoryPoint.hxx"

namespace tut {
    struct baseHitStopped {
        baseHitStopped() {
            // Run before each test.
            GenerateSIMG4Events("HitStopped");
        }
        ~baseHitStopped() {
            // Run after each test.
        }
    };

    // Declare the test
    typedef test_group<baseHitStopped>::object testHitStopped;
    test_group<baseHitStopped> groupHitStopped("HitStopped");

    // Check for stopped muons and pions in the event we know to have both
    template<> template<>
    void testHitStopped::test<1> () {
        int eventStopped = SimG4::StartEvents[7];
        TestHits::HitChecker checkHits = TestHits::HitChecker(SimG4::Events, 
                                                              eventStopped,
                                                              eventStopped+5);
        bool hasStopped = false;
        bool hasEntry = false;
        for (TestHits::HitChecker::const_iterator hitIter = checkHits.begin();
                                                  hitIter != checkHits.end();
                                                ++hitIter){
            // Ensure we are dealing with a monitor hit
            const COMET::IG4HitMonitor* hit 
                = dynamic_cast<const COMET::IG4HitMonitor*>(&hitIter);
            if (hit) {
                // Check if it is a muon/pion
                int pdgCode = hit->GetPDGEncoding();
                // Check if its an entry hit or a hit inside the volume
                int fluxType = hit->GetFluxType();
                if (fluxType == 0){
                    // Make sure hits inside the volume are from  pion/muon
                    ensure("Hits inside volume caused from muon or pion",
                           pdgCode == 13  || pdgCode == -211 ||
                           pdgCode == -13 || pdgCode == 211);
                    // Ensure all hits inside the volume are stopped hits
                    double magMom = TVector3(hit->GetMomentumX(),
                                             hit->GetMomentumY(),
                                             hit->GetMomentumZ()).Mag();
                    ensure_equals("Stopped particles have no momentum",
                                  magMom, 0.0);
                    hasStopped = true;
                } else if (fluxType == 1) hasEntry = true;
            }
        }
        ensure("Stopped particle found", hasStopped);
        ensure("Entry monitor particle found", hasEntry);
    }

};