UVtools/UVtools.Core/MeshFormats/AMFMeshFile.cs

/*
 *                     GNU AFFERO GENERAL PUBLIC LICENSE
 *                       Version 3, 19 November 2007
 *  Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
 *  Everyone is permitted to copy and distribute verbatim copies
 *  of this license document, but changing it is not allowed.
 */

using System.Collections.Generic;
using System.IO;
using System.IO.Compression;
using System.Numerics;
using UVtools.Core.Extensions;
using UVtools.Core.FileFormats;

namespace UVtools.Core.MeshFormats
{
    public class AMFMeshFile : MeshFile
    {
        #region Members
        private readonly Dictionary<Vector3, uint> _vertexCache = new(VertexCacheSize);
        private FileStream _triangleStream;
        #endregion

        #region Properties
        public static FileExtension FileExtension => new(typeof(AMFMeshFile), "amf", "Additive Manufacturing Format");
        #endregion

        #region Constructor
        public AMFMeshFile(string filePath, FileMode fileMode, MeshFileFormat fileFormat = MeshFileFormat.ASCII, FileFormat slicerFile = null) : base(filePath, fileMode, MeshFileFormat.ASCII, slicerFile) { }
        


        #endregion
        
        #region Methods
        public override void BeginWrite()
        {
            /* Create a stream to store the triangles (faces) as they come through */
            _triangleStream = new FileStream(PathExtensions.GetTempFilePathWithExtension("trig", $"{About.Software}_"), FileMode.Create, FileAccess.ReadWrite, FileShare.None, 81920, FileOptions.DeleteOnClose);
            
            MeshStream.WriteLineLF("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
            MeshStream.WriteLineLF("<amf unit=\"millimeter\" version=\"1.1\">");
            MeshStream.WriteLineLF($"\t<metadata type=\"name\">{FilenameWithoutExtension}</metadata>");
            MeshStream.WriteLineLF($"\t<metadata type=\"author\">{HeaderComment}</metadata>");
            MeshStream.WriteLineLF("\t<object id=\"0\">");
            MeshStream.WriteLineLF("\t\t<mesh>");
            MeshStream.WriteLineLF("\t\t\t<vertices>");
        }

        public override void WriteTriangle(Vector3 p1, Vector3 p2, Vector3 p3, Vector3 normal)
        {
            if (!_vertexCache.ContainsKey(p1))
            {
                MeshStream.WriteLineLF($"\t\t\t\t<vertex>");
                MeshStream.WriteLineLF($"\t\t\t\t\t<coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<x>{p1.X:F6}</x>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<y>{p1.Y:F6}</y>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<z>{p1.Z:F6}</z>");
                MeshStream.WriteLineLF($"\t\t\t\t\t</coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t</vertex>");
                _vertexCache.Add(p1, VertexCount);
                VertexCount++;
            }

            if (!_vertexCache.ContainsKey(p2))
            {
                MeshStream.WriteLineLF($"\t\t\t\t<vertex>");
                MeshStream.WriteLineLF($"\t\t\t\t\t<coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<x>{p2.X:F6}</x>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<y>{p2.Y:F6}</y>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<z>{p2.Z:F6}</z>");
                MeshStream.WriteLineLF($"\t\t\t\t\t</coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t</vertex>");
                _vertexCache.Add(p2, VertexCount);
                VertexCount++;
            }

            if (!_vertexCache.ContainsKey(p3))
            {
                MeshStream.WriteLineLF($"\t\t\t\t<vertex>");
                MeshStream.WriteLineLF($"\t\t\t\t\t<coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<x>{p3.X:F6}</x>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<y>{p3.Y:F6}</y>");
                MeshStream.WriteLineLF($"\t\t\t\t\t\t<z>{p3.Z:F6}</z>");
                MeshStream.WriteLineLF($"\t\t\t\t\t</coordinates>");
                MeshStream.WriteLineLF($"\t\t\t\t</vertex>");
                _vertexCache.Add(p3, VertexCount);
                VertexCount++;
            }

            _triangleStream.WriteLineLF($"\t\t\t\t<triangle>");
            _triangleStream.WriteLineLF($"\t\t\t\t\t<v1>{_vertexCache[p1]}</v1>");
            _triangleStream.WriteLineLF($"\t\t\t\t\t<v2>{_vertexCache[p2]}</v2>");
            _triangleStream.WriteLineLF($"\t\t\t\t\t<v3>{_vertexCache[p3]}</v3>");
            _triangleStream.WriteLineLF($"\t\t\t\t</triangle>");

            TriangleCount++;
            
            /* If we are getting close to the cache size. we do *not* want to go over the capacity as that will trigger
             * an allocation of a bigger buffer and copy of the kvp's */
            if (_vertexCache.Count >= VertexCacheSize - 10)
            {
                _vertexCache.Clear();
            }
        }

        public override void EndWrite()
        {
            _vertexCache.Clear();

            MeshStream.WriteLineLF("\t\t\t</vertices>");
            MeshStream.WriteLineLF("\t\t\t<volume>");
            MeshStream.WriteLineLF("\t\t\t\t<metadata type=\"name\">Model</metadata>");

            _triangleStream.Seek(0, SeekOrigin.Begin);
            _triangleStream.CopyTo(MeshStream);
            _triangleStream.Dispose();

            MeshStream.WriteLineLF("\t\t\t</volume>");
            MeshStream.WriteLineLF("\t\t</mesh>");
            MeshStream.WriteLineLF("\t</object>");
            MeshStream.WriteLineLF("</amf>");

            var tmpFile = PathExtensions.GetTempFilePathWithExtension("tmp", $"{About.Software}_");
            if (File.Exists(tmpFile)) File.Delete(tmpFile);
            using (var zip = ZipFile.Open(tmpFile, ZipArchiveMode.Create))
            {
                MeshStream.Seek(0, SeekOrigin.Begin);
                zip.PutFileContent(Filename, MeshStream, ZipArchiveMode.Create);
            }
            MeshStream.Dispose();
            
            File.Move(tmpFile, FilePath, true);
        }

        #endregion
    }
}