mvr-parser.go

package MVRParser

import (
	"archive/zip"
	"encoding/xml"
	"fmt"
	"io"

	GDTFReader "github.com/Patch2PDF/MVR-Parser/internal/gdtfreader"
	"github.com/Patch2PDF/MVR-Parser/internal/geometry3d"
	MVRXML "github.com/Patch2PDF/MVR-Parser/internal/types/mvrxml"
	MVRTypes "github.com/Patch2PDF/MVR-Parser/pkg/types"
)

// Takes a `.mvr` file zip reader and parses it, according to the given config
//
// may contain an entire mesh model of the stage, if configured as such
func ParseMVRZipReader(zipfile *zip.Reader, config MVRTypes.MVRParserConfig) (*MVRTypes.GeneralSceneDescription, error) {
	var mvrData MVRXML.GeneralSceneDescription

	// TODO: docs + tests

	// put all files in zip into the filemap
	var fileMap map[string]*zip.File = make(map[string]*zip.File)
	for _, file := range zipfile.File {
		fileMap[file.Name] = file
	}

	// parse mvr xml
	if fileMap["GeneralSceneDescription.xml"] == nil {
		return nil, fmt.Errorf("Invalid MVR: missing GeneralSceneDescription.xml")
	}
	xmlFile, err := fileMap["GeneralSceneDescription.xml"].Open()
	if err != nil {
		return nil, err
	}
	data, err := io.ReadAll(xmlFile)
	if err != nil {
		return nil, err
	}
	err = xml.Unmarshal(data, &mvrData)
	if err != nil {
		return nil, err
	}

	parseConfig := MVRXML.ParseConfigData{
		GDTFTaskMap: &map[string]*GDTFReader.GDTFTask{},
	}

	parsedData := mvrData.Parse(parseConfig)

	refPointers := MVRTypes.CreateRefPointersMap()

	GDTFReader.GetGDTFs(parseConfig.GDTFTaskMap, refPointers, fileMap, config)

	parsedData.CreateReferencePointer(refPointers)

	parsedData.ResolveReference(refPointers)

	if config.MeshHandling >= MVRTypes.ReadMeshesIntoModels {
		geometry3d.ReadMeshes(fileMap, parsedData)
	}

	if config.MeshHandling >= MVRTypes.BuildStageModel {
		parsedData.GenerateMeshes()
	}

	return parsedData, nil
}