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

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


from numpy import *
from pygolem_lite.modules import *
import time
import os, sys


card_setup = [ #[filename, dt]
#  ['Nidatap.lvm', 5e-6], # 200 kHz 
#  ['Nidatap_6143.lvm', 4e-6], # 250 kHz
  ['NIdata_5114_ch0_ascii.txt'] # 2.5 MHz
  ]


def prepare_data():
  for filename, dt in card_setup:
      try:
	data = loadtxt(filename)
	tvec = arange(len(data)) * dt
	save_adv(filename.rstrip(".lvm"), tvec, data)
	print "prepared data"
      except Exception, e:
	print "Failed saving: " + str(e) 
  

def plot_data(file_type):

    for name in ['NIbasic', 'NIbasic_6143', 'NIbasic_514']:
	#name = "NI Basic"
	# graph all
	#try:
	out = get_data(name, 'ch', "" ,xlim=[0,40],integrated=False , debug=True)
	multiplot(out, name , 'graph_'+name, (6,len(out)*2),  100,  'vertical', file_type)

	os.system('convert -resize 150x120\! graph_'+name+'.png icon.png')

	## integrated 
	out = get_data(name, 'ch', "",xlim=[0,40] , integrated=True , debug=True)
	multiplot(out, name+ " Integrated" , 'graph_'+name + '_int', (6,len(out)*2),  100,  'vertical', file_type)
	#except Exception, e:
	    #print "plotting failed", e

def main():
    if sys.argv[1] ==  "acquisition":
	prepare_data()
    if sys.argv[1] ==  "plots":
	plot_data('png')
	#plot_data('svg')
	saveconst('status', 0)
    


if __name__ == "__main__":
    main()