#include <algorithm>
#include <memory>
#include <iostream>
#include <tut.h>

#include <TH2F.h>
#include <TFile.h>

#include <TGeoManager.h>

#include <IGeomIdManager.hxx>
#include <IOADatabase.hxx>
#include <oaEvent_version.h>

#include "IGeomVisitor.cxx"

namespace tut {
    struct baseGeomVisitor {
        baseGeomVisitor() {
            // Run before each test.
            COMET::IOADatabase::Get().RegisterGeometryLookup(NULL);
            COMET::IOADatabase::Get().RegisterAlignmentLookup(NULL);
            if (!gGeoManager) {
                COMET::IOADatabase::Get().GeomId().ReadGeometry(
                    COMET::ISHAHashValue());
            }
            else {
                COMET::IOADatabase::Get().GeomId().ResetGeometry();
            }
            COMET::IOADatabase::Get().ClearGeometryCallbacks();
            COMET::IOADatabase::Get().SetGeometryOverride(
                COMET::IOADatabase::Get().GeomId().GetHash());
        }
        ~baseGeomVisitor() {
            // Run after each test.
            COMET::IOADatabase::Get().RegisterGeometryLookup(NULL);
            COMET::IOADatabase::Get().RegisterAlignmentLookup(NULL);
            COMET::IOADatabase::Get().ClearGeometryCallbacks();
        }
    };

    // Declare the test
    typedef test_group<baseGeomVisitor>::object testGeomVisitor;
    test_group<baseGeomVisitor> groupGeomVisitor("GeomVisitor");

    class IFalseVisitor: public COMET::IGeomVisitor {
    public:
        int calls;
        IFalseVisitor() {calls = 0;}
        virtual ~IFalseVisitor() {}
        bool VisitNode(int, const std::string& ,
                       const TGeoNode* ) {
            ++calls;
            return false;
        }
    };


    // Test the default constructor and destructor. 
    template<> template<>
    void testGeomVisitor::test<1> () {
        IFalseVisitor visitor;
        visitor.VisitGeometry();
        ensure_equals("Visitor only called once", visitor.calls,1);
    }

    class ITrueVisitor: public COMET::IGeomVisitor {
    public:
        int calls;
        ITrueVisitor() {calls = 0;}
        virtual ~ITrueVisitor() {}
        bool VisitNode(int, const std::string&,
                       const TGeoNode*) {
            ++calls;
            return true;
        }
    };

    // Test the default constructor and destructor. 
    template<> template<>
    void testGeomVisitor::test<2> () {
        ITrueVisitor visitor;
        visitor.VisitGeometry();
        ensure_greaterthan("Visitor for many nodes", 
                           visitor.calls, 1000);
        ensure_equals("Visitor saw all nodes", 
                      visitor.calls,
                      gGeoManager->CountNodes());
    }
};