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

from pygolem_lite import Shot, saveconst
from pygolem_lite.modules import *
from pygolem_lite.utilities import read_config

import numpy as np

import matplotlib
matplotlib.rcParams['backend'] = 'Agg'
matplotlib.rc('font',  size='10')

import matplotlib.pyplot as plt
import os, sys


R = 47.0                                   # [Ohm] resistance for U -> I


def calculate_M(U_up, U_down):
    return 0.4 * np.log(U_up / U_down) # should be I, but U/R / U/R would be the same


def make_plots1(probe_channels1, automatic_select_up):
    shot = Shot()
    t, data_st = shot['papouch_st']

    for up_ch, down_ch in probe_channels1:
        U_down = data_st[:,down_ch-1] # data columns is indexed from 0
        U_up = data_st[:,up_ch-1]
       # if automatic_select_up and U_down.mean() > U_up.mean():
          #  U_up, U_down = U_down, U_up   #swap
        M = calculate_M(U_up, U_down)
        #make_img1(t, M, up_ch, down_ch)
        data = get_data([t,M], "" , "M", hlines = [0])
        paralel_multiplot(data, "" ,  'M1-ch{}_ch{}'.format(up_ch, down_ch), [12, 6] , 100, 'vertical', 'png')


        
probe_channels1 = [
    (1, 2),
    (3, 4),
    (5, 6),
    (7, 8),
    (9,10),
    (11,12),
    ]

def make_plots2(probe_channels2, automatic_select_up):
    shot = Shot()
    t, data_ja = shot['papouch_ja']

    for up_ch, down_ch in probe_channels2:
        U_down = data_ja[:,down_ch-1] # data columns is indexed from 0
        U_up = data_ja[:,up_ch-1]
       # if automatic_select_up and U_down.mean() > U_up.mean():
          #  U_up, U_down = U_down, U_up   #swap
        M = calculate_M(U_up, U_down)
        data = get_data([t,M], "" , "M", hlines = [0])
        paralel_multiplot(data, "" ,  'M2-ch{}_ch{}'.format(up_ch, down_ch), [12, 6] , 100, 'vertical', 'png')

        
probe_channels2 = [
    (1, 2),
    (3, 4),
    ]
    
def graphs(file_type):
    shot = Shot()
    ShotNo=shot['shotno']
    param = {'ylim':[-0.5,0.5], 'reduction': True, 'ylabel':'U [V]'} 

    outDown = [get_data(['Papouch_St', 1], "" , **param),
               get_data(['Papouch_St', 3], "" , **param),
               get_data(['Papouch_St', 5], "" , **param),
               get_data(['Papouch_St', 7], "" , **param),
               get_data(['Papouch_Ja', 2], "" , **param),
               get_data(['Papouch_St', 9], "" , **param),
               get_data(['Papouch_St', 11], "" , **param),
               get_data(['Papouch_Ja', 1], "" , **param) ]
    
    paralel_multiplot(outDown, '\#'+str(int(ShotNo))+' Down'  , 'graphDown', (6,len(outDown)*2), 100, 'vertical', file_type)

    outUp = [get_data(['Papouch_St', 0], "M1" , **param),
             get_data(['Papouch_St', 2], "M2" , **param),
             get_data(['Papouch_St', 4], "M3" , **param),
             get_data(['Papouch_St', 6], "M4" , **param),
             get_data(['Papouch_Ja', 3], "M5" , **param),
             get_data(['Papouch_St', 8], "M6" , **param),
             get_data(['Papouch_St', 10], "M7" , **param),
             get_data(['Papouch_Ja', 0], "M8" , **param) ]
    


    multiplot(outUp, '\#'+str(int(ShotNo))+' Up'  , 'graphUp', (6,len(outUp)*2), 100, 'vertical', file_type)
	
if __name__ == '__main__':
   #    if sys.argv[1] ==  "plots":
    make_plots1(probe_channels1, automatic_select_up=True)
    make_plots2(probe_channels2, automatic_select_up=True)
    graphs('png')

    saveconst('status', 0)