radarsys_met Commit - r205:2b52673784ab

Task #769: Implementacion Scheduler para ABS, Task #717: Implementacion de start_device() y stop_device()...

Fiorella Quino -

r205:2b52673784ab

parent child

Context file:

r205:2b52673784ab

Collapse all files

apps/abs/models.py +45 -7

             from django.db import models
             from apps.main.models import Configuration
             from django.core.urlresolvers import reverse
             # Create your models here.
+            from celery.execute import send_task
+            from datetime import datetime
             import ast
             import socket
             import json
             import requests
             import struct
             import sys, time
             import multiprocessing
             antenna_default = json.dumps({
                                 "antenna_up": [[0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.5,0.5,0.5,0.5,1.0,1.0,1.0,1.0],
                                                 [0.5,0.5,0.5,0.5,1.0,1.0,1.0,1.0],
                                                 [0.5,0.5,0.5,0.5,1.0,1.0,1.0,1.0],
                                                 [0.5,0.5,0.5,0.5,1.0,1.0,1.0,1.0]
                                                 ]
                                                ,
                                 "antenna_down": [[0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.0,0.0,0.0,0.0,0.5,0.5,0.5,0.5],
                                                 [0.5,0.5,0.5,0.5,3.0,3.0,3.0,3.0],
                                                 [0.5,0.5,0.5,0.5,3.0,3.0,3.0,3.0],
                                                 [0.5,0.5,0.5,0.5,3.0,3.0,3.0,3.0],
                                                 [0.5,0.5,0.5,0.5,3.0,3.0,3.0,3.0]],
                             })
             tx_default = json.dumps({
                             "up": [[1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1]],
                           "down": [[1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1]],
                          })
             rx_default = json.dumps({
                             "up": [[1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1]],
                           "down": [[1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [1,1,1,1,0,0,0,0],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1],
                                    [0,0,0,0,1,1,1,1]],
                          })
             conf_default = {}
             status_default = {}
             default_messages = {}
             for i in range(1,65):
                 conf_default[str(i)] = ""
                 status_default[str(i)] = 0
                 default_messages[str(i)] = "Module "+str(i)
             ues_default = json.dumps({
                             "up": [0.533333,0.00000,1.06667,0.00000],
                             "down": [0.533333,0.00000,1.06667,0.00000]
                         })
             onlyrx_default = json.dumps({
                                 "up": False,
                                 "down": False
                             })
             def up_convertion(cadena):
                 valores = []
                 for c in cadena:
                     if c == 1.0: valores=valores+['000']
                     if c == 2.0: valores=valores+['001']
                     if c == 3.0: valores=valores+['010']
                     if c == 0.0: valores=valores+['011']
                     if c == 0.5: valores=valores+['100']
                     if c == 1.5: valores=valores+['101']
                     if c == 2.5: valores=valores+['110']
                     if c == 3.5: valores=valores+['111']
                 return valores
             def up_conv_bits(value):
                 if value == 1.0: bits="000"
                 if value == 2.0: bits="001"
                 if value == 3.0: bits="010"
                 if value == 0.0: bits="011"
                 if value == 0.5: bits="100"
                 if value == 1.5: bits="101"
                 if value == 2.5: bits="110"
                 if value == 3.5: bits="111"
                 return bits
             def down_convertion(cadena):
                 valores = []
                 for c in cadena:
                     if c == 1.0: valores=valores+['000']
                     if c == 2.0: valores=valores+['001']
                     if c == 3.0: valores=valores+['010']
                     if c == 0.0: valores=valores+['011']
                     if c == 0.5: valores=valores+['100']
                     if c == 1.5: valores=valores+['101']
                     if c == 2.5: valores=valores+['110']
                     if c == 3.5: valores=valores+['111']
                 return valores
             def down_conv_bits(value):
                 if value == 1.0: bits="000"
                 if value == 2.0: bits="001"
                 if value == 3.0: bits="010"
                 if value == 0.0: bits="011"
                 if value == 0.5: bits="100"
                 if value == 1.5: bits="101"
                 if value == 2.5: bits="110"
                 if value == 3.5: bits="111"
                 return bits
             def ip2position(module_number):
                 j=0
                 i=0
                 for x in range(0,module_number-1):
                     j=j+1
                     if j==8:
                         i=i+1
                         j=0
                 pos = [i,j]
                 return pos
             def fromBinary2Char(binary_string):
                 number = int(binary_string, 2)
                 #Plus 33 to avoid more than 1 characters values such as: '\x01'-'\x1f'
                 number = number + 33
                 char = chr(number)
                 return char
             def fromChar2Binary(char):
                 number = ord(char) - 33
                 #Minus 33 to get the real value
                 bits = bin(number)[2:]
                 #To ensure we have a string with 6bits
                 if len(bits) < 6:
                     bits = bits.zfill(6)
                 return bits
             OPERATION_MODES = (
                              (0, 'Manual'),
                              (1, 'Automatic'),
                          )
             class ABSConfiguration(Configuration):
                 active_beam     = models.CharField(verbose_name='Active Beam', max_length=20000, default="{}")
                 module_status   = models.CharField(verbose_name='Module Status', max_length=10000, default=json.dumps(status_default))
                 operation_mode  = models.PositiveSmallIntegerField(verbose_name='Operation Mode', choices=OPERATION_MODES, default = 0)
                 operation_value = models.FloatField(verbose_name='Periodic (seconds)', default="10", null=True, blank=True)
                 module_messages = models.CharField(verbose_name='Modules Messages', max_length=10000, default=json.dumps(default_messages))
                 class Meta:
                     db_table = 'abs_configurations'
                 def get_absolute_url_plot(self):
                     return reverse('url_plot_abs_patterns', args=[str(self.id)])
                 def parms_to_dict(self):
                     parameters = {}
                     parameters['device_id'] = self.device.id
                     parameters['name']      = self.name
                     parameters['beams']     = {}
                     beams = ABSBeam.objects.filter(abs_conf=self)
                     b=1
                     for beam in beams:
                         #absbeam = ABSBeam.objects.get(pk=beams[beam])
                         parameters['beams']['beam'+str(b)] = beam.parms_to_dict()#absbeam.parms_to_dict()
                         b+=1
                     return parameters
                 def dict_to_parms(self, parameters):
                     self.name = parameters['name']
                     absbeams  = ABSBeam.objects.filter(abs_conf=self)
                     beams     = parameters['beams']
                     if absbeams:
                         beams_number    = len(beams)
                         absbeams_number = len(absbeams)
                         if beams_number==absbeams_number:
                             i = 1
                             for absbeam in absbeams:
                                 absbeam.dict_to_parms(beams['beam'+str(i)])
                                 i = i+1
                         elif beams_number > absbeams_number:
                             i = 1
                             for absbeam in absbeams:
                                 absbeam.dict_to_parms(beams['beam'+str(i)])
                                 i=i+1
                             for x in range(i,beams_number+1):
                                 new_beam = ABSBeam(
                                            name     =beams['beam'+str(i)]['name'],
                                            antenna  =json.dumps(beams['beam'+str(i)]['antenna']),
                                            abs_conf = self,
                                            tx       =json.dumps(beams['beam'+str(i)]['tx']),
                                            rx       =json.dumps(beams['beam'+str(i)]['rx']),
                                            ues      =json.dumps(beams['beam'+str(i)]['ues']),
                                            only_rx  =json.dumps(beams['beam'+str(i)]['only_rx'])
                                            )
                                 new_beam.save()
                                 i=i+1
                         else: #beams_number < absbeams_number:
                             i = 1
                             for absbeam in absbeams:
                                 if i <= beams_number:
                                     absbeam.dict_to_parms(beams['beam'+str(i)])
                                     i=i+1
                                 else:
                                     absbeam.delete()
                     else:
                         for beam in beams:
                             new_beam = ABSBeam(
                                        name     =beams[beam]['name'],
                                        antenna  =json.dumps(beams[beam]['antenna']),
                                        abs_conf = self,
                                        tx       =json.dumps(beams[beam]['tx']),
                                        rx       =json.dumps(beams[beam]['rx']),
                                        ues      =json.dumps(beams[beam]['ues']),
                                        only_rx  =json.dumps(beams[beam]['only_rx'])
                                        )
                             new_beam.save()
                 def update_from_file(self, parameters):
                     self.dict_to_parms(parameters)
                     self.save()
                 def get_beams(self, **kwargs):
                     '''
                     This function returns ABS Configuration beams
                     '''
                     return ABSBeam.objects.filter(abs_conf=self.pk, **kwargs)
                 def clone(self, **kwargs):
                     beams = self.get_beams()
                     self.pk = None
                     self.id = None
                     for attr, value in kwargs.items():
                         setattr(self, attr, value)
                     self.save()
                     for beam in beams:
                         beam.clone(abs_conf=self)
                     #-----For Active Beam-----
                     active_beam = json.loads(self.active_beam)
                     new_beams = ABSBeam.objects.filter(abs_conf=self)
                     active_beam['active_beam'] = new_beams[0].id
                     self.active_beam = json.dumps(active_beam)
                     self.save()
                     #-----For Active Beam-----
