main.go

package GDTFMeshReader

import (
	"fmt"
	"io"
	"path/filepath"

	"github.com/Patch2PDF/GDTF-Mesh-Reader/v2/pkg/MeshTypes"
	FileHandlers "github.com/Patch2PDF/GDTF-Mesh-Reader/v2/pkg/file_handlers"
	Primitives "github.com/Patch2PDF/GDTF-Mesh-Reader/v2/pkg/primitives"
)

type ModelReaderConf struct {
	File          io.Reader
	Filename      *string
	PrimitiveType string
}

// Load Primitive Models
func LoadPrimitives() error {
	return Primitives.LoadPrimitives()
}

// Get Model by mesh file or PrimitiveType.
//
// Note: requires LoadPrimitives() to be run beforehand if you want to get a primitive
func GetModel(conf ModelReaderConf, desiredSize *MeshTypes.Vector) (*MeshTypes.Mesh, error) {
	var mesh *MeshTypes.Mesh
	var err error

	if conf.File != nil && conf.Filename != nil && *conf.Filename != "" {
		filetype := filepath.Ext(*conf.Filename)
		switch filetype {
		case ".gltf", ".glb":
			mesh, err = FileHandlers.LoadGLTF(conf.File, desiredSize)
			if err != nil {
				return nil, err
			}
		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 Primitives.Primitives[conf.PrimitiveType] == nil {
			return nil, fmt.Errorf("unknown primitive type %s", conf.PrimitiveType)
		}
		tempMesh := Primitives.Primitives[conf.PrimitiveType].Copy()
		mesh = &tempMesh
		if desiredSize != nil {
			mesh.ScaleToDimensions(desiredSize)
		}
	}

	return mesh, nil
}