B :^ @sdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z GdddZ ddZ ddd Z ed kre Zejre dd ejre d ejd dS)z .. module:: shutter :platform: unix :synopsis: This module is for communicating with the shutter instrument. .. codeauthor:: Michael Coughlin, Eric Coughlin Nc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)Shutterz? This is the class for communicating with the shutter cCsd|_||_d|_dS)N)statuscreate_connectionshutterstate)selfr/home/cbp/cbp/cbp/shutter.py__init__s zShutter.__init__c CsNyd}t|}d|_|Stk rH}zt|d|_Wdd}~XYnXdS)z^ This method creates a connection by using a child process. :return: z%picocom -b 57600 /dev/ttyACM.SHUTTER2 connectedz not connectedN)pexpectspawnr Exceptionprint)rshutter_commandchilderrr rs zShutter.create_connectionc CsHyd}t|}Wn0tk rB}zt|d|_Wdd}~XYnXdS)zP This method checks the status of the shutter :return: z%picocom -b 57600 /dev/ttyACM.SHUTTER2z not connectedN)r r rrr)rrrrrrr check_status*s zShutter.check_statuscCs|jdkr|jdS)zB This method closes the shutter :return: z not connectedN)rrclose)rrrr close_connection7s zShutter.close_connectioncCs`|jdkr\d}xL|sX|jjtjdgdd}|dkrLd|}|j|d}|d krqqWnd S) z This method is a duration run of the shutter, -1 can be used to keep shutter open indefinitely or 0 to close it. :param shutter: This is the length that the shutter will stay open in milliseconds. :return: z not connectedF )timeoutrz args {0:d} TN)rrexpectr ZTIMEOUTformatZsendline)rrdoneiZ argstringrrr run_shutter@s   zShutter.run_shuttercCs|jdkrd}t|ndS)zR This method compiles the shutter arduino code. :return: z not connectedz/cd /home/pi/Code/arduino/shutter/; ./compile.shN)rossystem)rZ steps_commandrrr do_compileVs  zShutter.do_compileN) __name__ __module__ __qualname____doc__r rrrrr!rrrr rs  rcCsLt}|jdddtd|jddddd |jd ddd |\}}|S) z6 Parse the options given on the command-line. z-sz --shutter)defaulttypez-cz --doCompile store_trueF)actionr'z --doShutter)optparse OptionParser add_optionint parse_args)parseroptsargsrrr parse_commandlinecs  r3compilecCs.t}|dkr|n|dkr*||dS)Nr4r)rr!r)runtypevalrrrr mainrs  r7__main__)r5r)r5r6)r4r)r%serialsystimeglobstructrnumpynpr+r rr3r7r"r1Z doCompileZ doShutterrrrrr  s0R