U Pe)9@s~dZdZddlZddlZddlZddlZddlmZddlm Z Gddde Z Gdd d Z Gd d d Z Gd d d ZdS)a Gesture recognition =================== This class allows you to easily create new gestures and compare them:: from kivy.gesture import Gesture, GestureDatabase # Create a gesture g = Gesture() g.add_stroke(point_list=[(1,1), (3,4), (2,1)]) g.normalize() # Add it to the database gdb = GestureDatabase() gdb.add_gesture(g) # And for the next gesture, try to find it! g2 = Gesture() # ... gdb.find(g2) .. warning:: You don't really want to do this: it's more of an example of how to construct gestures dynamically. Typically, you would need a lot more points, so it's better to record gestures in a file and reload them to compare later. Look in the examples/gestures directory for an example of how to do that. )GestureGestureDatabase GesturePoint GestureStrokeN)Vector)BytesIOc@s:eZdZdZddZddZddd Zd d Zd d ZdS)rz#Class to handle a gesture database.cCs g|_dSN)dbselfr 0/tmp/pip-unpacked-wheel-xzebddm3/kivy/gesture.py__init__1szGestureDatabase.__init__cCs|j|dS)z"Add a new gesture to the database.N)r append)r gesturer r r add_gesture4szGestureDatabase.add_gesture?TcCsJ|sdSd}|}|jD]"}|||}||kr0q|}|}q|sBdS||fS)z(Find a matching gesture in the database.N)r get_score)r rZminscorerotation_invariantbestZ bestscoregscorer r r find8s  zGestureDatabase.findcCs4t}t|}||tt|d}|S)z'Convert a gesture into a unique string. ) rpicklePicklerdumpbase64 b64encodezlibcompressgetvalue)r riopdatar r r gesture_to_strIs   zGestureDatabase.gesture_to_strcCs*ttt|}t|}|}|S)z%Convert a unique string to a gesture.)rr decompressr b64decoder Unpicklerload)r r$r"r#rr r r str_to_gestureQs zGestureDatabase.str_to_gestureN)rT) __name__ __module__ __qualname____doc__rrrr%r*r r r r r.s  rc@s$eZdZddZddZddZdS)rcCst||_t||_dS)z5Stores the x,y coordinates of a point in the gesture.N)floatxyr r0r1r r r r[s zGesturePoint.__init__cCs |j|9_|j|9_|S)z& Scales the point by the given factor.r0r1)r Zfactorr r r scale`szGesturePoint.scalecCsd|j|jfS)NzMouse_point: %f,%fr3r r r r __repr__fszGesturePoint.__repr__N)r+r,r-rr4r5r r r r rYsrc@s|eZdZdZddZeddZeddZedd Zed d Z d d Z ddZ ddZ dddZ dddZddZdS)rz- Gestures can be made up of multiple strokes.cCst|_t|_dS)z A stroke in the gesture.N)listpoints screenpointsr r r r rmszGestureStroke.__init__cCs&t|jdkrdSt|jdddjS)NrcSs|jSrr0ptr r r wz%GestureStroke.max_x..key)lenr7maxr0r r r r max_xsszGestureStroke.max_xcCs&t|jdkrdSt|jdddjS)NrcSs|jSrr9r:r r r r<}r=z%GestureStroke.min_x..r>)r@r7minr0r r r r min_xyszGestureStroke.min_xcCs&t|jdkrdSt|jdddjS)NrcSs|jSrr1r:r r r r<r=z%GestureStroke.max_y..r>)r@r7rAr1r r r r max_yszGestureStroke.max_ycCs&t|jdkrdSt|jdddjS)NrcSs|jSrrEr:r r r r<r=z%GestureStroke.min_y..r>)r@r7rCr1r r r r min_yszGestureStroke.min_ycCs&|jt|||j||fdS)zQ add_point(x=x_pos, y=y_pos) Adds a point to the stroke. N)r7rrr8r2r r r add_pointszGestureStroke.add_pointcsfdd|jD|_dS)zb scale_stroke(scale_factor=float) Scales the stroke down by scale_factor. csg|]}|qSr )r4.0r; scale_factorr r sz.GestureStroke.scale_stroke..N)r7)r rLr rKr scale_strokeszGestureStroke.scale_strokecCs.|j|j}|j|j}t||||S)z points_distance(point1=GesturePoint, point2=GesturePoint) Returns the distance between two GesturePoints. )r0r1mathsqrt)r Zpoint1Zpoint2r0r1r r r points_distances  zGestureStroke.points_distanceNcCsX|dkr|j}d}t|dkr"|Stt|dD] }||||||d7}q2|S)zlFinds the length of the stroke. If a point list is given, finds the length of that list. N)r7r@rangerQ)r point_listZgesture_lengthir r r stroke_lengths  zGestureStroke.stroke_length cCs4t|jdks||jdkr"dS||jt|}t}||jd|jd}d}|}|jddD]}|||}|dkrl|}||}||krl|j|j} |j|j} |||} | | |j} | | |j} |t | | ||jt|t|}qqlt||ks*t dt||f||_dS)zNormalizes strokes so that every stroke has a standard number of points. Returns True if stroke is normalized, False if it can't be normalized. sample_points controls the resolution of the stroke. rSrRFrNz>Invalid number of strokes points; got %d while it should be %dT) r@r7rWr/r6rrQr0r1r ValueError)r Z sample_pointsZtarget_stroke_sizeZ new_pointsprevZ src_distanceZ dst_distancecurrdZx_dirZy_dirZratioZto_xZto_yr r r normalize_strokes@       zGestureStroke.normalize_strokecCs,|jD] }|j|8_|j|8_qdS)z,Centers the stroke by offsetting the points.N)r7r0r1)r Zoffset_xZoffset_ypointr r r center_strokes zGestureStroke.center_stroke)N)rX)r+r,r-r.rpropertyrBrDrFrGrHrNrQrWr]r_r r r r rjs       .rc@seZdZdZdZd$ddZddZdd Zd%d d Zd&d dZ ddZ ddZ ddZ d'ddZ ddZddZddZddZd d!Zd"d#ZdS)(rzA python implementation of a gesture recognition algorithm by Oleg Dopertchouk: http://www.gamedev.net/reference/articles/article2039.asp Implemented by Jeiel Aranal (chemikhazi@gmail.com), released into the public domain. g?NcCs6d|_d|_d|_t|_|dkr,tj|_n||_dS)zu Gesture([tolerance=float]) Creates a new gesture with an optional matching tolerance value. rRN)widthheightgesture_productr6strokesrDEFAULT_TOLERANCE tolerance)r rfr r r rs zGesture.__init__c Cstdd|jD}tdd|jD}tdd|jD}tdd|jD}||}||_||}||_t||}|dkrdSd|}|jD]}||qd S) z( Scales down the gesture to a unit of 1.cSsg|] }|jqSr )rDrJstroker r r rMsz*Gesture._scale_gesture..cSsg|] }|jqSr )rBrgr r r rMscSsg|] }|jqSr )rGrgr r r rMscSsg|] }|jqSr )rFrgr r r rMsrRF?T)rCrdrArarbrN) r rDrBrGrFZx_lenZy_lenrLrhr r r _scale_gestures   zGesture._scale_gesturecCsd}d}d}|jD]J}tdd|jD}tdd|jD}||7}||7}|t|j7}q|dkrjdS||}||}|jD]}|||qdS)z+ Centers the Gesture.points of the gesture.rRrcSsg|] }|jqSr rErIr r r rMsz+Gesture._center_gesture..cSsg|] }|jqSr r9rIr r r rMsFT)rdsumr7r@r_)r Ztotal_xZtotal_yZ total_pointsrhZstroke_yZstroke_xr r r _center_gestures   zGesture._center_gesturecCs|jtt|ts"t|tr|D]n}t|trH|jdj|q&t|ts\t|trt|dkrpt d|jd |d|dq&t dq&n|dk rt d|jdS) zAdds a stroke to the gesture and returns the Stroke instance. Optional point_list argument is a list of the mouse points for the stroke. z#Stroke entry must have 2 values maxrrSzSThe point list should either be tuples of x and y or a list of GesturePoint objectsNz!point_list should be a tuple/list) rdrr isinstancer6tuplerr7r@rYrH TypeError)r rUr^r r r add_stroke*s   zGesture.add_strokerXcCs@|r|sd|_dS|jD]}||q |||_dS)zcRuns the gesture normalization algorithm and calculates the dot product with self. FN)rjrlrcrdr] dot_product)r Zstroke_samplesrhr r r normalize@s   zGesture.normalizecCst|jdks"t|jdjdkr&dSt|jdksHt|jdjdkrLdS|jdjd}t|j|jg}t|j|jg}||S)a" Extract the rotation to apply to a group of points to minimize the distance to a second group of points. The two groups of points are assumed to be centered. This is a simple version that just picks an angle based on the first point of the gesture. rSr)r@rdr7rr0r1angle)r Zdstptsr#targetsourcer r r get_rigid_rotationKs""zGesture.get_rigid_rotationc Cst|jt|jkrdSt|dddks8t|dddkr}g}|jD]$}t|j|jg|}||q||q | ||_ |Sr) rrdr7rr0r1rotaterrrrsrc)r rurrhtmpjvr r r r}ks     zGesture.rotateTcCsVt|trR|r"||}||}||}|dkr8|S|t|j|j}|SdS)zL Returns the matching score of the gesture against another gesture. rN)rorrxr}rsrOrPrc)r r|rrurr r r rvs     zGesture.get_scorecCsBt|tr:||}|d|jkr4|d|jkr4dSdSntSdS)z3 Allows easy comparisons between gesture instances.riTFN)rorrrfNotImplemented)r r|rr r r __eq__s   zGesture.__eq__cCs ||}|tkr|S| SdSr)rr)r r|resultr r r __ne__s zGesture.__ne__cCs tddS)Nz"Gesture cannot be evaluated with <rqr r|r r r __lt__szGesture.__lt__cCs tddS)Nz"Gesture cannot be evaluated with >rrr r r __gt__szGesture.__gt__cCs tddS)Nz#Gesture cannot be evaluated with <=rrr r r __le__szGesture.__le__cCs tddS)Nz#Gesture cannot be evaluated with >=rrr r r __ge__szGesture.__ge__)N)N)rX)T)r+r,r-r.rerrjrlrrrtrxrsr}rrrrrrrr r r r rs"    r)r.__all__rrrrOZ kivy.vectorrr"robjectrrrrr r r r s!  +|