models.py
272 lines
| 12.5 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): | |||
|
Fiorella Quino
|
r139 | JARS_NBITS = 32 | |
| 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) | |||
|
Fiorella Quino
|
r123 | ||
| class Meta: | |||
| db_table = 'jars_filters' | |||
|
Fiorella Quino
|
r136 | ||
| def __unicode__(self): | |||
| return u'%s' % (self.name) | |||
| def parms_to_dict(self): | |||
| parameters = {} | |||
|
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) | |||
|
Fiorella Quino
|
r136 | ||
| return parameters | |||
| def dict_to_parms(self, parameters): | |||
|
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'] | |||
|
Fiorella Quino
|
r123 | ||
|
Fiorella Quino
|
r118 | ||
|
Juan C. Espinoza
|
r6 | class JARSConfiguration(Configuration): | |
|
Fiorella Quino
|
r118 | ADC_RESOLUTION = 8 | |
| PCI_DIO_BUSWIDTH = 32 | |||
|
Fiorella Quino
|
r123 | HEADER_VERSION = 1103 | |
| BEGIN_ON_START = True | |||
| REFRESH_RATE = 1 | |||
|
Fiorella Quino
|
r118 | ||
|
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' | |||
|
Fiorella Quino
|
r123 | ||
| def parms_to_dict(self): | |||
| parameters = {} | |||
|
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) | |
|
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 | |
|
Fiorella Quino
|
r123 | ||
| return parameters | |||
|
Fiorella Quino
|
r136 | def add_parms_to_filter(self): | |
| self.filter_parms = self.filter.parms_to_dict() | |||
| self.save() | |||
|
Fiorella Quino
|
r123 | def dict_to_parms(self, parameters): | |
|
Fiorella Quino
|
r150 | ||
|
Juan C. Espinoza
|
r157 | self.name = parameters['name'] | |
| self.device.id = int(parameters['device_id']) | |||
|
Fiorella Quino
|
r150 | self.exp_type = int(parameters['exp_type']) | |
| if parameters['exptype'] == 'PDATA': | |||
| 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'] | |||
|
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
|
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'] | |
|
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']) | |
|
Fiorella Quino
|
r123 | ||
| def status_device(self): | |||
|
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() | |||
| return self.device.status | |||
| def stop_device(self): | |||
| answer = api.stop(self.device.ip_address,self.device.port_address) | |||
| self.device.status = int(answer[0]) | |||
| self.message = answer[2:] | |||
| self.device.save() | |||
| return self.device.status | |||
|
Fiorella Quino
|
r123 | ||
| def read_device(self): | |||
|
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 '' | |||
| #self.dict_to_parms(parms) | |||
| self.message = 'Current JARS configuration was read successfully.' | |||
| self.device.save() | |||
| return parms | |||
|
Fiorella Quino
|
r123 | ||
| def write_device(self): | |||
|
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) | |||
| 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:] | |||
| self.device.save() | |||
| return self.device.status | |||
|
Fiorella Quino
|
r158 | def start_device(self): | |
| self.write_device() | |||
|
Juan C. Espinoza
|
r157 | def echo(self): | |
| answer = api.echo(self.device.ip_address,self.device.port_address,'(=') | |||
| #print answer | |||
| self.device.status = int(answer[0]) | |||
| self.message = answer[2:] | |||
| self.device.save() | |||
| return #self.device.status | |||
| def update_from_file(self, parameters): | |||
| self.dict_to_parms(parameters) | |||
| self.save() | |||
| def get_absolute_url_import(self): | |||
| return reverse('url_import_jars_conf', args=[str(self.id)]) | |||
| def get_absolute_url_read(self): | |||
| return reverse('url_read_jars_conf', args=[str(self.id)]) |