a Df@sddlZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZgdZe e e efZdd Zd d Zd d ZddZddZddZddZddZ ddZ!ddZ"ddZ#ddZ$d d!Z%dS)"N)collectdshape)remove) DataShapeFixedVarEllipsisRecordUnitdate_ datetime_TypeVarto_numpy_dtype Categorical)floatingboolean) isdimension ishomogeneous istabularisfixedisscalarisrecord iscollection isnumeric isboolean isdatelikeisrealcCsFt|trt|}t|tr0t|dkr0|d}tt|d|ttfS)z Is this dshape a single dtype? >>> isscalar('int') True >>> isscalar('?int') True >>> isscalar('{name: string, amount: int}') False rrty) isinstancestrrrlengetattrr rdsr%l/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/datashader/datashape/predicates.pyrs rcCsBt|trt|}t|tr0t|dkr0|d}tt|d|tS)z Is this dshape a record type? >>> isrecord('{name: string, amount: int}') True >>> isrecord('int') False >>> isrecord('?{name: string, amount: int}') True rrr)rr rrr!r"r r#r%r%r&r$s rcCs t|tS)z Is a component a dimension? >>> from datashader.datashape import int32 >>> isdimension(Fixed(10)) True >>> isdimension(Var()) True >>> isdimension(int32) False )rdimension_typesr#r%r%r&r5s rcCs$t|}tttttt|dkS)z Does datashape contain only one dtype? >>> from datashader.datashape import int32 >>> ishomogeneous(int32) True >>> ishomogeneous('var * 3 * string') True >>> ishomogeneous('var * {name: string, amount: int}') False r)rr!setrrrrr#r%r%r&rCs rcCstt|jS)z&Number of dimensions of datashape )r!rshaper#r%r%r& _dimensionsRsr*cCslt|}t|dtrdSt|dtr,dSt|dtrNttt|djSt |dkrht| dSdS)z Contains no variable dimensions >>> isfixed('10 * int') True >>> isfixed('var * int') False >>> isfixed('10 * {name: string, amount: int}') True >>> isfixed('10 * {name: string, amounts: var * int}') False rNFrT) rrrrr allmaprtypesr!Zsubarrayr#r%r%r&rXs  rcCst|}t|dkot|jS)z A collection of records >>> istabular('var * {name: string, amount: int}') True >>> istabular('var * 10 * 3 * int') False >>> istabular('10 * var * int') False >>> istabular('var * (int64, string, ?float64)') False r)rr*rmeasurer#r%r%r&rps rcCst|trt|}t|dS)z Is a collection of items, has dimension >>> iscollection('5 * int32') True >>> iscollection('int32') False r)rr rrr#r%r%r&rs rcCsFt|}z t|}Wnty(YdS0t|to@t|tjSdS)z Has a numeric measure >>> isnumeric('int32') True >>> isnumeric('3 * ?real') True >>> isnumeric('string') False >>> isnumeric('var * {amount: ?int32}') False FN)launderr TypeErrorrr npZ issubdtypenumber)r$Znpdtyper%r%r&rs   rcCs.t|trt|}t|tr"|j}t|d|S)Nr)rr rrr.r"r#r%r%r&r/s   r/cCst|}t|to|tvS)z Has a numeric measure >>> isreal('float32') True >>> isreal('3 * ?real') True >>> isreal('string') False )r/rr rr#r%r%r&rs rcCs t|tvS)z Has a boolean measure >>> isboolean('bool') True >>> isboolean('3 * ?bool') True >>> isboolean('int') False )r/rr#r%r%r&rs rcCst|}|tkp|tkS)z Has a date or datetime measure >>> isdatelike('int32') False >>> isdatelike('3 * datetime') True >>> isdatelike('?datetime') True )r/r r r#r%r%r&rs r)&numpyr1utilrrZinternal_utilsrZ coretypesrrrr r r r r rrrZtypesetsrr__all__intr'rrrrr*rrrrr/rrrr%r%r%r&s& 4