+ nDjJ^RIHt^RIt^RIHtHt^RIHtHtH t H t H t ^RI t ^RIHt^RIt^RIHtHt^RIHt^RIHt] !R4t!R R 4t] !R R ]44t] !R R]44t]!RR]44t!RR]],4t!RR4t!RR]],4t !RR]4t!!RR]4t"]!RR7RRl4t#R#)) annotationsN)IterableIterator)GenericProtocolTypeVaroverloadruntime_checkable)definefields)ComponentDescriptor)catch_and_log_exceptionsTcN]tRt^tRtRRltRRltRRltR R R lltR t R#) DescribedComponentBatchz A `ComponentBatchLike` object with its associated `ComponentDescriptor`. Used by implementers of `AsComponents` to both efficiently expose their component data and assign the right tags given the surrounding context. c$V^8dQhRRRRRR/#)batchComponentBatchLike descriptorr returnNone)formats"V/data/cameron/venvs/s3viz/lib/python3.14/site-packages/rerun_sdk/rerun/_baseclasses.py __annotate__$DescribedComponentBatch.__annotate__s$&&0&>Q&VZ&c WnW nR#N_batch _descriptor)selfrrs&&&r__init__ DescribedComponentBatch.__init__s  %rcV^8dQhRR/#rrr r)rs"rrrs  &9 rc VP#)z1Returns the complete descriptor of the component.)r#r$s&rcomponent_descriptor,DescribedComponentBatch.component_descriptorsrcV^8dQhRR/#rrpa.Arrayr)rs"rrr#s,,,rc 6VPP4#)zq Returns a `pyarrow.Array` of the component data. Part of the [`rerun.ComponentBatchLike`][] logging interface. )r"as_arrow_arrayr*s&rr1&DescribedComponentBatch.as_arrow_array#s {{))++rNc V^8dQhRRRR/#rlengthsnpt.ArrayLike | NonerComponentColumnr)rs"rrr+sHH!5HHrc 0\VPWR7#a_ Partitions the component batch into multiple sub-batches, forming a column. This makes it possible to use `rr.send_columns` to send columnar data directly into Rerun. The returned columns will be partitioned into unit-length sub-batches by default. Use `ComponentColumn.partition` to repartition the data as needed. Parameters ---------- lengths: The offsets to partition the component at. If specified, `lengths` must sum to the total length of the component batch. If left unspecified, it will default to unit-length batches. Returns ------- The partitioned component batch as a column. r5)r7r#r$r5s&&r partition!DescribedComponentBatch.partition+s*t//GGrr!r ) __name__ __module__ __qualname____firstlineno____doc__r%r+r1r<__static_attributes__rrrrrs"& ,HHrrc&]tRt^CtRtRRltRtR#)rzSDescribes interface for objects that can be converted to batch of rerun Components.cV^8dQhRR/#r.r)rs"rrComponentBatchLike.__annotate__Gs   rc R#)z0Returns a `pyarrow.Array` of the component data.Nrr*s&rr1!ComponentBatchLike.as_arrow_arrayGs rrN)r>r?r@rArBr1rCrrrrrCs]  rrc&]tRt^LtRtRRltRtR#) AsComponentszIDescribes interface for interpreting an object as a bundle of Components.cV^8dQhRR/#rrzlist[DescribedComponentBatch]r)rs"rrAsComponents.__annotate__Ps  &C rc R#)z Returns an iterable of `ComponentBatchLike` objects. Each object in the iterable must adhere to the `ComponentBatchLike` interface. Nrr*s&ras_component_batches!AsComponents.as_component_batchesPs rrN)r>r?r@rArBrOrCrrrrJrJLsS  rrJcb]tRt^YtRtRRlt]RRl4t]RRl4tRR lt ]t R t R #) ArchetypezBase class for all archetypes.cV^8dQhRR/#rrstrr)rs"rrArchetype.__annotate__]s***rc aaa^RIHo\S4oRVVV3RllpRPV!44pV'dRSP RV R2#RSP R2#) r)pformatcV^8dQhRR/#)rrz Iterable[str]r)rs"rr'Archetype.__str__..__annotate__bs : :] :rc 3J<"\S4FpRVP9gK\SVP4pVfK2\VRR4pVfRpM4\P !RRS!V!4P 4R^R74pRVP R V 2xK R#5i) componentNr1z z\[\s+\[z[[T)compactindentz =)r metadatagetattrnameresub to_pylist)fldcompr1 comp_contentsclsrXr$s r fields_repr&Archetype.__str__..fields_reprbsc{#,,."42D| %,T3CT%JN%-(3 )+&gn6F6P6P6R\`ij.k) sxxj-99##s B#A>B#z, zrr.z( z )z())pprintrXtypejoinr>)r$rjargsrirXsf @@r__str__Archetype.__str__]sZ"4j : :(zz+-( c$s3 3b) )rcV^8dQhRR/#rTr)rs"rrrV}sEE#Erc VPPR4pRP.VPR^4RROVPN4#)z rerun_sdk..N)r? removeprefixrnrsplitr>)rimodules& r archetypeArchetype.archetype|sH,,\:xxC&--Q/4CcllCDDrcV^8dQhRR/#rTr)rs"rrrVs22S2rc PVP4PR^4R,#)rtru)ryrwris&rarchetype_short_nameArchetype.archetype_short_names }}%%c1-b11rcV^8dQhRR/#rLr)rs"rrrVs&Crc h.p\\V44FpRVP9gK\WP4pVfK1\ VP 4R,VP,VPVP4R7pVP\W444K V#)zm Return all the component batches that make up the archetype. Part of the `AsComponents` logging interface. r\:)component_typery) r rmr`rarbr r~_COMPONENT_TYPEryappendr)r$batchesrfrgdescrs& rrOArchetype.as_component_batchess $t*%Ccll*tXX.#/113c9CHHD'+';';"&.."2E NN#:4#GH&rrN) r>r?r@rArBrp classmethodryr~rO__repr__rCrrrrRrRYs@(*>EE 22*HrrRc]tRt^t$RtR]R&RRRllt]RRl4tRR lt R R lt ] R R l4t RRlt RtR#) BaseBatchNzpa.DataType | None_ARROW_DATATYPEc$V^8dQhRRRRRR/#)rdataT | Nonestrictz bool | Nonerrr)rs"rrBaseBatch.__annotate__s!$>$>X$>{$>d$>rc Ve\VPPVR7;_uu_4\V\P 4'd#VP VP8XdWnM VPWP4VnRRR4R#\VP4VnR# +'giL,;i)a Construct a new batch. This method must flexibly accept native data (which comply with type `T`). Subclasses must provide a type parameter specifying the type of the native data (this is automatically handled by the code generator). A value of None indicates that the component should be cleared and results in the creation of an empty array. The actual creation of the Arrow array is delegated to the `_native_to_pa_array()` method, which is not implemented by default. Parameters ---------- data: The data to convert into an Arrow array. strict: Whether to raise an exception if the data cannot be converted into an Arrow array. If None, the value defaults to the value of the `rerun.strict` global setting. Returns ------- The Arrow array encapsulating the data. N)r) r __class__r> isinstancepaArrayrmrpa_array_native_to_pa_array_empty_pa_array)r$rrs&&&rr%BaseBatch.__init__s4  )$..*A*A&QQdBHH--$))t?S?S2S$(M$($<$rc VfR#V!V4#)ax Primary method for creating Arrow arrays for components. The default value of None is preserved in the field to indicate that the optional field was not specified. If any value other than None is provided, it is passed through to `__init__`. Parameters ---------- data: The data to convert into an Arrow array. Returns ------- The Arrow array encapsulating the data. Nr)rirs&&r _converterBaseBatch._converters$ <t9 rc V^8dQhRRRR/#)rotherobjectrboolr)rs"rrrs//F/t/rc l\V\4'g\#VPVP8H#r )rrNotImplementedr)r$rs&&r__eq__BaseBatch.__eq__s(%++! !}}..rcV^8dQhRR/#rrintr)rs"rrr"""rc ,\VP4#r )lenrr*s&r__len__BaseBatch.__len__4==!!rc$V^8dQhRRRRRR/#)rrr data_type pa.DataTyperr/r)rs"rrrs!""!" ""rc \h)a- Converts native data into an Arrow array. Subclasses must provide an implementation of this method (via an override) if they are to be used as either an archetype's field (which should be the case for all components), or a (delegating) component's field (for datatypes). Datatypes which are used only within other datatypes may omit implementing this method, provided that the top-level datatype implements it. A hand-coded override must be provided for the code generator to implement this method. The override must be named `native_to_pa_array_override()` and exist as a static member of the `Ext` class located in `_ext.py`. `ColorExt.native_to_pa_array_override()` in `color_ext.py` is a good example of how to implement this method, in conjunction with the native type's converter (see `rgba__field_converter_override()`, used to construct the native `Color` object). Parameters ---------- data: The data to convert into an Arrow array. data_type: The Arrow data type of the data. Returns ------- The Arrow array encapsulating the data. )NotImplementedError)rrs&&rrBaseBatch._native_to_pa_arrays <"!rcV^8dQhRR/#r.r)rs"rrrsrc VP#za The component as an arrow batch. Part of the [`rerun.ComponentBatchLike`][] logging interface. rr*s&rr1BaseBatch.as_arrow_arrays }}rrr )r>r?r@rAr__annotations__r%rrrr staticmethodrr1rCrrrrrsR*.O'.)$>L,/ """>rrc]tRtRtRt]RRl4t]RRl4t]RRl4t]R R /R R ll4tR R R R /RRlltRRltRRltRRlt Rt R #)r7ia. A column of components that can be sent using `send_columns`. This is represented by a ComponentBatch array that has been partitioned into multiple segments. This is useful for reinterpreting a single contiguous batch as multiple sub-batches to use with the [`send_columns`][rerun.send_columns] API. c$V^8dQhRRRRRR/#)rrstr | ComponentDescriptorcomponent_batchrrrr)rs"rrComponentColumn.__annotate__s(-,  rc R#r r)r$rrs&&&rr%ComponentColumn.__init__s rc(V^8dQhRRRRRRRR/#) rrrrrr5 npt.ArrayLikerrr)rs"rrr"s2-,    rc R#r rr$rrr5s&&&$rr%r!rc(V^8dQhRRRRRRRR/#) rrrrroffsetsznpt.NDArray[np.int32]rrr)rs"rrr+s2-, '   rc R#r r)r$rrrs&&&$rr%r*rrr5Nc(V^8dQhRRRRRRRR/#) rrrrrr5r6rrr)rs"rrr4s2-, &   rc R#r rrs&&&$rr%r3rrrc ,V^8dQhRRRRRRRRR R /#) rrrrrr5r6rznpt.NDArray[np.int32] | Nonerrr)rs"rrr<s<5=5=-5=,5= & 5= . 5= 5=rc \V\4'd \V4pM\V\4'dTpVeVe \R4hWnW nVe W@nR#Ve\P!V4pVP^8wd \R4hVP4P\P4p\P!\P!^.\PR7\P!V\PR734VnR#\VP!44p\P"!V^,\PR7VnR#)a Construct a new component column. This makes it possible to use `rr.send_columns` to send columnar data directly into Rerun. The returned column will be partitioned into unit-length sub-batches by default. Use `ComponentColumn.partition` to repartition the data as needed. Parameters ---------- descriptor: The component descriptor for this component batch. component_batch: The component batch to partition into a column. lengths: The lengths of each partition. Must sum to the total length of the component batch. If left unspecified, it will default to unit-length batches. Mutually exclusive with `offsets`. offsets: Pre-computed int32 offsets array (including the leading 0). Mutually exclusive with `lengths`. Nz,Cannot specify both `lengths` and `offsets`.zLengths must be a 1D array.)dtype)rrUr ValueErrorrr_offsetsnparrayndimflattenastypeint32 concatenatecumsumrr1arange)r$rrr5rns&&&$$ rr%r<s > j# & &,Z8J  $7 8 8#J  7#6KL L$.  #M  hhw'G||q  !>??oo'..rxx8GNNBHHaS,I299U\dfdldlKm+noDMO2245AIIa!e288>>rc VPP4p\PP VP V4#r)rr1r ListArray from_arraysr)r$rs& rr1ComponentColumn.as_arrow_array{s3 $$335||'' u==rc V^8dQhRRRR/#)rr5rrr7r)rs"rrrsWWW?Wrc F\VPVPVR7#)a (Re)Partitions the column. This makes it possible to use `rr.send_columns` to send columnar data directly into Rerun. The returned columns will be partitioned into unit-length sub-batches by default. Use `ComponentColumn.partition` to repartition the data as needed. Parameters ---------- lengths: The offsets to partition the component at. Returns ------- The (re)partitioned column. r:)r7rrr;s&&rr<ComponentColumn.partitions&t0D0DgVVr)rrr) r>r?r@rArBr r%r+r1r<rCrrrr7r7s  )- 5= )- 5= 15 5=n>WWrr7cJ]tRtRtRtRRltRRltRRltR R ltR t R #) ComponentColumnListizu A collection of [`ComponentColumn`][rerun.ComponentColumn]s. Useful to partition and log multiple columns at once. c V^8dQhRRRR/#)rcolumnszIterable[ComponentColumn]rrr)rs"rr ComponentColumnList.__annotate__s&& 9&d&rc &\V4VnR#r )list_columns)r$rs&&rr%ComponentColumnList.__init__s W  rcV^8dQhRR/#)rrzIterator[ComponentColumn]r)rs"rrrs##3#rc ,\VP4#r )iterrr*s&r__iter__ComponentColumnList.__iter__sDMM""rcV^8dQhRR/#rr)rs"rrrrrc ,\VP4#r )rrr*s&rrComponentColumnList.__len__rrc V^8dQhRRRR/#)rr5rrrr)rs"rrrsVVV3FVrc r\VPUu.uFq"PV4NK up4#uupi)a/ (Re)Partitions the columns. This makes it possible to use `rr.send_columns` to send columnar data directly into Rerun. The returned columns will be partitioned into unit-length sub-batches by default. Use `ComponentColumn.partition` to repartition the data as needed. Parameters ---------- lengths: The offsets to partition the component at. If specified, `lengths` must sum to the total length of the component batch. If left unspecified, it will default to unit-length batches. Returns ------- The partitioned component batch as a column. )rrr<)r$r5cols&& rr<ComponentColumnList.partitions,*#dmm#TmsMM'$:m#TUU#Ts4)rN) r>r?r@rArBr%rrr<rCrrrrrs" &#"VVrrc>]tRtRtRRltRRltR RRlltR tR#) ComponentBatchMixinicV^8dQhRR/#rTr)rs"rr ComponentBatchMixin.__annotate__s$$$rc VP#zc Returns the name of the component. Part of the [`rerun.ComponentBatchLike`][] logging interface. )rr*s&rr"ComponentBatchMixin.component_types ###rc V^8dQhRRRR/#)rrr rrr)rs"rrrs99$79r?r@rArrr<rCrrrrrs$9MMrrcR]tRtRtRt]RRl4t]RRl4tRRltR t R #) ComponentMixiniz Makes components adhere to the `ComponentBatchLike` interface. A single component will always map to a batch of size 1. The class using the mixin must define the `_BATCH_TYPE` field, which should be a subclass of `BaseBatch`. cV^8dQhRR/#)rrrr)rs"rrComponentMixin.__annotate__s//;/rc .VPP#)z` The pyarrow type of this batch. Part of the [`rerun.ComponentBatchLike`][] logging interface. ) _BATCH_TYPErr}s&r arrow_typeComponentMixin.arrow_type...rcV^8dQhRR/#rTr)rs"rrrs//s/rc .VPP#r)rrr}s&rrComponentMixin.component_typerrcV^8dQhRR/#r.r)rs"rrrs999rc BVPV.4P4#r)rr1r*s&rr1ComponentMixin.as_arrow_arrays '6688rrN) r>r?r@rArBrrrr1rCrrrr r s9////99rr zcreating empty array)contextc V^8dQhRRRR/#)rrmrrr/r)rs"rrr s####+##(##rc  V\P!48Xd\P!^4#\V\P4'd%VP \ VP44#\V\P4'Ed]VPR8Xd\PPT^R\P!.\P!4R7P4^,\P!.\P!4R7P4^,.VUu.uFp\ VP 4NK upR7#\PPT^R\P!.\P!4R7P4^,.VUu.uFp\ VP 4NK upR7#\V\P"4'dN\P$P'VUu.uFp\ VP 4NK up\)V4R7#\P!.VR7#uupiuupiuupi)rdenseN)rm)rmlengthbufferschildren)r )rnullnullsr ExtensionType wrap_arrayr storage_type UnionTypemode UnionArray from_buffersrint8r rrm StructType StructArrayrr)rm field_types& rrr s rwwyxx{$(())t/@/@ABB$ %% 99 ==--HHRbggi088:1=HHRbhhj199;A> NRRTz/*//:TR.  ==--HHRbggi088:1=NRRTz/*//:TR. $ &&~~))^b*c^bPZ?:??+K^b*clpqulv)ww 88BT ""#SS +dsI7 I</J)$ __future__rrccollections.abcrrtypingrrrr r numpyr numpy.typingnptpyarrowrattrsr r rerun_bindingsr error_utilsrrrrrJrRrr7rrr rrrrr9s" .JJ .1 CL-H-H`     8    B BBJo odFWFWR%V(?3%VP"M,"MJ!9'!9H"89##:##r