a $j ; @sbddlmZddlZddlZddlmZddlmZmZddl m Z Gddde Z eGd d d Z Gd d d ejZGd ddejZGdddejZGdddejZddddZddddZdddddZd6ddd d!d"d#Zd7d$d$d%d$d&d'd(d$d)dd* d+d,Zed-eGd.d/d/Zed0eGd1d2d2Zed3eGd4d5d5ZdS)8) annotationsN) dataclass) event_class T_JSON_DICT)iocsBeZdZdZddddZedddddZfd d ZZS) MemoryDumpConfigz_ Configuration for memory dump. Used only when "memory-infra" category is enabled. dictreturncCs|SNselfr r 3E:\soccer\ml_env\Lib\site-packages\mycdp/tracing.pyto_jsonszMemoryDumpConfig.to_jsonjsonr cCs||Sr r clsrr r r from_jsonszMemoryDumpConfig.from_jsoncsdtS)NzMemoryDumpConfig({}))formatsuper__repr__r __class__r rrszMemoryDumpConfig.__repr__) __name__ __module__ __qualname____doc__r classmethodrr __classcell__r r rrrs rc@seZdZUdZded<dZded<dZded<dZded<dZded <dZ d ed <dZ d ed <dZ d ed <dZ ded<ddddZ edddddZdS) TraceConfigNtyping.Optional[str] record_modetyping.Optional[float]trace_buffer_size_in_kbtyping.Optional[bool]enable_samplingenable_systraceenable_argument_filterz!typing.Optional[typing.List[str]]included_categoriesexcluded_categoriessynthetic_delaysz!typing.Optional[MemoryDumpConfig]memory_dump_configrr cCst}|jdur|j|d<|jdur.|j|d<|jdurB|j|d<|jdurV|j|d<|jdurj|j|d<|jdurdd|jD|d<|jdurd d|jD|d <|jdurd d|jD|d <|j dur|j |d <|S)N recordModetraceBufferSizeInKbenableSamplingenableSystraceenableArgumentFiltercSsg|]}|qSr r .0ir r r Fz'TraceConfig.to_json..includedCategoriescSsg|]}|qSr r r4r r rr7Hr8excludedCategoriescSsg|]}|qSr r r4r r rr7Jr8syntheticDelaysmemoryDumpConfig) r r$r&r(r)r*r+r,r-r.r)rrr r rr9s(              zTraceConfig.to_jsonrcCs:||dddur t|dnd|dddur@t|dnd|dddur`t|dnd|dddurt|dnd|dddurt|dnd|dddurdd|dDnd|d ddurd d|d Dnd|d ddurd d|d Dnd|d ddur2t|d ndd S)Nr/r0r1r2r3r9cSsg|] }t|qSr strr4r r rr7lr8z)TraceConfig.from_json..r:cSsg|] }t|qSr r=r4r r rr7qr8r;cSsg|] }t|qSr r=r4r r rr7vr8r<) r$r&r(r)r*r+r,r-r.)getr>floatboolrrrr r rrOs:zTraceConfig.from_json)rrrr$__annotations__r&r(r)r*r+r,r-r.rr rr r r rr"!s          r"c@s:eZdZdZdZdZddddZedddd d Zd S) StreamFormatz Data format of a trace. Can be either the legacy JSON format or the protocol buffer format. Note that the JSON format will be deprecated soon. rprotor>r cCs|jSr valuerr r rrszStreamFormat.to_jsonrcCs||Sr r rr r rrszStreamFormat.from_jsonN) rrrrJSONPROTOrr rr r r rrCs rCc@s:eZdZdZdZdZddddZedddd d Zd S) StreamCompressionz8Compression type to use for traces returned via streams.nonegzipr>r cCs|jSr rErr r rrszStreamCompression.to_jsonrcCs||Sr r rr r rrszStreamCompression.from_jsonN) rrrrNONEZGZIPrr rr r r rrIs rIc@s>eZdZdZdZdZdZddddZeddd d d Z d S) MemoryDumpLevelOfDetailz Details exposed when memory request explicitly declared. Keep consistent with memory_dump_request_args.h and memory_instrumentation.mojom backgroundZlightZdetailedr>r cCs|jSr rErr r rrszMemoryDumpLevelOfDetail.to_jsonrcCs||Sr r rr r rrsz!MemoryDumpLevelOfDetail.from_jsonN) rrrrZ BACKGROUNDZLIGHTZDETAILEDrr rr r r rrMsrMc@s>eZdZdZdZdZdZddddZeddd d d Z d S) TracingBackenda Backend type to use for tracing. ``chrome`` uses the Chrome-integrated tracing service and is supported on all platforms. ``system`` is only supported on Chrome OS and uses the Perfetto system tracing service. ``auto`` chooses ``system`` when the perfettoConfig provided to Tracing.start specifies at least one non-Chrome data source; otherwise uses ``chrome``. autochromesystemr>r cCs|jSr rErr r rrszTracingBackend.to_jsonrcCs||Sr r rr r rrszTracingBackend.from_jsonN) rrrrZAUTOZCHROMEZSYSTEMrr rr r r rrOsrOz0typing.Generator[T_JSON_DICT, T_JSON_DICT, None]r ccsddi}|V}dS)zStop trace events collection.methodz Tracing.endNr cmd_dictrr r rendsrVz. categoriesr rTr r rget_categoriess rXr>)sync_idr ccs"t}||d<d|d}|V}dS)z| Record a clock sync marker in the trace. **EXPERIMENTAL** :param sync_id: The ID of this clock sync marker ZsyncIdzTracing.recordClockSyncMarkerrSparamsNr )rYr[rUrr r rrecord_clock_sync_markers r]r'z(typing.Optional[MemoryDumpLevelOfDetail]zCtyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, bool]]) deterministiclevel_of_detailr ccsVt}|dur||d<|dur*||d<d|d}|V}t|dt|dfS)a Request a global memory dump. **EXPERIMENTAL** :param deterministic: *(Optional)* Enables more deterministic results by forcing garbage collection :param level_of_detail: *(Optional)* Specifies level of details in memory dump. Defaults to "detailed". :returns: A tuple with the following items: 0. **dumpGuid** - GUID of the resulting global memory dump. 1. **success** - True iff the global memory dump succeeded. Nr^Z levelOfDetailzTracing.requestMemoryDumprZZdumpGuidsuccess)r rr>r?rA)r^r_r[rUrr r rrequest_memory_dumps rar#r%typing.Optional[StreamFormat]"typing.Optional[StreamCompression]ztyping.Optional[TraceConfig]ztyping.Optional[TracingBackend]) rWoptionsbuffer_usage_reporting_interval transfer_mode stream_formatstream_compression trace_configperfetto_configtracing_backendr c cst} |dur|| d<|dur&|| d<|dur6|| d<|durF|| d<|durZ|| d<|durn|| d<|dur|| d<|dur|| d <|dur|| d <d | d } | V} dS) aA Start trace events collection. :param categories: **(DEPRECATED)** **(EXPERIMENTAL)** *(Optional)* Category/tag filter :param options: **(DEPRECATED)** **(EXPERIMENTAL)** *(Optional)* Tracing options :param buffer_usage_reporting_interval: **(EXPERIMENTAL)** *(Optional)* If set, the agent will issue bufferUsage events at this interval, specified in milliseconds :param transfer_mode: *(Optional)* Whether to report trace events as series of dataCollected events or to save trace to a stream (defaults to ``ReportEvents``). :param stream_format: *(Optional)* Trace data format to use. This only applies when using ``ReturnAsStream`` transfer mode (defaults to ``json``). :param stream_compression: **(EXPERIMENTAL)** *(Optional)* Compression format to use. This only applies when using ``ReturnAsStream`` transfer mode (defaults to ``none``) :param trace_config: *(Optional)* :param perfetto_config: **(EXPERIMENTAL)** *(Optional)* Base64-encoded serialized perfetto.protos.TraceConfig protobuf message. When specified, the parameters `categories`, `options`, `traceConfig` are ignored. (Encoded as a base64 string when passed over JSON) :param tracing_backend: **(EXPERIMENTAL)** *(Optional)* Backend type (defaults to ``auto``) NrWrdZbufferUsageReportingIntervalZ transferModeZ streamFormatstreamCompressionZ traceConfigZperfettoConfigZtracingBackendz Tracing.startrZ)r r) rWrdrerfrgrhrirjrkr[rUrr r rstart s0%    rmzTracing.bufferUsagec@s>eZdZUdZded<ded<ded<eddddd Zd S) BufferUsagez **EXPERIMENTAL** r% percent_full event_countrFrrcCsh||dddur t|dnd|dddur@t|dnd|dddur`t|dnddS)NZ percentFullZ eventCountrF)rorprF)r?r@rr r rrZszBufferUsage.from_jsonNrrrrrBr rr r r rrnKs rnzTracing.dataCollectedc@s.eZdZUdZded<edddddZdS) DataCollectedz **EXPERIMENTAL** Contains a bucket of collected trace events. When tracing is stopped collected events will be sent as a sequence of dataCollected events followed by tracingComplete event. ztyping.List[dict]rFrrcCs|dd|dDdS)NcSsg|] }t|qSr r\r4r r rr7|r8z+DataCollected.from_json..rFrEr rr r rrzszDataCollected.from_jsonNrqr r r rrros rrzTracing.tracingCompletec@sFeZdZUdZded<ded<ded<ded <ed dd d d ZdS)TracingCompletez Signals that tracing is stopped, no trace buffers pending flush, and that all data was delivered via dataCollected events. rAdata_loss_occurredz typing.Optional[io.StreamHandle]streamrb trace_formatrcrhrrc Cs||t|d|dddur0tj|dnd|dddurRt|dnd|dddurtt|dnddS)NZdataLossOccurredruZ traceFormatrl)rtrurvrh)rAr?rZ StreamHandlerrCrIrr r rrs zTracingComplete.from_jsonNrqr r r rrss rs)NN) NNNNNNNNN) __future__renumtyping dataclassesrutilrrrr rr"EnumrCrIrMrOrVrXr]rarmrnrrrsr r r rsJ   ` "A"