#!/usr/bin/env run-groovy
/****************************************************************
 *    	Tutorial for CLAS12 analysis 22-sep-30	
 *		Coatjava Version: 	8.2.2	
 * **************************************************************
 ~							  	~
 * **************************************************************
 *	How to Run (on my account on ifarm1901):
 *      setenv JAVA_HOME /group/clas12/packages/jdk/11.0.2
 *      setenv COATJAVA /group/clas12/packages/coatjava/8.2.2
 *
 *      Usage: $COATJAVA/bin/run-groovy j2rootTutorial.groovy <hipo data file>.hipo
 *             data files exist in places like  the following:
 *                 /cache/clas12/rg-a/production/recon/fall2018/torus-1/pass1/v1/dst/train/ePiNCND/
 *
 *      Output: rootfile/tutorialHists1.root  (in rootfile area, you can creat your own or not)
 *              tutorialHists1.hipo (in area j2rootTutorial.groovy)
 *                  (delete these two files if you run the code again.)
 *
 *						       
 ****************************************************************/


// imports (probably don't need all of them) #####################
// for reconstruction data
import org.jlab.io.evio.*;
import org.jlab.io.hipo.*;

// for histograms
import org.jlab.groot.data.TDirectory;
import org.jlab.groot.data.H1F;
import org.jlab.groot.data.H2F;
import org.jlab.groot.ui.TCanvas;

// for vectors
import org.jlab.geom.prim.Vector3D;

// for Geometry and Swimming
import org.jlab.clas.physics.Particle;
import org.jlab.detector.base.GeometryFactory;
import org.jlab.detector.base.DetectorType;
import org.jlab.clas.fastmc.*;
import org.jlab.geom.prim.Line3D;
import org.jlab.geom.prim.Shape3D;
import org.jlab.geom.prim.Plane3D;
import org.jlab.geom.prim.Point3D;
import org.jlab.geom.detector.ec.ECSector;
import org.jlab.geom.detector.ec.ECSuperlayer;
import org.jlab.geom.detector.ec.ECLayer;
import org.jlab.geom.base.ConstantProvider;
import org.jlab.geom.detector.ec.ECFactory;
import org.jlab.geom.detector.ec.ECDetector;

// for Objects
import java.lang.Math;
import java.util.ArrayList;
import java.text.DecimalFormat;

// to create root output <---- really need these!!!!!!
import org.jlab.jroot.ROOTFile
import org.jlab.jroot.TNtuple

// create a histogram and directories in memory. ******
TDirectory histFile = new TDirectory();
H1F hpmag;
	        
histFile.mkdir("/hists");
histFile.cd("/hists");

hpmag = new H1F("hpmag",110,0,11);
hpmag.setFillColor(33);
hpmag.setTitle("p[0] magnitude");
hpmag.setTitleX("p [GeV/c]");

histFile.addDataSet(hpmag);  // add histogram to memory

// create ROOT output file.
def root_out = new ROOTFile("rootfile/tutorialHists1.root")

// create an ntuple *********
def myNT = root_out.makeNtuple('nt1', 'my title', 'x:y:z')

// First argument is the input file name
String inputFile = args[0];

// Create coatjava object to read in the data file
HipoDataSource reader = new HipoDataSource();
reader.open(inputFile);
System.out.println("-----------> Reading File: [" + inputFile + "]\n");

// create an event
HipoDataEvent event;

// keep track of counts
int nEvents = 1; // start at first event

// loop over all events (or as many as you want). $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$
int nEventsMax=2000;
System.out.println("-----------> Here we go...\n");

while(reader.hasEvent() && (nEvents<nEventsMax)){

  // get the event
  event = reader.getNextEvent();
  // check that it's a reasonable event and get the components.
  if(event.hasBank("REC::Particle")) {
    HipoDataBank reconBank = event.getBank("REC::Particle");
    int nTracks = reconBank.rows();
    double[] px = reconBank.getFloat("px");
    double[] py = reconBank.getFloat("py");
    double[] pz = reconBank.getFloat("pz");
    // get magnitude of \vec p.
    double pmag = Math.sqrt(Math.pow(px[0],2) + Math.pow(py[0],2) + Math.pow(pz[0],2));
    // fill the histogram
    hpmag.fill(pmag);
    // fill the ntuple with 'random' stuff.
    myNT.fill(px[0],py[0],Math.random());
  } // end of if on REC::Particle

// count the events.
  nEvents++;
} // end of the while loop over the events.

// write the histograms to a hipo file
histFile.writeFile("tutorialHists1.hipo");

// and write the histograms and ntuples to a root file.
root_out.mkdir('mytest');
root_out.cd('mytest');
root_out.writeDataSet(hpmag);

myNT.write();

root_out.close();