e{W.ddlZddlmZGddeZy)N)QtGuiceZdZdZddZdZdZdZd dZd!dZ d Z d Z d Z d Z d Zd dZd dZd dZd dZd dZd dZdZdZdZdZdZdZdZed"dZeddgddfdZy)#MeshDataa Class for storing and operating on 3D mesh data. May contain: - list of vertex locations - list of edges - list of triangles - colors per vertex, edge, or tri - normals per vertex or tri This class handles conversion between the standard [list of vertexes, list of faces] format (suitable for use with glDrawElements) and 'indexed' [list of vertexes] format (suitable for use with glDrawArrays). It will automatically compute face normal vectors as well as averaged vertex normal vectors. The class attempts to be as efficient as possible in caching conversion results and avoiding unnecessary conversions. Nc(d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_||?|j'|d||j)|d||j+|dyy|j'||j-|||j)|||j+|yyy)a- ============== ===================================================== **Arguments:** vertexes (Nv, 3) array of vertex coordinates. If faces is not specified, then this will instead be interpreted as (Nf, 3, 3) array of coordinates. faces (Nf, 3) array of indexes into the vertex array. edges [not available yet] vertexColors (Nv, 4) array of vertex colors. If faces is not specified, then this will instead be interpreted as (Nf, 3, 4) array of colors. faceColors (Nf, 4) array of face colors. ============== ===================================================== All arguments are optional. Nfacesindexed) _vertexes_vertexesIndexedByFaces_vertexesIndexedByEdges_faces_edges _vertexFaces _vertexEdges_vertexNormals_vertexNormalsIndexedByFaces _vertexColors_vertexColorsIndexedByFaces_vertexColorsIndexedByEdges _faceNormals_faceNormalsIndexedByFaces _faceColors_faceColorsIndexedByFaces_faceColorsIndexedByEdges _edgeColors_edgeColorsIndexedByEdges setVertexessetVertexColors setFaceColorssetFaces)selfvertexesredges vertexColors faceColorss ;/usr/lib/python3/dist-packages/pyqtgraph/opengl/MeshData.py__init__zMeshData.__init__sF"'+$'+$    #,0)!+/(+/(!*.')-&)-& )-&  }  7 ;+((w(G)&&z7&C*  * e$+((6)&&z2* c|jS)zReturn an array (Nf, 3) of vertex indexes, three per triangular face in the mesh. If faces have not been computed for this mesh, the function returns None. )rr"s r'rzMeshData.facesXs {{r)cR|j|j|jS)zDReturn an array (Nf, 3) of vertex indexes, two per edge in the mesh.)r _computeEdgesr+s r'r$zMeshData.edges_s" ;;     {{r)cx||_d|_d|_d|_|j d|_d|_y)zSet the (Nf, 3) array of faces. Each rown in the array contains three indexes into the vertex array, specifying the three corners of a triangular face.N)rrrr resetNormalsrr)r"rs r'r!zMeshData.setFaceses>   '+$ +/()-&r)c|4|j|j|j|jS|dk(rF|j.|j"|j|j|_|jSt d)a4Return an array (N,3) of the positions of vertexes in the mesh. By default, each unique vertex appears only once in the array. If indexed is 'faces', then the array will instead contain three vertexes per face in the mesh (and a single vertex may appear more than once in the array).r-Invalid indexing mode. Accepts: None, 'faces')r r _computeUnindexedVertexesr Exceptionr"r s r'r#zMeshData.vertexesqs ?~~%$*F*F*R..0>> !  ++38R/3~~djjl/K,// /KL Lr)c|4|*tj|tj|_d|_nD|dk(r4d|_|6tj|tj|_n t d|r|j yy)a. Set the array (Nv, 3) of vertex coordinates. If indexed=='faces', then the data must have shape (Nf, 3, 3) and is assumed to be already indexed as a list of faces. This will cause any pre-existing normal vectors to be cleared unless resetNormals=False. Ndtyperr1)npascontiguousarrayfloat32r r r3r/)r"vertsr r/s r'rzMeshData.setVertexess ? !#!5!5e2::!N+/D ( g !DN /1/C/CEQSQ[Q[/\,KL L      r)c<d|_d|_d|_d|_yN)rrrrr+s r'r/zMeshData.resetNormalss"",0) *.'r)c|jduS)zGReturn True if this object already has vertex positions indexed by faceN)r r+s r'hasFaceIndexedDatazMeshData.hasFaceIndexedDatas++477r)c|jduSr=)r r+s r'hasEdgeIndexedDatazMeshData.hasEdgeIndexedDatas++477r)c\|j|j|jfD]}|yy)z9Return True if this data set has vertex color informationTF)rrrr"vs r'hasVertexColorzMeshData.hasVertexColors7$$d&F&FHhHhi A} r)c\|j|j|jfD]}|yy)z7Return True if this data set has face color informationTF)rrrrCs r' hasFaceColorzMeshData.hasFaceColors7""D$B$BDDbDbc A} r)c|jO|jd}tj|dddf|dddfz |dddf|dddfz |_| |jS|dk(r|jmtj |jj dddftj}|jddtjddf|dd||_|jStd ) z Return an array (Nf, 3) of normal vectors for each face. If indexed='faces', then instead return an indexed array (Nf, 3, 3) (this is just the same array with each vector copied three times). Nrr rrr6r1) rr#r8crossremptyshaper:newaxisr3)r"r rDnormss r' faceNormalszMeshData.faceNormalss    $ g .A "1Q3!A#!A#q1v FD  ?$$ $  ..6$"3"3"9"9!">+?+?rzz"RD  4 4Q 78 3!&)u:?29D''/!%(yyay(q ,,.2##F+ 3 ?&& &  &&tzz|4 4KL Lr)c| |jS|dk(r:|j"|j|j|_|jStd)z Return an array (Nv, 4) of vertex colors. If indexed=='faces', then instead return an indexed array (Nf, 3, 4). rr1)rrrr3r4s r'r%zMeshData.vertexColorssZ ?%% %  //7373E3Edjjl3S033 3KL Lr)c|2tj|tj|_d|_y|dk(r2d|_tj|tj|_yt d)z Set the vertex color array (Nv, 4). If indexed=='faces', then the array will be interpreted as indexed and should have shape (Nf, 3, 4) Nr6rr1)r8r9r:rrr3r"colorsr s r'rzMeshData.setVertexColorssb ?!#!5!5fBJJ!OD /3D ,  !%D /1/C/CFRTR\R\/]D ,KL Lr)cr| |jS|dk(r|j|jv|jjd}tj|ddf|jj |_|jj |dd|jdd|jStd) z Return an array (Nf, 4) of face colors. If indexed=='faces', then instead return an indexed array (Nf, 3, 4) (note this is just the same array with each color repeated three times). NrrrJr6rIr1)rrrMr8rLr7reshaper3)r"r Nfs r'r&zMeshData.faceColorss ?## #  --5$:J:J:V%%++A.132q!*DL\L\LbLb1c.484D4D4L4LRQRTU4V..q111 1KL Lr)c|2tj|tj|_d|_y|dk(r2d|_tj|tj|_yt d)z Set the face color array (Nf, 4). If indexed=='faces', then the array will be interpreted as indexed and should have shape (Nf, 3, 4) Nr6rr1)r8r9r:rrr3r^s r'r zMeshData.setFaceColorssb ?!33F"**MD -1D *  #D -/-A-A&PRPZPZ-[D *KL Lr)c|j|jjdS|j|jjdSy)z9 Return the number of faces in the mesh. Nr)rrMr r+s r' faceCountzMeshData.faceCount#sK ;; ";;$$Q' '  ) ) 5//55a8 86r)c|jSr=)rr+s r' edgeColorszMeshData.edgeColors,sr)c 6|j}i}tj|jddtj|_g|_g|_d|_d|_ t|jdD]}||}t|jdD]}||}t|Dcgc]}t|dzc}}|j|d} | S|j j||jjgt|j dz } | ||<|j| j|| |j ||f<tj |j tj"|_ycc}w)Nrr6rgļBrI)r r8rLrMuintrr rrrrTtupleroundgetappendrUarrayr:) r"rr;ifacejptxpt2indexs r'r2z"MeshData._computeUnindexedVertexes4s] ,,hhu{{2Abgg>  "u{{1~& )A8D4::a=) )!WB7qU1T6]78 #t,=NN))"-%%,,R0/1E!&E#J!!%(//2#( AaC  ) )$.. C8s+F c\|jtt|jDcgc]}gc}|_t|jj dD]6}|j|}|D] }|j|j |"8|jScc}w)zf Return list mapping each vertex index to a list of face indexes that use the vertex. r)rrTrUr#rrMrn)r"rprqinds r'rSzMeshData.vertexFacesZs    $-23t}}3G-H I ID 4;;,,Q/0 5{{1~5C%%c*11!45 5   !Js B)c |jst|j}tj|dzdtj dfg}|jddddf|dd||jddddf|d|d|z|jdddf|d| ddf<|jdddf|d| ddf<|ddddf|ddddfkD}|d|dddddf|d|<tj |d|_y|j|j}tj|jdddftj }|jd}tj|dz|ddddf<|ddddfdz|ddddf<|ddddf|ddddf<|ddddfdz|ddddf<|ddddf|ddddf<|ddddf|ddddf<||_ytd) NrJrprr6rrIz6MeshData cannot generate edges--no faces in this data.) r?rUrr8rLrjuniquerr rMaranger3)r"nfr$maskr;s r'r-zMeshData._computeEdgesss&&&(T[[!BHHRT3*;)<=E#{{1RaR40E#Jq "&++a!e"4E#Jr!B$ !%QqS!1E#JstAv !%QqS!1E#JstAv :ac?U3Z!_4D$Sz$/$B$7E#Jt ))E*3/DK  ) ) 500EHHekk!na3277CEQB99R=1,E!Aa%L 1Qrsq5vq5r)