CodeWalker/CodeWalker.Core/World/Heightmaps.cs

using CodeWalker.Core.Utils;
using CodeWalker.GameFiles;
using Collections.Pooled;
using SharpDX;
using System;
using System.Collections.Generic;
using System.Text;

namespace CodeWalker.World
{
    public class Heightmaps : BasePathData
    {
        public volatile bool Inited = false;
        public GameFileCache GameFileCache;

        public List<HeightmapFile> HeightmapFiles = new List<HeightmapFile>();


        public Vector4[] GetNodePositions()
        {
            return NodePositions;
        }
        public EditorVertex[] GetTriangleVertices()
        {
            return TriangleVerts;
        }

        public Vector4[] NodePositions = [];
        public EditorVertex[] TriangleVerts = [];


        public void Init(GameFileCache gameFileCache, Action<string> updateStatus)
        {
            using var _ = new DisposableTimer("Heightmaps Init");
            Inited = false;

            GameFileCache = gameFileCache;


            HeightmapFiles.Clear();


            if (gameFileCache.EnableDlc)
            {
                LoadHeightmap("update\\update.rpf\\common\\data\\levels\\gta5\\heightmap.dat");
                LoadHeightmap("update\\update.rpf\\common\\data\\levels\\gta5\\heightmapheistisland.dat");
            }
            else
            {
                LoadHeightmap("common.rpf\\data\\levels\\gta5\\heightmap.dat");
            }


            BuildVertices();

            Inited = true;
        }

        private void LoadHeightmap(string filename)
        {
            var hmf = GameFileCache.RpfMan.GetFile<HeightmapFile>(filename);
            HeightmapFiles.Add(hmf);
        }



        public void BuildVertices()
        {

            using var vlist = new PooledList<EditorVertex>();
            using var nlist = new PooledList<Vector4>();

            foreach (var hmf in HeightmapFiles)
            {
                BuildHeightmapVertices(hmf, vlist, nlist);
            }

            if (vlist.Count > 0)
            {
                TriangleVerts = vlist.ToArray();
            }
            else
            {
                TriangleVerts = [];
            }
            if (nlist.Count > 0)
            {
                NodePositions = nlist.ToArray();
            }
            else
            {
                NodePositions = [];
            }

        }
        private void BuildHeightmapVertices(HeightmapFile hmf, PooledList<EditorVertex> vl, PooledList<Vector4> nl)
        {
            EditorVertex v1;
            EditorVertex v2;
            EditorVertex v3;
            EditorVertex v4;

            uint cgrn = (uint)new Color(0, 128, 0, 60).ToRgba();
            uint cyel = (uint)new Color(128, 128, 0, 200).ToRgba();

            var w = hmf.Width;
            var h = hmf.Height;
            var hmin = hmf.MinHeights;
            var hmax = hmf.MaxHeights;
            var min = hmf.BBMin;
            var max = hmf.BBMax;
            var siz = max - min;
            var step = siz / new Vector3(w - 1, h - 1, 255);

            for (int yi = 1; yi < h; yi++)
            {
                var yo = yi - 1;
                for (int xi = 1; xi < w; xi++)
                {
                    var xo = xi - 1;
                    var o1 = yo * w + xo;
                    var o2 = yo * w + xi;
                    var o3 = yi * w + xo;
                    var o4 = yi * w + xi;
                    v1 = new EditorVertex(min + step * new Vector3(xo, yo, hmin[o1]), cyel);
                    v2 = new EditorVertex(min + step * new Vector3(xi, yo, hmin[o2]), cyel);
                    v3 = new EditorVertex(min + step * new Vector3(xo, yi, hmin[o3]), cyel);
                    v4 = new EditorVertex(min + step * new Vector3(xi, yi, hmin[o4]), cyel);
                    vl.Add(v1); vl.Add(v2); vl.Add(v3);
                    vl.Add(v3); vl.Add(v2); vl.Add(v4);
                }
            }

            for (int yi = 1; yi < h; yi++)
            {
                var yo = yi - 1;
                for (int xi = 1; xi < w; xi++)
                {
                    var xo = xi - 1;
                    var o1 = yo * w + xo;
                    var o2 = yo * w + xi;
                    var o3 = yi * w + xo;
                    var o4 = yi * w + xi;
                    v1 = new EditorVertex(min + step * new Vector3(xo, yo, hmax[o1]), cgrn);
                    v2 = new EditorVertex(min + step * new Vector3(xi, yo, hmax[o2]), cgrn);
                    v3 = new EditorVertex(min + step * new Vector3(xo, yi, hmax[o3]), cgrn);
                    v4 = new EditorVertex(min + step * new Vector3(xi, yi, hmax[o4]), cgrn);
                    vl.Add(v1); vl.Add(v2); vl.Add(v3);
                    vl.Add(v3); vl.Add(v2); vl.Add(v4);
                }
            }

            for (int y = 0; y < h; y++)
            {
                for (int x = 0; x < w; x++)
                {
                    var o = y * w + x;
                    nl.Add(new Vector4(min + step * new Vector3(x, y, hmin[o]), 10));
                    nl.Add(new Vector4(min + step * new Vector3(x, y, hmax[o]), 10));
                }
            }


        }


    }
}