U h@shdZddlmZddlmZddlmZddlZddlmZddlm Z d d Z dd d Z 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) Author: Yao Feng Mail: yaofeng1995@gmail.com )absolute_import)division)print_functionN)time)mesh_core_cythonc Cst||gd}tj||gtjdd}tj||dgtjd}|tj}|tj}t||||||jd|jd|| dS)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 ~.AdtyperrN) npzerosint32float32astypecopyrZrasterize_triangles_coreshape)vertices triangleshw depth_bufferZtriangle_bufferZbarycentric_weightrM/tmp/pip-unpacked-wheel-5oclok7i/insightface/thirdparty/face3d/mesh/render.pyrasterize_triangless rr c Cs|dkr tj|||ftjd}n2|jd|krJ|jd|krJ|jd|ksNt|}tj||gtjddd}|tj}|tj}|tj}t} t ||||||jd|jd||| |S) a render mesh with colors Args: vertices: [nver, 3] triangles: [ntri, 3] colors: [nver, 3] h: height w: width c: channel BG: background image Returns: image: [h, w, c]. rendered image./rendering. Nr rrCr orderr) r r rrAssertionErrorrrrrrZrender_colors_core) rrcolorsrrcBGimagerstrrr render_colors4s,.r%nearestc Cs4| dkr tj|||ftjd} n2| jd|krJ| jd|krJ| jd|ksNt| } tj||gtjddd} |j\} } }|d krtd}n|d krtd}ntd}|tj}|tj}|tj}|tj}|tj}t | |||||| |jd|jd|jd|||| | ||| S) aK render mesh with texture map Args: vertices: [3, nver] triangles: [3, ntri] 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' Nr rrrrrrr&Zbilinear) r r rrrintrrrrZrender_texture_core)rrZtextureZ tex_coordsZ tex_trianglesrrr!Z mapping_typer"r#rZtex_hZtex_wZtex_cmtrrrrender_textureYsH.   r))r N)r r&N) __doc__ __future__rrrZnumpyr rZcythonrrr%r)rrrrs      %