|LeddlZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZgdZe e e efZdZd Zd Zd Zd Zd ZdZdZ dZ!dZ"dZ#dZ$dZ%dS)N)collectdshape)remove) DataShapeFixedVarEllipsisRecordUnitdate_ datetime_TypeVarto_numpy_dtype Categorical)floatingboolean) isdimension ishomogeneous istabularisfixedisscalarisrecord iscollection isnumeric isboolean isdatelikeisrealct|trt|}t|trt |dkr|d}tt |d|t tfS)z Is this dshape a single dtype? >>> isscalar('int') True >>> isscalar('?int') True >>> isscalar('{name: string, amount: int}') False rrty) isinstancestrrrlengetattrr rdss ?lib/python3.11/site-packages/datashader/datashape/predicates.pyrrsk"c BZZ"i  SWW\\ U gb$++dK-@ A AAct|trt|}t|trt |dkr|d}tt |d|t S)z Is this dshape a record type? >>> isrecord('{name: string, amount: int}') True >>> isrecord('int') False >>> isrecord('?{name: string, amount: int}') True rrr )r!r"rrr#r$r r%s r'rr$sf"c BZZ"i  SWW\\ U gb$++V 4 44r(c,t|tS)z Is a component a dimension? >>> from datashader.datashape import int32 >>> isdimension(Fixed(10)) True >>> isdimension(Var()) True >>> isdimension(int32) False )r!dimension_typesr%s r'rr5s b/ * **r(c t|}ttttt t |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%s r'rrCs> B s6+wx'<'<==>> ? ?1 DDr(cDtt|jS)z&Number of dimensions of datashape )r#rshaper%s r' _dimensionsr0Rs vbzz  r(ct|}t|dtrdSt|dtrdSt|dtr-t t t|djSt|dkr"t| 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) rr!rr r allmaprtypesr#subarrayr%s r'rrXs B"Q%!!t"Q%u"Q%  .3w1 ,,--- 2ww{{r{{1~~&&& 4r(cnt|}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)rr0rmeasurer%s r'rrps/ B r??a  8HRZ$8$88r(ctt|trt|}t|dS)z Is a collection of items, has dimension >>> iscollection('5 * int32') True >>> iscollection('int32') False r)r!r"rrr%s r'rrs3"c BZZ r!u  r(ct|} t|}t|tot j|tjS#t$rYdSwxYw)z Has a numeric measure >>> isnumeric('int32') True >>> isnumeric('3 * ?real') True >>> isnumeric('string') False >>> isnumeric('var * {amount: ?int32}') False F)launderrr!r np issubdtypenumber TypeError)r&npdtypes r'rrsk BJ $$"d##I gry(I(II uusA A"!A"ct|trt|}t|tr|j}t |d|S)Nr )r!r"rrr7r$r%s r'r:r:sJ"c BZZ"i   Z 2tR  r(c\t|}t|to|tvS)z Has a numeric measure >>> isreal('float32') True >>> isreal('3 * ?real') True >>> isreal('string') False )r:r!r rr%s r'rrs( B b$   2B(N2r(c.t|tvS)z Has a boolean measure >>> isboolean('bool') True >>> isboolean('3 * ?bool') True >>> isboolean('int') False )r:rr%s r'rrs 2;;' !!r(cLt|}|tkp |tkS)z Has a date or datetime measure >>> isdatelike('int32') False >>> isdatelike('3 * datetime') True >>> isdatelike('?datetime') True )r:r rr%s r'rrs" B ; )" /)r()&numpyr;utilrrinternal_utilsr coretypesrrr r r r r rrrrtypesetsrr__all__intr+rrrrr0rrrrr:rrrr(r'rLs!!!!!!!!""""""%%%%%%%%%%%%%%%%%%%%%%%%%%('''''''   h+BBB"555" + + + E E E!!! 0 9 9 9   JJJ,!!! 3 3 3 " " " * * * * *r(