1. using System;
  2. using System.Collections.Generic;
  4. namespace WowDBCExample
  5. {
  6. class DBC<T> : IEnumerable<T> where T : struct
  7. {
  8. private readonly WoWClientDB m_dbInfo;
  9. private readonly DBCFile m_fileHdr;
  11. public int MinIndex { get { return m_dbInfo.MinIndex; } }
  12. public int MaxIndex { get { return m_dbInfo.MaxIndex; } }
  13. public int NumRows { get { return m_dbInfo.NumRows; } }
  15. public T this[int index] { get { return WowMemory.Read<T>(GetRowPtr(index)); } }
  17. public bool HasRow(int index) { return GetRowPtr(index) != IntPtr.Zero; }
  19. /// <summary>
  20. /// Initializes a new instance of DBC class using specified memory address
  21. /// </summary>
  22. /// <param name="dbcBase">DBC memory address</param>
  23. public DBC(IntPtr dbcBase)
  24. {
  25. IntPtr addr = WowMemory.Rebase(dbcBase);
  26. m_dbInfo = WowMemory.Read<WoWClientDB>(addr);
  27. m_fileHdr = WowMemory.Read<DBCFile>(m_dbInfo.Data);
  28. }
  30. private IntPtr GetRowPtr(int index)
  31. {
  32. if (index < MinIndex || index > MaxIndex)
  33. return IntPtr.Zero;
  35. int actualIndex = index - MinIndex;
  37. int v5 = WowMemory.Read<int>(m_dbInfo.Unk1 + (4 * (actualIndex >> 5)));
  39. int a2a = (int)(actualIndex & 0x1Fu);
  41. if (((1 << a2a) & v5) != 0)
  42. {
  43. int bitsSet = CountBitsSet(v5 << (31 - a2a));
  44. int entry = bitsSet + GetArrayEntryBySizeType(m_dbInfo.Unk3, actualIndex >> 5) - 1;
  46. if (m_dbInfo.Unk2 == 0)
  47. {
  48. entry = GetArrayEntryBySizeType(m_dbInfo.Rows, entry);
  49. }
  50. return m_dbInfo.FirstRow + m_fileHdr.RecordSize * entry;
  51. }
  53. return IntPtr.Zero;
  54. }
  56. private int CountBitsSet(int a1)
  57. {
  58. return 0x1010101 *
  59. ((((a1 - ((a1 >> 1) & 0x55555555)) & 0x33333333) +
  60. (((a1 - ((a1 >> 1) & 0x55555555)) >> 2) & 0x33333333) +
  61. ((((a1 - ((a1 >> 1) & 0x55555555)) & 0x33333333) +
  62. (((a1 - ((a1 >> 1) & 0x55555555)) >> 2) & 0x33333333)) >> 4)) & 0x0F0F0F0F) >> 24;
  63. }
  65. private int GetArrayEntryBySizeType(IntPtr arrayPtr, int index)
  66. {
  67. if (m_dbInfo.RowEntrySize == 2)
  68. return WowMemory.Read<short>(arrayPtr + (2 * index));
  69. else
  70. return WowMemory.Read<int>(arrayPtr + (4 * index));
  71. }
  73. public IEnumerator<T> GetEnumerator()
  74. {
  75. for (uint i = 0; i < NumRows; ++i)
  76. yield return WowMemory.Read<T>(m_dbInfo.FirstRow + (int)(i * m_fileHdr.RecordSize));
  77. }
  79. System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
  80. {
  81. return GetEnumerator();
  82. }
  83. }
  84. }