mesh_builder.go

package draco

// #include "draco_api.h"
import "C"
import (
	"unsafe"

	"github.com/flywave/go3d/vec2"
	"github.com/flywave/go3d/vec3"
	"github.com/flywave/go3d/vec4"

	vec2d "github.com/flywave/go3d/float64/vec2"
	vec3d "github.com/flywave/go3d/float64/vec3"
	vec4d "github.com/flywave/go3d/float64/vec4"
)

type MeshBuilder struct {
	ref *C.struct__draco_mesh_builder_t
}

func (m *MeshBuilder) Free() {
	if m.ref != nil {
		C.draco_mesh_builder_free(m.ref)
		m.ref = nil
	}
}

func NewMeshBuilder() *MeshBuilder {
	m := &MeshBuilder{ref: C.draco_new_mesh_builder()}
	// runtime.SetFinalizer(m, (*MeshBuilder).free)
	return m
}

func (m *MeshBuilder) Start(size int) {
	C.draco_mesh_builder_start(m.ref, C.int(size))
}

func (m *MeshBuilder) SetAttribute(numPoints int, src interface{}, att GeometryAttrType) uint32 {
	var ncomp int8
	var dt DataType
	var pt unsafe.Pointer
	switch data := src.(type) {
	case []int8:
		ncomp = 1
		dt = DT_INT8
		pt = unsafe.Pointer(&data[0])
	case []uint8:
		ncomp = 1
		dt = DT_UINT8
		pt = unsafe.Pointer(&data[0])
	case []int16:
		ncomp = 1
		dt = DT_INT16
		pt = unsafe.Pointer(&data[0])
	case []uint16:
		ncomp = 1
		dt = DT_UINT16
		pt = unsafe.Pointer(&data[0])
	case []int32:
		ncomp = 1
		dt = DT_INT32
		pt = unsafe.Pointer(&data[0])
	case []uint32:
		ncomp = 1
		dt = DT_UINT32
		pt = unsafe.Pointer(&data[0])
	case []int64:
		ncomp = 1
		dt = DT_INT64
		pt = unsafe.Pointer(&data[0])
	case []uint64:
		ncomp = 1
		dt = DT_UINT64
		pt = unsafe.Pointer(&data[0])
	case []float32:
		ncomp = 1
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case []float64:
		ncomp = 1
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case []bool:
		ncomp = 1
		dt = DT_BOOL
		pt = unsafe.Pointer(&data[0])
	case [][2]int8:
		ncomp = 2
		dt = DT_INT8
		pt = unsafe.Pointer(&data[0])
	case [][2]uint8:
		ncomp = 2
		dt = DT_UINT8
		pt = unsafe.Pointer(&data[0])
	case [][2]int16:
		ncomp = 2
		dt = DT_INT16
		pt = unsafe.Pointer(&data[0])
	case [][2]uint16:
		ncomp = 2
		dt = DT_UINT16
		pt = unsafe.Pointer(&data[0])
	case [][2]int32:
		ncomp = 2
		dt = DT_INT32
		pt = unsafe.Pointer(&data[0])
	case [][2]uint32:
		ncomp = 2
		dt = DT_UINT32
		pt = unsafe.Pointer(&data[0])
	case [][2]int64:
		ncomp = 2
		dt = DT_INT64
		pt = unsafe.Pointer(&data[0])
	case [][2]uint64:
		ncomp = 2
		dt = DT_UINT64
		pt = unsafe.Pointer(&data[0])
	case [][2]float32:
		ncomp = 2
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case []vec2.T:
		ncomp = 2
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case [][2]float64:
		ncomp = 2
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case []vec2d.T:
		ncomp = 2
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case [][2]bool:
		ncomp = 2
		dt = DT_BOOL
		pt = unsafe.Pointer(&data[0])
	case [][3]int8:
		ncomp = 3
		dt = DT_INT8
		pt = unsafe.Pointer(&data[0])
	case [][3]uint8:
		ncomp = 3
		dt = DT_UINT8
		pt = unsafe.Pointer(&data[0])
	case [][3]int16:
		ncomp = 3
		dt = DT_INT16
		pt = unsafe.Pointer(&data[0])
	case [][3]uint16:
		ncomp = 3
		dt = DT_UINT16
		pt = unsafe.Pointer(&data[0])
	case [][3]int32:
		ncomp = 3
		dt = DT_INT32
		pt = unsafe.Pointer(&data[0])
	case [][3]uint32:
		ncomp = 3
		dt = DT_UINT32
		pt = unsafe.Pointer(&data[0])
	case [][3]int64:
		ncomp = 3
		dt = DT_INT64
		pt = unsafe.Pointer(&data[0])
	case [][3]uint64:
		ncomp = 3
		dt = DT_UINT64
		pt = unsafe.Pointer(&data[0])
	case [][3]float32:
		ncomp = 3
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case []vec3.T:
		ncomp = 3
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case [][3]float64:
		ncomp = 3
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case []vec3d.T:
		ncomp = 3
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case [][3]bool:
		ncomp = 3
		dt = DT_BOOL
		pt = unsafe.Pointer(&data[0])
	case [][4]int8:
		ncomp = 4
		dt = DT_INT8
		pt = unsafe.Pointer(&data[0])
	case [][4]uint8:
		ncomp = 4
		dt = DT_UINT8
		pt = unsafe.Pointer(&data[0])
	case [][4]int16:
		ncomp = 4
		dt = DT_INT16
		pt = unsafe.Pointer(&data[0])
	case [][4]uint16:
		ncomp = 4
		dt = DT_UINT16
		pt = unsafe.Pointer(&data[0])
	case [][4]int32:
		ncomp = 4
		dt = DT_INT32
		pt = unsafe.Pointer(&data[0])
	case [][4]uint32:
		ncomp = 4
		dt = DT_UINT32
		pt = unsafe.Pointer(&data[0])
	case [][4]int64:
		ncomp = 4
		dt = DT_INT64
		pt = unsafe.Pointer(&data[0])
	case [][4]uint64:
		ncomp = 4
		dt = DT_UINT64
		pt = unsafe.Pointer(&data[0])
	case [][4]float32:
		ncomp = 4
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case []vec4.T:
		ncomp = 4
		dt = DT_FLOAT32
		pt = unsafe.Pointer(&data[0])
	case [][4]float64:
		ncomp = 4
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case []vec4d.T:
		ncomp = 4
		dt = DT_FLOAT64
		pt = unsafe.Pointer(&data[0])
	case [][4]bool:
		ncomp = 4
		dt = DT_BOOL
		pt = unsafe.Pointer(&data[0])
	}
	attr_id := C.draco_mesh_set_attribute(C.int(numPoints), m.ref, pt, C.uint(att), C.schar(ncomp), C.uint(dt))
	return uint32(attr_id)
}

func (m *MeshBuilder) GetMesh() *Mesh {
	mesh := &Mesh{PointCloud{ref: C.draco_mesh_builder_get(m.ref)}}
	// runtime.SetFinalizer(mesh, (*Mesh).free)
	return mesh
}