#!/usr/bin/python2
# -*- coding: utf-8 -*-



from pygolem_lite import  Shot
from numpy import vstack
from pygolem_lite.modules import save_adv_par
import time, os, sys

def getdata():
    Data = Shot()
    CD_trigger = Data['tcd']
    #print CD_trigger
    #exit()

    from  isf_read import isf_read
    channel_matrix = []
    t = None
    for ch in ['ch1', 'ch2', 'ch3', 'ch4']:
        try:
            (data,tvec,header) = isf_read(ch+'.isf')
        except IOError:
            print 'file %s was not found'%(ch+'.isf')
            continue

	config = ""
	config += 'Time resolution:' + str(1/header.x_increment * 1e-6) + ' MHz\n'
	config += 'Bit depth:' + str(header.bit_num) + ' bit\n'
	config += "Range:" + str(0.5*header.y_multipt_const*2**header.bit_num) + header.y_unit + '\n'
	config += 'Setting:'+header.setting + '\n'

	f = open(ch + '.cfg', 'w')
	f.write(config)
	f.close()
	os.remove(ch+'.isf')

	t = tvec+CD_trigger
	channel_matrix.append(data)
        print  CD_trigger, t[0],t[-1]
        attr = [d for d in dir(header) if d[0]!= '_']
        for a in attr:
            print a, ':', getattr(header, a)

    print t
    #exit()
    save_adv_par('tektronix3014', t, vstack(channel_matrix).T)

    print "done"
    
	
def graphs(file_type):

    import matplotlib   
    matplotlib.rcParams['backend'] = 'Agg'
    matplotlib.rc('font',  size='10')
    matplotlib.rc('text', usetex=True)  # FIXME !! nicer but slower !!!

    name = "Tektronix"
    
    from pygolem_lite.modules import get_data, paralel_multiplot,saveconst,multiplot


    for i in range(1, 4+1):
        ch_name = 'ch%i' % i
        output_name = 'graph%i' % i
        out = get_data(ch_name, ch_name, 'U [V]', reduction = False)
        paralel_multiplot(out, name , output_name,  (9,3), 100, 'vertical', file_type )
        # integrated
        out = get_data(ch_name, ch_name, "U t [Vs]" , integrated=True )
        paralel_multiplot(out, name + ' Integrated' , output_name + '_intgr', (9,3), 100,  'vertical', file_type)

    
    #os.system("convert -resize 150x120\!  graph1.png icon.png")
    
    # icon
    #out = get_data('ch2', '', '', xlabel = "")
    #paralel_multiplot(out, "" , 'icon', (4,3), 40)
    saveconst('status', 0)


def main():
    try:
	if sys.argv[1] ==  "acquisition":
	    getdata()
	elif sys.argv[1] ==  "plots":
	    graphs('png')
    except IOError:
	print "Missing Tektronix data "
    

if __name__ == "__main__":
    main()