+                    #-----For Device Status---
+                    self.device.status = 3
+                    self.device.save()
+                    #-----For Device Status---
                     return self
+                def start_device(self):
+                    if self.device.status == 3:
+                        try:
+                            #self.write_device()
+                            send_task('task_change_beam', [self.id],)
+                            self.message = 'ABS running'
+                        except Exception as e:
+                            self.message = str(e)
+                            return False
+                        return True
+                    else:
+                        self.message = 'Please, select Write ABS Device first.'
+                        return False
+                def stop_device(self):
+                    self.device.status = 2
+                    self.device.save()
+                    self.message = 'ABS has been stopped.'
+                    self.save()
+                    return True
                 def module_conf(self, module_num, beams):
                     """
                     This function creates beam configurations for one abs module.
                     """
                     ip_address  = self.device.ip_address
                     ip_address  = ip_address.split('.')
                     module_seq  = (ip_address[0],ip_address[1],ip_address[2])
                     dot         = '.'
                     module_ip   = dot.join(module_seq)+'.'+str(module_num)
                     module_port = self.device.port_address
                     write_route = 'http://'+module_ip+':'+str(module_port)+'/write'
                     #header = 'JROABSCeCnModCnMod01000108SNDFexperimento1.ab1'
                     #module = 'ABS_'+str(module_num)
                     bs  = '' #{}
                     i=1
                     for beam in beams:
                         #bs[i] = fromBinary2Char(beam.module_6bits(module_num))
                         bs = bs + fromBinary2Char(beam.module_6bits(module_num))
                         i=i+1
                     beams = bs
                     parameters = {}
                     parameters['beams']  = beams #json.dumps(beams)
                     print parameters
                     answer = ''
                     try:
                         #r_write = requests.post(write_route, parameters, timeout=0.5)
                         r_write = requests.post(write_route, json = parameters, timeout=0.5)
                         answer  = r_write.json()
                         #self.message = answer['message']
                     except:
                         #self.message = "Could not write ABS parameters"
                         return False
                     return answer
                 def write_device(self):
                     """
                     This function sends the beams list to every abs module.
                     It needs 'module_conf' function
                     """
                     beams = ABSBeam.objects.filter(abs_conf=self)
                     connected_modules = ast.literal_eval(self.module_status)
                     suma_connected_modules = 0
                     for c in connected_modules:
                         suma_connected_modules = suma_connected_modules+connected_modules[c]
                     if not suma_connected_modules > 0 :
                         self.message = "No ABS Module detected."
                         return False
                     #-------------Write each abs module-----------
                     if beams:
                         beams_status = ast.literal_eval(self.module_status)
                         disconnected_modules = 0
                         for i in range(1,65): #(62,65)
                             #--------------JUEVES-------------
                             if beams_status[str(i)] != 0:
                                 try:
                                     answer = self.module_conf(i,beams)
                                     if answer:
                                         if answer['status']:
                                             beams_status[str(i)] = 3
                                 except:
                                     beams_status[str(i)] = 1
                                     pass
                             else:
                                 disconnected_modules += 1
                     else:
                         self.message = "ABS Configuration does not have beams"
                         return False
                     #self.device.status = 1
                     ##
                     #-------------Jueves-------------
                     if disconnected_modules == 64:
                         self.message = "Could not write ABS Modules"
