U ‰±ËhJã@s„ddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z d d „Z d d „Z d d„Zdd„Zdd„ZdS)é)Úabsolute_import)Údivision)Úprint_functionN)Úio)Útimeé)Úmesh_core_cythoncCsdS)z read mesh r©)Úobj_namer r úI/tmp/pip-unpacked-wheel-5oclok7i/insightface/thirdparty/face3d/mesh/io.pyÚread_objsr cCs4| d¡ddkr t ||¡nt |d|¡dS)z3 Args: vertices: shape = (nver, 3) Ú.éÿÿÿÿZascz.ascN)ÚsplitÚnpZsavetxt)ÚpathÚverticesr r r Ú write_ascsrc Csô| ¡}|d7}| d¡ddkr*|d}t|dƒ¶}t|jdƒD]R}d ||df||df||d f||df||df||d f¡}| |¡qD|j\}}t|jdƒD]4}d  ||d f||df||df¡}| |¡q°W5QRXd S) zÇ Save 3D face model with texture represented by colors. Args: obj_name: str vertices: shape = (nver, 3) triangles: shape = (ntri, 3) colors: shape = (nver, 3) rr rÚobjú.objÚwrzv {} {} {} {} {} {} éz f {} {} {} N)ÚcopyrÚopenÚrangeÚshapeÚformatÚwrite) r rÚ trianglesÚcolorsÚfÚiÚsÚkZntrir r r Úwrite_obj_with_colorss D  &r$c Cs°| d¡ddkr|d}| dd¡}| dd¡}| ¡}|d7}t|dƒ}d  tj |¡¡}| |¡t |j d ƒD]4} d  || d f|| df|| d f¡}| |¡qzt |j d ƒD].} d  || d fd|| df¡}| |¡q¾| d¡t |j d ƒD]T} d || d f|| d f|| df|| df|| d f|| d f¡}| |¡qW5QRXt|dƒ,}| d¡d tj |¡¡}| |¡W5QRXt ||ƒdS)au Save 3D face model with texture represented by texture map. Ref: https://github.com/patrikhuber/eos/blob/bd00155ebae4b1a13b08bf5a991694d682abbada/include/eos/core/Mesh.hpp Args: obj_name: str vertices: shape = (nver, 3) triangles: shape = (ntri, 3) texture: shape = (256,256,3) uv_coords: shape = (nver, 3) max value<=1 r rrrú.mtlú _texture.pngrrz mtllib {} rz v {} {} {} rz vt {} {} zusemtl FaceTexture zf {}/{} {}/{} {}/{} únewmtl FaceTexture ú map_Kd {} N) rÚreplacerrrÚosrÚabspathrrrÚimsave) r rrÚtextureÚ uv_coordsÚmtl_nameÚ texture_namer r"r!r r r Úwrite_obj_with_texture=s0    &    D  r1c Cs| d¡ddkr|d}| dd¡}| dd¡}| ¡}|d7}| tj¡ ¡| tj¡ ¡| tj¡ ¡}}}t t  |¡t  t j   |¡¡|||||j d|j d|j d¡ t|d ƒ,}| d ¡d  t j   |¡¡} | | ¡W5QRXt ||¡d S) a} Save 3D face model with texture. Ref: https://github.com/patrikhuber/eos/blob/bd00155ebae4b1a13b08bf5a991694d682abbada/include/eos/core/Mesh.hpp Args: obj_name: str vertices: shape = (nver, 3) triangles: shape = (ntri, 3) colors: shape = (nver, 3) texture: shape = (256,256,3) uv_coords: shape = (nver, 3) max value<=1 r rrrr%r&rrrr'r(N)rr)rZastyperZfloat32rZ"write_obj_with_colors_texture_coreÚstrÚencoder*rr+rrrrrr,) r rrrr-r.r/r0r r"r r r Úwrite_obj_with_colors_textureps   4@  r4)Ú __future__rrrZnumpyrr*ZskimagerrZcythonrr rr$r1r4r r r r Ús       3