summaryrefslogtreecommitdiff
path: root/src/generic/util/refcounted_base.h
blob: 0a800256dd34889b81c81756de18fce01c00ba92 (plain)
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
// refcounted_base.h   -*-c++-*-
//
//   Copyright (C) 2008-2009 Daniel Burrows
//
//   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; see the file COPYING.  If not, write to
//   the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
//   Boston, MA 02111-1307, USA.

#ifndef REFCOUNTED_BASE_H
#define REFCOUNTED_BASE_H

#include <sigc++/trackable.h>

#include <cwidget/generic/threads/threads.h>

namespace aptitude
{
  namespace util
  {
    /** \brief A class meant to be wrapped in cwidget::ref_ptr objects. */
    class refcounted_base_not_threadsafe : public sigc::trackable
    {
      mutable int refcount;

    public:
      /** \brief Create an object with a reference count of 0. */
      refcounted_base_not_threadsafe() : refcount(0) { }
      virtual ~refcounted_base_not_threadsafe();

      /** \brief Increment the reference count (normally ref_ptr will do this). */
      void incref() { ++refcount; }
      /** \brief Decrement the reference count (normally ref_ptr will do this). */
      void decref()
      {
	--refcount;
	if(refcount == 0)
	  delete this;
      }
    };

    /** \brief A class meant to be wrapped in cwidget::ref_ptr objects.
     *
     *  This variant is threadsafe, which means it's quite a bit more
     *  expensive to copy around.
     */
    class refcounted_base_threadsafe : public sigc::trackable
    {
      mutable int refcount;
      cwidget::threads::mutex m;

    public:
      /** \brief Create an object with a reference count of 0. */
      refcounted_base_threadsafe() : refcount(0) { }
      virtual ~refcounted_base_threadsafe();

      /** \brief Increment the reference count (normally ref_ptr will do this). */
      void incref()
      {
	cwidget::threads::mutex::lock l(m);
	++refcount;
      }
      /** \brief Decrement the reference count (normally ref_ptr will do this). */
      void decref()
      {
	int new_refcount;
	{
	  cwidget::threads::mutex::lock l(m);
	  --refcount;
	  new_refcount = refcount;
	}
	if(new_refcount == 0)
	  delete this;
      }
    };
  }
}

#endif