Subjects.cs

using System.Runtime.CompilerServices;
using KeyHashProof.Types;

namespace KeyHashProof;

/// <summary>
/// Сборка словарей и поиск по ним. У каждого метода NoInlining: иначе
/// компилятор встроит его в тело замера и часть работы удалит.
///
/// Все шесть словарей строятся из одних и тех же данных: число повторяется,
/// строка уникальна. Различаются только объявления ключей.
/// </summary>
internal static class Subjects
{
    /// <summary>Словарь с ключом, где первым идёт число.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<NumberFirst, int> BuildNumberFirst(string[] texts)
    {
        Dictionary<NumberFirst, int> map = new(texts.Length);
        for (int i = 0; i < texts.Length; i++)
        {
            map[new NumberFirst { A = Payloads.SharedNumber, B = texts[i] }] = i;
        }

        return map;
    }

    /// <summary>Словарь с ключом, где первой идёт строка.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<TextFirst, int> BuildTextFirst(string[] texts)
    {
        Dictionary<TextFirst, int> map = new(texts.Length);
        for (int i = 0; i < texts.Length; i++)
        {
            map[new TextFirst { A = Payloads.SharedNumber, B = texts[i] }] = i;
        }

        return map;
    }

    /// <summary>Словарь с ключом, у которого есть сравнение, но нет своего хеша.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<EquatableOnly, int> BuildEquatableOnly(string[] texts)
    {
        Dictionary<EquatableOnly, int> map = new(texts.Length);
        for (int i = 0; i < texts.Length; i++)
        {
            map[new EquatableOnly { A = Payloads.SharedNumber, B = texts[i] }] = i;
        }

        return map;
    }

    /// <summary>Словарь с полностью описанным ключом.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<FullKey, int> BuildFull(string[] texts)
    {
        Dictionary<FullKey, int> map = new(texts.Length);
        for (int i = 0; i < texts.Length; i++)
        {
            map[new FullKey { A = Payloads.SharedNumber, B = texts[i] }] = i;
        }

        return map;
    }

    /// <summary>Словарь с ключом-записью.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<RecordKey, int> BuildRecord(string[] texts)
    {
        Dictionary<RecordKey, int> map = new(texts.Length);
        for (int i = 0; i < texts.Length; i++)
        {
            map[new RecordKey(Payloads.SharedNumber, texts[i])] = i;
        }

        return map;
    }

    /// <summary>Словарь с ключом без ссылок внутри.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static Dictionary<NoReference, int> BuildNoReference(int count)
    {
        Dictionary<NoReference, int> map = new(count);
        for (int i = 0; i < count; i++)
        {
            map[new NoReference { A = Payloads.SharedNumber, B = i }] = i;
        }

        return map;
    }

    // ---------- поиск ----------

    /// <summary>Поиск по словарю с ключом, где первым идёт число.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupNumberFirst(Dictionary<NumberFirst, int> map, string[] texts, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new NumberFirst { A = Payloads.SharedNumber, B = texts[probes[i]] }, out _))
            {
                found++;
            }
        }

        return found;
    }

    /// <summary>Поиск по словарю с ключом, где первой идёт строка.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupTextFirst(Dictionary<TextFirst, int> map, string[] texts, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new TextFirst { A = Payloads.SharedNumber, B = texts[probes[i]] }, out _))
            {
                found++;
            }
        }

        return found;
    }

    /// <summary>Поиск по словарю с ключом без своего хеша.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupEquatableOnly(Dictionary<EquatableOnly, int> map, string[] texts, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new EquatableOnly { A = Payloads.SharedNumber, B = texts[probes[i]] }, out _))
            {
                found++;
            }
        }

        return found;
    }

    /// <summary>Поиск по словарю с полностью описанным ключом.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupFull(Dictionary<FullKey, int> map, string[] texts, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new FullKey { A = Payloads.SharedNumber, B = texts[probes[i]] }, out _))
            {
                found++;
            }
        }

        return found;
    }

    /// <summary>Поиск по словарю с ключом-записью.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupRecord(Dictionary<RecordKey, int> map, string[] texts, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new RecordKey(Payloads.SharedNumber, texts[probes[i]]), out _))
            {
                found++;
            }
        }

        return found;
    }

    /// <summary>Поиск по словарю с ключом без ссылок.</summary>
    [MethodImpl(MethodImplOptions.NoInlining)]
    internal static int LookupNoReference(Dictionary<NoReference, int> map, int[] probes)
    {
        int found = 0;
        for (int i = 0; i < probes.Length; i++)
        {
            if (map.TryGetValue(new NoReference { A = Payloads.SharedNumber, B = probes[i] }, out _))
            {
                found++;
            }
        }

        return found;
    }
}