# This file is part of MAUS: http://micewww.pp.rl.ac.uk/projects/maus
#
# MAUS is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# MAUS is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with MAUS.  If not, see <http://www.gnu.org/licenses/>.
#

"""Tests for ReduceCppGlobalPID"""
# pylint: disable = C0103,E1101
import os
import json
import unittest
import Configuration
import datetime
import ROOT
import MAUS

# pylint: disable = R0904
class ReduceCppGlobalPIDTestCase(unittest.TestCase):
    """Tests for ReduceCppGlobalPID"""
    config0 = json.loads(Configuration.Configuration().getConfigJSON())

    @classmethod
    def setUpClass(cls): # pylint: disable = C0103
        """Initialize a reducer object"""
        cls.reducer = MAUS.ReduceCppGlobalPID()

    def test_process(self): # pylint: disable = R0914
        """Check ReduceCppGlobalPID process function"""
        self.config0['pid_beam_setting'] = 'test_200MeV'
        self.config0['pid_config'] = 'commissioning'
        self.config0['pid_beamline_polarity'] = 'positive'
        now = datetime.datetime.now()
        self.config0['unique_identifier'] = now.strftime("%Y_%m_%dT%H_%M_%S_%f")
        test5 = ('%s/src/reduce/ReduceCppGlobalPID/muon_input.root' 
                                            % os.environ.get("MAUS_ROOT_DIR"))
        self.reducer.birth(json.dumps(self.config0))
        infile = ROOT.TFile(test5)
        t1 = infile.Get("Spill")
        data_ptr = ROOT.MAUS.Data()
        t1.SetBranchAddress("data", data_ptr)
        for i in range(1, t1.GetEntries()):
            t1.GetEntry(i) # Update the spill pointed to by data_ptr
            spill = data_ptr.GetSpill()
            if spill.GetDaqEventType() == "physics_event":
                self.reducer.process(data_ptr)
                
        self.reducer.death()
        self.assertTrue("pid_beam_setting" in self.config0)
        self.assertTrue("unique_identifier" in self.config0)
        mu_hypothesis = "test_200MeV_mu_plus"
        e_hypothesis = "test_200MeV_e_plus"
        pi_hypothesis = "test_200MeV_pi_plus"
        timestamp = self.config0["unique_identifier"]
        # Check birth created PDF file
        mu_PDFdirectory = ('%s/files/PID/%s_%s' 
                    % (os.environ.get("MAUS_ROOT_DIR"),
                       mu_hypothesis, timestamp))
        e_PDFdirectory = ('%s/files/PID/%s_%s' 
                    % (os.environ.get("MAUS_ROOT_DIR"),
                       e_hypothesis, timestamp))
        pi_PDFdirectory = ('%s/files/PID/%s_%s' 
                    % (os.environ.get("MAUS_ROOT_DIR"),
                       pi_hypothesis, timestamp))
        PDFfile1 = ('%s/diffTOF1TOF2_%s.root' 
                    % (mu_PDFdirectory, mu_hypothesis))
        self.assertTrue(os.path.exists(PDFfile1))
        rootPDFfile1 = ROOT.TFile(PDFfile1)
        #check histogram exists in file
        hist1name = "diffTOF1TOF2_test_200MeV_mu_plus"
        hist1 = rootPDFfile1.Get(hist1name)
        ## Check process has added info from 3 tracks to histogram:
        ## N.B. for ComPIDVarA the behavior to spread one event over
        ## all bins is on, so there should be as many additional
        ## entries as there are bins.
        self.assertEqual(hist1.GetEntries(), (hist1.GetSize() + 3))
        os.system('rm -rf %s' % mu_PDFdirectory)
        os.system('rm -rf %s' % e_PDFdirectory)
        os.system('rm -rf %s' % pi_PDFdirectory)

    def tearDown(self): #pylint: disable = C0103
        """Death is being checked in each test, so this is empty"""

if __name__ == '__main__':
    unittest.main()