aboutsummaryrefslogtreecommitdiff
path: root/tests/DummyData.h
blob: eb0c84ddefa779a193c036e4fbdbb2c611a7c2c3 (plain)
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
/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef TRINITY_DUMMYDATA_H
#define TRINITY_DUMMYDATA_H

#include "Common.h"
#include "Define.h"
#include "DB2Store.h"

#include <string_view>

struct ItemTemplate;

class UnitTestDataLoader
{
    public:
        template <typename T, uint32 T::*ID>
        class DB2
        {
            class LoaderGuard
            {
                public:
                    LoaderGuard(DB2& d) : _d(d) {}
                    ~LoaderGuard() { _d.Dump(); }

                    T& Add() { return _d._storage.emplace_back(); }
                private:
                    DB2& _d;
            };

            public:
                DB2(DB2Storage<T>& store) : _store(store) {}
                LoaderGuard Loader() { return {*this}; }
                void Dump()
                {
                    delete[] _store._indexTable;
                    for (T const& entry : _storage)
                        if (entry.*ID >= _store._indexTableSize)
                            _store._indexTableSize = entry.*ID + 1;
                    _store._indexTable = new char*[_store._indexTableSize];
                    for (size_t i = 0; i < _store._indexTableSize; ++i)
                        _store._indexTable[i] = nullptr;
                    for (T& entry : _storage)
                        _store._indexTable[entry.*ID] = reinterpret_cast<char*>(&entry);
                }

            private:
                std::vector<T> _storage;
                DB2Storage<T>& _store;
        };

        static void LoadAchievementTemplates();
        static void LoadItemTemplates();

    private:
        static ItemTemplate& GetItemTemplate(uint32 id, std::string_view name);
        static void SetItemLocale(uint32 id, LocaleConstant locale, std::string_view name);
};

#endif