+                        self.device.status = 0
                         return False
                     else:
                         self.message = "ABS Beams List have been sent to ABS Modules"
                         beams[0].set_as_activebeam()
+                    self.device.status = 3
                     self.module_status = json.dumps(beams_status)
-                    self.save()
                     self.save()
                     return True
                 def read_module(self, module):
                     """
                     Read out-bits (up-down) of 1 abs module NOT for Configuration
                     """
                     parameters = {}
                     ip_address  = self.device.ip_address
                     ip_address  = ip_address.split('.')
                     module_seq  = (ip_address[0],ip_address[1],ip_address[2])
                     dot         = '.'
                     module_ip   = dot.join(module_seq)+'.'+str(module)
                     module_port = self.device.port_address
                     read_route = 'http://'+module_ip+':'+str(module_port)+'/read'
                     print(read_route)
                     answer = ''
                     module_bits = ''
                     try:
                         r_write = requests.get(read_route, timeout=0.7)
                         answer  = r_write.json()
                         self.message = answer['message']
                         module_bits    = answer['allbits']
                     except:
                         #message = "Could not read ABS parameters"
                         answer  = r_write.json()
                         return 0
                     return module_bits
                 def absmodule_status(self):
                     """
                     This function gets the status of each abs module. It sends GET method to Web Application
                     in Python Bottle.
                     This function updates "module_status" field from ABSconfiguration.
                     """
                     ip_address  = self.device.ip_address
                     ip_address  = ip_address.split('.')
                     module_seq  = (ip_address[0],ip_address[1],ip_address[2])
                     dot         = '.'
                     module_port = self.device.port_address
                     modules_status = json.loads(self.module_status)
                     module_messages = json.loads(self.module_messages)
                     for i in range(1,65):
                         module_ip   = dot.join(module_seq)+'.'+str(i)
                         print module_ip
                         route = 'http://'+module_ip+':'+str(module_port)+'/status'
                         try:
                             r = requests.get(route, timeout=0.5)#, timeout=0.7)
                             answer = r.json()
                             modules_status[str(i)] = answer['status']
                             module_messages[str(i)] = answer['message']
                         except:
                             modules_status[str(i)] = 0
                             pass
                     self.module_status   = json.dumps(modules_status)
                     self.module_messages = json.dumps(module_messages)
                     self.save()
                     return
                 def connected_modules(self):
                     """
                     This function returns the number of connected abs-modules without updating.
                     """
                     modules_status = json.loads(self.module_status)
                     num = 0
                     for status in modules_status:
                         if modules_status[status] != 0:
                             num += 1
                     return num
                 def status_device(self):
                     """
                     This function returns the status of all abs-modules as one.
                     If at least one module is connected, its answer is "1"
                     """
                     self.absmodule_status()
                     connected_modules = self.connected_modules()
                     if connected_modules>0:
                         self.message = 'ABS modules Status have been updated.'
                         return 1
                     self.message = 'No ABS module is connected.'
                     return 0
                 def write_module(self, module):
                     """
                     Send configuration to one abs module
                     """
                     parameters = {}
                     ip_address  = self.abs_conf.device.ip_address
                     ip_address  = ip_address.split('.')
                     module_seq  = (ip_address[0],ip_address[1],ip_address[2])
                     dot         = '.'
                     module_ip   = dot.join(module_seq)+'.'+str(module)
                     module_port = self.abs_conf.device.port_address
                     write_route = 'http://'+module_ip+':'+str(module_port)+'/write'
                     #print write_route
                     #header = 'JROABSCeCnModCnMod01000108SNDFexperimento1.ab1'
                     #module = 'ABS_'+str(module)
                     beams  = '!`*3<ENW'#{1: '001000', 2: '010001', 3: '010010', 4: '000011', 5: '101100', 6: '101101',
                              # 7: '110110', 8: '111111', 9: '000000', 10: '001001', 11: '010010', 12: '011011'}
                     #parameters['header'] = header
                     parameters['module'] = module
                     parameters['beams']  = json.dumps(beams)
                     answer = ''
                     try:
                         r_write = requests.post(write_route, parameters, timeout=0.5)
                         answer  = r_write.json()
                         self.message = answer['message']
                     except:
                         self.message = "Could not write ABS parameters"
                         return 0
                     #self.device.status = int(answer['status'])
                     return 1
                 def beam_selector(self, module, beam_pos):
                     """
                     This function selects the beam number for one absmodule.
                     """
                     if beam_pos > 0:
                         beam_pos = beam_pos - 1
                     else:
                         beam_pos = 0
                     #El indice del apunte debe ser menor que el numero total de apuntes
                     #El servidor tcp en el embebido comienza a contar desde 0
                     beams_list    = ABSBeam.objects.filter(abs_conf=self)
                     if len(beams_list) < beam_pos:
                         return 0
                     flag = 1
                     if beam_pos>9:
                         flag = 2
                     module_address = ('192.168.1.'+str(module), 5500)
                     header   = 'JROABSCeCnModCnMod0100000'
                     numbers  = len(str(beam_pos))
                     function = 'CHGB'
                     message_tx = header+str(numbers)+function+str(beam_pos)+'0'
                     # Create the datagram socket
                     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                     #sock.connect(module_address)
                     try:
                         sock.connect(module_address)
                         sock.send(message_tx)
                         sock.close()
                         print("Writing abs module:"+module_address[0]+"...")
                     except:
                         sock = None
                         print("Problem writing abs module:"+module_address[0])
                         return 0
                     return 1
                 def change_beam(self, beam_pos):
                     """
                     This function selects the beam number for all absmodules.
                     """
                     for i in range(1,65):
                         try:
                             self.beam_selector(i,beam_pos)
                         except:
                             print("Problem with module: 192.168.1."+str(i))
                             self.message = "Problem with module: 192.168.1."+str(i)
                             #return 0
                     return 1
                 def send_beam_num(self, beam_pos):
                     """
                     This function connects to a multicast group and sends the beam number
                     to all abs modules.
                     """
                     if beam_pos > 0:
                         beam_pos = beam_pos - 1
                     else:
                         beam_pos = 0
                     #El indice del apunte debe ser menor que el numero total de apuntes
                     #El servidor tcp en el embebido comienza a contar desde 0
                     beams_list    = ABSBeam.objects.filter(abs_conf=self)
                     if len(beams_list) < beam_pos:
                         return 0
                     flag = 1
                     if beam_pos>9:
                         flag = 2
                     header   = 'JROABSCeCnModCnMod0100000'
                     flag  = str(flag)
                     function = 'CHGB'
                     message_tx = header+flag+function+str(beam_pos)+'0'
                     multicast_group = '224.3.29.71'
                     server_address = ('',10000)
                     # Create the socket
                     sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
                     # Bind to the server address
                     sock.bind(server_address)
                     # Telling the OS add the socket to the multicast on all interfaces
                     group = socket.inet_aton(multicast_group)
                     mreq  = struct.pack('4sL', group, socket.INADDR_ANY)
                     sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
                     #print 'sending acknowledgement to all: \n' + message_tx
                     sock.sendto(message_tx, (multicast_group, 10000))
                     sock.close()
                     sock = None
                     return 1
                 def test1(self):
                     t1 = time.time()
                     t2 = 0
                     while (t2-t1)<100:#300
                         t2 = time.time()
                         self.send_beam_num(2)
                         time.sleep(0.04)
                         self.send_beam_num(1)
                         time.sleep(0.04)
                     return
                 def change_procs_test1(self, module):
                 	for i in range (1,300):#300
                 	    beam_pos = 1
                 	    module_address = ('192.168.1.'+str(module), 5500)
                 	    header   = 'JROABSCeCnModCnMod0100000'
                 	    numbers  = len(str(beam_pos))
                 	    function = 'CHGB'
                 	    message_tx = header+str(numbers)+function+str(beam_pos)+'0'
                 	    # Create the datagram socket
                 	    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                 	    sock.connect(module_address)
                 	    sock.send(message_tx)
                 	    #t = sock.recv(1024)
                 	    sock.close()
                 	    sock = None
                 	    time.sleep(0.04)
                 	    beam_pos = 0
                 	    numbers  = len(str(beam_pos))
                 	    message_tx = header+str(numbers)+function+str(beam_pos)+'0'
                 	    # Create the datagram socket
                 	    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                 	    sock.connect(module_address)
                 	    sock.send(message_tx)
                 	    sock.close()
                 	    sock = None
                 	    time.sleep(0.04)
                 def multi_procs_test1(self):
                     """
                     This function sends the beam number to all abs modules using multiprocessing.
                     """
                     #if __name__ == "__main__":
                     size = 10000000   # Number of random numbers to add
                     procs =  65 # (Number-1) of processes to create (absmodule)
                     # Create a list of jobs and then iterate through
                     # the number of processes appending each process to
                     # the job list
                     jobs = []
                     for i in range(1, procs):
                         process = multiprocessing.Process(target=self.change_procs_test1,args=(i,))
                         jobs.append(process)
                         #print jobs
                         # Start the processes (i.e. calculate the random number lists)
                     for j in jobs:
                             #time.sleep(0.4)
                             #print j
                             j.start()
                         # Ensure all of the processes have finished
                     for j in jobs:
                             j.join()
                     print("List processing complete.")
                     return 1
                 def multi_procs_test2(self):
                     """
                     This function use multiprocessing python library. Importing Pool we can select
                     the number of cores we want to use
                     After 'nproc' linux command, we know how many cores computer has.
                     NOT WORKING
                     """
                     import multiprocessing
                     pool = multiprocessing.Pool(3) # cores
                     tasks = []
                     procs = 65
                     for i in range(62, procs):
                         tasks.append( (i,))
                     #print tasks
                     #pool.apply_async(self.change_procs_test1, 62)
                     results = [pool.apply( self.change_procs_test1, t ) for t in tasks]
                     #for result in results:
                         #result.get()
                         #(plotNum, plotFilename) = result.get()
                         #print("Result: plot %d written to %s" % (plotNum, plotFilename) )
                     return 1
                 def multi_procs_test3(self):
                     """
                     This function use multiprocessing python library. Importing Pool we can select
                     the number of cores we want to use
                     After 'nproc' linux command, we know how many cores computer has.
                     """
                     from multiprocessing import Pool
                     import time
                     def f(x):
                         return x*x
                     tasks = []
                     procs = 65
                     pool = Pool(processes=4)
                     #for i in range(62, procs):
                     #result = pool.map(f, range(62,65))
                     it = pool.imap(self.change_procs_test1, range(62,65))
                     #result.get(timeout=5)
                     return 1
                 def f(x):
                     print x
                     return
                 def multi_procs_test4(self):
                     import multiprocessing as mp
                     num_workers = mp.cpu_count()
                     pool = mp.Pool(num_workers)
                     procs = 65
                     for i in range(62, procs):
                     #for task in tasks:
                         pool.apply_async(self.f, args = (i,))
                     pool.close()
                     pool.join()
                     return 1
                 def get_absolute_url_import(self):
                     return reverse('url_import_abs_conf', args=[str(self.id)])
             class ABSBeam(models.Model):
                 name         = models.CharField(max_length=60, default='Beam')
                 antenna      = models.CharField(verbose_name='Antenna', max_length=1000, default=antenna_default)
                 abs_conf     = models.ForeignKey(ABSConfiguration, null=True, verbose_name='ABS Configuration')
                 tx           = models.CharField(verbose_name='Tx', max_length=1000, default=tx_default)
                 rx           = models.CharField(verbose_name='Rx', max_length=1000, default=rx_default)
                 s_time       = models.TimeField(verbose_name='Star Time', default='00:00:00')
                 e_time       = models.TimeField(verbose_name='End Time', default='23:59:59')
                 modules_conf = models.CharField(verbose_name='Modules', max_length=2000, default=json.dumps(conf_default))
                 ues          = models.CharField(verbose_name='Ues', max_length=100, default=ues_default)
                 only_rx      = models.CharField(verbose_name='Only RX', max_length=40, default=onlyrx_default)
                 class Meta:
                     db_table = 'abs_beams'
                 def __unicode__(self):
                     return u'%s' % (self.name)
                 def parms_to_dict(self):
                     #Update data
                     self.modules_6bits()
                     parameters = {}
                     parameters['name']          = self.name
                     parameters['antenna']       = ast.literal_eval(self.antenna)
                     parameters['abs_conf']      = self.abs_conf.name
                     parameters['tx']            = ast.literal_eval(self.tx)
                     parameters['rx']            = ast.literal_eval(self.rx)
                     parameters['s_time']        = self.s_time.strftime("%H:%M:%S")
                     parameters['e_time']        = self.e_time.strftime("%H:%M:%S")
                     parameters['configuration'] = ast.literal_eval(self.modules_conf)
                     parameters['ues']           = ast.literal_eval(self.ues)
                     parameters['only_rx']       = json.loads(self.only_rx)
                     return parameters
                 def dict_to_parms(self, parameters):
                     self.name     = parameters['name']
                     self.antenna  = json.dumps(parameters['antenna'])
                     #self.abs_conf = parameters['abs_conf']
                     self.tx       = json.dumps(parameters['tx'])
                     self.rx       = json.dumps(parameters['rx'])
                     #self.s_time = parameters['s_time']
                     #self.e_time = parameters['e_time']
                     self.ues      = json.dumps(parameters['ues'])
                     self.only_rx     = json.dumps(parameters['only_rx'])
                     self.modules_6bits()
                     self.save()
                 def clone(self, **kwargs):
                     self.pk = None
                     self.id = None
                     for attr, value in kwargs.items():
                         setattr(self, attr, value)
                     self.save()
                     return self
                 def module_6bits(self, module):
                     """
                     This function reads antenna pattern and choose 6bits (upbits-downbits) for one abs module
                     """
                     if module > 64:
                         beam_bits = ""
                         return beam_bits
                     data      = ast.literal_eval(self.antenna)
                     up_data   = data['antenna_up']
                     down_data = data['antenna_down']
                     pos        = ip2position(module)
                     up_value   = up_data[pos[0]][pos[1]]
                     down_value = down_data[pos[0]][pos[1]]
                     up_bits   = up_conv_bits(up_value)
                     down_bits = down_conv_bits(down_value)
                     beam_bits = up_bits+down_bits
                     return beam_bits
                 def modules_6bits(self):
                     """
                     This function returns 6bits from every abs module (1-64) in a dict
                     """
                     modules_configuration = ast.literal_eval(self.modules_conf)
                     for i in range(1,65):
                         modules_configuration[str(i)] =  self.module_6bits(i)
                     self.modules_conf = json.dumps(modules_configuration)
                     self.save()
                     return self.modules_conf
                 def set_as_activebeam(self):
                     """
                     This function set this beam as the active beam of its ABS Configuration.
                     """
                     self.abs_conf.active_beam = json.dumps({'active_beam': self.id})
                     self.abs_conf.save()
                     return
                 @property
                 def get_upvalues(self):
                     """
                     This function reads antenna pattern and show the up-value of one abs module
                     """
                     data      = ast.literal_eval(self.antenna)
                     up_data   = data['antenna_up']
                     up_values = []
                     for data in up_data:
                         for i in range(0,8):
                             up_values.append(data[i])
                     return up_values
                 @property
                 def antenna_upvalues(self):
                     """
                     This function reads antenna pattern and show the up - values of one abs beam
                     in a particular order
                     """
                     data      = ast.literal_eval(self.antenna)
                     up_data   = data['antenna_up']
                     return up_data
                 @property
                 def antenna_downvalues(self):
                     """
                     This function reads antenna pattern and show the down - values of one abs beam
                     in a particular order
                     """
                     data      = ast.literal_eval(self.antenna)
                     down_data   = data['antenna_down']
                     return down_data
                 @property
                 def get_downvalues(self):
                     """
                     This function reads antenna pattern and show the down-value of one abs module
                     """
                     data      = ast.literal_eval(self.antenna)
                     down_data = data['antenna_down']
                     down_values = []
                     for data in down_data:
                         for i in range(0,8):
                             down_values.append(data[i])
                     return down_values
                 @property
                 def get_up_ues(self):
                     """
                     This function shows the up-ues-value of one beam
                     """
                     data      = ast.literal_eval(self.ues)
                     up_ues = data['up']
                     return up_ues
                 @property
                 def get_down_ues(self):
                     """
                     This function shows the down-ues-value of one beam
                     """
                     data      = ast.literal_eval(self.ues)
                     down_ues = data['down']
                     return down_ues
                 @property
                 def get_up_onlyrx(self):
                     """
                     This function shows the up-onlyrx-value of one beam
                     """
                     data      = json.loads(self.only_rx)
                     up_onlyrx = data['up']
                     return up_onlyrx
                 @property
                 def get_down_onlyrx(self):
                     """
                     This function shows the down-onlyrx-value of one beam
                     """
                     data      = json.loads(self.only_rx)
                     down_onlyrx = data['down']
                     return down_onlyrx
                 @property
                 def get_tx(self):
                     """
                     This function shows the tx-values of one beam
                     """
                     data = json.loads(self.tx)
                     return data
                 @property
                 def get_uptx(self):
                     """
                     This function shows the up-tx-values of one beam
                     """
                     data = json.loads(self.tx)
                     up_data = data['up']
                     up_values = []
                     for data in up_data:
                         for i in range(0,8):
                             up_values.append(data[i])
                     return up_values
                 @property
                 def get_downtx(self):
                     """
                     This function shows the down-tx-values of one beam
                     """
                     data = json.loads(self.tx)
                     down_data = data['down']
                     down_values = []
                     for data in down_data:
                         for i in range(0,8):
                             down_values.append(data[i])
                     return down_values
                 @property
                 def get_rx(self):
                     """
                     This function shows the rx-values of one beam
                     """
                     data = json.loads(self.rx)
                     return data
                 @property
                 def get_uprx(self):
                     """
                     This function shows the up-rx-values of one beam
                     """
                     data = json.loads(self.rx)
                     up_data = data['up']
                     up_values = []
                     for data in up_data:
                         for i in range(0,8):
                             up_values.append(data[i])
                     return up_values
                 @property
                 def get_downrx(self):
                     """
                     This function shows the down-rx-values of one beam
                     """
                     data = json.loads(self.rx)
                     down_data = data['down']
                     down_values = []
                     for data in down_data:
                         for i in range(0,8):
                             down_values.append(data[i])
                     return down_values

