import Tkinter as tk
import pexpect
import numpy as np
from array import *

##############################################################################
##############################################################################


class Modulo:

    

    def __init__(self, Nmr_modulo):
         self.child = pexpect.spawn('./N957Run '+ Nmr_modulo)    
         #self.Controle['Nmr_serial']=Serial 
         self.Data=array('L',[0]*8192)
         self.Controle=dict(pause=1,Nmr_serial=0,Nmr_canais=512, ON=0)
         self.Livetime = dict(ADC_Conversion=0 , Time=0. , DeadTime=0.)
       


    def start(self):
	self.child.send('s')
        msg=(self.child.readline().strip("\r\n"))

        return msg


    def set_fator(self, canais):  # Define o fator para conversao do numero de canais

        self.Nmr_canais=canais
        if int(canais) == 8192:   # 8192 pontos
            self.child.send('a')   
        elif int(canais) == 4096: # 4096 pontos
            self.child.send('b') 
        elif int(canais) == 2048: # 2048 pontos
            self.child.send('c')
        elif int(canais) == 1024: # 1024 pontos
            self.child.send('d')
        elif int(canais) == 512:  # 512 pontos
            self.child.send('e')
        elif int(canais) == 256:  # 256 pontos
            self.child.send('f')
        elif int(canais) == 128:  # 128 pontos
            self.child.send('g')
        elif int(canais) == 64:   # 64 pontos
            self.child.send('h')

    def load(self): 
        self.child.send('y')

    def LLDT(self, lldt):

        self.child.send('j')
        self.child.send(chr(int(lldt)/10))
        self.child.send(chr(int(lldt)%10))

    def Read_Data(self):

        msg=(self.child.readline().strip("\r\n"))   
        if  msg == 'OK':
            self.Data = eval(self.child.readline().strip("\r\n"))
            self.Livetime['ADC_Conversion'] = int(self.child.readline().strip("\r\n"))
            self.Livetime['Time'] = float(self.child.readline().strip("\r\n"))
            self.Livetime['DeadTime'] = (self.Livetime['Time'] - float(self.child.readline().strip("\r\n")))*100/self.Livetime['Time']
  
        return msg

    def dump(self): # Sinal para programa coletor enviar os dados
	self.child.send('i')

    def reset(self):
        self.child.send('r')
	for k in range(self.Controle['Nmr_canais']):
            self.Data[k] = 0
        

    def pause(self):
        self.child.send('p')