// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Collections.Immutable;

namespace Microsoft.Diagnostics.DataContractReader.Data;

[CdacType(nameof(DataType.CMiniMdRW))]
internal sealed partial class CMiniMdRW : IData<CMiniMdRW>
{
    [FieldAddress] public partial TargetPointer Schema { get; }
    [Field] public partial uint TableCount { get; }
    [Field(UnderlyingBoolType = typeof(uint))] public partial bool All4ByteColumns { get; }
    [FieldAddress] public partial TargetPointer Tables { get; }
    [FieldAddress] public partial TargetPointer StringHeap { get; }
    [FieldAddress] public partial TargetPointer BlobHeap { get; }
    [FieldAddress] public partial TargetPointer UserStringHeap { get; }
    [FieldAddress] public partial TargetPointer GuidHeap { get; }
    [CustomInit(nameof(InitTableSegments))] public partial ImmutableArray<TargetPointer> TableSegments { get; }

    private partial ImmutableArray<TargetPointer> InitTableSegments(Target target, TargetPointer address)
    {
        int tableCount = checked((int)TableCount);
        uint tableStride = TableRW.GetSize(target);

        var tableSegments = new TargetPointer[tableCount];
        for (int i = 0; i < tableCount; i++)
        {
            tableSegments[i] = Tables + (ulong)i * tableStride;
        }
        return ImmutableArray.Create(tableSegments);
    }
}