o پi-@sddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZmZe eZdeeBdBdefd d ZdOdefddZdefddZdefddZdedefddZdedefddZddZdedefddZ  dPdedBdedBdefdd ZdQd#d$Z d%d&Z!defd'd(Z"d)edefd*d+Z#d)edefd,d-Z$de%eeffd.d/Z&d0ede'efd1d2Z( 3dRd4ed0ed5e)dedBfd6d7Z*d8eddfd9d:Z+d8eddfd;d<Z,d=eddfd>d?Z-    dSd@edAedBedBdCedBdedBdedBddfdDdEZ.dFej/de%eefdBfdGdHZ0dFej/de%eeffdIdJZ1d@ede%eeffdKdLZ2  dPd@edAedBedBdCedBddf dMdNZ3dS)TN)Path)Imageget_bool_env_var) init_logger)RequestPerfRecordget_diffusion_perf_log_dir image_pathreturncCs*|durdSt|to|dp|dS)zCheck if image_path is a URL.NFzhttp://zhttps://) isinstancestr startswith)r rY/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/test/test_utils.py is_image_urls  r 127.0.0.1:u@c Csrttjtj'}||z|||fWWddSty.YWddSw1s2wYdS)NTF)socketAF_INET SOCK_STREAM settimeoutconnectOSError)hostporttimeoutsrrr probe_ports  rcCstdS)NSGLANG_IS_IN_CIrrrrris_in_ci)sr c Cs~tjdd}|s d}zt|ddd}Wn ttfy'd}Ynwtr5d|d}|dSd|d}|dS) NCUDA_VISIBLE_DEVICES0,ri'ii Ni) osenvirongetintsplitstrip ValueError IndexErrorr ) cuda_devicesfirst_device_id base_portrrrget_dynamic_server_port-s  r/datacCs t|dkrdS|dddkS)z9Check if data represents a valid MP4 file by magic bytes.Fsftyp)lenr0rrris_mp4>s r5cC |dS)Nsr r4rrris_jpegE r8cCr6)NsPNG  r7r4rrris_pngJr9r:cCs |dddko|dddkS)Nr2sRIFFr1 sWEBPrr4rrris_webpOs r< output_format backgroundcCs<|pd}|dvr|dkrdS|S|pddkrdSdS)a=Infer expected image format based on request parameters. Args: output_format: The output_format parameter from the request (png/jpeg/webp/jpg) background: The background parameter from the request (transparent/opaque/auto) Returns: Expected file extension: "jpg", "png", or "webp" >jpgpngjpegwebprBr@auto transparentrA)lower)r=r>fmtrrrget_expected_image_formatTs rHr@?cCs^t|}d}t|kr"t|||drdSt|t|kstd|d|d|)N)rTzPort :z not ready. Last error: )timersleep TimeoutError)rrdeadlineintervalendlast_errrrr wait_for_portgs    rScCs||j||fdS)N) assertEqualsize)utimagewidthheightrrrcheck_image_sizeqsrZcCst}|s tdt|S)zGGets the performance log directory from the centralized sglang utility.zmPerformance logging is disabled (SGLANG_PERF_LOG_DIR is empty), but a test tried to access the log directory.)r RuntimeErrorr) log_dir_strrrrget_perf_log_dirvs r]log_dircCs|jddd|dS)NT)parentsexist_okzperformance.log)mkdir)r^rrr_ensure_log_pathsrbcCs,t|}|r |td||S)z>Delete the perf log file so tests can watch for fresh entries.z'[server-test] Monitoring perf log at %s)rbexistsunlinkloggerinfoas_posixr^log_pathrrrclear_perf_logs rjcCst}t|}||fS)zAConvenience helper to resolve and clear the perf log in one call.)r]rjrhrrrprepare_perf_logsrkric Cs|sgSg}|jddd0}|D]$}|}|sqzt|}|tdi|Wqtjy6YqwWd|S1sBwY|S)Nrzutf-8)encodingr)rcopenr)jsonloadsappendrJSONDecodeError)rirecordsfhline record_dictrrrread_perf_logss&    rw>@ request_idrcCstd|t|}t|kr1t|}|D] }|j|kr%|Sqtdt|kstjdddkr expected_ext file_sizeactual_filenamefheaderimgrXrYrrrvalidate_image_filesD          "rcapcCs<|tj}|tj}|dks|dkrdSt|t|fS)zGet video dimensions from metadata properties. Args: cap: OpenCV VideoCapture object Returns: Tuple of (width, height) if successful, None if metadata is invalid rN)r&cv2CAP_PROP_FRAME_WIDTHCAP_PROP_FRAME_HEIGHTr')rrXrYrrr#_get_video_dimensions_from_metadatas rcCsB|\}}|r |durtd|jdd\}}t|t|fS)zGet video dimensions by reading the first frame. Args: cap: OpenCV VideoCapture object Returns: Tuple of (width, height) Nz,Unable to read video frame to get dimensions)rr*shaper')rretframerYrXrrr _get_video_dimensions_from_frames rcCsFt|}zt|}|dur|W|St|W|S|w)zGet video dimensions (width, height) from a video file. Tries to get dimensions from metadata first, falls back to reading first frame. Returns: Tuple of (width, height) N)r VideoCapturerreleaser)rr dimensionsrrrget_video_dimensions.s  rc Cs(tj|s Jd||dsJd|tj|}|dks*Jd|tj|}||ks?Jd|d|dt|d }|d }t|sUJd |Wd n1s_wY|d ur|d urt |\}} ||ksJd |d|| |ksJd|d| d Sd Sd S)zUValidate video output file: existence, extension, size, filename, format, dimensions.zVideo file does not exist: z.mp4zExpected .mp4 extension, got: rzVideo file is empty: rrrr zFile is not a valid MP4: NzVideo width mismatch: expected rz Video height mismatch: expected ) r$rrcrrrrnrr5r) rrrrrrrr actual_width actual_heightrrrvalidate_video_fileDs,        r)rrr)NN)rrrIrJ)rx)NNNN)4rror$rrLpathlibrrPILr*sglang.multimodal_gen.runtime.utils.commonr1sglang.multimodal_gen.runtime.utils.logging_utilsr/sglang.multimodal_gen.runtime.utils.perf_loggerrr__name__rer boolrrr r'r/bytesr5r8r:r<rHrSrZr]rbrjtuplerklistrwfloatrrrrrrrrrrrrrrs             4