aboutsummaryrefslogtreecommitdiff
path: root/src/Ryujinx.Graphics.Texture/BCnEncoder.cs
blob: 253ba305cd89fbb9b6648d114e517d4bcc581d18 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
using Ryujinx.Common;
using Ryujinx.Common.Memory;
using Ryujinx.Graphics.Texture.Encoders;
using System;
using System.Buffers;

namespace Ryujinx.Graphics.Texture
{
    public static class BCnEncoder
    {
        private const int BlockWidth = 4;
        private const int BlockHeight = 4;

        public static IMemoryOwner<byte> EncodeBC7(Memory<byte> data, int width, int height, int depth, int levels, int layers)
        {
            int size = 0;

            for (int l = 0; l < levels; l++)
            {
                int w = BitUtils.DivRoundUp(Math.Max(1, width >> l), BlockWidth);
                int h = BitUtils.DivRoundUp(Math.Max(1, height >> l), BlockHeight);

                size += w * h * 16 * Math.Max(1, depth >> l) * layers;
            }

            IMemoryOwner<byte> output = ByteMemoryPool.Rent(size);
            Memory<byte> outputMemory = output.Memory;

            int imageBaseIOffs = 0;
            int imageBaseOOffs = 0;

            for (int l = 0; l < levels; l++)
            {
                int w = BitUtils.DivRoundUp(width, BlockWidth);
                int h = BitUtils.DivRoundUp(height, BlockHeight);

                for (int l2 = 0; l2 < layers; l2++)
                {
                    for (int z = 0; z < depth; z++)
                    {
                        BC7Encoder.Encode(
                            outputMemory[imageBaseOOffs..],
                            data[imageBaseIOffs..],
                            width,
                            height,
                            EncodeMode.Fast | EncodeMode.Multithreaded);

                        imageBaseIOffs += width * height * 4;
                        imageBaseOOffs += w * h * 16;
                    }
                }

                width = Math.Max(1, width >> 1);
                height = Math.Max(1, height >> 1);
                depth = Math.Max(1, depth >> 1);
            }

            return output;
        }
    }
}