apps/abs/tasks.py +5 -14

             from __future__ import absolute_import
             from apps.main.models import Configuration
             from .models import ABSBeam
             import json
             from datetime import timedelta, datetime
             from celery.task import task
-            """
+            @task(name='task_change_beam')
-            @task
-            def beam_task(id_conf):
-                abs_conf  = Configuration.objects.get(pk=id_conf)
-                task_change_beam(abs_conf.pk)
-                return task_change_beam(abs_conf.pk)
-            """
-            @task
             def task_change_beam(id_conf):
                 abs_conf = Configuration.objects.get(pk=id_conf)
                 beams_list = ABSBeam.objects.filter(abs_conf=abs_conf)
                 active_beam = json.loads(abs_conf.active_beam)
                 run_every = timedelta(seconds=abs_conf.operation_value)
                 now = datetime.utcnow()
                 date = now + run_every
+                if abs_conf.device.status != 3:
+                    return abs_conf.device.status
                 if abs_conf.operation_mode == 0:  #Manual Mode
                     return 1
                 if active_beam:
                     current_beam = ABSBeam.objects.get(pk=active_beam['active_beam'])
                     i=0
                     for beam in beams_list:
                         if beam == current_beam:
                             i+=1
                             break
                         i+=1
                     if i < len(beams_list):
                         next_beam = beams_list[i]
                         abs_conf.send_beam_num(i+1)
                         next_beam.set_as_activebeam()
                         task = task_change_beam.apply_async((abs_conf.pk,), eta=date)
                         print next_beam
                     else:
                         abs_conf.send_beam_num(1)
                         beams_list[0].set_as_activebeam()
                         task = task_change_beam.apply_async((abs_conf.pk,), eta=date)
