o `j5 @s&dZddlmZmZGdddZdS)aGenerate flat encode/decode functions for a StructField + version. Given a StructField and a protocol version, generates Python functions that encode/decode directly with zero dispatch overhead - no intermediate SimpleField/ArrayField/StructField method calls. Usage: from kafka.protocol.schemas.fields.codegen import CodegenContext # Encode: see StructField.compiled_encode_into() # Decode: see StructField.compiled_decode_from()  pack_into unpack_fromc@sBeZdZdZddZdddZddZd d Zd d Zd dZ dS)CodegenContextz!Shared state for code generation.cCsg|_ttd|_d|_dS)Nrr)linesrrglobs _var_counterselfr c/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/protocol/schemas/fields/codegen.py__init__s  zCodegenContext.__init__vcCs|jd7_d||jS)N_)r)r prefixr r r next_varszCodegenContext.next_varcCs|j||dS)N)rappend)r indentliner r r emitszCodegenContext.emitcCs$||d|||d|dS)aEmit an inline capacity check before a write of up to ``nbytes`` bytes. Generated encode functions keep three locals in sync: * ``buf`` - the destination bytearray (``out.buf``), * ``pos`` - the current write offset, * ``_cap`` - ``len(buf)``, the cached capacity. These are not set up per fragment: they are declared once by the ``def _encode(item, out):`` preamble in ``StructField.encode_into__optimized_context`` (the sole generator of the compiled encode function), and every emitted fragment - including this one - is spliced into that function body. Emit fragments are therefore only valid inside that body; they cannot stand alone. ``nbytes`` is the MAXIMUM number of bytes the following write can consume (an ``int`` for fixed/varint fields, or a string expression such as ``'len(_bv1)'`` for a variable payload). The fast path is a single comparison; only on overflow do we sync ``out.pos``, grow via :meth:`EncodeBuffer.ensure`, and re-read the (possibly reallocated) buffer back into ``buf``/``_cap``. Because :meth:`EncodeBuffer.ensure` may rebind ``out.buf``, code that instead delegates to a runtime ``encode_into`` (e.g. tagged fields) must re-bind ``buf``/``_cap`` itself afterwards - ``emit_reserve`` only covers writes emitted inline. zif pos + %s > _cap:zA out.pos = pos; out.ensure(%s); buf = out.buf; _cap = len(buf)N)r)r rnbytesr r r emit_reserveszCodegenContext.emit_reservecCs d|jS)N )joinrr r r r source>s zCodegenContext.sourcecCsftd|jD]}td|d|j|qtdt|jD]\}}t|ddd|q dS)NzGLOBALS:z =z SOURCE:rz<4 )printr enumerater)r varirr r r rAs zCodegenContext.printN)r) __name__ __module__ __qualname____doc__r rrrrrr r r r rs  rN)r%structrrrr r r r s