forked from cataclysmbnteam/Cataclysm-BN
-
Notifications
You must be signed in to change notification settings - Fork 0
/
creature_tracker.h
101 lines (87 loc) · 3.71 KB
/
creature_tracker.h
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
#pragma once
#ifndef CATA_SRC_CREATURE_TRACKER_H
#define CATA_SRC_CREATURE_TRACKER_H
#include <cstddef>
#include <memory>
#include <set>
#include <unordered_map>
#include <vector>
#include "memory_fast.h"
#include "point.h"
#include "type_id.h"
class JsonIn;
class JsonOut;
class monster;
class Creature_tracker
{
private:
void add_to_faction_map( shared_ptr_fast<monster> critter );
class weak_ptr_comparator
{
public:
bool operator()( const weak_ptr_fast<monster> &lhs,
const weak_ptr_fast<monster> &rhs ) const {
return lhs.lock().get() < rhs.lock().get();
}
};
std::unordered_map<mfaction_id, std::set<weak_ptr_fast<monster>, weak_ptr_comparator>>
monster_faction_map_;
/**
* Creatures that get removed via @ref remove are stored here until the end of the turn.
* This keeps the objects valid and they can still be accessed instead of causing UB.
*/
std::vector<shared_ptr_fast<monster>> removed_;
public:
Creature_tracker();
~Creature_tracker();
/**
* Returns the monster at the given location.
* If there is no monster, it returns a `nullptr`.
* Dead monsters are ignored and not returned.
*/
shared_ptr_fast<monster> find( const tripoint &pos ) const;
/**
* Returns a temporary id of the given monster (which must exist in the tracker).
* The id is valid until monsters are added or removed from the tracker.
* The id remains valid through serializing and deserializing.
* Use @ref from_temporary_id to get the monster pointer back. (The later may
* return a nullptr if the given id is not valid.)
*/
int temporary_id( const monster &critter ) const;
shared_ptr_fast<monster> from_temporary_id( int id );
/**
* Adds the given monster to the tracker. @p critter must not be null.
* If the operation succeeded, the monster pointer is now managed by this tracker.
* @return Whether the operation was successful. It may fail if there is already
* another monster at the location of the new monster.
*/
bool add( shared_ptr_fast<monster> critter );
size_t size() const;
/** Updates the position of the given monster to the given point. Returns whether the operation
* was successful. */
bool update_pos( const monster &critter, const tripoint &new_pos );
/** Removes the given monster from the Creature tracker, adjusting other entries as needed. */
void remove( const monster &critter );
void clear();
void rebuild_cache();
/** Swaps the positions of two monsters */
void swap_positions( monster &first, monster &second );
/** Kills 0 hp monsters. Returns if it killed any. */
bool kill_marked_for_death();
/** Removes dead monsters from. Their pointers are invalidated. */
void remove_dead();
const std::vector<shared_ptr_fast<monster>> &get_monsters_list() const {
return monsters_list;
}
void serialize( JsonOut &jsout ) const;
void deserialize( JsonIn &jsin );
const decltype( monster_faction_map_ ) &factions() const {
return monster_faction_map_;
}
private:
std::vector<shared_ptr_fast<monster>> monsters_list;
std::unordered_map<tripoint, shared_ptr_fast<monster>> monsters_by_location;
/** Remove the monsters entry in @ref monsters_by_location */
void remove_from_location_map( const monster &critter );
};
#endif // CATA_SRC_CREATURE_TRACKER_H