-                        print next_beam
+                        print beams_list[0]
                         i=0
                 else:
                     abs_conf.send_beam_num(1)
                     beams_list[0].set_as_activebeam()
                     task = task_change_beam.apply_async((abs_conf.pk,), eta=date)
                 return 2

apps/abs/views.py +3 -3

             from django.shortcuts import render_to_response
             from django.template import RequestContext
             from django.shortcuts import redirect, render, get_object_or_404
             from django.contrib import messages
             from django.conf import settings
             from django.http import HttpResponse
             from django.core.urlresolvers import reverse
             from datetime import datetime
             from time import sleep
             import os
-            from apps.main.models import Device, Configuration
+            from apps.main.models import Device, Configuration, Experiment
             from apps.main.views import sidebar
             from .models import ABSConfiguration, ABSBeam
             from .forms import ABSConfigurationForm, ABSBeamEditForm, ABSBeamAddForm, ABSImportForm
             from .utils.overJroShow import overJroShow
             from .utils.OverJRO import OverJRO
             # Create your views here.
             import json, ast
             def get_values_from_form(form_data):
                 sublistup     = []
                 sublistdown   = []
                 subtxlistup   = []
                 subtxlistdown = []
                 subrxlistup   = []
                 subrxlistdown = []
                 up_values_list     = []
                 down_values_list   = []
                 up_txvalues_list   = []
                 down_txvalues_list = []
                 up_rxvalues_list   = []
                 down_rxvalues_list = []
                 values_list = {}
                 cont = 1
                 for i in range(1,65):
                     x = float(form_data['abs_up'+str(i)])
                     y = float(form_data['abs_down'+str(i)])
                     sublistup.append(x)
                     sublistdown.append(y)
                     if str(i) in form_data.getlist('uptx_checks'):
                         subtxlistup.append(1)
                     else:
                         subtxlistup.append(0)
                     if str(i) in form_data.getlist('downtx_checks'):
                         subtxlistdown.append(1)
                     else:
                         subtxlistdown.append(0)
                     if str(i) in form_data.getlist('uprx_checks'):
                         subrxlistup.append(1)
                     else:
                         subrxlistup.append(0)
                     if str(i) in form_data.getlist('downrx_checks'):
                         subrxlistdown.append(1)
                     else:
                         subrxlistdown.append(0)
                     cont = cont+1
                     if cont == 9:
                         up_values_list.append(sublistup)
                         down_values_list.append(sublistdown)
                         sublistup   = []
                         sublistdown = []
                         up_txvalues_list.append(subtxlistup)
                         down_txvalues_list.append(subtxlistdown)
                         subtxlistup   = []
                         subtxlistdown = []
                         up_rxvalues_list.append(subrxlistup)
                         down_rxvalues_list.append(subrxlistdown)
                         subrxlistup   = []
                         subrxlistdown = []
                         cont = 1
                 list_uesup   = []
                 list_uesdown = []
                 for i in range(1,5):
                     if form_data['ues_up'+str(i)] == '':
                         list_uesup.append(0.0)
                     else:
                         list_uesup.append(float(form_data['ues_up'+str(i)]))
                     if form_data['ues_down'+str(i)] == '':
                         list_uesdown.append(0.0)
                     else:
                         list_uesdown.append(float(form_data['ues_down'+str(i)]))
                 onlyrx_list = form_data.getlist('onlyrx')
                 only_rx = {}
                 if '1' in onlyrx_list:
                     only_rx['up'] = True
                 else:
                     only_rx['up'] = False
                 if '2' in onlyrx_list:
                     only_rx['down'] = True
                 else:
                     only_rx['down'] = False
                 antenna = {'antenna_up': up_values_list, 'antenna_down': down_values_list}
                 tx      = {'up': up_txvalues_list, 'down': down_txvalues_list}
                 rx      = {'up': up_rxvalues_list, 'down': down_rxvalues_list}
                 ues     = {'up': list_uesup, 'down': list_uesdown}
                 name    = str(form_data['beam_name'])
                 beam_data = {'name': name, 'antenna': antenna, 'tx': tx, 'rx': rx, 'ues': ues, 'only_rx': only_rx}
                 return beam_data
             def abs_conf(request, id_conf):
                 conf           = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beams          = ABSBeam.objects.filter(abs_conf=conf)
                 active_beam_id = json.loads(conf.active_beam)
                 #------------Colors for Active Beam:-------------
                 modules_status = json.loads(conf.module_status)
                 module_messages = json.loads(conf.module_messages)
                 color_status   = {}
                 for status in modules_status:
                     if modules_status[status] == 3:    #Running background-color: #00cc00;
                         color_status[status] = 'class=text-success'#'bgcolor=#00cc00'
                     elif modules_status[status] == 1:  #Connected background-color: #ee902c;
                         color_status[status] = 'class=text-warning'#'bgcolor=#ee902c'
                     else:                              #Disconnected background-color: #ff0000;
                         color_status[status] = 'class=text-danger'#'bgcolor=#FF0000'
                 #------------------------------------------------
                 kwargs = {}
                 #kwargs['status'] = conf.device.get_status_display()
                 kwargs['connected_modules'] = str(conf.connected_modules())+'/64'
                 kwargs['dev_conf'] = conf
                 kwargs['dev_conf_keys'] = ['name',]
                 kwargs['title'] = 'ABS Configuration'
                 kwargs['suptitle'] = 'Details'
