:ShjUddlZddlZddlmZmZmZmZmZddlZ ddl m Z ddl m Z mZejejejeZejed<dZdZGd d eZGd d eZGd deZdS)N)ListOptionalTupleTypeUnion)gss)PipelineTransformerMixin)levelloggerg?g?ceZdZdS)FailedToFindAlignmentExceptionN)__name__ __module__ __qualname__X/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/ffsubsync/aligners.pyrrsDrrceZdZd deeddfdZdejdejdejfdZdejdejddfdZ dd e ddfd Z de ee eefffd ZdS) FFTAlignerNmax_offset_samplesreturnc>||_d|_d|_d|_dS)NF)r best_offset_ best_score_ get_score_)selfrs r__init__zFFTAligner.__init__s%1C+/,0 %rconvolve substringctj|jSfd}tdd||j <td||jd<S)NcNtdz |ztz SN)len)offsetrr s r_offset_to_indexzNFFTAligner._eliminate_extreme_offsets_from_solutions.._offset_to_index&s$x==1$v-I> >rz-inf)npcopyrfloat)rrr r's `` r)_eliminate_extreme_offsets_from_solutionsz4FFTAligner._eliminate_extreme_offsets_from_solutionss78$$  " *O ? ? ? ? ? ?BGv=##T%<$<===>@Ef !!$"9::<<=rcttj|}t|dz |z t|z |_|||_dSr#)intr(argmaxr%rr)rrr best_idxs r_compute_argmaxzFFTAligner._compute_argmax-sOry**++MMA-83y>>I#H-rF get_scorec d||fD\}}td||g\}}tjt|t|zd}t dtj|z}|t|z t|z }t jt jt j |t|z|}t jt j t j|t j t||zd}t j t j ||z} | || ||||_|S)Ncg|];}t|tr"ttt|n|z"FFTAligner.fit..3sL   ",As!3!3 :DS!      rcfdtj|tzdz S)Nr$)r(arrayastyper*)r9s rz FFTAligner.fit..8s&a"(1++,,U333a7rr<r)r7mathlogr%r-ceilr(fftappendzerosfliprealifftr0r+r) r refstringr r1 total_bits total_length extra_zerossubftrefftrs rfitzFFTAligner.fit2s  +    9 # 7 7)Y9O   9Xc)nns9~~=qAA 1 * 5 5566 "S^^3c)nnD  29RXkC NN.J%K%KYWWXX GBIi#i..;2N)O)OPPRS T T  726;;uu}5566   : :8Y O O    $ rc:|jr|j|jfS|jSN)rrr)r_s r transformzFFTAligner.transformIs& ? %#T%66 6$ $rrQ)F)rrrrr-rr(ndarrayr+r0boolrOrrr*rSrrrrrs&&8C=&D&&&&   /1z      . .rz.d.... 4L.%uS%s *;%;<%%%%%%rrceZdZ d deeeefdeeddfdZdZ dee e e fddfdZ de e eefe ffd ZdS) MaxScoreAlignerN base_alignersrtinrc||_||d|_n$tt||z|_t |t r||j|_n||_||_g|_dS)N)r) rYrabsr-r4typerXmax_offset_seconds_scores)rrXrY sample_rater]s rrzMaxScoreAligner.__init__Qs%*  "4"<59D # #&)#.@;.N*O*O&P&PD # lD ) ) -,8L#'#:---D  !-D 1CAC rcPfd}t|ttS)Nc(|}|j}j|d}td|d|d||rj||f|d S)NTr1z)got score %.0f (offset %d) for ratio %.3frr$) fit_transformrYrXr infor^rD)framerate_ratio is_last_itersubpiper scorerIr subpipe_makers ropt_funcz)MaxScoreAligner.fit_gss..opt_funcgs#mO44G--dj99I%3394E KK;aa      6 ##UG$4555!H9 r)rMIN_FRAMERATE_RATIOMAX_FRAMERATE_RATIO)rrIrirjs``` rfit_gsszMaxScoreAligner.fit_gssfsC        H)+>??? rsubpipescTt|ts|g}|D]}t|r|||(t |dr||j}n|}|j|j ||d|f|S)NrSTrb) r4r6callablermhasattrrSrYr^rDrXrc)rrIrnrgr s rrOzMaxScoreAligner.fitzs(D)) " zH  G   $ Y000+.. $#--dj99 # L  %33!94      rcj}j ttfd|}t |dkr't djt|d\\}}}||f|fS)NcLt|ddjkS)Nrr$)r[r)r9rs rr?z+MaxScoreAligner.transform..sQqT!W1H!HrrzZSynchronization failed; consider passing --max-offset-seconds with a number larger than {}c|ddS)Nrr)xs rr?z+MaxScoreAligner.transform..sQqT!Wr)key) r^rr6filterr%rformatr]max)rrRscoresrhr&rgs` rrSzMaxScoreAligner.transforms  " .HHHH&QQF v;;!  0VD344  $'v3D3D#E#E#E v''r)NNN)rrrrrrrr5rrmr rrOrr*rSrrrrWrWPs $ DDJZ(889D}D  DDDD*(#(4>)A#B . (uU5%<%8(%BC ( ( ( ( ( (rrW)loggingr@typingrrrrrnumpyr(ffsubsync.golden_section_searchrffsubsync.sklearn_shimr r basicConfigINFO getLoggerrr Logger__annotations__rkrl ExceptionrrrWrrrrsU 55555555555555//////========',''''**844444     Y   5%5%5%5%5%!5%5%5%pN(N(N(N(N(&N(N(N(N(N(r