Version: 8.3.0
AutoRefCnt.hxx
Go to the documentation of this file.
1 // Copyright (C) 2006-2016 CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19 
20 #ifndef __AUTOREFCNT_HXX__
21 #define __AUTOREFCNT_HXX__
22 
23 #include "Exception.hxx"
24 
25 namespace YACS
26 {
27  namespace BASES
28  {
29  template<class T>
30  class AutoRefCnt
31  {
32  public:
33  AutoRefCnt(const AutoRefCnt& other):_ptr(0) { referPtr(other._ptr); }
34  AutoRefCnt(T *ptr=0):_ptr(ptr) { }
35  ~AutoRefCnt() { destroyPtr(); }
36  bool operator==(const AutoRefCnt& other) const { return _ptr==other._ptr; }
37  bool operator==(const T *other) const { return _ptr==other; }
38  AutoRefCnt &operator=(const AutoRefCnt& other) { if(_ptr!=other._ptr) { destroyPtr(); referPtr(other._ptr); } return *this; }
39  AutoRefCnt &operator=(T *ptr) { if(_ptr!=ptr) { destroyPtr(); _ptr=ptr; } return *this; }
40  T *operator->() { return _ptr ; }
41  const T *operator->() const { return _ptr; }
42  T& operator*() { return *_ptr; }
43  const T& operator*() const { return *_ptr; }
44  operator T *() { return _ptr; }
45  operator const T *() const { return _ptr; }
46  T *retn() { if(_ptr) _ptr->incrRef(); return _ptr; }
47  private:
48  void referPtr(T *ptr) { _ptr=ptr; if(_ptr) _ptr->incrRef(); }
49  void destroyPtr() { if(_ptr) _ptr->decrRef(); }
50  private:
51  T *_ptr;
52  };
53 
54  template<class T, class U>
55  typename YACS::BASES::AutoRefCnt<U> DynamicCast(typename YACS::BASES::AutoRefCnt<T>& autoSubPtr) throw()
56  {
57  T *subPtr(autoSubPtr);
58  U *ptr(dynamic_cast<U *>(subPtr));
59  typename YACS::BASES::AutoRefCnt<U> ret(ptr);
60  if(ptr)
61  ptr->incrRef();
62  return ret;
63  }
64 
65  template<class T, class U>
66  typename YACS::BASES::AutoRefCnt<U> DynamicCastSafe(typename YACS::BASES::AutoRefCnt<T>& autoSubPtr)
67  {
68  T *subPtr(autoSubPtr);
69  U *ptr(dynamic_cast<U *>(subPtr));
70  if(subPtr && !ptr)
71  throw Exception("DynamicCastSafe : U is not a subtype of T !");
72  typename YACS::BASES::AutoRefCnt<U> ret(ptr);
73  if(ptr)
74  ptr->incrRef();
75  return ret;
76  }
77 
78  template<class T>
79  class AutoCppPtr
80  {
81  public:
82  AutoCppPtr(T *ptr=0):_ptr(ptr) { }
83  ~AutoCppPtr() { destroyPtr(); }
84  AutoCppPtr &operator=(T *ptr) { if(_ptr!=ptr) { destroyPtr(); _ptr=ptr; } return *this; }
85  T *operator->() { return _ptr ; }
86  const T *operator->() const { return _ptr; }
87  T& operator*() { return *_ptr; }
88  const T& operator*() const { return *_ptr; }
89  operator T *() { return _ptr; }
90  operator const T *() const { return _ptr; }
91  private:
92  void destroyPtr() { delete _ptr; }
93  private:
94  T *_ptr;
95  };
96  }
97 }
98 
99 #endif