-                kwargs['no_play'] = True
+                #kwargs['no_play'] = True
                 kwargs['button'] = 'Edit Configuration'
                 #------------------Active Beam-----------------------
                 try:
                     active_beam_id = active_beam_id['active_beam']
                     active_beam = ABSBeam.objects.get(pk=active_beam_id)
                     kwargs['active_beam'] = active_beam
                     for beam in beams:
                         if beam.id == active_beam.id:
                             beam.color_status    = color_status
                             beam.module_messages = module_messages
                 except:
                     active_beam = ''
                 #----------------------------------------------------
                 kwargs['beams']    = beams
                 kwargs['modules_status'] = modules_status
                 kwargs['color_status']   = color_status
                 kwargs['module_messages'] = module_messages
-                kwargs['only_stop'] = True
+                #kwargs['only_stop'] = True
                 ###### SIDEBAR ######
                 kwargs.update(sidebar(conf=conf))
                 return render(request, 'abs_conf.html', kwargs)
             def abs_conf_edit(request, id_conf):
                 conf     = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beams = ABSBeam.objects.filter(abs_conf=conf)
                 print beams
                 if request.method=='GET':
                     form = ABSConfigurationForm(instance=conf)
                 if request.method=='POST':
                     form = ABSConfigurationForm(request.POST, instance=conf)
                     if form.is_valid():
                         conf = form.save(commit=False)
                         conf.save()
                         return redirect('url_abs_conf', id_conf=conf.id)
                 ###### SIDEBAR ######
                 kwargs = {}
                 kwargs['dev_conf'] = conf
                 #kwargs['id_dev'] = conf.id
                 kwargs['id_conf'] = conf.id
                 kwargs['form'] = form
                 kwargs['abs_beams'] = beams
                 kwargs['title'] = 'Device Configuration'
                 kwargs['suptitle'] = 'Edit'
                 kwargs['button'] = 'Save'
                 kwargs['edit'] = True
                 return render(request, 'abs_conf_edit.html', kwargs)
             def import_file(request, id_conf):
                 conf = get_object_or_404(ABSConfiguration, pk=id_conf)
                 if request.method=='POST':
                     form = ABSImportForm(request.POST, request.FILES)
                     if form.is_valid():
                         try:
                             parms = conf.import_from_file(request.FILES['file_name'])
                             if parms:
                                 conf.update_from_file(parms)
                                 messages.success(request, 'Configuration "%s" loaded succesfully' % request.FILES['file_name'])
                                 return redirect(conf.get_absolute_url_edit())
                         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 = ABSImportForm()
                 kwargs = {}
                 kwargs['form'] = form
                 kwargs['title'] = 'ABS Configuration'
                 kwargs['suptitle'] = 'Import file'
                 kwargs['button'] = 'Upload'
                 kwargs['previous'] = conf.get_absolute_url()
                 return render(request, 'abs_import.html', kwargs)
             def send_beam(request, id_conf, id_beam):
                 conf = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beam = get_object_or_404(ABSBeam, pk=id_beam)
                 beams_list    = ABSBeam.objects.filter(abs_conf=conf)
                 #To set this beam as an Active Beam
                 beam.set_as_activebeam()
                 #To send beam position to abs-modules
                 i = 0
                 for b in beams_list:
                     if b.id == int(id_beam):
                         break
                     else:
                         i += 1
                 beam_pos = i + 1 #Estandarizar
                 print 'Position: ',beam_pos
                 conf.send_beam_num(beam_pos)
                 return redirect('url_abs_conf', conf.id)
             def add_beam(request, id_conf):
                 conf     = get_object_or_404(ABSConfiguration, pk=id_conf)
                 confs    = Configuration.objects.all()
                 if request.method=='GET':
                     #form = ABSBeamEditForm()
                     form = ABSBeamAddForm()
                 if request.method=='POST':
                     form = ABSBeamAddForm(request.POST)
                     beam_data = get_values_from_form(request.POST)
                     new_beam = ABSBeam(
                                name    =beam_data['name'],
                                antenna =json.dumps(beam_data['antenna']),
                                abs_conf=conf,
                                tx      =json.dumps(beam_data['tx']),
                                rx      =json.dumps(beam_data['rx']),
                                ues     =json.dumps(beam_data['ues']),
                                only_rx =json.dumps(beam_data['only_rx'])
                                )
                     new_beam.save()
                     #---Update 6bits configuration and add beam to abs configuration beams list.
                     new_beam.modules_6bits()
                     #new_beam.add_beam2list()
                     messages.success(request, 'Beam: "%s" has been added.'  % new_beam.name)
                     return redirect('url_edit_abs_conf', conf.id)
                 ###### SIDEBAR ######
                 kwargs = {}
                 #kwargs['dev_conf'] = conf.device
                 #kwargs['id_dev'] = conf.device
                 kwargs['id_conf'] = conf.id
                 kwargs['form'] = form
                 kwargs['title'] = 'ABS Beams'
                 kwargs['suptitle'] = 'Add Beam'
                 kwargs['button'] = 'Add'
                 kwargs['no_sidebar'] = True
                 #kwargs['previous'] = conf.get_absolute_url_edit()
                 kwargs['edit'] = True
                 return render(request, 'abs_add_beam.html', kwargs)
             def edit_beam(request, id_conf, id_beam):
                 conf = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beam = get_object_or_404(ABSBeam, pk=id_beam)
                 if request.method=='GET':
                     form = ABSBeamEditForm(initial={'beam': beam})
                 if request.method=='POST':
                     form = ABSBeamEditForm(request.POST)
                     beam_data = get_values_from_form(request.POST)
                     beam.dict_to_parms(beam_data)
                     beam.save()
                     messages.success(request, 'Beam: "%s" has been updated.'  % beam.name)
                     return redirect('url_edit_abs_conf', conf.id)
                 ###### SIDEBAR ######
                 kwargs = {}
                 kwargs['id_conf'] = conf.id
                 kwargs['form'] = form
                 kwargs['title'] = 'ABS Beams'
                 kwargs['suptitle'] = 'Edit Beam'
                 kwargs['button'] = 'Save'
                 kwargs['no_sidebar'] = True
                 #kwargs['previous'] = conf.get_absolute_url_edit()
                 kwargs['edit'] = True
                 return render(request, 'abs_edit_beam.html', kwargs)
             def remove_beam(request, id_conf, id_beam):
                 conf     = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beam     = get_object_or_404(ABSBeam, pk=id_beam)
                 if request.method=='POST':
                     if beam:
                         try:
                             beam.remove_beamfromlist()
                             beam.delete()
                             messages.success(request, 'Beam: "%s" has been deleted.'  % beam)
                         except:
                             messages.error(request, 'Unable to delete beam: "%s".' % beam)
                     return redirect('url_edit_abs_conf', conf.id)
                 ###### SIDEBAR ######
                 kwargs = {}
                 kwargs['object'] = beam
                 kwargs['delete'] = True
                 kwargs['title'] = 'Delete'
                 kwargs['suptitle'] = 'Beam'
                 kwargs['previous'] = conf.get_absolute_url_edit()
                 return render(request, 'confirm.html', kwargs)
             def plot_patterns(request, id_conf, id_beam=None):
                 kwargs = {}
                 conf = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beams = ABSBeam.objects.filter(abs_conf=conf)
                 if id_beam:
                     beam  = get_object_or_404(ABSBeam, pk=id_beam)
                     kwargs['beam']       = beam
                 ###### SIDEBAR ######
                 kwargs['dev_conf']   = conf.device
                 kwargs['id_dev']     = conf.device
                 kwargs['id_conf']    = conf.id
                 kwargs['abs_beams']  = beams
                 kwargs['title']      = 'ABS Patterns'
                 kwargs['suptitle']   = conf.name
                 kwargs['no_sidebar'] = True
                 return render(request, 'abs_patterns.html', kwargs)
             def plot_pattern(request, id_conf, id_beam, antenna):
                 if antenna=='down':
                     sleep(3)
                 conf = get_object_or_404(ABSConfiguration, pk=id_conf)
                 beam = get_object_or_404(ABSBeam, pk=id_beam)
                 name = conf.experiment.name
                 just_rx = 1 if json.loads(beam.only_rx)[antenna] else 0
                 phases = json.loads(beam.antenna)['antenna_{}'.format(antenna)]
                 gain_tx = json.loads(beam.tx)[antenna]
                 gain_rx = json.loads(beam.rx)[antenna]
                 ues = json.loads(beam.ues)[antenna]
                 newOverJro = overJroShow(name)
                 fig = newOverJro.plotPattern2(datetime.today(), phases, gain_tx, gain_rx, ues, just_rx)
                 response=HttpResponse(content_type='image/png')
                 fig.canvas.print_png(response)
                 return response

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages