| 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 | 
 | 
|---|