stl.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

// vim: shiftwidth=4 tabstop=4 noexpandtab

package main

import (
	"io/ioutil"
	"log"
	"encoding/binary"
	"math"
)

type BinaryStl struct {
	header []byte
	numberOfTriangles uint32
	surface Surface
}

func ReadBinaryStlFile(filePath string) (BinaryStl, error) {

	fileContent, err := ioutil.ReadFile(filePath)
	if err != nil {
		return BinaryStl{}, err
	}

	model := BinaryStl{}
	model.surface = Surface{}

	model.header = fileContent[0:80]
	log.Printf("STL header: '%s'\n", string(model.header))

	model.numberOfTriangles = binary.LittleEndian.Uint32(fileContent[80:84])
	log.Printf("STL model has %d triangles\n", model.numberOfTriangles)

	var i uint32
	for i = 0; i < model.numberOfTriangles; i++ {  // for each expected triangle
		start := 84 + i*50  // 50 bytes is length of one triangle
		end := 84 + (i+1)*50
		model.surface.triangles = append(model.surface.triangles,
			ParseBinaryStlTriangle(fileContent[start:end]))
	}

	return model,nil
}

func ParseBinaryStlTriangle(data []byte) *Triangle {  // FIXME: This function should only accept 50 byte slices/arrays

	triangle := new(Triangle)
	triangle.a = new(Point)
	triangle.b = new(Point)
	triangle.c = new(Point)

	triangle.a.x = math.Float32frombits(binary.LittleEndian.Uint32(data[12:16]))
	triangle.a.y = math.Float32frombits(binary.LittleEndian.Uint32(data[16:20]))
	triangle.a.z = math.Float32frombits(binary.LittleEndian.Uint32(data[20:24]))

	triangle.b.x = math.Float32frombits(binary.LittleEndian.Uint32(data[24:28]))
	triangle.b.y = math.Float32frombits(binary.LittleEndian.Uint32(data[28:32]))
	triangle.b.z = math.Float32frombits(binary.LittleEndian.Uint32(data[32:36]))

	triangle.c.x = math.Float32frombits(binary.LittleEndian.Uint32(data[36:40]))
	triangle.c.y = math.Float32frombits(binary.LittleEndian.Uint32(data[40:44]))
	triangle.c.z = math.Float32frombits(binary.LittleEndian.Uint32(data[44:48]))

	return triangle
}