o q::j"@s(ddlmZddlZddlZddlZddlZddlZddlmZddl m Z e r.ddl m Z dZ dZdZed Zed Zed Zd;ddZedZedZedZdd$d%Zd?d)d*Zd@dAd.d/Z 0 0 1dBdCd6d7ZdDd8d9Z e!d:kre dSdS)E) annotationsN)dedent) TYPE_CHECKING)IterablezV// // WARNING: This file is automatically generated! Please edit onnx.in.proto. // zB // For using protobuf-lite option optimize_for = LITE_RUNTIME; onnxz\s*//\s*#if\s+ONNX-ML\s*$z\s*//\s*#endif\s*$z\s*//\s*#else\s*$lines Iterable[str]onnx_mlboolreturnccsd}|D]G}t|r|dksJd}qt|r#|dks Jd}qt|r1|dvs.Jd}q|dkr9|Vq|dkrC|rC|Vq|dkrL|sL|VqdS)Nr)r r )IF_ONNX_ML_REGEXmatchELSE_ONNX_ML_REGEXENDIF_ONNX_ML_REGEX)rr in_iflinerU/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/gen_proto.py process_ifs)s(        rz#(\s*)import\s*"([^"]*)\.proto";\s*$z\{PACKAGE_NAME\}z(.*)\-ml package_namestrccs|tk}|D]@}|rt|nd}|r@|d}t|}|r,|dd|d}n|d|}|dd|dVqt||VqdS)Nr r _-mlimport "z.proto";)DEFAULT_PACKAGE_NAME IMPORT_REGEXrgroupML_REGEXPACKAGE_NAME_REGEXsub)rr need_renamerm include_namemlrrrprocess_package_nameCs  r&z"(\s*)syntax\s*=\s*"proto2"\s*;\s*$z(\s*)optional\s(.*)$ccs|D]?}t|}|r|ddVqt|}|r(|d|dVqt|}|r?|dd|ddVq|VqdS)Nr zsyntax = "proto3";r rz .proto3";)PROTO_SYNTAX_REGEXrrOPTIONAL_REGEXr)rrr#rrrconvert_to_proto3Ws    r) protoc_path proto3_path include_pathcpp_out python_outNonecCs>td|||d|g}|d|d|gtj|dddS)NzGenerate pb3 code using z-Iz --cpp_outz --python_outT)check)printextend subprocessrun)r*r+r,r-r. build_argsrrrgen_proto3_codens r6sourceprotointcCsJ|}t||d}t||d}|dkrt|}n|dksJtj|S)N)r )rr ) splitlinesrr&r)oslinesepjoin)r7r8r rrrrr translatews     r?fpardir str | NonecCs*|durtjtjt}tj||SN)r<pathrealpathdirname__file__r>)r@rArrrqualifysrHFstemoutput do_onnx_mllitec Cs~t|d}|tk}|}d|vrd}|r%|r|d|dn|d} n |r.|d|n|} t| d|d} t| d|d} td |t|d d } | } td | t| d dd d}|t|t| d||d|r||tWdn1swYtd | t| d dd d}|t|t| d||d|r|tWdn1swY|rt j | }t j |}t || |||t t j || d}|D]}td|t |q|r<|rt|d|d}n t|d|d}td |t|d dd d}|d|d| dWdn 1s7wYWdn 1sGwYt|ddd|d}|rjt| ddd|d}n|r{t|ddd|d}n t|ddd|d}td|t|d d d }|tdt j t j |d d!WddS1swYdS)"Nz .in.proto onnx-dataFrrz.proto)rAz.proto3z Processing zutf-8)encodingzWriting wrI)newlinerOr )r8r rr:z .proto3.*z Removing z-ml.pb.hz.pb.hz #pragma once z #include "z.pb.h" -z_pb.pyz_pb2.pyz _ml_pb2.pyz generating za # This file is generated by 'gen_proto.py'. DO NOT EDIT! from .rz" import * # noqa )rHrr1openreadwriteautogen_headerr? LITE_OPTIONr<rDrFr6globr>removereplacersplitextbasename)rJrrKrLrMr*proto_inr" import_ml proto_baser8proto3finr7fout porto3_dirbase_dir pb3_filespb3_file proto_headerpb_pypb2_pyr@rrrconverts         '$rjc Cstjdd}|jddddd|jdd d d d |jd dd dd |jddtjtjtdd|jdddd|jddgddd|}tj |j sVt |j |j D]}t ||j|j |j|j|jdqYdS)Nz/Generates .proto file variations from .in.proto) descriptionz-pz --packagerz@package name in the generated proto files (default: %(default)s))defaulthelpz-mz--ml store_truezML mode)actionrmz-lz--litez-generate lite proto to use with protobuf-litez-oz--outputz'output directory (default: %(default)s)z --protoc_pathrIz)path to protoc for proto3 file validationstems*)rzonnx-operatorsrNz3list of .in.proto file stems (default: %(default)s))nargsrlrm)rrKrLrMr*)argparseArgumentParser add_argumentr<rDrErFrG parse_argsexistsrKmakedirsrprjpackager%rMr*)parserargsrJrrrmainsX  r|__main__)rrr r r r)rrrrr r)rrr r) r*rr+rr,rr-rr.rr r/) r7rr8r9r r rrr rrC)r@rrArBr r)FFrI)rJrrrrKrrLr rMr r*rr r/)r r/)" __future__rrsrXr<rer3textwraprtypingrcollections.abcrrVrWrcompilerrrrrr rr&r'r(r)r6r?rHrjr|__name__rrrrsF                   W0