import json

from django.contrib import messages
from django.utils.safestring import mark_safe
from django.shortcuts import render, redirect, get_object_or_404, HttpResponse

from apps.main.models import Configuration, Experiment, Device
from apps.main.views import sidebar

from .models import RCConfiguration, RCLine, RCLineType, RCLineCode
from .forms import RCConfigurationForm, RCLineForm, RCLineViewForm, RCLineEditForm, RCImportForm, RCLineCodesForm
from .utils import plot_pulses


def conf(request, conf_id):

    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    
    lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel')
    
    for line in lines:
        params = json.loads(line.params)
        line.form = RCLineViewForm(extra_fields=params, line=line)
        if 'params' in params:
            line.subforms = [RCLineViewForm(extra_fields=fields, line=line, subform=True) for fields in params['params']]
    
    kwargs = {}
    kwargs['dev_conf'] = conf
    kwargs['rc_lines'] = lines
    kwargs['dev_conf_keys'] = ['name', 'ipp', 'ntx', 'clock_in', 'clock_divider', 'clock', 
                               'time_before', 'time_after', 'sync', 'sampling_reference', 'control_tx', 'control_sw']
    
    kwargs['title'] = 'RC Configuration'
    kwargs['suptitle'] = 'Details'
    
    kwargs['button'] = 'Edit Configuration'
    ###### SIDEBAR ######
    kwargs.update(sidebar(conf=conf))
    
    return render(request, 'rc_conf.html', kwargs)


