#include <cometEventLoop.hxx>
#include <iostream>
#include <IFieldManager.hxx>
#include <IOADatabase.hxx>
#include <TGeoManager.h>


class ITestEventLoop: public COMET::ICOMETEventLoopFunction {
public:
    ITestEventLoop() {
        first = true;
    }

    virtual ~ITestEventLoop() {};

    virtual void BeginFile(COMET::IVInputFile *const){
        if(first){
            //test field with command line: testEventLoop.exe -b c data_file.root
            //or testEventLoop.exe -b [u|n|fieldFile] data_file.root
            double xyz[3] = {6410.*unit::mm, 0.*unit::mm, 7650.*unit::mm};
            double B[6];
            try{
              COMET::IFieldManager::GetObject()->GetFieldValue(xyz,B);
              std::cout<<"ITestEventLoop:xyz ("<<xyz[0]<<","<<xyz[1]<<","<<xyz[2]<<")mm Bxyz ("<<B[0]/unit::tesla<<","<<B[1]/unit::tesla<<","<<B[2]/unit::tesla<<")Tesla"<<std::endl;
            } catch (const std::exception& e){
               std::cout << e.what() << '\n';
            }

            try{
              if(COMET::IOADatabase::Get().GetGeometry()) {
                std::cout<< "GetGeometry()->NNodes " << COMET::IOADatabase::Get().GetGeometry()->GetNNodes() <<std::endl;
              }else{
                std::cout<< "No Geometry loaded" << std::endl;
              }
            } catch (const std::exception& e){
              std::cout << e.what() << '\n';
            }

          first = false;
        }
    }

private:
    bool first;
};

int main(int argc, char **argv) {
    ITestEventLoop userCode;
    cometEventLoop(argc,argv,userCode);
}