U h(@svdZddlmZddlmZddlmZddlZddlmZddZd d Z d d Z dddZ dddZ dddZ dS)a functions about rendering mesh(from 3d obj to 2d image). only use rasterization render here. Note that: 1. Generally, render func includes camera, light, raterize. Here no camera and light(I write these in other files) 2. Generally, the input vertices are normalized to [-1,1] and cetered on [0, 0]. (in world space) Here, the vertices are using image coords, which centers on [w/2, h/2] with the y-axis pointing to oppisite direction. Means: render here only conducts interpolation.(I just want to make the input flexible) Preparation knowledge: z-buffer: https://cs184.eecs.berkeley.edu/lecture/pipeline Author: Yao Feng Mail: yaofeng1995@gmail.com )absolute_import)division)print_functionN)timecCs|}|dddf|dddf}|dddf|dddf}||dddf}t|j|}t|j|}t|j|}t|j|} t|j|} || ||dkrd} nd|| ||} | ||| | } || ||| } | dk| dk@| | dk@S)a# Judge whether the point is in the triangle Method: http://blackpawn.com/texts/pointinpoly/ Args: point: (2,). [u, v] or [x, y] tri_points: (3 vertices, 2 coords). three vertices(2d points) of a triangle. Returns: bool: true for in triangle NrnpdotT)point tri_pointstpv0v1v2dot00dot01dot02dot11dot12 inverDenouvrS/tmp/pip-unpacked-wheel-5oclok7i/insightface/thirdparty/face3d/mesh_numpy/render.py isPointInTris   rcCs|}|dddf|dddf}|dddf|dddf}||dddf}t|j|}t|j|}t|j|}t|j|} t|j|} || ||dkrd} nd|| ||} | ||| | } || ||| } d| | }| }| }|||fS)a0 Get the weights of the position Methods: https://gamedev.stackexchange.com/questions/23743/whats-the-most-efficient-way-to-find-barycentric-coordinates -m1.compute the area of the triangles formed by embedding the point P inside the triangle -m2.Christer Ericson's book "Real-Time Collision Detection". faster.(used) Args: point: (2,). [u, v] or [x, y] tri_points: (3 vertices, 2 coords). three vertices(2d points) of a triangle. Returns: w0: weight of v0 w1: weight of v1 w2: weight of v3 rNrrr)r r rrrrrrrrrrrrw0w1w2rrrget_point_weight;s$    r c Cst||gd}tj||gtjdd}tj||dgtjd}t|jdD]}||ddf}tttt ||dfd} t tt t||df|d} tttt ||dfd} t tt t||df|d} | | ksN| | krqNt| | dD]} t| | dD]}t | |g||ddfsLq(t | |g||ddf\}}}|||ddf|||ddf|||ddf}|||| fkr(|||| f<|||| f<t |||g||| ddf<q(qqN|||fS)a Args: vertices: [nver, 3] triangles: [ntri, 3] h: height w: width Returns: depth_buffer: [h, w] saves the depth, here, the bigger the z, the fronter the point. triangle_buffer: [h, w] saves the tri id(-1 for no triangle). barycentric_weight: [h, w, 3] saves corresponding barycentric weight. # Each triangle has 3 vertices & Each vertex has 3 coordinates x, y, z. # h, w is the size of rendering ~.A)ZdtyperrNr)r zerosint32Zfloat32rangeshapemaxintceilminfloorrr array)vertices triangleshw depth_buffertriangle_bufferbarycentric_weightitriuminumaxvminvmaxrrrrr point_depthrrrrasterize_trianglesds,"&"& <  (r;r"cCs|jd|jdkstt||||\}}}t|dg} t|d|g} | ddddtjf} ||| ddfddf} | | } t| d} t| |||g} | S)a render mesh with colors(rasterize triangle first) Args: vertices: [nver, 3] triangles: [ntri, 3] colors: [nver, 3] h: height w: width c: channel Returns: image: [h, w, c]. rendering. rNr)r&AssertionErrorr;r ZreshapeZnewaxissum)r-r.colorsr/r0cr1r2r3Ztriangle_buffer_flatZbarycentric_weight_flatZweightZ colors_flatimagerrrrender_colors_rass  rBc Cs&|jd|jdkstt|||f}t||gd}t|jdD]}||ddf} tttt|| dfd} ttt t|| df|d} tttt|| dfd} ttt t|| df|d} | | ksH| | krqHt| | dD] }t| | dD]}t ||g|| ddfsHq$t ||g|| ddf\}}}||| ddf||| ddf||| ddf}||||fkr$||||f<||| dddf||| dddf||| dddf|||ddf<q$qqH|S)z render mesh with colors Args: vertices: [nver, 3] triangles: [ntri, 3] colors: [nver, 3] h: height w: width Returns: image: [h, w, c]. rr!Nrr) r&r=r r#r%r'r(r)r*r+rr )r-r.r?r/r0r@rAr1r4r5r6r7r8r9rrrrrr:rrr render_colorss* "&"& < `rCnearestc # Cs|jd|jdkst|j\} } } t|||f} t||gd} t|jdD]}||ddf}||ddf}tttt||dfd}ttt t||df|d}tttt||dfd}ttt t||df|d}||ksT||krqTt||dD]}t||dD]}t ||g||ddfsfq@t ||g||ddf\}}}|||ddf|||ddf|||ddf}|| ||fkr@|| ||f<|||dddf|||dddf|||dddf}tt|dt | dd|d<tt|dt | dd|d<|dkrt |tj}||d|dddf}n.|dkr|tt |dtt |dddf}|tt |dtt|dddf}|tt|dtt |dddf}|tt|dtt|dddf} |dt |d}!|dt |d}"|d|"d|!||"d|!|d|"|!| |"|!}|| ||ddf<q@q,qT| S) aK render mesh with texture map Args: vertices: [nver], 3 triangles: [ntri, 3] texture: [tex_h, tex_w, 3] tex_coords: [ntexcoords, 3] tex_triangles: [ntri, 3] h: height of rendering w: width of rendering c: channel mapping_type: 'bilinear' or 'nearest' rr!NrrgrDZbilinear)r&r=r r#r%r'r(r)r*r+rr floatroundZastyper$)#r-r.ZtextureZ tex_coordsZ tex_trianglesr/r0r@Z mapping_typeZtex_hZtex_w_rAr1r4r5Ztex_trir6r7r8r9rrrrrr:Ztex_xyZ tex_valueulurdlZdrZydZxdrrrrender_texturesJ  "&"& < H    ....@rK)r")r")r"rD)__doc__ __future__rrrZnumpyr rrr r;rBrCrKrrrrs    $)/  +