#include <iostream>
#include <fstream>
#include <time.h>
#include <unistd.h>
#include <getopt.h>

#include <TROOT.h>
#include <TFile.h>
#include <TRint.h>
#include <TSystem.h>

#include "ICOMETLog.hxx"
#include "IRooTrackerHistogram.hxx"

void usage(void) {
    std::cout
    <<"Usage: oaRooTracker_DrawHistogram.exe [options] file\n"
      "  where file specifier is the path to a oaRooTracker histogram file\n"
      "  <filepath>/<oaRooTrackerHistogramFile>\n"
      "The input file histogram is are drawn and saved to a new output file. \n"
      "To draw the input histogram, the following projections are used: \n"
      " * 1D projections in each dimension\n" 
      " * 2D projections in:\n" 
      "   * XY, YZ, XZ spacial dimensions\n" 
      "   * XY, YZ, XZ momentum dimensions\n" 
      "   * TE 4-component dimensions\n" 
      "Options: \n"
      "    -o <name>  Name of the output file\n" 
      "    -s         Flag to skip the 2D projections.  Useful for very high\n" 
      "               dimentionality histograms" << std::endl;
}

int main(int argc, char **argv) {
    // Get the default options
    std::string inputFile;
    std::string outFileName = "addedRTHists.root";
    // Option to only draw the 1D histograms
    bool only1DHistos = false;

    // Set the options
    int c;
    while ((c = getopt(argc, argv, "I:o:hs")) != -1)
        switch (c){
            // Choose the output name
            case 'o':
                outFileName= optarg;
                break;
            // Only draw the 1D part of the histogram
            case 's':
                only1DHistos = true;
                break;
            // Print the help menu
            case 'h':
                usage();
                return 0;
            // Let getopt deal with the rest
            case '?':
                return 1;
            default:
                abort();
        }

    // Prepare the input file
    if (argc - optind != 1) {
        COMETNamedError("RooTrackHist", "Too many positional arguments!" 
                                        " Only one input file can be drawn");
        return 1;
    }

    // Input file is the last positional argument
    inputFile = argv[argc-1];

    //Configure COMETLog levels from input config file
    TString configPath = "${PWD}/cometlog.config";
    gSystem->ExpandPathName(configPath);
    COMET::ICOMETLog::Configure(configPath.Data());

    // Try to draw it
    TFile* outFile;
    try {
        // Open the output
        outFile = TFile::Open(outFileName.c_str(), "RECREATE");
        // Open the file to draw
        COMET::IRooTrackerHistogram* toDraw = 
            COMET::IRooTrackerHistogram::Open(inputFile);
        toDraw->Draw(outFile, only1DHistos);
    } catch (COMET::ECannotDrawHistogram& ex) {
        COMETNamedError("IRooTrackerHistogram", "Cannot open file " <<
                inputFile << " and save its drawn contents in " <<
                outFileName.c_str());
        return 1;
    }
    // Close the merged file
    outFile->Close();
    return 0;
}