#!/usr/bin/env python
# -*- coding:UTF-8 -*-
"""
    This file will add some timer options to a additional panel to gmoccapy
    and demonstrates at the same time the possibilities you have introducing
    your own handler files and functions to that screen, showing the
    possibilities to modify the layout and behavior

    Copyright 2021 Norbert Schechner (nieson@web.de)
    Copyright 2021 Hans Unzner (hansunzner@gmail.com)

    This program 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 2 of the License, or
    (at your option) any later version.

    This program 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 this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

"""

import hal_glib                           # needed to make our own hal pins
import hal                                # needed to make our own hal pins
from gladevcp.persistence import IniFile  # we use this one to save the states of the widgets on shut down and restart
from gladevcp.persistence import widget_defaults
from gladevcp.persistence import select_widgets
import gtk
from gmoccapy import preferences
from gmoccapy import getiniinfo

#from time import time
import time
import gobject                            # needed to add the timer for periodic

import numpy as np
from numpy import loadtxt, savetxt        # load numpy array from csv file
from datetime import datetime

import subprocess


class CounterClass:

    def __init__(self, halcomp, builder, useropts):

        self.builder = builder
        self.halcomp = halcomp

        self.defaults = { IniFile.vars : { "machine_time"               : 0 ,
                                           "spindle_time"               : 0 ,
                                           "running_time"               : 0 ,
                                           "num_periodes_machine"       : 0 ,
                                           "num_periodes_spindle"       : 0 ,
                                           "total_travel_x"             : 0.0 ,
                                           "total_travel_y"             : 0.0 ,
                                           "total_travel_z"             : 0.0 ,
                                           "total_travel_all"           : 0.0 ,
                                         },

                        }
        self.running_time_prog = 0
        self.running_time_today = 0
        self.spindle_time_today = 0
        self.parts = 0
        get_ini_info = getiniinfo.GetIniInfo()
        prefs = preferences.preferences(get_ini_info.get_preference_file_path())
        theme_name = prefs.getpref("gtk_theme", "Follow System Theme", str)
        if theme_name == "Follow System Theme":
            theme_name = gtk.settings_get_default().get_property("gtk-theme-name")
        gtk.settings_get_default().set_string_property("gtk-theme-name", theme_name, "")

        self.ini_filename = __name__ + ".var"
        self.ini = IniFile(self.ini_filename, self.defaults, self.builder)
        self.ini.restore_state(self)

        # lets make our pins
        self.travel_x = hal_glib.GPin(halcomp.newpin("travel_x", hal.HAL_FLOAT, hal.HAL_IN))
        self.travel_y = hal_glib.GPin(halcomp.newpin("travel_y", hal.HAL_FLOAT, hal.HAL_IN))
        self.travel_z = hal_glib.GPin(halcomp.newpin("travel_z", hal.HAL_FLOAT, hal.HAL_IN))
        self.travel_all = hal_glib.GPin(halcomp.newpin("travel_all", hal.HAL_FLOAT, hal.HAL_IN))
        
        hal_glib.GPin(halcomp.newpin("periode_time_machine", hal.HAL_U32, hal.HAL_IN))
        hal_glib.GPin(halcomp.newpin("elapsed_periodes_machine", hal.HAL_U32, hal.HAL_OUT))
        self.halcomp["elapsed_periodes_machine"] = self.num_periodes_machine

        hal_glib.GPin(halcomp.newpin("alarm_machine", hal.HAL_BIT, hal.HAL_OUT))
        self.halcomp["alarm_machine"] = False
        self.reset_alarm_machine = hal_glib.GPin(halcomp.newpin("reset_alarm_machine", hal.HAL_BIT, hal.HAL_IN))
        self.reset_alarm_machine.connect("value_changed", self._reset_alarm)

        hal_glib.GPin(halcomp.newpin("periode_time_spindle", hal.HAL_U32, hal.HAL_IN))
        hal_glib.GPin(halcomp.newpin("elapsed_periodes_spindle", hal.HAL_U32, hal.HAL_OUT))
        self.halcomp["elapsed_periodes_spindle"] = self.num_periodes_spindle

        hal_glib.GPin(halcomp.newpin("alarm_spindle", hal.HAL_BIT, hal.HAL_OUT))
        self.halcomp["alarm_spindle"] = False
        self.reset_alarm_spindle = hal_glib.GPin(halcomp.newpin("reset_alarm_spindle", hal.HAL_BIT, hal.HAL_IN))
        self.reset_alarm_spindle.connect("value_changed", self._reset_alarm)

        # get all widgets and connect them
        self.lbl_machine_time = self.builder.get_object("lbl_machine_time")
        self.lbl_machine_time.connect("destroy", self._on_destroy)
        self.lbl_spindle_time = self.builder.get_object("lbl_spindle_time")
        self.lbl_running_time = self.builder.get_object("lbl_running_time")
        self.lbl_running_time_prog = self.builder.get_object("lbl_running_time_prog")
        self.lbl_travel_x = self.builder.get_object("lbl_travel_x")
        self.lbl_travel_y = self.builder.get_object("lbl_travel_y")
        self.lbl_travel_z = self.builder.get_object("lbl_travel_z")
        self.lbl_travel_all = self.builder.get_object("lbl_travel_all")
        self.lbl_parts = self.builder.get_object("lbl_parts")
        

        # we need to set the label values
        self.lbl_machine_time.set_text(self._get_label_text_time(self.machine_time))        
        self.lbl_spindle_time.set_text(self._get_label_text_time(self.spindle_time))        
        self.lbl_running_time.set_text(self._get_label_text_time(self.running_time))
        self.lbl_running_time_prog.set_text(self._get_label_text_time(self.running_time_prog))
        self.lbl_travel_x.set_text(self._get_label_text_travel(self.total_travel_x))
        self.lbl_travel_y.set_text(self._get_label_text_travel(self.total_travel_y))
        self.lbl_travel_z.set_text(self._get_label_text_travel(self.total_travel_z))
        self.lbl_travel_all.set_text(self._get_label_text_travel(self.total_travel_all))
        self.lbl_parts.set_text(str(self.parts))

        gobject.timeout_add(1000, self._periodic )  # time between calls to the function, in milliseconds

        self.load_stats_from_file("stats.txt")
        running_time_today = self.stats[-1,1]
        spindle_time_today = self.stats[-1,2]        

    def _reset_alarm(self, pin):
        print(pin.name)
        if not pin.get():
            return
        if pin.name == "reset_alarm_machine":
            self.num_periodes_machine += 1
            self.halcomp["elapsed_periodes_machine"] = self.num_periodes_machine
            self.halcomp["alarm_machine"] = False
        if pin.name == "reset_alarm_spindle":
            self.num_periodes_spindle += 1
            self.halcomp["elapsed_periodes_spindle"] = self.num_periodes_spindle
            self.halcomp["alarm_spindle"] = False


    def on_hal_status_file_loaded(self, widget, filename):
        if filename:
            self.parts = 0
            self.lbl_parts.set_text(str(self.parts))


    def on_hal_status_interp_run(self, emc_stat):
        if hal.get_value("halui.program.is-running"):
            self.parts += 1
            self.lbl_parts.set_text(str(self.parts))
            self.running_time_prog = 0
            self.lbl_running_time_prog.set_text(self._get_label_text_time(self.running_time_prog))


    def _periodic(self):
        if hal.get_value("halui.machine.is-on"):
            self.machine_time += 1
            self.lbl_machine_time.set_text(self._get_label_text_time(self.machine_time))
            if self.machine_time > self.halcomp["periode_time_machine"] * 60 * (self.num_periodes_machine + 1): 
                if not self.halcomp["alarm_machine"]:
                    self.halcomp["alarm_machine"] = True
            self.lbl_travel_x.set_text(self._get_label_text_travel(self.total_travel_x + self.travel_x.get()/1000))
            self.lbl_travel_y.set_text(self._get_label_text_travel(self.total_travel_y + self.travel_y.get()/1000))
            self.lbl_travel_z.set_text(self._get_label_text_travel(self.total_travel_z + self.travel_z.get()/1000))
            self.lbl_travel_all.set_text(self._get_label_text_travel(self.total_travel_all + self.travel_all.get()/1000))

        if hal.get_value("halui.spindle.0.is-on"):
            self.spindle_time += 1
            self.spindle_time_today += 1
            self.lbl_spindle_time.set_text(self._get_label_text_time(self.spindle_time))
            if self.spindle_time > self.halcomp["periode_time_spindle"] * 60 * (self.num_periodes_spindle + 1): 
                if not self.halcomp["alarm_spindle"]:
                    self.halcomp["alarm_spindle"] = True

        if hal.get_value("halui.program.is-running") or hal.get_value("motion.current-vel") != 0:
            self.running_time += 1
            self.running_time_today += 1
            self.lbl_running_time.set_text(self._get_label_text_time(self.running_time))            

        if hal.get_value("halui.program.is-running"):
            self.running_time_prog += 1
            self.lbl_running_time_prog.set_text(self._get_label_text_time(self.running_time_prog))

        return True


    def _get_label_text_time(self, seconds):
        hours = int(seconds / 3600)
        minutes = str(int((seconds % 3600)/60)).zfill(2)
        seconds = str(int(seconds % 60)).zfill(2)
        lbl_text = "{0:d}:{1}:{2}".format(hours, minutes, seconds)
        return lbl_text


    def _get_label_text_travel(self, travel):
        lbl_text = "{0:.3f} m".format(travel)
        return lbl_text       


    def load_stats_from_file(self, filename):
        # load data
        try:
            self.stats = loadtxt(filename, delimiter=',', ndmin=2)
            print("Last date from file: {}".format(datetime.utcfromtimestamp(self.stats[-1,0]).strftime('%Y-%m-%d')))
        except IOError:
            print("No file found")
            self.stats = np.array([[time.mktime(datetime.now().timetuple()),0,0,0]])
        # check if last entry is from today
        self.last = datetime.utcfromtimestamp(self.stats[-1,0]).strftime('%Y-%m-%d')
        if datetime.now().strftime("%Y-%m-%d") == self.last:
            print("Last entry is from today (or new file created with current date)")
        else:
            print("New entry for today")
            # python3: data = np.append(data,[[datetime.timestamp(datetime.now()),0,0,0]], axis=0) 
            self.stats = np.append(self.stats,[[time.mktime(datetime.now().timetuple()),0,0,0]], axis=0)

    def on_view_stats_clicked(self, widget):
        self.save_stats('stats.txt')
        subprocess.call('python view_stats.py', shell=True)

    def _on_destroy(self, obj, data = None):
        self.total_travel_x += self.travel_x.get()/1000
        self.total_travel_y += self.travel_y.get()/1000
        self.total_travel_z += self.travel_z.get()/1000
        self.total_travel_all += self.travel_all.get()/1000
        print("Destroy, save state")
        self.ini.save_state(self)
        self.save_stats('stats.txt')


    def save_stats(self, filename):
        self.stats[-1,1] += self.running_time_today
        self.stats[-1,2] += self.spindle_time_today
        self.stats[-1,3] += self.travel_all.get()
        savetxt(filename, self.stats, delimiter=', ', fmt='%8d')


def get_handlers(halcomp, builder, useropts):
    return[CounterClass(halcomp, builder, useropts)]
