aboutsummaryrefslogtreecommitdiff
path: root/src/Spv.Generator/LiteralInteger.cs
blob: 013ca3b9455fcaf7a3ce935f3904c1ed85f325d2 (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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
using System;
using System.IO;

namespace Spv.Generator
{
    public class LiteralInteger : IOperand, IEquatable<LiteralInteger>
    {
        [ThreadStatic]
        private static GeneratorPool<LiteralInteger> _pool;

        internal static void RegisterPool(GeneratorPool<LiteralInteger> pool)
        {
            _pool = pool;
        }

        internal static void UnregisterPool()
        {
            _pool = null;
        }

        public OperandType Type => OperandType.Number;

        private enum IntegerType
        {
            UInt32,
            Int32,
            UInt64,
            Int64,
            Float32,
            Float64,
        }

        private IntegerType _integerType;
        private ulong _data;

        public ushort WordCount { get; private set; }

        public LiteralInteger() { }

        private static LiteralInteger New()
        {
            return _pool.Allocate();
        }

        private LiteralInteger Set(ulong data, IntegerType integerType, ushort wordCount)
        {
            _data = data;
            _integerType = integerType;

            WordCount = wordCount;

            return this;
        }

        public static implicit operator LiteralInteger(int value) => New().Set((ulong)value, IntegerType.Int32, 1);
        public static implicit operator LiteralInteger(uint value) => New().Set(value, IntegerType.UInt32, 1);
        public static implicit operator LiteralInteger(long value) => New().Set((ulong)value, IntegerType.Int64, 2);
        public static implicit operator LiteralInteger(ulong value) => New().Set(value, IntegerType.UInt64, 2);
        public static implicit operator LiteralInteger(float value) => New().Set(BitConverter.SingleToUInt32Bits(value), IntegerType.Float32, 1);
        public static implicit operator LiteralInteger(double value) => New().Set(BitConverter.DoubleToUInt64Bits(value), IntegerType.Float64, 2);
        public static implicit operator LiteralInteger(Enum value) => New().Set((ulong)(int)(object)value, IntegerType.Int32, 1);

        // NOTE: this is not in the standard, but this is some syntax sugar useful in some instructions (TypeInt ect)
        public static implicit operator LiteralInteger(bool value) => New().Set(Convert.ToUInt64(value), IntegerType.Int32, 1);

        public static LiteralInteger CreateForEnum<T>(T value) where T : Enum
        {
            return value;
        }

        public void WriteOperand(BinaryWriter writer)
        {
            if (WordCount == 1)
            {
                writer.Write((uint)_data);
            }
            else
            {
                writer.Write(_data);
            }
        }

        public override bool Equals(object obj)
        {
            return obj is LiteralInteger literalInteger && Equals(literalInteger);
        }

        public bool Equals(LiteralInteger cmpObj)
        {
            return Type == cmpObj.Type && _integerType == cmpObj._integerType && _data == cmpObj._data;
        }

        public override int GetHashCode()
        {
            return DeterministicHashCode.Combine(Type, _data);
        }

        public bool Equals(IOperand obj)
        {
            return obj is LiteralInteger literalInteger && Equals(literalInteger);
        }

        public override string ToString() => $"{_integerType} {_data}";
    }
}