U ‰±ËhÒã@spdZddlmZddlmZddlmZddlZddlZddlm Z dd„Z d d „Z d d „Z d d„Z dd„ZdS)z» io: read&write mesh 1. read obj as array(TODO) 2. write arrays to obj Preparation knowledge: representations of 3d face: mesh, point cloud... storage format: obj, ply, bin, asc, mat... é)Úabsolute_import)Údivision)Úprint_functionN)ÚiocCsdS)z read mesh r©)Úobj_namerrúO/tmp/pip-unpacked-wheel-5oclok7i/insightface/thirdparty/face3d/mesh_numpy/io.pyÚread_objsr cCs4| d¡ddkr t ||¡nt |d|¡dS)z3 Args: vertices: shape = (nver, 3) Ú.éÿÿÿÿZascz.ascN)ÚsplitÚnpZsavetxt)ÚpathÚverticesrrrÚ 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) ér r Úobjú.objÚwrúv {} {} {} {} {} {} éz f {} {} {} N)Úcopyr ÚopenÚrangeÚshapeÚformatÚwrite) rrÚ trianglesÚcolorsÚfÚiÚsÚkZntrirrrÚwrite_obj_with_colors$s 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 f|| 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 r rrú.mtlú _texture.pngrrú mtllib {} rz v {} {} {} rú vt {} {} úusemtl FaceTexture úf {}/{} {}/{} {}/{} únewmtl FaceTexture ú map_Kd {} N) r ÚreplacerrrÚosrÚabspathrrrÚimsave) rrrÚtextureÚ uv_coordsÚmtl_nameÚ texture_namerr!r rrrÚwrite_obj_with_textureCs0    &   D  r4c CsÌ| d¡ddkr|d}| dd¡}| dd¡}| ¡}|d7}t|dƒ,}d  tj |¡¡} | | ¡t |j d ƒD]R} d  || d f|| df|| d f|| d f|| df|| d f¡} | | ¡qzt |j d ƒD]*} d  || d f|| df¡} | | ¡qÜ| d¡t |j d ƒD]T} d || d f|| d f|| df|| df|| d f|| d f¡} | | ¡q W5QRXt|dƒ,}| d¡d tj |¡¡} | | ¡W5QRXt   ||¡dS)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 r rrr$r%rrr&rrrr'r(r)r*r+N) r r,rrrr-rr.rrrrr/) rrrrr0r1r2r3rr!r rrrÚwrite_obj_with_colors_texturews0    D   D  r5)Ú__doc__Ú __future__rrrZnumpyr r-Zskimagerr rr#r4r5rrrrÚs     4