models.py
878 lines
| 29.7 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | import ast | ||
|
Juan C. Espinoza
|
r25 | import json | ||
|
Juan C. Espinoza
|
r175 | import requests | ||
|
Juan C. Espinoza
|
r45 | import numpy as np | ||
|
Juan C. Espinoza
|
r175 | from base64 import b64encode | ||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r0 | from django.db import models | ||
|
Juan C. Espinoza
|
r79 | from django.core.urlresolvers import reverse | ||
|
Juan C. Espinoza
|
r23 | from django.core.validators import MinValueValidator, MaxValueValidator | ||
|
Juan C. Espinoza
|
r6 | from apps.main.models import Configuration | ||
|
Juan C. Espinoza
|
r107 | from devices.rc import api | ||
|
Juan C. Espinoza
|
r111 | from .utils import RCFile | ||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r0 | # Create your models here. | ||
|
Juan C. Espinoza
|
r6 | |||
|
Juan C. Espinoza
|
r23 | LINE_TYPES = ( | ||
|
Juan C. Espinoza
|
r45 | ('none', 'Not used'), | ||
|
Juan C. Espinoza
|
r23 | ('tr', 'Transmission/reception selector signal'), | ||
| ('tx', 'A modulating signal (Transmission pulse)'), | ||||
| ('codes', 'BPSK modulating signal'), | ||||
| ('windows', 'Sample window signal'), | ||||
| ('sync', 'Synchronizing signal'), | ||||
| ('flip', 'IPP related periodic signal'), | ||||
| ('prog_pulses', 'Programmable pulse'), | ||||
|
Juan C. Espinoza
|
r107 | ('mix', 'Mixed line'), | ||
|
Juan C. Espinoza
|
r23 | ) | ||
|
Juan C. Espinoza
|
r79 | SAMPLING_REFS = ( | ||
| ('none', 'No Reference'), | ||||
|
Juan C. Espinoza
|
r157 | ('begin_baud', 'Begin of the first baud'), | ||
|
Juan C. Espinoza
|
r79 | ('first_baud', 'Middle of the first baud'), | ||
| ('sub_baud', 'Middle of the sub-baud') | ||||
| ) | ||||
| DAT_CMDS = { | ||||
| # Pulse Design commands | ||||
| 'DISABLE' : 0, # Disables pulse generation | ||||
| 'ENABLE' : 24, # Enables pulse generation | ||||
|
Juan C. Espinoza
|
r172 | 'DELAY_START' : 40, # Write delay status to memory | ||
|
Juan C. Espinoza
|
r79 | 'FLIP_START' : 48, # Write flip status to memory | ||
| 'SAMPLING_PERIOD' : 64, # Establish Sampling Period | ||||
|
Juan C. Espinoza
|
r172 | 'TX_ONE' : 72, # Output '0' in line TX | ||
| 'TX_ZERO' : 88, # Output '0' in line TX | ||||
| 'SW_ONE' : 104, # Output '0' in line SW | ||||
|
Juan C. Espinoza
|
r79 | 'SW_ZERO' : 112, # Output '1' in line SW | ||
| 'RESTART': 120, # Restarts CR8 Firmware | ||||
| 'CONTINUE' : 253, # Function Unknown | ||||
| # Commands available to new controllers | ||||
| # In Pulse Design Executable, the clock divisor code is written as 12 at the start, but it should be written as code 22(below) just before the final enable. | ||||
| 'CLOCK_DIVISOR_INIT' : 12, # Specifies Clock Divisor. Legacy command, ignored in the actual .dat conversion | ||||
| 'CLOCK_DIVISOR_LAST' : 22, # Specifies Clock Divisor (default 60 if not included) syntax: 255,22 254,N-1. | ||||
|
Juan C. Espinoza
|
r172 | 'CLOCK_DIVIDER' : 8, | ||
|
Juan C. Espinoza
|
r79 | } | ||
|
Juan C. Espinoza
|
r6 | class RCConfiguration(Configuration): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r138 | ipp = models.FloatField(verbose_name='IPP [Km]', validators=[MinValueValidator(1), MaxValueValidator(9000)], default=300) | ||
|
Juan C. Espinoza
|
r172 | ntx = models.PositiveIntegerField(verbose_name='Number of TX', validators=[MinValueValidator(1), MaxValueValidator(400)], default=1) | ||
|
Juan C. Espinoza
|
r112 | clock_in = models.FloatField(verbose_name='Clock in [MHz]', validators=[MinValueValidator(1), MaxValueValidator(80)], default=1) | ||
|
Juan C. Espinoza
|
r45 | clock_divider = models.PositiveIntegerField(verbose_name='Clock divider', validators=[MinValueValidator(1), MaxValueValidator(256)], default=1) | ||
|
Juan C. Espinoza
|
r112 | clock = models.FloatField(verbose_name='Clock Master [MHz]', blank=True, default=1) | ||
| time_before = models.PositiveIntegerField(verbose_name='Time before [μS]', default=12) | ||||
| time_after = models.PositiveIntegerField(verbose_name='Time after [μS]', default=1) | ||||
|
Juan C. Espinoza
|
r45 | sync = models.PositiveIntegerField(verbose_name='Synchro delay', default=0) | ||
|
Juan C. Espinoza
|
r79 | sampling_reference = models.CharField(verbose_name='Sampling Reference', choices=SAMPLING_REFS, default='none', max_length=40) | ||
| control_tx = models.BooleanField(verbose_name='Control Switch TX', default=False) | ||||
| control_sw = models.BooleanField(verbose_name='Control Switch SW', default=False) | ||||
|
Juan C. Espinoza
|
r112 | total_units = models.PositiveIntegerField(default=0) | ||
|
Juan C. Espinoza
|
r107 | mix = models.BooleanField(default=False) | ||
|
Juan C. Espinoza
|
r6 | |||
| class Meta: | ||||
| db_table = 'rc_configurations' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def get_absolute_url_plot(self): | ||
| return reverse('url_plot_rc_pulses', args=[str(self.id)]) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def get_absolute_url_import(self): | ||
| return reverse('url_import_rc_conf', args=[str(self.id)]) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r112 | @property | ||
| def ipp_unit(self): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r112 | return '{} ({})'.format(self.ipp, int(self.ipp*self.km2unit)) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | @property | ||
| def us2unit(self): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | return self.clock_in/self.clock_divider | ||
| @property | ||||
| def km2unit(self): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | return 20./3*(self.clock_in/self.clock_divider) | ||
|
Juan C. Espinoza
|
r85 | def clone(self, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | lines = self.get_lines() | ||
| self.pk = None | ||||
| self.id = None | ||||
| for attr, value in kwargs.items(): | ||||
| setattr(self, attr, value) | ||||
| self.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | for line in lines: | ||
| line.clone(rc_configuration=self) | ||||
|
Juan C. Espinoza
|
r172 | |||
| return self | ||||
|
Juan C. Espinoza
|
r85 | |||
|
Juan C. Espinoza
|
r107 | def get_lines(self, **kwargs): | ||
|
Juan C. Espinoza
|
r79 | ''' | ||
|
Juan C. Espinoza
|
r172 | Retrieve configuration lines | ||
|
Juan C. Espinoza
|
r79 | ''' | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | return RCLine.objects.filter(rc_configuration=self.pk, **kwargs) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r79 | def clean_lines(self): | ||
| ''' | ||||
| ''' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | empty_line = RCLineType.objects.get(name='none') | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | for line in self.get_lines(): | ||
| line.line_type = empty_line | ||||
| line.params = '{}' | ||||
| line.save() | ||||
| def parms_to_dict(self): | ||||
| ''' | ||||
|
Juan C. Espinoza
|
r172 | ''' | ||
|
Juan C. Espinoza
|
r111 | ignored = ('parameters', 'type', 'polymorphic_ctype', 'configuration_ptr', | ||
| 'created_date', 'programmed_date') | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | data = {} | ||
| for field in self._meta.fields: | ||||
|
Juan C. Espinoza
|
r111 | if field.name in ignored: | ||
| continue | ||||
|
Juan C. Espinoza
|
r79 | data[field.name] = '{}'.format(field.value_from_object(self)) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r147 | data['device_id'] = data.pop('device') | ||
|
Juan C. Espinoza
|
r79 | data['lines'] = [] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | for line in self.get_lines(): | ||
|
Juan C. Espinoza
|
r172 | line_data = json.loads(line.params) | ||
|
Juan C. Espinoza
|
r79 | if 'TX_ref' in line_data and line_data['TX_ref'] not in (0, '0'): | ||
|
Juan C. Espinoza
|
r172 | line_data['TX_ref'] = RCLine.objects.get(pk=line_data['TX_ref']).get_name() | ||
|
Juan C. Espinoza
|
r79 | if 'code' in line_data: | ||
| line_data['code'] = RCLineCode.objects.get(pk=line_data['code']).name | ||||
| line_data['type'] = line.line_type.name | ||||
|
Juan C. Espinoza
|
r157 | line_data['name'] = line.get_name() | ||
|
Juan C. Espinoza
|
r79 | data['lines'].append(line_data) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | data['delays'] = self.get_delays() | ||
| data['pulses'] = self.get_pulses() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | return data | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | def dict_to_parms(self, data): | ||
| ''' | ||||
| ''' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | self.name = data['name'] | ||
|
Juan C. Espinoza
|
r148 | self.ipp = float(data['ipp']) | ||
| self.ntx = int(data['ntx']) | ||||
|
Juan C. Espinoza
|
r172 | self.clock_in = float(data['clock_in']) | ||
|
Juan C. Espinoza
|
r148 | self.clock_divider = int(data['clock_divider']) | ||
| self.clock = float(data['clock']) | ||||
|
Juan C. Espinoza
|
r107 | self.time_before = data['time_before'] | ||
| self.time_after = data['time_after'] | ||||
| self.sync = data['sync'] | ||||
| self.sampling_reference = data['sampling_reference'] | ||||
|
Juan C. Espinoza
|
r119 | self.total_units = self.ipp*self.ntx*self.km2unit | ||
|
Juan C. Espinoza
|
r114 | self.save() | ||
|
Juan C. Espinoza
|
r107 | self.clean_lines() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | lines = [] | ||
|
Juan C. Espinoza
|
r172 | positions = {'tx':0, 'tr':0} | ||
| for i, line_data in enumerate(data['lines']): | ||||
| name = line_data.pop('name', '') | ||||
|
Juan C. Espinoza
|
r107 | line_type = RCLineType.objects.get(name=line_data.pop('type')) | ||
| if line_type.name=='codes': | ||||
| code = RCLineCode.objects.get(name=line_data['code']) | ||||
|
Juan C. Espinoza
|
r172 | line_data['code'] = code.pk | ||
|
Juan C. Espinoza
|
r107 | line = RCLine.objects.filter(rc_configuration=self, channel=i) | ||
| if line: | ||||
| line = line[0] | ||||
| line.line_type = line_type | ||||
| line.params = json.dumps(line_data) | ||||
| else: | ||||
|
Juan C. Espinoza
|
r172 | line = RCLine(rc_configuration=self, line_type=line_type, | ||
|
Juan C. Espinoza
|
r107 | params=json.dumps(line_data), | ||
| channel=i) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | if line_type.name=='tx': | ||
| line.position = positions['tx'] | ||||
| positions['tx'] += 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | if line_type.name=='tr': | ||
| line.position = positions['tr'] | ||||
| positions['tr'] += 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | line.save() | ||
| lines.append(line) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | for line, line_data in zip(lines, data['lines']): | ||
| if 'TX_ref' in line_data: | ||||
| params = json.loads(line.params) | ||||
| if line_data['TX_ref'] in (0, '0'): | ||||
| params['TX_ref'] = '0' | ||||
| else: | ||||
|
Juan C. Espinoza
|
r113 | params['TX_ref'] = [l.pk for l in lines if l.line_type.name=='tx' and line_data['TX_ref'] in l.get_name()][0] | ||
|
Juan C. Espinoza
|
r107 | line.params = json.dumps(params) | ||
| line.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def get_delays(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | pulses = [line.pulses_as_points() for line in self.get_lines()] | ||
|
Juan C. Espinoza
|
r79 | points = [tup for tups in pulses for tup in tups] | ||
| points = set([x for tup in points for x in tup]) | ||||
| points = list(points) | ||||
|
Juan C. Espinoza
|
r172 | points.sort() | ||
| if points[0]!=0: | ||||
|
Juan C. Espinoza
|
r79 | points.insert(0, 0) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | return [points[i+1]-points[i] for i in range(len(points)-1)] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | def get_pulses(self, binary=True): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | pulses = [line.pulses_as_points() for line in self.get_lines()] | ||
|
Juan C. Espinoza
|
r107 | points = [tup for tups in pulses for tup in tups] | ||
| points = set([x for tup in points for x in tup]) | ||||
| points = list(points) | ||||
|
Juan C. Espinoza
|
r172 | points.sort() | ||
|
Juan C. Espinoza
|
r111 | line_points = [line.pulses_as_points() for line in self.get_lines()] | ||
|
Juan C. Espinoza
|
r79 | line_points = [[(x, x+y) for x,y in tups] for tups in line_points] | ||
|
Juan C. Espinoza
|
r172 | line_points = [[t for x in tups for t in x] for tups in line_points] | ||
|
Juan C. Espinoza
|
r79 | states = [[1 if x in tups else 0 for tups in line_points] for x in points] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | if binary: | ||
| states.reverse() | ||||
| states = [int(''.join([str(x) for x in flips]), 2) for flips in states] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | return states[:-1] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def add_cmd(self, cmd): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | if cmd in DAT_CMDS: | ||
| return (255, DAT_CMDS[cmd]) | ||||
|
Juan C. Espinoza
|
r172 | |||
| def add_data(self, value): | ||||
|
Juan C. Espinoza
|
r79 | return (254, value-1) | ||
|
Juan C. Espinoza
|
r175 | |||
| def parms_to_binary(self, dat=True): | ||||
|
Juan C. Espinoza
|
r79 | ''' | ||
| Create "dat" stream to be send to CR | ||||
| ''' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | data = bytearray() | ||
|
Juan C. Espinoza
|
r79 | # create header | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('DISABLE')) | ||
| data.extend(self.add_cmd('CONTINUE')) | ||||
| data.extend(self.add_cmd('RESTART')) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | if self.control_sw: | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('SW_ONE')) | ||
|
Juan C. Espinoza
|
r79 | else: | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('SW_ZERO')) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | if self.control_tx: | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('TX_ONE')) | ||
|
Juan C. Espinoza
|
r79 | else: | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('TX_ZERO')) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | # write divider | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('CLOCK_DIVIDER')) | ||
| data.extend(self.add_data(self.clock_divider)) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | # write delays | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('DELAY_START')) | ||
|
Juan C. Espinoza
|
r79 | # first delay is always zero | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(1)) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | delays = self.get_delays() | ||
|
Juan C. Espinoza
|
r172 | |||
| for delay in delays: | ||||
| while delay>252: | ||||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(253)) | ||
|
Juan C. Espinoza
|
r79 | delay -= 253 | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(int(delay))) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | # write flips | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('FLIP_START')) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | states = self.get_pulses(binary=False) | ||
| for flips, delay in zip(states, delays): | ||||
|
Juan C. Espinoza
|
r79 | flips.reverse() | ||
|
Juan C. Espinoza
|
r172 | flip = int(''.join([str(x) for x in flips]), 2) | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(flip+1)) | ||
|
Juan C. Espinoza
|
r79 | while delay>252: | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(1)) | ||
|
Juan C. Espinoza
|
r79 | delay -= 253 | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | # write sampling period | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('SAMPLING_PERIOD')) | ||
|
Juan C. Espinoza
|
r107 | wins = self.get_lines(line_type__name='windows') | ||
|
Juan C. Espinoza
|
r79 | if wins: | ||
| win_params = json.loads(wins[0].params)['params'] | ||||
| if win_params: | ||||
| dh = int(win_params[0]['resolution']*self.km2unit) | ||||
| else: | ||||
| dh = 1 | ||||
| else: | ||||
| dh = 1 | ||||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_data(dh)) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | # write enable | ||
|
Juan C. Espinoza
|
r175 | data.extend(self.add_cmd('ENABLE')) | ||
| if not dat: | ||||
| return data | ||||
| return '\n'.join(['{}'.format(x) for x in data]) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def update_from_file(self, filename): | ||
| ''' | ||||
| Update instance from file | ||||
| ''' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | f = RCFile(filename) | ||
|
Juan C. Espinoza
|
r107 | self.dict_to_parms(f.data) | ||
|
Juan C. Espinoza
|
r111 | self.update_pulses() | ||
|
Juan C. Espinoza
|
r107 | |||
| def update_pulses(self): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | for line in self.get_lines(): | ||
|
Juan C. Espinoza
|
r119 | line.update_pulses() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | def plot_pulses2(self, km=False): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | import matplotlib.pyplot as plt | ||
| from bokeh.resources import CDN | ||||
| from bokeh.embed import components | ||||
|
Juan C. Espinoza
|
r172 | from bokeh.mpl import to_bokeh | ||
| from bokeh.models.tools import WheelZoomTool, ResetTool, PanTool, HoverTool, SaveTool | ||||
| lines = self.get_lines() | ||||
|
Juan C. Espinoza
|
r111 | N = len(lines) | ||
|
Juan C. Espinoza
|
r172 | npoints = self.total_units/self.km2unit if km else self.total_units | ||
|
Juan C. Espinoza
|
r175 | fig = plt.figure(figsize=(12, 2+N*0.5)) | ||
|
Juan C. Espinoza
|
r111 | ax = fig.add_subplot(111) | ||
|
Juan C. Espinoza
|
r172 | labels = ['IPP'] | ||
|
Juan C. Espinoza
|
r111 | for i, line in enumerate(lines): | ||
|
Juan C. Espinoza
|
r172 | labels.append(line.get_name(channel=True)) | ||
|
Juan C. Espinoza
|
r157 | l = ax.plot((0, npoints),(N-i-1, N-i-1)) | ||
|
Juan C. Espinoza
|
r172 | points = [(tup[0], tup[1]-tup[0]) for tup in line.pulses_as_points(km=km) if tup!=(0,0)] | ||
| ax.broken_barh(points, (N-i-1, 0.5), | ||||
|
Juan C. Espinoza
|
r111 | edgecolor=l[0].get_color(), facecolor='none') | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | n = 0 | ||
| f = ((self.ntx+50)/100)*5 if ((self.ntx+50)/100)*10>0 else 2 | ||||
| for x in np.arange(0, npoints, self.ipp if km else self.ipp*self.km2unit): | ||||
| if n%f==0: | ||||
| ax.text(x, N, '%s' % n, size=10) | ||||
| n += 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | labels.reverse() | ||
|
Juan C. Espinoza
|
r172 | ax.set_yticks(range(len(labels))) | ||
|
Juan C. Espinoza
|
r111 | ax.set_yticklabels(labels) | ||
|
Juan C. Espinoza
|
r116 | ax.set_xlabel = 'Units' | ||
|
Juan C. Espinoza
|
r175 | plot = to_bokeh(fig, use_pandas=False) | ||
|
Juan C. Espinoza
|
r172 | plot.tools = [PanTool(dimensions=['width']), WheelZoomTool(dimensions=['width']), ResetTool(), SaveTool()] | ||
| plot.toolbar_location="above" | ||||
|
Juan C. Espinoza
|
r111 | return components(plot, CDN) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | def plot_pulses(self, km=False): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | from bokeh.plotting import figure | ||
| from bokeh.resources import CDN | ||||
| from bokeh.embed import components | ||||
| from bokeh.models import FixedTicker, PrintfTickFormatter | ||||
| from bokeh.models.tools import WheelZoomTool, ResetTool, PanTool, HoverTool, SaveTool | ||||
| from bokeh.models.sources import ColumnDataSource | ||||
|
Juan C. Espinoza
|
r85 | |||
|
Juan C. Espinoza
|
r175 | lines = self.get_lines().reverse() | ||
| N = len(lines) | ||||
| npoints = self.total_units/self.km2unit if km else self.total_units | ||||
| ipp = self.ipp if km else self.ipp*self.km2unit | ||||
| hover = HoverTool(tooltips=[("Line", "@name"), | ||||
| ("IPP", "@ipp"), | ||||
| ("X", "@left")]) | ||||
| tools = [PanTool(dimensions=['width']), | ||||
| WheelZoomTool(dimensions=['width']), | ||||
| hover, SaveTool()] | ||||
| plot = figure(width=1000, | ||||
| height=40+N*50, | ||||
| y_range = (0, N), | ||||
| tools=tools, | ||||
| toolbar_location='above', | ||||
| toolbar_sticky=False,) | ||||
| plot.xaxis.axis_label = 'Km' if km else 'Units' | ||||
| plot.xaxis[0].formatter = PrintfTickFormatter(format='%d') | ||||
| plot.yaxis.axis_label = 'Pulses' | ||||
| plot.yaxis[0].ticker=FixedTicker(ticks=list(range(N))) | ||||
| plot.yaxis[0].formatter = PrintfTickFormatter(format='Line %d') | ||||
| for i, line in enumerate(lines): | ||||
| points = [tup for tup in line.pulses_as_points(km=km) if tup!=(0,0)] | ||||
| source = ColumnDataSource(data = dict( | ||||
| bottom = [i for tup in points], | ||||
| top = [i+0.5 for tup in points], | ||||
| left = [tup[0] for tup in points], | ||||
| right = [tup[1] for tup in points], | ||||
| ipp = [int(tup[0]/ipp) for tup in points], | ||||
| name = [line.get_name() for tup in points] | ||||
| )) | ||||
| plot.quad( | ||||
| bottom = 'bottom', | ||||
| top = 'top', | ||||
| left = 'left', | ||||
| right = 'right', | ||||
| source = source, | ||||
| fill_alpha = 0, | ||||
| #line_color = 'blue', | ||||
| ) | ||||
| plot.line([0, npoints], [i, i])#, color='blue') | ||||
| return components(plot, CDN) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | def status_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | try: | ||
| req = requests.get(self.device.url) | ||||
| payload = req.json() | ||||
| if payload['status']=='ok': | ||||
| self.device.status = 3 | ||||
| else: | ||||
| self.device.status = 1 | ||||
| except: | ||||
| self.device.status = 0 | ||||
| self.device.save() | ||||
| return self.device.status | ||||
| def reset_device(self): | ||||
| payload = bytearray() | ||||
| payload.extend(self.add_cmd('RESTART')) | ||||
| data = b64encode(payload) | ||||
| req = requests.put(self.device.url, data) | ||||
| if data==req.text.encode('utf8'): | ||||
| return 1 | ||||
| else: | ||||
|
Juan C. Espinoza
|
r107 | return 0 | ||
|
Juan C. Espinoza
|
r175 | |||
| def stop_device(self): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | payload = bytearray() | ||
| payload.extend(self.add_cmd('DISABLE')) | ||||
| data = b64encode(payload) | ||||
| req = requests.put(self.device.url, data) | ||||
| if data==req.text.encode('utf8'): | ||||
| return 1 | ||||
| else: | ||||
| return 0 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | def start_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r175 | payload = bytearray() | ||
| payload.extend(self.add_cmd('ENABLE')) | ||||
| data = b64encode(payload) | ||||
| req = requests.put(self.device.url, data) | ||||
| if data==req.text.encode('utf8'): | ||||
| return 1 | ||||
| else: | ||||
|
Juan C. Espinoza
|
r107 | return 0 | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | def write_device(self): | ||
|
Juan C. Espinoza
|
r175 | |||
| data = b64encode(self.parms_to_binary(dat=False)) | ||||
| req = requests.put(self.device.url, data) | ||||
| print(req.text) | ||||
| if data==req.text.encode('utf8'): | ||||
| return 1 | ||||
| else: | ||||
|
Juan C. Espinoza
|
r107 | return 0 | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | |||
|
Juan C. Espinoza
|
r23 | class RCLineCode(models.Model): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r25 | name = models.CharField(max_length=40) | ||
|
Juan C. Espinoza
|
r23 | bits_per_code = models.PositiveIntegerField(default=0) | ||
| number_of_codes = models.PositiveIntegerField(default=0) | ||||
| codes = models.TextField(blank=True, null=True) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | class Meta: | ||
| db_table = 'rc_line_codes' | ||||
|
Juan C. Espinoza
|
r25 | ordering = ('name',) | ||
|
Juan C. Espinoza
|
r172 | |||
| def __str__(self): | ||||
| return u'%s' % self.name | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r107 | |||
|
Juan C. Espinoza
|
r23 | class RCLineType(models.Model): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | name = models.CharField(choices=LINE_TYPES, max_length=40) | ||
| description = models.TextField(blank=True, null=True) | ||||
| params = models.TextField(default='[]') | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | class Meta: | ||
| db_table = 'rc_line_types' | ||||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
|
Juan C. Espinoza
|
r23 | return u'%s - %s' % (self.name.upper(), self.get_name_display()) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | class RCLine(models.Model): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | rc_configuration = models.ForeignKey(RCConfiguration, on_delete=models.CASCADE) | ||
|
Juan C. Espinoza
|
r23 | line_type = models.ForeignKey(RCLineType) | ||
| channel = models.PositiveIntegerField(default=0) | ||||
| position = models.PositiveIntegerField(default=0) | ||||
| params = models.TextField(default='{}') | ||||
|
Juan C. Espinoza
|
r45 | pulses = models.TextField(default='') | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | class Meta: | ||
| db_table = 'rc_lines' | ||||
|
Juan C. Espinoza
|
r45 | ordering = ['channel'] | ||
|
Juan C. Espinoza
|
r172 | |||
| def __str__(self): | ||||
|
Juan C. Espinoza
|
r79 | if self.rc_configuration: | ||
| return u'%s - %s' % (self.rc_configuration, self.get_name()) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | def clone(self, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | self.pk = None | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | for attr, value in kwargs.items(): | ||
| setattr(self, attr, value) | ||||
|
Juan C. Espinoza
|
r172 | |||
| self.save() | ||||
|
Juan C. Espinoza
|
r85 | |||
| return self | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def get_name(self, channel=False): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r23 | chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' | ||
|
Juan C. Espinoza
|
r172 | s = '' | ||
|
Juan C. Espinoza
|
r45 | if self.line_type.name in ('tx',): | ||
|
Juan C. Espinoza
|
r111 | s = chars[self.position] | ||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name in ('codes', 'windows', 'tr'): | ||
|
Juan C. Espinoza
|
r111 | if 'TX_ref' in json.loads(self.params): | ||
| pk = json.loads(self.params)['TX_ref'] | ||||
| if pk in (0, '0'): | ||||
|
Juan C. Espinoza
|
r172 | s = ','.join(chars[l.position] for l in self.rc_configuration.get_lines(line_type__name='tx')) | ||
|
Juan C. Espinoza
|
r111 | else: | ||
| ref = RCLine.objects.get(pk=pk) | ||||
| s = chars[ref.position] | ||||
|
Juan C. Espinoza
|
r113 | s = '({})'.format(s) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | s = '{}{}'.format(self.line_type.name.upper(), s) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | if channel: | ||
| return '{} {}'.format(s, self.channel) | ||||
|
Juan C. Espinoza
|
r25 | else: | ||
|
Juan C. Espinoza
|
r157 | return s | ||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r107 | def get_lines(self, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
| return RCLine.objects.filter(rc_configuration=self.rc_configuration, **kwargs) | ||||
|
Juan C. Espinoza
|
r45 | def pulses_as_array(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r112 | y = np.zeros(self.rc_configuration.total_units) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | for tup in ast.literal_eval(self.pulses): | ||
| y[tup[0]:tup[1]] = 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | return y.astype(np.int8) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def pulses_as_points(self, km=False): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | if km: | ||
| unit2km = 1/self.rc_configuration.km2unit | ||||
| return [(tup[0]*unit2km, tup[1]*unit2km) for tup in ast.literal_eval(self.pulses)] | ||||
| else: | ||||
| return ast.literal_eval(self.pulses) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def get_win_ref(self, params, tx_id, km2unit): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | ref = self.rc_configuration.sampling_reference | ||
|
Juan C. Espinoza
|
r107 | codes = [line for line in self.get_lines(line_type__name='codes') if int(json.loads(line.params)['TX_ref'])==int(tx_id)] | ||
|
Juan C. Espinoza
|
r172 | |||
| if codes: | ||||
|
Juan C. Espinoza
|
r107 | tx_width = float(json.loads(RCLine.objects.get(pk=tx_id).params)['pulse_width'])*km2unit/len(json.loads(codes[0].params)['codes'][0]) | ||
|
Juan C. Espinoza
|
r79 | else: | ||
| tx_width = float(json.loads(RCLine.objects.get(pk=tx_id).params)['pulse_width'])*km2unit | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | if ref=='first_baud': | ||
| return int(1 + (tx_width + 1)/2 + params['first_height']*km2unit - params['resolution']*km2unit) | ||||
| elif ref=='sub_baud': | ||||
| return int(1 + params['first_height']*km2unit - params['resolution']*km2unit/2) | ||||
| else: | ||||
| return 0 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | def update_pulses(self): | ||
|
Juan C. Espinoza
|
r79 | ''' | ||
|
Juan C. Espinoza
|
r172 | Update pulses field | ||
|
Juan C. Espinoza
|
r79 | ''' | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | km2unit = self.rc_configuration.km2unit | ||
| us2unit = self.rc_configuration.us2unit | ||||
|
Juan C. Espinoza
|
r45 | ipp = self.rc_configuration.ipp | ||
|
Juan C. Espinoza
|
r175 | ntx = int(self.rc_configuration.ntx) | ||
|
Juan C. Espinoza
|
r79 | ipp_u = int(ipp*km2unit) | ||
|
Juan C. Espinoza
|
r116 | total = ipp_u*ntx if self.rc_configuration.total_units==0 else self.rc_configuration.total_units | ||
|
Juan C. Espinoza
|
r111 | y = [] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | if self.line_type.name=='tr': | ||
|
Juan C. Espinoza
|
r111 | tr_params = json.loads(self.params) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | if tr_params['TX_ref'] in ('0', 0): | ||
| txs = self.get_lines(line_type__name='tx') | ||||
|
Juan C. Espinoza
|
r45 | else: | ||
|
Juan C. Espinoza
|
r115 | txs = RCLine.objects.filter(pk=tr_params['TX_ref']) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | for tx in txs: | ||
| params = json.loads(tx.params) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | if float(params['pulse_width'])==0: | ||
| continue | ||||
|
Juan C. Espinoza
|
r172 | delays = [float(d)*km2unit for d in params['delays'].split(',') if d] | ||
|
Juan C. Espinoza
|
r111 | width = float(params['pulse_width'])*km2unit+int(self.rc_configuration.time_before*us2unit) | ||
| before = 0 | ||||
| after = int(self.rc_configuration.time_after*us2unit) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | y_tx = self.points(ntx, ipp_u, width, | ||
| delay=delays, | ||||
| before=before, | ||||
| after=after, | ||||
| sync=self.rc_configuration.sync) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | ranges = params['range'].split(',') | ||
|
Juan C. Espinoza
|
r172 | |||
| if len(ranges)>0 and ranges[0]!='0': | ||||
|
Juan C. Espinoza
|
r111 | y_tx = self.mask_ranges(y_tx, ranges) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | tr_ranges = tr_params['range'].split(',') | ||
|
Juan C. Espinoza
|
r172 | |||
| if len(tr_ranges)>0 and tr_ranges[0]!='0': | ||||
|
Juan C. Espinoza
|
r111 | y_tx = self.mask_ranges(y_tx, tr_ranges) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | y.extend(y_tx) | ||
|
Juan C. Espinoza
|
r119 | |||
|
Juan C. Espinoza
|
r172 | self.pulses = str(y) | ||
|
Juan C. Espinoza
|
r116 | y = self.array_to_points(self.pulses_as_array()) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='tx': | ||
|
Juan C. Espinoza
|
r172 | params = json.loads(self.params) | ||
| delays = [float(d)*km2unit for d in params['delays'].split(',') if d] | ||||
|
Juan C. Espinoza
|
r111 | width = float(params['pulse_width'])*km2unit | ||
|
Juan C. Espinoza
|
r172 | |||
| if width>0: | ||||
|
Juan C. Espinoza
|
r111 | before = int(self.rc_configuration.time_before*us2unit) | ||
| after = 0 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | y = self.points(ntx, ipp_u, width, | ||
| delay=delays, | ||||
| before=before, | ||||
| after=after, | ||||
|
Juan C. Espinoza
|
r172 | sync=self.rc_configuration.sync) | ||
|
Juan C. Espinoza
|
r111 | ranges = params['range'].split(',') | ||
|
Juan C. Espinoza
|
r172 | |||
| if len(ranges)>0 and ranges[0]!='0': | ||||
|
Juan C. Espinoza
|
r111 | y = self.mask_ranges(y, ranges) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='flip': | ||
|
Juan C. Espinoza
|
r111 | n = float(json.loads(self.params)['number_of_flips']) | ||
| width = n*ipp*km2unit | ||||
| y = self.points(int((ntx+1)/(2*n)), ipp_u*n*2, width) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='codes': | ||
| params = json.loads(self.params) | ||||
| tx = RCLine.objects.get(pk=params['TX_ref']) | ||||
| tx_params = json.loads(tx.params) | ||||
|
Juan C. Espinoza
|
r172 | delays = [float(d)*km2unit for d in tx_params['delays'].split(',') if d] | ||
| f = int(float(tx_params['pulse_width'])*km2unit/len(params['codes'][0])) | ||||
| codes = [(np.fromstring(''.join([s*f for s in code]), dtype=np.uint8)-48).astype(np.int8) for code in params['codes']] | ||||
|
Juan C. Espinoza
|
r111 | codes = [self.array_to_points(code) for code in codes] | ||
| n = len(codes) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | for i, tup in enumerate(tx.pulses_as_points()): | ||
| code = codes[i%n] | ||||
| y.extend([(c[0]+tup[0], c[1]+tup[0]) for c in code]) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | ranges = tx_params['range'].split(',') | ||
|
Juan C. Espinoza
|
r172 | if len(ranges)>0 and ranges[0]!='0': | ||
|
Juan C. Espinoza
|
r111 | y = self.mask_ranges(y, ranges) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='sync': | ||
| params = json.loads(self.params) | ||||
|
Juan C. Espinoza
|
r111 | n = ipp_u*ntx | ||
|
Juan C. Espinoza
|
r45 | if params['invert'] in ('1', 1): | ||
|
Juan C. Espinoza
|
r111 | y = [(n-1, n)] | ||
|
Juan C. Espinoza
|
r45 | else: | ||
|
Juan C. Espinoza
|
r111 | y = [(0, 1)] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='prog_pulses': | ||
| params = json.loads(self.params) | ||||
| if int(params['periodic'])==0: | ||||
| nntx = 1 | ||||
|
Juan C. Espinoza
|
r111 | nipp = ipp_u*ntx | ||
| else: | ||||
| nntx = ntx | ||||
| nipp = ipp_u | ||||
|
Juan C. Espinoza
|
r172 | |||
| if 'params' in params and len(params['params'])>0: | ||||
|
Juan C. Espinoza
|
r111 | for p in params['params']: | ||
|
Juan C. Espinoza
|
r172 | y_pp = self.points(nntx, nipp, | ||
| p['end']-p['begin'], | ||||
| before=p['begin']) | ||||
| y.extend(y_pp) | ||||
|
Juan C. Espinoza
|
r45 | elif self.line_type.name=='windows': | ||
|
Juan C. Espinoza
|
r172 | params = json.loads(self.params) | ||
|
Juan C. Espinoza
|
r45 | if 'params' in params and len(params['params'])>0: | ||
|
Juan C. Espinoza
|
r111 | tr_params = json.loads(self.get_lines(line_type__name='tr')[0].params) | ||
| tr_ranges = tr_params['range'].split(',') | ||||
| for p in params['params']: | ||||
|
Juan C. Espinoza
|
r172 | y_win = self.points(ntx, ipp_u, | ||
| p['resolution']*p['number_of_samples']*km2unit, | ||||
|
Juan C. Espinoza
|
r111 | before=int(self.rc_configuration.time_before*us2unit)+self.get_win_ref(p, params['TX_ref'], km2unit), | ||
| sync=self.rc_configuration.sync) | ||||
|
Juan C. Espinoza
|
r172 | |||
| if len(tr_ranges)>0 and tr_ranges[0]!='0': | ||||
|
Juan C. Espinoza
|
r111 | y_win = self.mask_ranges(y_win, tr_ranges) | ||
|
Juan C. Espinoza
|
r172 | |||
| y.extend(y_win) | ||||
|
Juan C. Espinoza
|
r107 | elif self.line_type.name=='mix': | ||
| values = self.rc_configuration.parameters.split('-') | ||||
|
Juan C. Espinoza
|
r116 | confs = [RCConfiguration.objects.get(pk=value.split('|')[0]) for value in values] | ||
|
Juan C. Espinoza
|
r107 | modes = [value.split('|')[1] for value in values] | ||
|
Juan C. Espinoza
|
r112 | ops = [value.split('|')[2] for value in values] | ||
| delays = [value.split('|')[3] for value in values] | ||||
| masks = [value.split('|')[4] for value in values] | ||||
| mask = list('{:8b}'.format(int(masks[0]))) | ||||
| mask.reverse() | ||||
| if mask[self.channel] in ('0', '', ' '): | ||||
|
Juan C. Espinoza
|
r116 | y = np.zeros(confs[0].total_units, dtype=np.int8) | ||
|
Juan C. Espinoza
|
r112 | else: | ||
|
Juan C. Espinoza
|
r116 | y = confs[0].get_lines(channel=self.channel)[0].pulses_as_array() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | for i in range(1, len(values)): | ||
| mask = list('{:8b}'.format(int(masks[i]))) | ||||
|
Juan C. Espinoza
|
r172 | mask.reverse() | ||
|
Juan C. Espinoza
|
r107 | if mask[self.channel] in ('0', '', ' '): | ||
| continue | ||||
|
Juan C. Espinoza
|
r116 | Y = confs[i].get_lines(channel=self.channel)[0].pulses_as_array() | ||
|
Juan C. Espinoza
|
r107 | delay = float(delays[i])*km2unit | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r116 | if modes[i]=='P': | ||
| if delay>0: | ||||
|
Juan C. Espinoza
|
r172 | if delay<self.rc_configuration.ipp*km2unit and len(Y)==len(y): | ||
|
Juan C. Espinoza
|
r116 | y_temp = np.empty_like(Y) | ||
| y_temp[:delay] = 0 | ||||
| y_temp[delay:] = Y[:-delay] | ||||
| elif delay+len(Y)>len(y): | ||||
| y_new = np.zeros(delay+len(Y), dtype=np.int8) | ||||
| y_new[:len(y)] = y | ||||
| y = y_new | ||||
| y_temp = np.zeros(delay+len(Y), dtype=np.int8) | ||||
| y_temp[-len(Y):] = Y | ||||
| elif delay+len(Y)==len(y): | ||||
| y_temp = np.zeros(delay+len(Y)) | ||||
| y_temp[-len(Y):] = Y | ||||
| elif delay+len(Y)<len(y): | ||||
| y_temp = np.zeros(len(y), dtype=np.int8) | ||||
| y_temp[delay:delay+len(Y)] = Y | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r116 | if ops[i]=='OR': | ||
| y = y | y_temp | ||||
| elif ops[i]=='XOR': | ||||
| y = y ^ y_temp | ||||
| elif ops[i]=='AND': | ||||
| y = y & y_temp | ||||
| elif ops[i]=='NAND': | ||||
| y = y & ~y_temp | ||||
|
Juan C. Espinoza
|
r172 | else: | ||
|
Juan C. Espinoza
|
r116 | y = np.concatenate([y, Y]) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r112 | total = len(y) | ||
|
Juan C. Espinoza
|
r111 | y = self.array_to_points(y) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r45 | else: | ||
|
Juan C. Espinoza
|
r111 | y = [] | ||
|
Juan C. Espinoza
|
r172 | |||
| if self.rc_configuration.total_units != total: | ||||
|
Juan C. Espinoza
|
r112 | self.rc_configuration.total_units = total | ||
|
Juan C. Espinoza
|
r119 | self.rc_configuration.save() | ||
|
Juan C. Espinoza
|
r172 | |||
| self.pulses = str(y) | ||||
|
Juan C. Espinoza
|
r111 | self.save() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | @staticmethod | ||
| def array_to_points(X): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | d = X[1:]-X[:-1] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | up = np.where(d==1)[0] | ||
| if X[0]==1: | ||||
| up = np.concatenate((np.array([-1]), up)) | ||||
| up += 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | dw = np.where(d==-1)[0] | ||
| if X[-1]==1: | ||||
| dw = np.concatenate((dw, np.array([len(X)-1]))) | ||||
| dw += 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | return [(tup[0], tup[1]) for tup in zip(up, dw)] | ||
| @staticmethod | ||||
| def mask_ranges(Y, ranges): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | y = [(0, 0) for __ in Y] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | for index in ranges: | ||
| if '-' in index: | ||||
| args = [int(a) for a in index.split('-')] | ||||
|
Juan C. Espinoza
|
r172 | y[args[0]-1:args[1]] = Y[args[0]-1:args[1]] | ||
|
Juan C. Espinoza
|
r111 | else: | ||
|
Juan C. Espinoza
|
r172 | y[int(index-1)] = Y[int(index-1)] | ||
|
Juan C. Espinoza
|
r111 | return y | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | @staticmethod | ||
| def points(ntx, ipp, width, delay=[0], before=0, after=0, sync=0): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | delays = len(delay) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r111 | Y = [(ipp*x+before+delay[x%delays], ipp*x+width+before+delay[x%delays]+after) for x in range(ntx)] | ||
|
Juan C. Espinoza
|
r172 | |||
| return Y | ||||