o soi!@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_iflinerB/home/ubuntu/.local/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|gt|dS)NzGenerate pb3 code using z-Iz --cpp_outz --python_out)printextend subprocess check_call)r*r+r,r-r. build_argsrrrgen_proto3_codens r5sourceprotointcCsJ|}t||d}t||d}|dkrt|}n|dksJtj|S)N)r )rr ) splitlinesrr&r)oslinesepjoin)r6r7r rrrrr translatews     r>fpardir str | NonecCs*|durtjtjt}tj||SN)r;pathrealpathdirname__file__r=)r?r@rrrqualifysrGFstemoutput 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)r@z.proto3z Processing zutf-8)encodingzWriting wrH)newlinerNr )r7r rr9z .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 z[ # This file is generated by setup.py. DO NOT EDIT! from .rz" import * # noqa )rGrr0openreadwriteautogen_headerr> LITE_OPTIONr;rCrEr5globr=removereplacersplitextbasename)rIrrJrKrLr*proto_inr" import_ml proto_baser7proto3finr6fout porto3_dirbase_dir pb3_filespb3_file proto_headerpb_pypb2_pyr?rrrconverts         '$ric 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)actionrlz-lz--litez-generate lite proto to use with protobuf-litez-oz--outputz'output directory (default: %(default)s)z --protoc_pathrHz)path to protoc for proto3 file validationstems*)rzonnx-operatorsrMz3list of .in.proto file stems (default: %(default)s))nargsrkrl)rrJrKrLr*)argparseArgumentParser add_argumentr;rCrDrErF parse_argsexistsrJmakedirsroripackager%rLr*)parserargsrIrrrmainsX  r{__main__)rrr r r r)rrrrr r)rrr r) r*rr+rr,rr-rr.rr r/) r6rr7r8r r rrr rrB)r?rr@rAr r)FFrH)rIrrrrJrrKr rLr r*rr r/)r r/)" __future__rrrrWr;rer2textwraprtypingrcollections.abcrrUrVrcompilerrrrrr rr&r'r(r)r5r>rGrir{__name__rrrrsF                   W0