1 | #ifndef PREFCOUNT_H
|
---|
2 | #define PREFCOUNT_H
|
---|
3 |
|
---|
4 | // Classe d'objet avec comptage de référence.
|
---|
5 | // Le nombre de référence est incrémenté lors de la création et lors d'un Attach
|
---|
6 | // (inutile d'appeler Attach si on crée l'objet).
|
---|
7 | // Si on cree l'objet avec isTmp = true, alors le possesseur est temporaire,
|
---|
8 | // et en tout cas, possede le PRefCount en lecture seule.
|
---|
9 | // Par exemple, une fonction qui retourne un objet de type PRefCounted le créera
|
---|
10 | // en mode temporaire. Un constructeur par copie appelé derrière ne dupliquera pas
|
---|
11 | // les données.
|
---|
12 | // HasTmpOnly retourne true si un objet n'est référencé que de façon temporaire.
|
---|
13 |
|
---|
14 | #ifdef TRACE_REFCOUNT
|
---|
15 | #define DBG_DESTROY cerr << "PRefCounted destroyed" << endl;
|
---|
16 | #define DBG_ATTACH cerr << "PRefCounted attached" << endl;
|
---|
17 | #define DBG_DETACH cerr << "PRefCounted detached" << endl;
|
---|
18 | #else
|
---|
19 | #define DBG_DESTROY
|
---|
20 | #define DBG_ATTACH
|
---|
21 | #define DBG_DETACH
|
---|
22 | #endif
|
---|
23 |
|
---|
24 |
|
---|
25 | class PRefCounted {
|
---|
26 | public:
|
---|
27 | PRefCounted(bool isTmp=false)
|
---|
28 | :refcount(1), tmpcount(isTmp?1:0) {}
|
---|
29 | virtual ~PRefCounted() {DBG_DESTROY}
|
---|
30 |
|
---|
31 | void Attach(bool isTemp=false)
|
---|
32 | {if (isTemp && HasTmpOnly()) tmpcount++;
|
---|
33 | DBG_ATTACH
|
---|
34 | refcount++;}
|
---|
35 | void Detach() {refcount--;
|
---|
36 | DBG_DETACH
|
---|
37 | if (!refcount) delete this;}
|
---|
38 |
|
---|
39 | int RefCount() const {return refcount;}
|
---|
40 | bool HasTmpOnly() const {return tmpcount == refcount;}
|
---|
41 |
|
---|
42 | private:
|
---|
43 | int refcount;
|
---|
44 | int tmpcount;
|
---|
45 | };
|
---|
46 |
|
---|
47 | // Macro pour les constructeurs par copie d'objets référençant des PRefCounted
|
---|
48 | // Il faut que le PRefCounted ait une méthode T* Clone(bool isTmp);
|
---|
49 |
|
---|
50 | template <class refcount>
|
---|
51 | void PRefCountCopy(refcount*& dest, refcount const* src, bool isTmp=false) {
|
---|
52 | if (dest) dest->Detach();
|
---|
53 | if (!src) {
|
---|
54 | dest=0;
|
---|
55 | } else if (src->HasTmpOnly()) {
|
---|
56 | dest = const_cast<refcount*>(src);
|
---|
57 | dest->Attach(isTmp);
|
---|
58 | } else {
|
---|
59 | dest = src->Clone(isTmp);
|
---|
60 | }
|
---|
61 | }
|
---|
62 |
|
---|
63 |
|
---|
64 | #endif
|
---|
65 |
|
---|