def conf_edit(request, conf_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    
    lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel')
        
    for line in lines:        
        params = json.loads(line.params)                
        line.form = RCLineEditForm(conf=conf, line=line, extra_fields=params)
        line.subform = False
        
        if 'params' in params:
            line.subforms = [RCLineEditForm(extra_fields=fields, line=line, subform=i) for i, fields in enumerate(params['params'])]
            line.subform = True
    
    if request.method=='GET':
        
        form = RCConfigurationForm(instance=conf)
                
    elif request.method=='POST':
        
        line_data = {}
        conf_data = {}
        extras = []
        
        #classified post fields
        for label,value in request.POST.items():
            if label=='csrfmiddlewaretoken':
                continue
            
            if label.count('|')==0:
                conf_data[label] = value
                continue
            
            elif label.split('|')[0]<>'-1':
                extras.append(label)
                continue 
            
            x, pk, name = label.split('|')            
            
            if name=='codes':
                value = [s for s in value.split('\r\n') if s]
            
            if pk in line_data:
                line_data[pk][name] = value
            else:
                line_data[pk] = {name:value}
        
        #update conf
        form = RCConfigurationForm(conf_data, instance=conf)

        if form.is_valid():
            
            form.save()
        
            #update lines fields 
            extras.sort()
            for label in extras:
                x, pk, name = label.split('|')
                if pk not in line_data:
                    line_data[pk] = {}
                if 'params' not in line_data[pk]:
                    line_data[pk]['params'] = []
                if len(line_data[pk]['params'])<int(x)+1:
                    line_data[pk]['params'].append({})
                line_data[pk]['params'][int(x)][name] = float(request.POST[label])
                
            for pk, params in line_data.items():
                line = RCLine.objects.get(pk=pk)
                if line.line_type.name in ('windows', 'prog_pulses'):
                    if 'params' not in params:
                        params['params'] = []
                line.params = json.dumps(params)
                line.save()
            
            #update pulses field
            for line in conf.get_lines():
                if line.line_type.name=='tr':
                    continue
                line.update_pulses()
            
            for tr in conf.get_lines('tr'):
                tr.update_pulses()                    
            
            messages.success(request, 'RC Configuration successfully updated')
            
            return redirect(conf.get_absolute_url())    
    
    kwargs = {}
    kwargs['dev_conf'] = conf
    kwargs['form'] = form
    kwargs['rc_lines'] = lines
    kwargs['edit'] = True
    
    kwargs['title'] = 'RC Configuration'
    kwargs['suptitle'] = 'Edit'    
    kwargs['button'] = 'Update'
    kwargs['previous'] = conf.get_absolute_url()
    
    return render(request, 'rc_conf_edit.html', kwargs)


def add_line(request, conf_id, line_type_id=None, code_id=None):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    
    if request.method=='GET':
        if line_type_id:
            line_type = get_object_or_404(RCLineType, pk=line_type_id)
            
            if code_id:
                form = RCLineForm(initial={'rc_configuration':conf_id, 'line_type': line_type_id, 'code_id': code_id},
                                  extra_fields=json.loads(line_type.params))
            else:
                form = RCLineForm(initial={'rc_configuration':conf_id, 'line_type': line_type_id},
                                  extra_fields=json.loads(line_type.params))
        else:
            line_type = {'id':0}
            form = RCLineForm(initial={'rc_configuration':conf_id})
        
    if request.method=='POST':
        
        line_type = get_object_or_404(RCLineType, pk=line_type_id)
        form = RCLineForm(request.POST,
                          extra_fields=json.loads(line_type.params))
        
        if form.is_valid():
            form.save()
            form.instance.update_pulses()
            return redirect('url_edit_rc_conf', conf.id)
          
    kwargs = {}
    kwargs['form'] = form
    kwargs['title'] = 'RC Configuration'
    kwargs['suptitle'] = 'Add Line'
    kwargs['button'] = 'Add'
    kwargs['previous'] = conf.get_absolute_url_edit()
    kwargs['dev_conf'] = conf
    kwargs['line_type'] = line_type
    
    return render(request, 'rc_add_line.html', kwargs)

def edit_codes(request, conf_id, line_id, code_id=None):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    line = get_object_or_404(RCLine, pk=line_id)    
    params = json.loads(line.params)
    
    if request.method=='GET':
        if code_id:
            code = get_object_or_404(RCLineCode, pk=code_id)
            form = RCLineCodesForm(instance=code)
        else:
            initial = {'code': params['code'], 
                       'codes': params['codes'] if 'codes' in params else [],
                       'number_of_codes': len(params['codes']) if 'codes' in params else 0,
                       'bits_per_code': len(params['codes'][0]) if 'codes' in params else 0,
                       }
            form = RCLineCodesForm(initial=initial)
    
    if request.method=='POST':
        form = RCLineCodesForm(request.POST)
        if form.is_valid():
            params['code'] = request.POST['code']
            params['codes'] = [s for s in request.POST['codes'].split('\r\n') if s]
            line.params = json.dumps(params)
            line.save()
            messages.success(request, 'Line: "%s" has been updated.'  % line)
            return redirect('url_edit_rc_conf', conf.id)
    
    kwargs = {}
    kwargs['form'] = form
    kwargs['title'] = line
    kwargs['suptitle'] = 'Edit' 
    kwargs['button'] = 'Update'
    kwargs['dev_conf'] = conf
    kwargs['previous'] = conf.get_absolute_url_edit()
    kwargs['line'] = line
    
    return render(request, 'rc_edit_codes.html', kwargs)

def add_subline(request, conf_id, line_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    line = get_object_or_404(RCLine, pk=line_id)
    
    if request.method == 'POST':
        if line:
            params = json.loads(line.params)
            subparams = json.loads(line.line_type.params)
            if 'params' in subparams:
                dum = {}
                for key, value in subparams['params'].items():
                    dum[key] = value['value']
                params['params'].append(dum)
                line.params = json.dumps(params)
                line.save()
            return redirect('url_edit_rc_conf', conf.id)

    kwargs = {}
    
    kwargs['title'] = 'Add new'
    kwargs['suptitle'] = '%s to %s' % (line.line_type.name, line)
    
    return render(request, 'confirm.html', kwargs)

def remove_line(request, conf_id, line_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    line = get_object_or_404(RCLine, pk=line_id)
    
    if request.method == 'POST':
        if line:
            try:
                channel = line.channel
                line.delete()
                for ch in range(channel+1, RCLine.objects.filter(rc_configuration=conf).count()+1):
                    l = RCLine.objects.get(rc_configuration=conf, channel=ch)
                    l.channel = l.channel-1
                    l.save()
                messages.success(request, 'Line: "%s" has been deleted.'  % line)
            except:
                messages.error(request, 'Unable to delete line: "%s".' % line) 

        return redirect('url_edit_rc_conf', conf.id)
            
    kwargs = {}
    
    kwargs['object'] = line
    kwargs['delete_view'] = True
    kwargs['title'] = 'Confirm delete'
    kwargs['previous'] = conf.get_absolute_url_edit()
    return render(request, 'confirm.html', kwargs)


def remove_subline(request, conf_id, line_id, subline_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    line = get_object_or_404(RCLine, pk=line_id)
    
    if request.method == 'POST':
        if line:
            params = json.loads(line.params)
            params['params'].remove(params['params'][int(subline_id)-1])
            line.params = json.dumps(params)
            line.save()

            return redirect('url_edit_rc_conf', conf.id)

    kwargs = {}
    
    kwargs['object'] = line
    kwargs['object_name'] = line.line_type.name
    kwargs['delete_view'] = True
    kwargs['title'] = 'Confirm delete'
    
    return render(request, 'confirm.html', kwargs)
    

def update_lines_position(request, conf_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    
    if request.method=='POST':
        ch = 0
        for item in request.POST['items'].split('&'):            
            line = RCLine.objects.get(pk=item.split('=')[-1])
            line.channel = ch
            line.save()
            ch += 1          
            
        lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel')
    
        for line in lines:
            params = json.loads(line.params)                
            line.form = RCLineEditForm(conf=conf, line=line, extra_fields=params)
        
            if 'params' in params:
                line.subform = True
                line.subforms = [RCLineEditForm(extra_fields=fields, line=line, subform=i) for i, fields in enumerate(params['params'])]
        
        html = render(request, 'rc_lines.html', {'dev_conf':conf, 'rc_lines':lines, 'edit':True})
        data = {'html': html.content}
        
        return HttpResponse(json.dumps(data), content_type="application/json")
    return redirect('url_edit_rc_conf', conf.id)

    
def import_file(request, conf_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    if request.method=='POST':
        form = RCImportForm(request.POST, request.FILES)
        if form.is_valid():
            #try:
            if True:
                
                conf.update_from_file(request.FILES['file_name'])
                conf.save()
                
                for line in conf.get_lines():
                    if line.line_type.name=='tr':
                        continue
                    line.update_pulses()
                
                for tr in conf.get_lines('tr'):
                    tr.update_pulses()                            
                
                messages.success(request, 'Configuration "%s" loaded succesfully' % request.FILES['file_name'])
                return redirect(conf.get_absolute_url())
            
            #except Exception as e:    
            #    messages.error(request, 'Error parsing file: "%s" - %s' % (request.FILES['file_name'], e))
        
    else:
        messages.warning(request, 'Your current configuration will be replaced')
        form = RCImportForm()    
    
    kwargs = {}
    kwargs['form'] = form
    kwargs['title'] = 'RC Configuration'
    kwargs['suptitle'] = 'Import file'
    kwargs['button'] = 'Upload'
    kwargs['previous'] = conf.get_absolute_url()
    
    return render(request, 'rc_import.html', kwargs)
    

def view_pulses(request, conf_id):
    
    conf = get_object_or_404(RCConfiguration, pk=conf_id)
    lines = RCLine.objects.filter(rc_configuration=conf)
    
    unit = (conf.clock/conf.clock_divider)*3./20
    
    N = int(conf.ipp*(conf.clock/conf.clock_divider)*20./3)*conf.ntx
        
    script, div = plot_pulses(unit, N, lines) 
    
    kwargs = {}
        
    kwargs['title'] = 'RC Pulses'
    kwargs['suptitle'] = conf.name
    kwargs['div'] = mark_safe(div)
    kwargs['script'] = mark_safe(script)
                   
    return render(request, 'rc_pulses.html', kwargs)