B ^K?@sdZddlZddlZddlZddlZddlZddlmZddlm Z dZ dZ dZ ddd d d gd d ddddddgddddddddddddd d!gd"d#d$gd%d&d'd(d)d*gd+d,d-d.gd/d0d1d2d3d4d5d6d7d8g d9Z d:d;Zdd?ZGd@dAdAeZdS)Bz- pyrow.py Interface to concept2 indoor rower N)USBError) csafe_cmdig?zWait for min speedzWait for accelerationZDriveZDwellingZRecoveryzJust Row / no splitszJust Row / splitszFixed Distance / splitszFixed Distance / no splitszFixed Time / no splitszFixed Time IntervalzFixed Distance IntervalzVariable Intervalz Waiting beginz Workout rowzCountdown pausez Interval restzWork time invervalzWork distance intervalz Rest end timezRest end distancez Time end restzDistance end restz Workout endzWorkout terminatezWorkout loggedz Workout rearmTimeDistanceZRestZStandardzForce/VelocityZPaceboatz Per StrokeZSimpleZTargetz Time/MetersZPaceZWattsZCaloriesErrorZReadyZIdlezHave IDzN/AzIn UseZPauseZFinishedZManualZOffline) strokestateZ workouttypeZ workoutstateinttypeZ displaytypeZdisplayunitstypestatuscCs8t|tst|||kr&|ks4nt|ddS)zH Checks that value is an integer and within the specified range z outside of rangeT) isinstanceint TypeError ValueError)valueZlabelZminimumZmaximumr/var/underergo/pyrow/pyrow.py checkvaluegs   rc CsP|rLxF|D]:}|tkryt|||||<Wqtk rFYqXqW|S)z' Makes data_dict values pretty )keys ERG_MAPPING IndexError) data_dictprettykeyrrr get_prettyqs rc Cs^ytjjdtd}Wn4tk rH}z|jdkr8tdWdd}~XYnX|dkrZtd|S)z7 Returns list of pyusb Devices which are ergs. T)Zfind_allZidVendorzUSB busyNzErgs not found)usbZcorefind C2_VENDOR_IDrerrnoConnectionRefusedErrorr )Zergserrrrs rc@speZdZdZddZeddZdddZdd d Zdd d Z dd dZ dddZ ddZ dddZ ddZdS)PyErgz- Manages low-level erg communication c Csddlm}tjdkrPy*|tr.|tn|dtjWnYnXtj |ty | Wn0t k r}z|d|Wdd}~XYnX||_ |d}|d}|dj|_|dj|_tj|_dS) z> Configures usb connection and sets erg value r)warnwin32z"DEBUG: usb kernel driver not on {}z1DEBUG: usb error whilst setting configuration, {}N)rr)warningsr!sysplatformZis_kernel_driver_active INTERFACEZdetach_kernel_driverformatrutilZclaim_interfaceZset_configurationrergZbEndpointAddress inEndpoint outEndpointdatetimenow_PyErg__lastsend)selfr*r!rZ configurationZifacerrr__init__s&        zPyErg.__init__cOs t||S)N)r)argskwargsrrr _checkvalueszPyErg._checkvalueFc Csdddddddg}|r&|dd d g||}d d d ddddddddg }||}ddg}||}i}||d<||d<||d<|dd|ddd|d<|dd|ddd|d <|dd|d!<|dd|d"<|d"r"d#|ddd$d%|d&<|ddd'd(|d)<nd*\|d&<|d)<|dd|d+<|dd|d,<|r|ddd-}|dd|d|d.<|d d|d/<|ddd0@|d1<t||}|S)2a Returns values from the monitor that relate to the current workout, optionally returns force plot data and stroke state. (* required) time: time in seconds distance: distance in meters spm: strokes per minute power: power in watts pace: seconds/500m calhr: calories burned per hour calories: calories burned heartrate: heartrate status if heartrate: forceplot: force plot data strokestate ZCSAFE_PM_GET_WORKTIMEZCSAFE_PM_GET_WORKDISTANCEZCSAFE_GETCADENCE_CMDZCSAFE_GETPOWER_CMDZCSAFE_GETCALORIES_CMDZCSAFE_GETHRCUR_CMDZCSAFE_PM_GET_DRAGFACTORCSAFE_PM_GET_FORCEPLOTDATA CSAFE_PM_GET_STROKESTATEZCSAFE_GETODOMETER_CMDZCSAFE_PM_GET_RESTTIMEZCSAFE_GETPACE_CMDZCSAFE_GETUSERINFO_CMDZCSAFE_GETHORIZONTAL_CMDZCSAFE_GETPROGRAM_CMDCSAFE_GETSERIAL_CMDZCSAFE_GETUNITS_CMDCSAFE_GETID_CMDCSAFE_GETVERSION_CMDCSAFE_GETSTATUS_CMDZCSAFE_PM_GET_STROKESTATSrresultsresults2results3rr#gY@timeg$@distanceZspmZpowergffffff@gUUUUUU?ipacegOe @gr@Zcalhr)rrZcaloriesZ heartrate forceplotrr )extendsendr) r0rCrcommandr<r=r>Zmonitor datapointsrrr get_monitorsF        zPyErg.get_monitorcCsrdddg}||}i}|ddd}|dd|d|d<|dd|d<|d dd @|d <t||}|S) z: Returns force plot data and stroke state r5r6r7rrBr#rCrr;rDr )rFr)r0rrGr<rCrHrrr get_forceplot s   zPyErg.get_forceplotcCsdddddg}||}i}|dd|d<|dd|d<|dd|d <|dd|d <|dd|d <|d dd @|d<t||}|S)z. Returns overall workout data r9ZCSAFE_PM_GET_WORKOUTTYPEZCSAFE_PM_GET_WORKOUTSTATEZCSAFE_PM_GET_INTERVALTYPEZ!CSAFE_PM_GET_WORKOUTINTERVALCOUNTrZuseridtypestaterZintcountr;rDr )rFr)r0rrGr<Z workoutdatarrr get_workouts  zPyErg.get_workoutcCsddddg}||}i}|dd|d<|dd|d<|dd|d <|dd |d <|dd |d <|dd|d<|dd|d<|dd|d<|dd|d<|ddd@|d<t||}|S)zI Returns all erg data that is not related to the workout r:r8ZCSAFE_GETCAPS_CMDrZmfgidr#ZcidrBZmodelZ hwversionZ swversionserialZmaxrxZmaxtxZ mininterframer;rDr )rFr)r0rrGr<Zergdatarrrget_erg1s   z PyErg.get_ergcCs6dg}||}i}|ddd@|d<t||}|S)z/ Returns the status of the erg r;rrDr )rFr)r0rrGr<r rrr get_statusLs   zPyErg.get_statuscCsFtj}d|j|j|jg}|d|jd|j|jg| |dS)zK Sets the erg clock to the computers current time and date ZCSAFE_SETTIME_CMDZCSAFE_SETDATE_CMDilN) r-r.hourminutesecondrEyearmonthdayrF)r0r.rGrrr set_clockYs zPyErg.set_clockNc Cs||dgg}|dkr*||dddn|dkrt|dkrJ|ddt|dkrb|dd||dddd ||dd dd ||dd dd |ddkr|ddkr|dd krtd|d|d|d|dgn*|dkr||ddd|d|dg|dk r|dk r|dkrt|d}|dd|dd|d} t| ddd} ||dtd| | d|dd|gnT|dk r|dkrt|dd} ||dtd| ||dd|gntd|dk rttd|d d!}n|dk r4tt|d"d#}|dk rN|d$|d%g|dkr\d}|d&|dd'g||dS)(a If machine is in the ready state, function will set the workout and display the start workout screen Choose one of: program: workout program 0 to 15 workout_time: workout time as a list, [hours, minutes, seconds] distance: meters If workout_time or distance, optional: split One of the following for pace boat (optional): pace: seconds calpace: calories per hour powerpace: watts ZCSAFE_RESET_CMDNZProgramrrDr#rBz Time Hours z Time Minutes;z Time SecondszWorkout too shortZCSAFE_SETTWORK_CMDrdiPZCSAFE_SETHORIZONTAL_CMD$i<g?z Split TimeiZCSAFE_PM_SET_SPLITDURATIONzSplit distancez!Cannot set split for current goalgffffff@g@@rNgr@gOe @ZCSAFE_SETPOWER_CMDXZCSAFE_SETPROGRAM_CMDZCSAFE_GOINUSE_CMD) rFr4leninsertr rEr maxround) r0ZprogramZ workout_timer@splitrAZcalpaceZ powerpacerGZtime_rawZminsplitrrr set_workoutdsR     $         zPyErg.set_workoutc Cstj}||j}|j|jd}|tkr:tt|t |}y|j j |j |dd}Wn4t k r}z|j dkrtdWdd}~XYnXtj|_g}xn|sy"|j j|j|dd} t| }Wqtk r }z t| t|t||Wdd}~XYqXqW|S)zb Converts and sends message to erg; receives, converts, and returns ergs response g.Ai)timeoutzUSB device disconectedN)r-r.r/seconds microseconds MIN_FRAME_GAPr?sleeprwriter*r,rrConnectionErrorreadr+ Exceptionprint) r0messager.deltaZdeltarawZcsafelengthrZresponseZ transmissionrrrrFs.     z PyErg.send)FF)F)F)F)F)NNNNNNN)__name__ __module__ __qualname____doc__r1 staticmethodr4rIrJrMrQrRrYrhrFrrrrr s&  K      Kr )rzr-r?r%Zusb.corerZusb.utilrZ pyrow.csaferrrmr'rrrrobjectr rrrr s