Jerdak · February 13, 2022 16:32 · davidB · Feb 6, 2015
diff --git a/example_usage.py b/example_usage.py
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 from __future__ import print_function
 import mesh_pb2 as pobj

 def dump_mesh(mesh):
    """ Dump mesh data up to, at most, 10 vertices
        and 10 faces.
        
        The 10 vert/face limit is simply to limit
        the amount of data on screen.
    """
    print("Mesh name: " + mesh.name)
    print("Num. Verts: %d"%(len(mesh.vertices)))
    print("Num. Faces: %d"%(len(mesh.faces)))
    for index,vertex in enumerate(mesh.vertices):
        print("V[%d]: %f %f %f"%(index,vertex.x,vertex.y,vertex.z)) 
        if index == 10:
            break
    for index,face in enumerate(mesh.faces):
        print("f[%d count: %d]: %d %d %d"%(index,face.poly_type,face.vert_index[0],face.vert_index[1],face.vert_index[2])) 
        if index == 10:
            break

 def test_protobuf_read():
  """ Test protocol buffer reader
  """
  with open("tmp.pb","rb") as fr:
    mesh = pobj.Mesh()
    mesh.ParseFromString(fr.read())

    dump_mesh(mesh)
    
 def test_protobuf_write():
  """ Test protocol buffer write
  
      Assumes you have a valid `load_mesh` 
      function.  
  """
  with open("tmp.pb","wb") as fw:
      # load vertex and face data however you like
      verts,faces = load_mesh("test.ply")
      mesh = pobj.Mesh()
      mesh.name = "protobj_py_test"

      for v in verts:
          #protobuf repeated message fields are RepeatedMessageFields and are created with add()
          mv = mesh.vertices.add()
          mv.x = v[0]
          mv.y = v[1]
          mv.z = v[2]
      for f in faces:
          mf = mesh.faces.add()

          # protobuf repeated non-message fields are treated as RepeatedScalarFieldContainers
          # so use 'append', not 'add'
          mf.vert_index.append(f[0])
          mf.vert_index.append(f[1])
          mf.vert_index.append(f[2])
            
 test_protobuf_write()
 test_protobuf_read()
diff --git a/mesh.proto b/mesh.proto
 package protobj;

 // 4D Vertex
 message Vertex {
    optional float x = 1 [default=0];
    optional float y = 2 [default=0];
    optional float z = 3 [default=0];
    optional float w = 4 [default=1];
 }

 // RGB color (each color in [0,255])
 message Color {
    optional uint32 r = 1 [default=255];
    optional uint32 g = 2 [default=255];
    optional uint32 b = 3 [default=255];
 }

 // Texture coordinate 
 message UV {
    optional float u = 1 [default=0];
    optional float v = 2 [default=0];
    optional float w = 3 [default=1];
 }

 // Polygonal face
 message Face {
    optional uint32 poly_type = 1    [default=3];        //polygonal type = number of sides
    repeated uint32 vert_index = 2   [packed=true];      //vertex array index
    repeated uint32 uv_index = 3     [packed=true];      //uv array index
    repeated uint32 norm_index = 4   [packed=true];      //normal array index
    repeated uint32 order = 5        [packed=true];      //Original order (to preserve original array order)  
 }

 // Mesh data
 message Mesh {
    optional string name = 1;
    optional string comments = 2;

    // Core
    repeated Vertex vertices = 3;   //Vertex array, Should *never* be empty
    repeated Face faces = 4;        //Face array, can be empty
    repeated Vertex normals = 5;    //Normal array, can be empty
    repeated Color colors = 6;      //Color array, can be empty
    repeated UV uvs = 7;            //UV array, can be empty

    // Extension support
    extensions 2000 to max;
 }
diff --git a/protobj.md b/protobj.md
	#!/usr/bin/python
	# -- coding: utf-8 --
	from __future__ import print_function
	import mesh_pb2 as pobj

	def dump_mesh(mesh):
	""" Dump mesh data up to, at most, 10 vertices
	and 10 faces.

	The 10 vert/face limit is simply to limit
	the amount of data on screen.
	"""
	print("Mesh name: " + mesh.name)
	print("Num. Verts: %d"%(len(mesh.vertices)))
	print("Num. Faces: %d"%(len(mesh.faces)))
	for index,vertex in enumerate(mesh.vertices):
	print("V[%d]: %f %f %f"%(index,vertex.x,vertex.y,vertex.z))
	if index == 10:
	break
	for index,face in enumerate(mesh.faces):
	print("f[%d count: %d]: %d %d %d"%(index,face.poly_type,face.vert_index[0],face.vert_index[1],face.vert_index[2]))
	if index == 10:
	break

	def test_protobuf_read():
	""" Test protocol buffer reader
	"""
	with open("tmp.pb","rb") as fr:
	mesh = pobj.Mesh()
	mesh.ParseFromString(fr.read())

	dump_mesh(mesh)

	def test_protobuf_write():
	""" Test protocol buffer write

	Assumes you have a valid `load_mesh`
	function.
	"""
	with open("tmp.pb","wb") as fw:
	# load vertex and face data however you like
	verts,faces = load_mesh("test.ply")
	mesh = pobj.Mesh()
	mesh.name = "protobj_py_test"

	for v in verts:
	#protobuf repeated message fields are RepeatedMessageFields and are created with add()
	mv = mesh.vertices.add()
	mv.x = v[0]
	mv.y = v[1]
	mv.z = v[2]
	for f in faces:
	mf = mesh.faces.add()

	# protobuf repeated non-message fields are treated as RepeatedScalarFieldContainers
	# so use 'append', not 'add'
	mf.vert_index.append(f[0])
	mf.vert_index.append(f[1])
	mf.vert_index.append(f[2])

	test_protobuf_write()
	test_protobuf_read()
	package protobj;

	// 4D Vertex
	message Vertex {
	optional float x = 1 [default=0];
	optional float y = 2 [default=0];
	optional float z = 3 [default=0];
	optional float w = 4 [default=1];
	}

	// RGB color (each color in [0,255])
	message Color {
	optional uint32 r = 1 [default=255];
	optional uint32 g = 2 [default=255];
	optional uint32 b = 3 [default=255];
	}

	// Texture coordinate
	message UV {
	optional float u = 1 [default=0];
	optional float v = 2 [default=0];
	optional float w = 3 [default=1];
	}

	// Polygonal face
	message Face {
	optional uint32 poly_type = 1 [default=3]; //polygonal type = number of sides
	repeated uint32 vert_index = 2 [packed=true]; //vertex array index
	repeated uint32 uv_index = 3 [packed=true]; //uv array index
	repeated uint32 norm_index = 4 [packed=true]; //normal array index
	repeated uint32 order = 5 [packed=true]; //Original order (to preserve original array order)
	}

	// Mesh data
	message Mesh {
	optional string name = 1;
	optional string comments = 2;

	// Core
	repeated Vertex vertices = 3; //Vertex array, Should never be empty
	repeated Face faces = 4; //Face array, can be empty
	repeated Vertex normals = 5; //Normal array, can be empty
	repeated Color colors = 6; //Color array, can be empty
	repeated UV uvs = 7; //UV array, can be empty

	// Extension support
	extensions 2000 to max;
	}