#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_1 filename_2\n"
        " Check the difference between two geometry files by inspecting a random set of points in each geometry.\n"
        "\n"
        " -n <n_points>     Set the number of points to sample\n"
        " -o <filename>     Provide a name for the output file (if used) to store the different points\n"
        " -s                Toggle whether or not to save the output to a root file (default: off)\n"
        " -p                Toggle whether or not to print the output to screen (default: on)\n"
        " -a                Toggle whether or not to print out every changed point (default: off)\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,filename_two;
    std::string test_volume;
    std::string point_selector_options="random";
    bool print_diff_summary=true;
    bool print_diff_all=false;
    bool save_diff=false;
    std::string output_file="geometry_diff.root";
    std::string pointfile = "";
    int n_points=1000;
    for (;;) {
        char c = getopt(argc, argv, "h?n:o:spaS:t:f:");
        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 'o':
                {
                    output_file=optarg;
                    continue;
                }
            case 's': 
                save_diff=!save_diff;
                continue;
            case 'p':
                print_diff_summary=!print_diff_summary;
                continue;
            case 'a':
                print_diff_all=!print_diff_all;
                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;

    const int n_args=argc - optind;
    if (n_args<2){
        cout<<"Error: Need the name of two input files containing geometry files"<<endl;
        return 2;
    }else if (n_args>2){
        cout<<"Warning:More than two files provided on command line, using first two arguments only."<<endl;
    }
    filename_one=argv[optind];
    filename_two=argv[optind+1];

    cout<<"\n";
    cout<<"Checking first geometry in: "<<filename_one<<endl;
    COMET::ICOMETInput* file_one=OpenFile(filename_one);
    TGeoManager* geom_one=
        COMET::GeomTools::FetchGeometry(file_one,filename_one,test_volume);
    if(!geom_one) return 3;

    COMET::GeomTools::PointSelector* pointSelector=
        COMET::GeomTools::BuildPointSelector(point_selector_options,
                                             n_points,
                                             geom_one);

    COMET::GeomTools::PointDetailsList pointDetails;
    bool allGood=ScanFirstGeometry(geom_one,pointSelector,pointDetails);
    if(!allGood){
        COMETError("Problem scanning first geometry");
        return 5;
    }

    cout<<"\nChecking second geometry in: "<<filename_two<<endl;
    COMET::ICOMETInput* file_two=OpenFile(filename_two);
    TGeoManager* geom_two=COMET::GeomTools::FetchGeometry(file_two,filename_two);
    if(!geom_two) return 4;

    COMET::GeomTools::PointDiffLists pointDiffs;
    allGood=ScanSecondGeometry(geom_two,pointDetails,pointDiffs);
    if(!allGood){
        COMETError("Problem scanning second geometry");
        return 6;
    }

    if(print_diff_summary || print_diff_all)
        PrintDiff(pointDiffs, print_diff_all);
    if(save_diff)  {
        allGood=SaveEverything(output_file,pointDiffs,pointDetails);
        if(!allGood) return 7;
    }

    return 0;
}