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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
|
/*
* 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 TRINITYCORE_COLLECTION_MGR_H
#define TRINITYCORE_COLLECTION_MGR_H
#include "Define.h"
#include "DatabaseEnvFwd.h"
#include "EnumFlag.h"
#include "ObjectGuid.h"
#include <boost/dynamic_bitset_fwd.hpp>
#include <map>
#include <unordered_map>
#include <unordered_set>
class Item;
class WorldSession;
struct ItemModifiedAppearanceEntry;
enum class CollectionItemState : uint8
{
Unchanged,
New,
Changed,
Removed
};
enum HeirloomPlayerFlags
{
HEIRLOOM_FLAG_NONE = 0x00,
HEIRLOOM_FLAG_UPGRADE_LEVEL_1 = 0x01,
HEIRLOOM_FLAG_UPGRADE_LEVEL_2 = 0x02,
HEIRLOOM_FLAG_UPGRADE_LEVEL_3 = 0x04,
HEIRLOOM_FLAG_UPGRADE_LEVEL_4 = 0x08,
HEIRLOOM_FLAG_UPGRADE_LEVEL_5 = 0x10,
HEIRLOOM_FLAG_UPGRADE_LEVEL_6 = 0x20,
};
enum HeirloomItemFlags
{
HEIRLOOM_ITEM_FLAG_NONE = 0x00,
HEIRLOOM_ITEM_FLAG_SHOW_ONLY_IF_KNOWN = 0x01,
HEIRLOOM_ITEM_FLAG_PVP = 0x02
};
struct HeirloomData
{
HeirloomData(uint32 _flags = 0, uint32 _bonusId = 0) : flags(_flags), bonusId(_bonusId) { }
uint32 flags;
uint32 bonusId;
};
enum class ToyFlags : uint32
{
None = 0,
Favorite = 0x01,
HasFanfare = 0x02
};
DEFINE_ENUM_FLAG(ToyFlags);
typedef std::map<uint32, EnumFlag<ToyFlags>> ToyBoxContainer;
typedef std::map<uint32, HeirloomData> HeirloomContainer;
enum MountStatusFlags : uint8
{
MOUNT_STATUS_NONE = 0x00,
MOUNT_NEEDS_FANFARE = 0x01,
MOUNT_IS_FAVORITE = 0x02
};
typedef std::map<uint32, MountStatusFlags> MountContainer;
typedef std::unordered_map<uint32, uint32> MountDefinitionMap;
enum class WarbandSceneCollectionFlags : uint8
{
None = 0x00,
Favorite = 0x01,
HasFanfare = 0x02
};
DEFINE_ENUM_FLAG(WarbandSceneCollectionFlags);
struct WarbandSceneCollectionItem
{
EnumFlag<WarbandSceneCollectionFlags> Flags = WarbandSceneCollectionFlags::None;
CollectionItemState State = CollectionItemState::Unchanged;
};
using WarbandSceneCollectionContainer = std::map<uint32, WarbandSceneCollectionItem>;
class TC_GAME_API CollectionMgr
{
public:
explicit CollectionMgr(WorldSession* owner);
CollectionMgr(CollectionMgr const&) = delete;
CollectionMgr(CollectionMgr&&) = delete;
CollectionMgr& operator=(CollectionMgr const&) = delete;
CollectionMgr& operator=(CollectionMgr&&) = delete;
~CollectionMgr();
static void LoadMountDefinitions();
static void LoadWarbandSceneDefinitions();
// Account-wide toys
void LoadToys();
void LoadAccountToys(PreparedQueryResult result);
void SaveAccountToys(LoginDatabaseTransaction trans);
void ToySetFavorite(uint32 itemId, bool favorite);
void ToyClearFanfare(uint32 itemId);
bool AddToy(uint32 itemId, bool isFavourite, bool hasFanfare);
bool UpdateAccountToys(uint32 itemId, bool isFavourite, bool hasFanfare);
bool HasToy(uint32 itemId) const { return _toys.contains(itemId); }
ToyBoxContainer const& GetAccountToys() const { return _toys; }
void OnItemAdded(Item* item);
// Account-wide heirlooms
void LoadHeirlooms();
void LoadAccountHeirlooms(PreparedQueryResult result);
void SaveAccountHeirlooms(LoginDatabaseTransaction trans);
void AddHeirloom(uint32 itemId, uint32 flags);
bool HasHeirloom(uint32 itemId) const { return _heirlooms.contains(itemId); }
void UpgradeHeirloom(uint32 itemId, int32 castItem);
void CheckHeirloomUpgrades(Item* item);
bool UpdateAccountHeirlooms(uint32 itemId, uint32 flags);
uint32 GetHeirloomBonus(uint32 itemId) const;
HeirloomContainer const& GetAccountHeirlooms() const { return _heirlooms; }
// Account-wide mounts
void LoadMounts();
void LoadAccountMounts(PreparedQueryResult result);
void SaveAccountMounts(LoginDatabaseTransaction trans);
bool AddMount(uint32 spellId, MountStatusFlags flags, bool factionMount = false, bool learned = false);
void MountSetFavorite(uint32 spellId, bool favorite);
void SendSingleMountUpdate(std::pair<uint32, MountStatusFlags> mount);
MountContainer const& GetAccountMounts() const { return _mounts; }
// Appearances
void LoadItemAppearances();
void LoadAccountItemAppearances(PreparedQueryResult knownAppearances, PreparedQueryResult favoriteAppearances);
void SaveAccountItemAppearances(LoginDatabaseTransaction trans);
void AddItemAppearance(Item* item);
void AddItemAppearance(uint32 itemId, uint32 appearanceModId = 0);
void AddTransmogSet(uint32 transmogSetId);
bool IsSetCompleted(uint32 transmogSetId) const;
void RemoveTemporaryAppearance(Item* item);
// returns pair<hasAppearance, isTemporary>
std::pair<bool, bool> HasItemAppearance(uint32 itemModifiedAppearanceId) const;
std::unordered_set<ObjectGuid> GetItemsProvidingTemporaryAppearance(uint32 itemModifiedAppearanceId) const;
// returns ItemAppearance::ID, not ItemModifiedAppearance::ID
std::unordered_set<uint32> GetAppearanceIds() const;
void SetAppearanceIsFavorite(uint32 itemModifiedAppearanceId, bool apply);
void SendFavoriteAppearances() const;
// Illusions
void LoadTransmogIllusions();
void LoadAccountTransmogIllusions(PreparedQueryResult knownTransmogIllusions);
void SaveAccountTransmogIllusions(LoginDatabaseTransaction trans);
void AddTransmogIllusion(uint32 transmogIllusionId);
bool HasTransmogIllusion(uint32 transmogIllusionId) const;
// Warband Scenes
void LoadWarbandScenes();
void LoadAccountWarbandScenes(PreparedQueryResult knownWarbandScenes);
void SaveAccountWarbandScenes(LoginDatabaseTransaction trans);
void AddWarbandScene(uint32 warbandSceneId);
bool HasWarbandScene(uint32 warbandSceneId) const;
void SetWarbandSceneIsFavorite(uint32 warbandSceneId, bool apply);
WarbandSceneCollectionContainer const& GetWarbandScenes() const { return _warbandScenes; }
void SendWarbandSceneCollectionData() const;
private:
bool CanAddAppearance(ItemModifiedAppearanceEntry const* itemModifiedAppearance) const;
void AddItemAppearance(ItemModifiedAppearanceEntry const* itemModifiedAppearance);
void AddTemporaryAppearance(ObjectGuid const& itemGuid, ItemModifiedAppearanceEntry const* itemModifiedAppearance);
WorldSession* _owner;
ToyBoxContainer _toys;
HeirloomContainer _heirlooms;
MountContainer _mounts;
std::unique_ptr<boost::dynamic_bitset<uint32>> _appearances;
std::unordered_map<uint32, std::unordered_set<ObjectGuid>> _temporaryAppearances;
std::unordered_map<uint32, CollectionItemState> _favoriteAppearances;
std::unique_ptr<boost::dynamic_bitset<uint32>> _transmogIllusions;
WarbandSceneCollectionContainer _warbandScenes;
};
#endif // TRINITYCORE_COLLECTION_MGR_H
|
