#include <ICOMETLog.hxx>
#include <IOADatabase.hxx>
#include <IGeomIdManager.hxx>
#include <ICOMETInput.hxx>

#include <TRandom3.h>
#include <TPolyMarker3D.h>
#include <TPointSet3D.h>
#include <TVector3.h>
#include <TGeoNode.h>
#include <TGeoVolume.h>
#include <TGeoShape.h>
#include <TGeoBBox.h>
#include <TGeoManager.h>

#include <cstdlib>
#include <ios>
#include <sstream>
#include <string>
#include <map>
#include <utility>
#include <IGeomTools.hxx>

using std::cout;
using std::endl;

void Usage(const char* prog){
    cout<<"Usage: "<<prog<<" [options] filename\n"
        " Check the properties of the geometry at a given global coordinate point.\n"
        "\n"
        " -n <n_points>     Set the number of random points to sample\n"
        " -S <selector>     Set the point selector type.  Accepted arguments are:\n"
        "                       random = randomly select points\n"
        "                       beamline = select points along the beamline\n"
        "                       fromfile = select points given in text file\n"
        " -t <test_volume>  Set the volume to check for changes. Defaults to the World volume\n"
        "                   Only works for the \"random\" selector.\n" 
        " -f <file_name>    Set the input filename to search for points for the \"fromfile\" selector\n"
        "                   The input filename format is X Y Z, one per line, in mm like:\n"
        "                   10 11 12\n"
        "                   11.5 13 12.5\n"
        "                   25 -13 -12.5\n"
        <<endl;
}

COMET::ICOMETInput* OpenFile(const std::string& filename){
    return new COMET::ICOMETInput(filename.c_str());
}

int main(int argc,char** argv) {
    COMET::ICOMETLog::Configure();

    std::string filename_one;
    std::string test_volume;
    std::string point_selector_options="random";
    std::string pointfile = "";
    int n_points=1000;
    for (;;) {
        char c = getopt(argc, argv, "h?n:S:f:t:");
        std::stringstream stream;
        if(optarg) stream.str(optarg);
        if (c<0) break;
        switch (c) {
            case 'n':
                {
                    stream>>n_points;
                    break;
                }
            case 't':
                {
                    test_volume=optarg;
                    continue;
                }
            case 'S':
                {
                    point_selector_options=optarg;
                    continue;
                }
            case 'f':
                {
                    pointfile=optarg;
                    continue;
                }
            case 'h':case '?':
                Usage(argv[0]);
                return 0;
            default:
                cout<<"Error: Unknown option: "<<c<<endl<<endl;
                Usage(argv[0]);
                return 1;
        }
        if(stream.fail() || ! stream.eof()){
            cout<<"Error: Bad option value: '"<<optarg<<"' for option '"<<c <<"'"<<endl<<endl;
            Usage(argv[0]);
            return 1;
        }
    }
    // Append any additional arguments in the correct order
    point_selector_options += " " + pointfile;

    // Count the number of position arguments
    const int n_args=argc - optind;
    // Throw an error if there are not input files named
    if (n_args<1){
        cout<<"Error: Need the name of an input file containing geometry"<<endl;
        return 2;
    }
    // Warn if there are too many input files named
    else if (n_args>1){
        cout<<"Warning: More than one file provided on command line,"
              " using first argument only."<<endl;
    }
    // Set the input file to the positional argument
    filename_one=argv[optind];

    // Open the input file
    cout<<"\nChecking first geometry in: "<<filename_one<<endl;
    COMET::ICOMETInput* file_one=OpenFile(filename_one);
    // Fetch the geometry
    TGeoManager* geom_one=
        COMET::GeomTools::FetchGeometry(file_one,filename_one,test_volume);
    if(!geom_one) return 3;

    // Build the point selector
    COMET::GeomTools::PointSelector* pointSelector=
        COMET::GeomTools::BuildPointSelector(point_selector_options,
                                             n_points,
                                             geom_one);
    // If its not built, return an error
    if (!pointSelector) return 4;

    // Build the list of point details from the geometry file
    COMET::GeomTools::PointDetailsList pointDetails;
    bool allGood=ScanFirstGeometry(geom_one,pointSelector,pointDetails);
    if(!allGood){
        COMETError("Problem scanning first geometry");
        return 5;
    }

    // Dump the description
    PrintDetailList(pointDetails, true, false);
    return 0;
}