##// END OF EJS Templates
ABS » abscontrol
RSS
imanay -
r248:249
parent child
Show More
Show file before | Show file after | Hide comments
@@ -1,482 +1,503
1 1 import os
2 2 import library3
3 3 import time
4 4 from threading import Thread
5 5 import threading
6 6 import Queue
7 7
8 8 class ABSServer:
9 9
10 10 def __init__(self,ipSource="localhost", ipDestino="192.168.1.117", portDestino=7000, ipDestino2="192.168.1.11", portDestino2=5500, rx_buffer = "default"):
11 11
12 12
13 13 self.ipSource = ipSource
14 14 self.ipDestino = ipDestino
15 15 self.portDestino = portDestino
16 16
17 17 self.ipDestino2 = ipDestino2
18 18 self.portDestino2 = portDestino2
19 19
20 20 self.tx_buffer = "default"
21 21 self.rx_buffer = rx_buffer
22 22 self.enaModules = []
23 23 self.bits = []
24 24 self.phase = []
25 25 self.txFile = []
26 26 self.rxFile = []
27 27
28 28 self.createObjects()
29 29
30 30 print "Checking control modules, please wait ..."
31 31 self.enaModules = self.checkAntenna()
32 32 print '\nThis modules are present : ' + str(self.enaModules) + '\n'
33 33 # print "Starting automatic control module status."
34 34 self.__StartingAutomaticControlModules()
35 35 # self.__AutomaticControlModules()
36 36
37 37 def createObjects(self):
38 38
39 39 asServer = True
40 40 self.commServerObj = library3.TCPComm("Central_Control", "CeCnMod", self.ipDestino, "CnMod01", self.portDestino, asServer)
41 41 self.commClientObj = library3.TCPComm("Central_Control", "CeCnMod", self.ipDestino2, "CnMod01", self.portDestino2)
42 42 self.wFiles = library3.FilesStuff()
43 43
44 44 def waitRequest(self):
45 45
46 46 #Using rx buffer
47 47 # ipSource, ipDestino, cmd, self.rx_buffer = self.commServerObj.waitData()
48 48 ipSource, ipDestino, cmd, rx_buffer = self.commServerObj.waitData()
49 49
50 50 if cmd == "SNDF":
51 51 datarpta = self.__sendFile2Modules(cmd = cmd, rx_buffer = rx_buffer)
52 52
53 53 if cmd == "GETF":
54 54 datarpta = self.__getFileFromModules(cmd = cmd, rx_buffer = rx_buffer)
55 55
56 56 if cmd == "CHGB":
57 57 datarpta = self.__changeBeam(cmd = cmd, rx_buffer = rx_buffer)
58 58
59 59 if cmd == "ANST":
60 60 datarpta = self.__getControlModuleStatus_old(cmd = cmd, rx_buffer = rx_buffer)
61 61
62 62 if cmd == "BGPH":
63 63 datarpta = self.__getControlModuleBigPhase(cmd = cmd, rx_buffer = rx_buffer)
64 64
65 65 if cmd == "LWPH":
66 66 datarpta = self.__getControlModuleLowPhase(cmd = cmd, rx_buffer = rx_buffer)
67 67
68 68 if cmd == "NTST":
69 69 datarpta = self.__getConnectionStatus(cmd = cmd, rx_buffer = rx_buffer)
70 70
71 71 if cmd == "MNTR":
72 72 datarpta = self.__getControlModuleStatus_new(cmd = cmd, rx_buffer = rx_buffer)
73 73
74 74 self.commServerObj.sendData(cmd=cmd, data=datarpta, ipDestino = ipSource)
75 75
76 76 def __checkModule(self, address):
77 77
78 78 cmd = "ping -c 1 -w 1 192.168.1."+ str(address) + " >> /dev/null"
79 79 status = os.system(cmd)
80 80
81 81 if status == 256:
82 82 return False
83 83
84 84 return True
85 85
86 86 def __writeReport(self, enaModules):
87 87
88 88 status_array = ["Status of modules\n"]
89 89 status_array.append("----------------\n")
90 90
91 91 for address in range(1,65):
92 92 if address in enaModules:
93 93 status_array.append("192.168.1." + str(address) + " [1 1]\n")
94 94 else:
95 95 status_array.append("192.168.1." + str(address) + " [0 0]\n")
96 96
97 97 filename = "module_status.txt"
98 98 self.__writeFile(filename,status_array)
99 99 # f = open("module_status.txt","w")
100 100 # f.writelines(status_array)
101 101 # f.close()
102 102
103 103
104 104 def __CreateFile(self, filename):
105 105
106 106 fobj = open(filename,"w")
107 107 fobj.close()
108 108
109 109 def __writeNewFile(self, filename, data):
110 110
111 111 fobj = open(filename,"w")
112 112 fobj.writelines(data)
113 113 fobj.close()
114 114
115 115 def __writeFile(self, filename, data):
116 116
117 117 fobj = open(filename,"a")
118 118 fobj.writelines(data)
119 119 fobj.close()
120 120
121 121 def __checkAntenna(self):
122 122
123 123 """
124 124 Direccion de los modulos de las antenas:
125 125
126 126 Norte : 01-16
127 127 Este : 17-32
128 128 Oeste: : 33-48
129 129 Sur : 49-64
130 130
131 131 """
132 132
133 133 enaModules2 = []
134 134
135 135 for address in range(1,65):
136 136 if self.checkModule(address):
137 137 enaModules2.append(address)
138 138
139 139 self.__writeReport(enaModules2)
140 140 return enaModules2
141 141
142 142 def checkModule(self, arg, queue):
143 143 cmd = "ping -c 1 -w 1 192.168.1."+ str(arg) + " >> /dev/null"
144 144 status = os.system(cmd)
145 145 if status == 256:
146 146 result = "failed"
147 147 else:
148 148 result = "ok"
149 149 queue.put({arg: result})
150 150
151 151 def checkAntenna(self):
152 152
153 153 iD = range(1,65)
154 154 q = Queue.Queue()
155 155 threads = []
156 156 tOut = []
157 157 modules = []
158 158
159 159 for argument in iD:
160 160 t = Thread(target=self.checkModule, args=(argument, q))
161 161 t.start()
162 162 threads.append(t)
163 163
164 164 for t in threads:
165 165 t.join()
166 166
167 167 for _ in range(len(iD)):
168 168 tOut.append(q.get())
169 169
170 170 for i in tOut:
171 171 if i.values()[0] == 'ok':
172 172 modules.append(i.keys()[0])
173 173
174 174 return modules
175 175
176 176 def __ConnectionWithControlModules(self,data,cmd,id):
177 177
178 178 self.commClientObj.open_socket()
179 179 ip = "192.168.1." + str(id)
180 180 self.commClientObj.sendData(cmd, data, ip)
181 181 ipSource, ipDestino, cmd, tmp = self.commClientObj.waitData()
182 182 self.commClientObj.close_socket()
183 183
184 184 return tmp
185 185
186 186 def abs2ControlModuleFormatFile(self, filename):
187 187
188 188 #From matriz to control module format
189 189 self.wFiles.toCentralControlFormat(filename)
190 190 FileName = "CentralControlFormat.txt"
191 191 F_Obj = open(FileName,"r")
192 192 FileList = F_Obj.readlines()
193 193 F_Obj.close()
194 194 FileStr = "".join(FileList)
195 195
196 196 return FileStr
197 197
198 198 def __All2Blocks(self,input):
199 199
200 200 rx_frame_lst = input.split('\n',2)
201 201
202 202 header = rx_frame_lst[0] + "\n"
203 203 control_modules_str = rx_frame_lst[2]
204 204 control_modules_lst = control_modules_str.split("------\n")
205 205
206 206 return header, control_modules_lst
207 207
208 208
209 209 def __sendFile2Modules(self,cmd, rx_buffer):
210 210
211 211 # rx_buffer_lst = self.rx_buffer.split('\n',1)
212 212 rx_buffer_lst = rx_buffer.split('\n',1)
213 213 #Getting the filename from the begining of data
214 214 filename = rx_buffer_lst[0]
215 215 tmp = rx_buffer_lst[1]
216 216 self.__writeFile(filename,tmp)
217 217 data = self.abs2ControlModuleFormatFile(filename)
218 218 #Needed for the loop
219 219 header, control_modules_lst = self.__All2Blocks(data)
220 220 correct = 0
221 221
222 222 for id in range(1,65):
223 223
224 224 if id not in self.enaModules:
225 225 continue
226 226
227 227 if self.__ConnectionWithControlModules(header + control_modules_lst[id-1], cmd, id) == "OK":
228 228 correct = correct + 1
229 229
230 230 if correct == len(self.enaModules):
231 231 rpta = "OK"
232 232 else:
233 233 rpta = "Failure"
234 234
235 235 return rpta
236 236
237 237 def __getFileFromModules(self, cmd, rx_buffer):
238 238
239 239 for id in range(1,65):
240 240 if id not in self.enaModules:
241 241 continue
242 242
243 243 file = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
244 244 del self.rxFile[id-1]
245 245 self.rxFile.insert(id-1, file)
246 246
247 247 return self.rxFile
248 248
249 249 def __changeBeam(self, cmd, rx_buffer):
250 250
251 251 correct = 0
252 252 # enaModules = self.checkAntenna()
253 253 # enaModules = [11,12,13,14]
254 254
255 255 for id in range(1,65):
256 256 if id not in self.enaModules:
257 257 continue
258 258
259 259 if self.__ConnectionWithControlModules(rx_buffer,cmd,id) == "OK":
260 260 correct = correct + 1
261 261
262 262 if correct == len(self.enaModules):
263 263 rpta = "OK"
264 264 else:
265 265 rpta = "Failure"
266 266
267 267 return rpta
268 268
269 269 def __getControlModuleStatus(self, cmd, rx_buffer):
270 270
271 271 # all_blocks = ""
272 272 # all_blocks = []
273 273 # enaModules = self.checkAntenna()
274 274 # enaModules = [11,12,13,14]
275 275
276 276 for id in range(1,65):
277 277 if id not in self.enaModules:
278 278 continue
279 279
280 280 bits = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
281 281 del self.bits[id-1]
282 282 self.bits.insert(id-1, bits)
283 283
284 284 # all_blocks.append(one_block)
285 285 #Using tx buffer
286 286
287 287 return self.bits
288 288
289 289 def __getControlModuleStatus_new(self, cmd, rx_buffer):
290 290
291 291 all_mnt = []
292 292 for id in range(1,65):
293 293 if id not in self.enaModules:
294 294 continue
295 295
296 296 mnt = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
297 297 all_mnt.append(mnt)
298 298
299 299 __function1(type = rx_buffer, in_lst = all_mnt)
300 300 #return "".join(all_mnt)
301 301
302 302 def __function1(self, type, in_lst):
303 303
304 304 if type == "0":
305 305 out_str = "".join(in_lst)
306 306 elif type == "1":
307 307 out_str = "".join(in_lst)
308 308 elif type == "2":
309 309 out_str = "".join(in_lst)
310 310 elif type == "3":
311 311 TwoComparation(in_lst = in_lst)
312 312 else:
313 313 ThreeComparation(in_lst = in_lst)
314 314
315 315 return out_str
316 316
317 317 def __TwoComparation(self, in_lst):
318 318 cm_up =[]
319 319 cm_dw =[]
320 320 relay_up =[]
321 321 relay_dw =[]
322 322 for i in in_lst:
323 323 relay_up.append(i[14:17])
324 324 relay_dw.append(i[17:20])
325 cm_up.append(i[21:24])
326 cm_dw.append(i[24:27])
327
328 comp_up = []
329 comp_dw = []
330
331 for i in range(len(relay_up)):
332 if cmp(relay_up[i], cm_up[i]) == 0:
333 comp_up.append(0)
334 else:
335 comp_up.append(1)
336
337 for i in range(len(relay_dw)):
338 if cmp(relay_dw[i], cm_dw[i]) == 0:
339 comp_dw.append(0)
340 else:
341 comp_dw.append(1)
342
343 print comp_up
344 print comp_dw
345
325 346
326 347
327 348 def __ThreeComparation(self, in_lst):
328 349 pass
329 350
330 351 def __getControlModuleBigPhase(self, cmd, rx_buffer):
331 352
332 353 # all_blocks = ""
333 354 all_blocks = []
334 355 # enaModules = self.checkAntenna()
335 356 # enaModules = [11,12,13,14]
336 357
337 358 for id in range(1,65):
338 359 if id not in self.enaModules:
339 360 continue
340 361
341 362 one_block = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
342 363
343 364 # all_blocks = all_blocks + one_block
344 365 all_blocks.append(one_block)
345 366 #Using tx buffer
346 367 return all_blocks
347 368
348 369 def __getControlModuleLowPhase(self, cmd, rx_buffer):
349 370
350 371 # all_blocks = ""
351 372 # all_blocks = []
352 373 # enaModules = self.checkAntenna()
353 374 # enaModules = [11,12,13,14]
354 375
355 376 for id in range(1,65):
356 377 if id not in self.enaModules:
357 378 continue
358 379
359 380 phase = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
360 381 del self.phase[id-1]
361 382 self.phase.insert(id-1, phase)
362 383 # all_blocks = all_blocks + one_block
363 384 # all_blocks.append(one_block)
364 385 #Using tx buffer
365 386 # return all_blocks
366 387
367 388 def __getConnectionStatus(self, cmd, rx_buffer):
368 389
369 390 ena = self.checkAntenna()
370 391 print ena
371 392 self.enaModules = ena
372 393
373 394 blockLst = []
374 395
375 396 for id in range(1,65):
376 397 if id not in self.enaModules:
377 398 continue
378 399
379 400 blockStr = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
380 401 blockLst.append(blockStr + ", 192.168.1." + str(id) + "\n")
381 402 #Using tx buffer
382 403 all_blocks = "".join(blockLst)
383 404
384 405 return all_blocks
385 406
386 407 def getConnectionStatus(self, cmd):
387 408
388 409 ena = self.checkAntenna()
389 410 self.enaModules = ena
390 411
391 412 blockLst = []
392 413
393 414 for id in range(1,65):
394 415 if id not in self.enaModules:
395 416 continue
396 417
397 418 blockStr = self.__ConnectionWithControlModules(self.rx_buffer,cmd,id)
398 419 blockLst.append(blockStr + ", 192.168.1." + str(id) + "\n")
399 420 #Using tx buffer
400 421 self.tx_buffer = "".join(blockLst)
401 422 print self.tx_buffer
402 423
403 424 return self.tx_buffer
404 425
405 426 def __getControlModuleStatus_old(self, cmd, rx_buffer):
406 427
407 428 all_blocks = ""
408 429 # enaModules = self.checkAntenna()
409 430 # enaModules = [11,12,13,14]
410 431
411 432 for id in range(1,65):
412 433 if id not in self.enaModules:
413 434 continue
414 435
415 436 one_block = self.__ConnectionWithControlModules(rx_buffer,cmd,id)
416 437
417 438 all_blocks = all_blocks + one_block
418 439 #Using tx buffer
419 440 print all_blocks
420 441 self.tx_buffer = all_blocks
421 442
422 443 return all_blocks
423 444
424 445 def __StartingAutomaticControlModules(self):
425 446
426 447 #Starting file
427 448 self.__CreateFile("Monitoring.txt")
428 449 # data = "MOD.\t BITS\t\t PHASE\n"
429 450 # self.__writeFile("Monitoring.txt", data)
430 451 #Starting lists
431 452 self.txFile = list("------\n------\n------\n" for i in range(64))
432 453 self.rxFile = list("------\n------\n------\n" for i in range(64))
433 454 self.bits = list("------\n------\n------\n" for i in range(64))
434 455 self.phase = list("----- -----\n----- -----\n----- -----\n" for i in range(64))
435 456
436 457 def __AutomaticControlModules(self):
437 458
438 459 # cmd = "GETF"
439 460 # rx_buffer = "experimento1.ab1" + "\n"
440 461 # self.__getFileFromModules(cmd = cmd, rx_buffer = rx_buffer)
441 462 #
442 463 # print self.rxFile
443 464
444 465 cmd = "ANST"
445 466 rx_buffer = "1"
446 467 self.__getControlModuleStatus(cmd = cmd, rx_buffer = rx_buffer)
447 468
448 469 cmd = "LWPH"
449 470 rx_buffer = "0"
450 471 self.__getControlModuleLowPhase(cmd = cmd, rx_buffer = rx_buffer)
451 472 print "Saving file..."
452 473
453 474
454 475 print self.bits
455 476 print self.phase
456 477
457 478 self.__WritingMonitoringFile()
458 479
459 480 threading.Timer(60, self.__AutomaticControlModules).start()
460 481
461 482 def __WritingMonitoringFile(self):
462 483 filename = "Monitoring.txt"
463 484 data = '===============================' + '\n'
464 485 self.__writeFile(filename, data)
465 486 data = time.strftime('\t' + "%d%b%Y %I:%M:%S %p" + '\n', time.localtime())
466 487 self.__writeFile(filename, data)
467 488 data = "MOD.\t BITS\t\t PHASE\n"
468 489 self.__writeFile(filename, data)
469 490 data = '===============================' + '\n'
470 491 self.__writeFile(filename, data)
471 492 for i in range(64):
472 493 tmp = self.bits[i].split('\n',3)
473 494 self.__writeFile(filename, ' ' + str(i + 1) + '\t\t' + tmp[2])
474 495 tmp = self.phase[i].split('\n',3)
475 496 self.__writeFile(filename, '\t\t' + tmp[2] + '\n')
476 497
477 498 if __name__ == '__main__':
478 499
479 500 absObj = ABSServer()
480 501
481 502 while 1:
482 503 absObj.waitRequest() No newline at end of file
General Comments 0
You need to be logged in to leave comments. Login now