GetModel()

main.go

@@ -1 +1,50 @@
 package GDTFMeshReader
+
+import (
+	"fmt"
+	"io"
+	"path/filepath"
+
+	"github.com/Patch2PDF/GDTF-Mesh-Reader/pkg/MeshTypes"
+	FileHandlers "github.com/Patch2PDF/GDTF-Mesh-Reader/pkg/file_handlers"
+	Primitives "github.com/Patch2PDF/GDTF-Mesh-Reader/pkg/primitives"
+)
+
+type ModelReaderConf struct {
+	File          *io.Reader
+	Filename      *string
+	PrimitiveType string
+}
+
+func GetModel(conf ModelReaderConf, desiredSize MeshTypes.Vector) (*MeshTypes.Mesh, error) {
+	var mesh *MeshTypes.Mesh
+
+	if conf.PrimitiveType == "undefined" && conf.File != nil && conf.Filename != nil && *conf.Filename != "" {
+		filetype := filepath.Ext(*conf.Filename)
+		switch filetype {
+		case ".gltf", ".glb":
+			meshes, err := FileHandlers.LoadGLTF(*conf.File, desiredSize)
+			if err != nil {
+				return nil, err
+			}
+			mesh = meshes[0]
+		case ".3ds":
+			data, err := io.ReadAll(*conf.File)
+			if err != nil {
+				return nil, err
+			}
+			mesh, err = FileHandlers.Load3DS(&data, &desiredSize)
+			if err != nil {
+				return nil, err
+			}
+		default:
+			return nil, fmt.Errorf("unknown model type %s", filetype)
+		}
+	} else if conf.PrimitiveType != "undefined" && Primitives.Primitives[conf.PrimitiveType] != nil {
+		tempMesh := Primitives.Primitives[conf.PrimitiveType].Copy()
+		mesh = &tempMesh
+		mesh.ScaleToDimensions(&desiredSize)
+	}
+
+	return mesh, nil
+}

pkg/file_handlers/gltf.go

@@ -10,7 +10,7 @@ import (
 	"github.com/qmuntal/gltf"
 )
 
-func LoadGLTF(file io.ReadCloser, desiredSize Types.Vector) ([]*Types.Mesh, error) {
+func LoadGLTF(file io.Reader, desiredSize Types.Vector) ([]*Types.Mesh, error) {
 	var doc gltf.Document
 	gltf.NewDecoder(file).Decode(&doc)