o ip$ @sxddlZddlZddlmZdejdeeeffddZ    d>dejdedBdedBd edBd edBf d d Z  d?d ddejdee e fdBdee e fdBde de f ddZ dddejdejde de de de f ddZ  d?ddde dee e fdBdee e fdBde de f ddZ    d>dejdedBdedBd edBd edBf dd Z  d?dejd!e dBd"e dBfd#d$Z  d?dejd%edBd&edBfd'd(Zd)d*Z  d?d+ejd!e dBd"e dBddfd,d-Z .  d@d/e fd0d1Zdejddfd2d3Zd4ee e fde fd5d6Zddd d7d8e dee e fdBdee e fdBde ddf d9d:Zd;e de fd= 1). We require C <= limit, where limit = max(max_rel, 1.0 / min_rel). Returns the computed closeness factor C. rzInvalid image dimensions?z%Aspect ratios must be close: ar1/ar2=.2gz, allowed range u–z (limit z).)rminr max) r"r#r$r%rw1h1w2h2ar1ar2 closenesslimitr r r &validate_images_aspect_ratio_closeness4s&  r2 aspect_ratiocCst|}t||||d|S)zQParses 'X:Y' and validates it against optional bounds. Returns the numeric ratio.r)_parse_aspect_ratio_stringr)r3rrrr r r r validate_aspect_ratio_stringSsr5videoc Csz|\}}Wnty!}z td|WYd}~dSd}~ww|dur5||kr5td|d|d|durH||krHtd|d|d|dur[||kr[td|d|d|durn||krptd|d|ddSdS)Nz%Error getting dimensions of video: %szVideo width must be at least rrzVideo width must be at most zVideo height must be at least zVideo height must be at most )get_dimensions Exceptionloggingerrorr )r6rrrrrrer r r validate_video_dimensions`s  r< min_duration max_durationc Csz|}Wnty}z td|WYd}~dSd}~wwd}|dur7|||kr7td|d|d|durL|||krNtd|d|ddSdS)Nz#Error getting duration of video: %sg-C6?z Video duration must be at least s, got szVideo duration must be at most ) get_durationr8r9r:r )r6r=r>durationr;epsilonr r r validate_video_durationws  rDmin_frame_countmax_frame_countc Csz|}Wnty}z td|WYd}~dSd}~ww|dur2||kr2td|d||durD||krFtd|d|dSdS)Nz&Error getting frame count of video: %sz#Video frame count must be at least z, got z"Video frame count must be at most )get_frame_countr8r9r:r )r6rErF frame_countr;r r r validate_video_frame_counts  rIcCs,t|tjr|jdkr|jdSdSt|S)Nrrr) isinstancetorchTensorndimr r )imagesr r r get_number_of_imagess rOaudiocCst|d}t|djd|}d|}|dur-|||kr-td|d||dd|durE|||krGtd |d||dddSdS) N sample_ratewaveformr&z Audio duration must be at least r?z.2fr@zAudio duration must be at most )intr r )rPr=r>srdurepsr r r validate_audio_durations rXpromptstringcCs|dur td|d|r|}|r*t||kr*td|d|dt|d|rBt||krDtd|d|dt|ddSdS)NzField 'z' cannot be empty.z' cannot be shorter than z characters; was z characters long.z Field 'z cannot be longer than )r8stripr )rZstrip_whitespace field_name min_length max_lengthr r r validate_stringsr`cCs"|}|dvrtd|dS)z(Validates video container format is MP4.)mp4zmov,mp4,m4a,3gp,3g2,mj2z*Only MP4 container format supported. Got: N)get_container_formatr )r6container_formatr r r validate_container_format_is_mp4srdrcCs6|\}}|dks |dkrtd|d|d||S)NrzRatios must be positive, got :.)r )reabr r r _ratio_from_tuplesrjrr cCs|durt|nd}|durt|nd}|dur%|dur%||kr%||}}|durJ|r0||krJn||krJ|r8dnd}td|dd|d|dd|duro|rU||krsn||krq|r]dnd}td|dd|d|dddSdSdS) zFValidate a numeric aspect ratio against optional min/max ratio bounds.Ns     !