wiNhddlmZddlmZddlZddlZeGddZdedefdZ dS) ) dataclass)TupleNceZdZUdZejed<ejed<ejed<ejed<eed<eed<eed<eed <e eed <d Z d Z d Z dejfdZ edZdejdejfdZdededdfdZdS)DifferentiableProjectiveCamerazE Implements a batch, differentiable, standard pinhole camera originxyzwidthheightx_fovy_fovshapecj|jjd|jjdcxkr-|jjdcxkr|jjdksnJ|jjd|jjdcxkr4|jjdcxkr|jjdcxkrdksnJt |jjt |jjcxkrBt |jjcxkr$t |jjcxkrdksnJdS)Nr)rrr r rlenselfs q/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/pipelines/shap_e/camera.py __post_init__z,DifferentiableProjectiveCamera.__post_init__&s8v|A$&,q/\\\\TV\!_\\\\ HYZ[H\\\\\\\v|A$&,q/aaaaTV\!_aaaa HYZ[H\aaaa`aaaaaaa46<  C $5$5iiiiTV\9J9JiiiicRVR]RcNdNdiiiihiiiiiiiiic~tjtj|j|jgtjSN)dtype)torch from_numpynparrayr r float32rs r resolutionz)DifferentiableProjectiveCamera.resolution+s-$*dk)B"* U U UVVVrc~tjtj|j|jgtjSr)rrrr r rr!rs rfovz"DifferentiableProjectiveCamera.fov.s-$*dj)A T T TUUUrreturnctj|j|jz}tj||jztj||jdgd}|S)z> :return: coords of shape (width * height, 2) trunc) rounding_moderaxis)raranger r stackdiv)r pixel_indicescoordss rget_image_coordsz/DifferentiableProjectiveCamera.get_image_coords1sc T[4:%=>>  * -7KKK      rcb|j^}}ttj|}|}t j|d||zg|j}||}| |||j z|j zdd}|S)Nrrr) rintrprodr0r broadcast_to unsqueezeget_camera_raysviewr r )r batch_size inner_shapeinner_batch_sizer/rayss r camera_raysz*DifferentiableProjectiveCamera.camera_rays?s#': [rw{3344&&((#F$4$4Q$7$7*GW:W9gZ`Zf9ghh##F++yy%5 %Cdj%PRSUVWW rr/c j|j^}}}|dksJ||jjdksJ||dd}|}|}||dz z dzdz }|t j|dz z}||dd}|j|dd|j |dd|ddddddfzz|j |dd|ddddddfzz} | | ddz } t j t j |j|dd|| jddg| gd} | j|g|ddRS) NrrrrT)dimkeepdim)r?)rrr7r"r$floatrtanr rr normr,r4) rr/r8rn_coordsflatresr$fracs directionsr;s rr6z.DifferentiableProjectiveCamera.get_camera_raysLs'-|$ UH1}}}}T[.q11111{{:r1--oohhjjq)Q.2 #'*** :r1-- FKK Aq ) )fkk*a++eAAAqqq"1"Ho= >fkk*a++eAAAqqq!""Ho= >   *//b$/"G"GG {"4;#3#3J1#E#E T^TdefTgijGkll     ty2e2Q22222rc ||jz||jzks Jdt|j|j|j|j|||j|jS)zf Creates a new camera for the resized view assuming the aspect ratio does not change. z#The aspect ratio should not change.)rrr r r r r r) r r rrrr r r r)rr r s r resize_imagez+DifferentiableProjectiveCamera.resize_imageisgt{"ftz&9999;`999-;fff**    rN)__name__ __module__ __qualname____doc__rTensor__annotations__r2rArrr"r$r0propertyr<r6rJrrrrsC L |OOO |OOO |OOO JJJ KKK LLL LLL :jjj WWWVVV %,      X 3el3u|3333: # s 7W      rrsizer%cg}g}g}g}tjddtjzdD]}tjtj|tj|dg}|tjtj|dzz}| dz}tjtj|tj| dg}tj||} | || || | | |ttj tj |dtj tj |dtj tj |dtj tj |d||d d d t|f S) Nrr)numggr)gffffff?r) rrr r r r r rr)rlinspacepir sincossqrtsumcrossappendrrrr,rAr) rSoriginsxsyszsthetar rrr s rcreate_pan_camerasrezsG B B BQBE r222   HbfUmmRVE]]D9 : : RWRVAqD\\ " ""a HbfUmmbfUmm^S9 : : HQNNv !  !  ! )q 9 9 9::@@BB  28BQ/// 0 0 6 6 8 8  28BQ/// 0 0 6 6 8 8  28BQ/// 0 0 6 6 8 8#b''l    r) dataclassesrtypingrnumpyrrrr2rerRrrris"!!!!!  ` ` ` ` ` ` `  ` FS%Cr