#include "TFile.h"
#include "TTree.h"
#include "TGraph.h"
#include "TLegend.h"
#include "TCanvas.h"
#include "TH1D.h"
#include "TObjArray.h"
#include "TBranch.h"
#include <iostream>

void aarootscript(int argument = 0) {
    std::cout << "\n\n\n";
    std::cout << "=========================================\n";
    std::cout << "=========   ANASEN Root Script  =========\n";
    std::cout << "=========================================\n";

    TFile *f = new TFile("SimAnasen1.root");
    TTree *tree1 = (TTree*)f->Get("tree");
    if (!tree1) {
        std::cerr << "Error: tree1 not found in the file!" << std::endl;
        return;
    }
    TTree *tree2 = (TTree*)f->Get("tree2");

    TTreeReader reader("tree");
    TTreeReaderValue<double> branchValue(reader, "Tb");
    double totalSum = 0;
    while (reader.Next()) {
        totalSum += *branchValue;
    }
    std::cout << "Total Sum: " << totalSum << std::endl;

    TTreeReader reader2("tree2");
    TTreeReaderValue<double> branchValue2(reader2, "Tb");
    double totalSum2 = 0;
    int zeroCount = 0;
    while (reader2.Next()) {
        totalSum2 += *branchValue2;
        if (*branchValue2 == 0) {
            zeroCount++;
        }
    }
    std::cout << "Total Sum: " << totalSum2 << std::endl;

    std::cout << "Difference: " << totalSum - totalSum2 << std::endl;

    std::cout << "Zero Count: " << zeroCount << std::endl;

    std::cout << "Making histograms..." << std::endl;

    gErrorIgnoreLevel = 2001;
    gROOT->ProcessLine(".x histcomp.C");

    std::cout << "=========================================\n";

    if (argument == 1) {
        if (tree1) {
            gROOT->ProcessLine("tree->Print();");
        } else {
            std::cout << "Tree1 not found!" << std::endl;
        }
    }

    if (argument == 2) {
        if (tree2) {
            gROOT->ProcessLine("tree2->Print();");
        } else {
            std::cout << "Tree2 not found!" << std::endl;
        }
    }
    std::cout << "\n\n\n";

}