models.py
273 lines
| 12.2 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r6 | from django.db import models | ||
| from apps.main.models import Configuration | ||||
Fiorella Quino
|
r118 | from django.core.validators import MinValueValidator, MaxValueValidator | ||
|
Juan C. Espinoza
|
r157 | from django.core.urlresolvers import reverse | ||
| from devices.jars import api | ||||
|
Fiorella Quino
|
r123 | |||
| from apps.rc.models import RCConfiguration | ||||
|
Juan C. Espinoza
|
r157 | |||
| import json | ||||
|
Juan C. Espinoza
|
r6 | # Create your models here. | ||
|
Fiorella Quino
|
r123 | EXPERIMENT_TYPE = ( | ||
| (0, 'RAW_DATA'), | ||||
| (1, 'PDATA'), | ||||
| ) | ||||
| DATA_TYPE = ( | ||||
| (0, 'SHORT'), | ||||
| (1, 'FLOAT'), | ||||
| ) | ||||
| class JARSfilter(models.Model): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r139 | JARS_NBITS = 32 | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r139 | name = models.CharField(max_length=60, unique=True, default='') | ||
|
Fiorella Quino
|
r144 | clock = models.FloatField(verbose_name='Clock In (MHz)',validators=[MinValueValidator(5), MaxValueValidator(75)], null=True, default=60) | ||
|
Fiorella Quino
|
r139 | mult = models.PositiveIntegerField(verbose_name='Multiplier',validators=[MinValueValidator(1), MaxValueValidator(20)], default=5) | ||
| fch = models.DecimalField(verbose_name='Frequency (MHz)', validators=[MinValueValidator(0), MaxValueValidator(150)], max_digits=19, decimal_places=16, null=True, default=49.9200) | ||||
| fch_decimal = models.BigIntegerField(verbose_name='Frequency (Decimal)',validators=[MinValueValidator(0), MaxValueValidator(2**JARS_NBITS-1)], null=True, default=721554505) | ||||
| filter_fir = models.PositiveIntegerField(verbose_name='FIR Filter',validators=[MinValueValidator(1), MaxValueValidator(20)], default = 6) | ||||
| filter_2 = models.PositiveIntegerField(verbose_name='Filter 2',validators=[MinValueValidator(1), MaxValueValidator(20)], default = 10) | ||||
| filter_5 = models.PositiveIntegerField(verbose_name='Filter 5',validators=[MinValueValidator(1), MaxValueValidator(20)], default = 1) | ||||
| speed = models.PositiveIntegerField(verbose_name='Speed',validators=[MinValueValidator(0), MaxValueValidator(100000)], default = 0) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | class Meta: | ||
| db_table = 'jars_filters' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | def __unicode__(self): | ||
| return u'%s' % (self.name) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | def parms_to_dict(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | parameters = {} | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r139 | parameters['name'] = self.name | ||
| parameters['clock'] = float(self.clock) | ||||
| parameters['mult'] = int(self.mult) | ||||
| parameters['fch'] = float(self.fch) | ||||
| parameters['fch_decimal'] = int(self.fch) | ||||
| parameters['filter_fir'] = int(self.filter_fir) | ||||
| parameters['filter_2'] = int(self.filter_2) | ||||
| parameters['filter_5'] = int(self.filter_5) | ||||
| parameters['speed'] = int(self.speed) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | return parameters | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | def dict_to_parms(self, parameters): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r139 | self.name = parameters['name'] | ||
| self.clock = parameters['clock'] | ||||
| self.mult = parameters['mult'] | ||||
| self.fch = parameters['fch'] | ||||
| self.fch_decimal = parameters['fch_decimal'] | ||||
| self.filter_fir = parameters['filter_fir'] | ||||
| self.filter_2 = parameters['filter_2'] | ||||
| self.filter_5 = parameters['filter_5'] | ||||
| self.speed = parameters['speed'] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r118 | |||
|
Juan C. Espinoza
|
r6 | class JARSConfiguration(Configuration): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r118 | ADC_RESOLUTION = 8 | ||
| PCI_DIO_BUSWIDTH = 32 | ||||
|
Fiorella Quino
|
r123 | HEADER_VERSION = 1103 | ||
| BEGIN_ON_START = True | ||||
| REFRESH_RATE = 1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r139 | #rc = models.ForeignKey(RCConfiguration, on_delete=models.CASCADE, null=True) | ||
|
Fiorella Quino
|
r123 | exp_type = models.PositiveIntegerField(verbose_name='Experiment Type', choices=EXPERIMENT_TYPE, default=0) | ||
|
Fiorella Quino
|
r125 | cards_number = models.PositiveIntegerField(verbose_name='Number of Cards', validators=[MinValueValidator(1), MaxValueValidator(4)], default = 1) | ||
| channels_number = models.PositiveIntegerField(verbose_name='Number of Channels', validators=[MinValueValidator(1), MaxValueValidator(8)], default = 5) | ||||
|
Fiorella Quino
|
r123 | channels = models.CharField(verbose_name='Channels', max_length=15, default = '1,2,3,4,5') | ||
|
Juan C. Espinoza
|
r157 | rd_directory = models.CharField(verbose_name='Raw Data Directory', max_length=200, default='', blank=True, null=True) | ||
| pd_directory = models.CharField(verbose_name='Process Data Directory', max_length=200, default='', blank=True, null=True) | ||||
| #raw_data_blocks = models.PositiveIntegerField(verbose_name='Raw Data Blocks', validators=[MaxValueValidator(5000)], default=120) | ||||
|
Fiorella Quino
|
r123 | data_type = models.PositiveIntegerField(verbose_name='Data Type', choices=DATA_TYPE, default=0) | ||
|
Fiorella Quino
|
r125 | acq_profiles = models.PositiveIntegerField(verbose_name='Acquired Profiles', validators=[MaxValueValidator(5000)], default=400) | ||
| profiles_block = models.PositiveIntegerField(verbose_name='Profiles Per Block', validators=[MaxValueValidator(5000)], default=400) | ||||
|
Juan C. Espinoza
|
r157 | ftp_interval = models.PositiveIntegerField(verbose_name='FTP Interval', default=60) | ||
|
Fiorella Quino
|
r125 | fftpoints = models.PositiveIntegerField(verbose_name='FFT Points',default=16) | ||
|
Juan C. Espinoza
|
r157 | cohe_integr_str = models.PositiveIntegerField(verbose_name='Coh. Int. Stride',validators=[MinValueValidator(1)], default=30) | ||
|
Fiorella Quino
|
r141 | cohe_integr = models.PositiveIntegerField(verbose_name='Coherent Integrations',validators=[MinValueValidator(1)], default=30) | ||
|
Fiorella Quino
|
r130 | incohe_integr = models.PositiveIntegerField(verbose_name='Incoherent Integrations',validators=[MinValueValidator(1)], default=30) | ||
|
Fiorella Quino
|
r135 | filter = models.ForeignKey(JARSfilter, on_delete=models.CASCADE, null=True) | ||
|
Fiorella Quino
|
r130 | spectral_number = models.PositiveIntegerField(verbose_name='# Spectral Combinations',validators=[MinValueValidator(1)], default=1) | ||
| spectral = models.CharField(verbose_name='Combinations', max_length=5000, default = '[0, 0],') | ||||
|
Fiorella Quino
|
r118 | create_directory = models.BooleanField(verbose_name='Create Directory Per Day', default=True) | ||
|
Fiorella Quino
|
r123 | include_expname = models.BooleanField(verbose_name='Experiment Name in Directory', default=True) | ||
|
Juan C. Espinoza
|
r157 | #acq_link = models.BooleanField(verbose_name='Acquisition Link', default=True) | ||
| #view_raw_data = models.BooleanField(verbose_name='View Raw Data', default=True) | ||||
|
Fiorella Quino
|
r125 | save_ch_dc = models.BooleanField(verbose_name='Save Channels DC', default=True) | ||
|
Juan C. Espinoza
|
r157 | save_data = models.BooleanField(verbose_name='Save Data', default=True) | ||
|
Fiorella Quino
|
r150 | filter_parms = models.CharField(max_length=10000, default='{}') | ||
|
Juan C. Espinoza
|
r6 | |||
| class Meta: | ||||
| db_table = 'jars_configurations' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | def parms_to_dict(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | parameters = {} | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | parameters['device_id'] = self.device.id | ||
|
Fiorella Quino
|
r123 | parameters['name'] = self.name | ||
|
Fiorella Quino
|
r150 | #parameters['rc'] = self.rc.name | ||
|
Fiorella Quino
|
r123 | parameters['exp_type'] = self.exp_type | ||
|
Fiorella Quino
|
r150 | parameters['exptype'] = EXPERIMENT_TYPE[self.exp_type][1] | ||
|
Fiorella Quino
|
r123 | parameters['cards_number'] = self.cards_number | ||
| parameters['channels_number'] = self.channels_number | ||||
| parameters['channels'] = self.channels | ||||
| parameters['rd_directory'] = self.rd_directory | ||||
|
Juan C. Espinoza
|
r157 | #parameters['raw_data_blocks'] = self.raw_data_blocks | ||
|
Fiorella Quino
|
r123 | parameters['data_type'] = self.data_type | ||
|
Juan C. Espinoza
|
r157 | parameters['cohe_integr_str'] = self.cohe_integr_str | ||
|
Fiorella Quino
|
r123 | parameters['acq_profiles'] = self.acq_profiles | ||
| parameters['profiles_block'] = self.profiles_block | ||||
|
Juan C. Espinoza
|
r157 | parameters['ftp_interval'] = self.ftp_interval | ||
|
Fiorella Quino
|
r150 | parameters['fftpoints'] = self.fftpoints | ||
| parameters['cohe_integr'] = self.cohe_integr | ||||
| #parameters['incohe_integr'] = self.incohe_integr | ||||
|
Fiorella Quino
|
r123 | parameters['filter'] = self.filter.name | ||
|
Juan C. Espinoza
|
r157 | parameters['filter_parms'] = self.filter_parms | ||
|
Fiorella Quino
|
r150 | #parameters['spectral_number'] = self.spectral_number | ||
| #parameters['spectral'] = self.spectral | ||||
|
Fiorella Quino
|
r123 | parameters['create_directory'] = bool(self.create_directory) | ||
| parameters['include_expname'] = bool(self.include_expname) | ||||
|
Juan C. Espinoza
|
r157 | #parameters['acq_link'] = bool(self.acq_link) | ||
| #parameters['view_raw_data'] = bool(self.view_raw_data) | ||||
|
Fiorella Quino
|
r150 | parameters['save_ch_dc'] = bool(self.save_ch_dc) | ||
|
Juan C. Espinoza
|
r157 | parameters['save_data'] = bool(self.save_data) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r150 | if parameters['exptype'] == 'PDATA': | ||
| parameters['incohe_integr'] = self.incohe_integr | ||||
| parameters['spectral_number'] = self.spectral_number | ||||
| parameters['spectral'] = self.spectral | ||||
|
Juan C. Espinoza
|
r157 | parameters['pd_directory'] = self.pd_directory | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | return parameters | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r136 | def add_parms_to_filter(self): | ||
| self.filter_parms = self.filter.parms_to_dict() | ||||
| self.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | def dict_to_parms(self, parameters): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | self.name = parameters['name'] | ||
| self.device.id = int(parameters['device_id']) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r150 | self.exp_type = int(parameters['exp_type']) | ||
|
Juan C. Espinoza
|
r172 | if parameters['exptype'] == 'PDATA': | ||
|
Fiorella Quino
|
r150 | self.incohe_integr = parameters['incohe_integr'] | ||
| self.spectral_number = parameters['spectral_number'] | ||||
|
Juan C. Espinoza
|
r157 | self.spectral = parameters['spectral'] | ||
| self.pd_directory = parameters['pd_directory'] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r150 | self.cards_number = int(parameters['cards_number']) | ||
| self.channels_number = int(parameters['channels_number']) | ||||
| self.channels = parameters['channels'] | ||||
| self.rd_directory = parameters['rd_directory'] | ||||
|
Juan C. Espinoza
|
r157 | #self.raw_data_blocks = parameters['raw_data_blocks'] | ||
|
Fiorella Quino
|
r150 | self.data_type = parameters['data_type'] | ||
|
Juan C. Espinoza
|
r157 | self.cohe_integr_str = parameters['cohe_integr_str'] | ||
|
Fiorella Quino
|
r150 | self.acq_profiles = parameters['acq_profiles'] | ||
| self.profiles_block = parameters['profiles_block'] | ||||
|
Juan C. Espinoza
|
r157 | self.ftp_interval = parameters['ftp_interval'] | ||
|
Fiorella Quino
|
r150 | self.fftpoints = parameters['fftpoints'] | ||
| self.cohe_integr = parameters['cohe_integr'] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | filter_name = parameters['filter'] | ||
| self.filter = JARSfilter.objects.get(name=filter_name) | ||||
|
Fiorella Quino
|
r150 | self.add_parms_to_filter() | ||
|
Juan C. Espinoza
|
r157 | self.filter_parms = parameters['filter_parms'] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r150 | self.create_directory = bool(parameters['create_directory']) | ||
| self.include_expname = bool(parameters['include_expname']) | ||||
|
Juan C. Espinoza
|
r157 | #self.acq_link = bool(parameters['acq_link']) | ||
| #self.view_raw_data = bool(parameters['view_raw_data']) | ||||
|
Fiorella Quino
|
r150 | self.save_ch_dc = bool(parameters['save_ch_dc']) | ||
|
Juan C. Espinoza
|
r157 | self.save_data = bool(parameters['save_data']) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | def status_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | answer = api.status(self.device.ip_address,self.device.port_address) | ||
| self.device.status = int(answer[0]) | ||||
| self.message = answer[2:] | ||||
| self.device.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | return self.device.status | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def stop_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | answer = api.stop(self.device.ip_address,self.device.port_address) | ||
| self.device.status = int(answer[0]) | ||||
| self.message = answer[2:] | ||||
| self.device.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | return self.device.status | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | def read_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | answer = api.read(self.device.ip_address,self.device.port_address) | ||
| self.device.status = int(answer[0]) | ||||
| try: | ||||
| data = json.loads(answer[2:]) | ||||
| parms = data['configurations']['jars'] | ||||
| except: | ||||
| self.device.status = 0 | ||||
| self.device.save() | ||||
| self.message = 'Could not read JARS configuration.' | ||||
| return '' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | #self.dict_to_parms(parms) | ||
| self.message = 'Current JARS configuration was read successfully.' | ||||
| self.device.save() | ||||
| return parms | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r123 | def write_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | data = self.experiment.parms_to_dict() | ||
| data = json.loads(data) | ||||
| data['configurations']['dds'] ='' | ||||
| data['configurations']['cgs'] ='' | ||||
| data['configurations']['rc']['pulses']='' | ||||
| data['configurations']['rc']['delays']='' | ||||
| json_data = json.dumps(data) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | answer = api.configure(self.device.ip_address,self.device.port_address,json_data) | ||
| #print answer | ||||
| self.device.status = int(answer[0]) | ||||
| self.message = answer[2:] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | self.device.save() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | return self.device.status | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r158 | def start_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r158 | self.write_device() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def echo(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | answer = api.echo(self.device.ip_address,self.device.port_address,'(=') | ||
| #print answer | ||||
| self.device.status = int(answer[0]) | ||||
| self.message = answer[2:] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | self.device.save() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | return #self.device.status | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def update_from_file(self, parameters): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | self.dict_to_parms(parameters) | ||
| self.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def get_absolute_url_import(self): | ||
| return reverse('url_import_jars_conf', args=[str(self.id)]) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r157 | def get_absolute_url_read(self): | ||
|
Juan C. Espinoza
|
r172 | return reverse('url_read_jars_conf', args=[str(self.id)]) | ||