summaryrefslogtreecommitdiff
path: root/src/generic/util/dynamic_set_transform.h
blob: 19211fc9143fb7eb1d06c9ae92531b2f9b130dce (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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/** \file dynamic_set_transform.h */       // -*-c++-*-

// Copyright (C) 2010 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 DYNAMIC_SET_TRANSFORM_H
#define DYNAMIC_SET_TRANSFORM_H

#include "dynamic_set.h"
#include "enumerator_transform.h"

#include <boost/function.hpp>
#include <boost/make_shared.hpp>
#include <boost/shared_ptr.hpp>

#include <sigc++/signal.h>
#include <sigc++/slot.h>

namespace aptitude
{
  namespace util
  {
    /** \brief A wrapper that passes each element in a dynamic set
     *  through a function before passing it to client code.
     *
     *  \tparam From   The type of the values stored in the
     *                 underlying dynamic_set.
     *  \tparam To     The type of value produced by the transform.
     */
    template<typename From, typename To>
    class dynamic_set_transform : public dynamic_set<To>
    {
      boost::shared_ptr<dynamic_set<From> > wrapped_set;
      boost::function<To (From)> f;

      sigc::signal<void, To> signal_inserted;
      sigc::signal<void, To> signal_removed;

      void handle_inserted(const From &from);
      void handle_removed(const From &from);

    public:
      /** \warning Should only be used by create(). */
      dynamic_set_transform(const boost::shared_ptr<dynamic_set<From> > &_wrapped_set,
                            const boost::function<To (From)> &_f);

      static boost::shared_ptr<dynamic_set_transform>
      create(const boost::shared_ptr<dynamic_set<From> > &wrapped_set,
             const boost::function<To (From)> &f);

      std::size_t size();
      boost::shared_ptr<enumerator<To> > enumerate();
      sigc::connection connect_inserted(const sigc::slot<void, To> &slot);
      sigc::connection connect_removed(const sigc::slot<void, To> &slot);
    };

    template<typename From, typename To>
    dynamic_set_transform<From, To>::dynamic_set_transform(const boost::shared_ptr<dynamic_set<From> > &_wrapped_set,
                                                           const boost::function<To (From)> &_f)
      : wrapped_set(_wrapped_set),
        f(_f)
    {
      wrapped_set->connect_inserted(sigc::mem_fun(*this, &dynamic_set_transform::handle_inserted));
      wrapped_set->connect_removed(sigc::mem_fun(*this, &dynamic_set_transform::handle_removed));
    }

    template<typename From, typename To>
    boost::shared_ptr<dynamic_set_transform<From, To> >
    dynamic_set_transform<From, To>::create(const boost::shared_ptr<dynamic_set<From> > &wrapped_set,
                                            const boost::function<To (From)> &f)
    {
      return boost::make_shared<dynamic_set_transform<From, To> >(wrapped_set, f);
    }

    template<typename From, typename To>
    std::size_t dynamic_set_transform<From, To>::size()
    {
      return wrapped_set->size();
    }

    template<typename From, typename To>
    boost::shared_ptr<enumerator<To> > dynamic_set_transform<From, To>::enumerate()
    {
      return enumerator_transform<From, To>::create(wrapped_set->enumerate(), f);
    }

    template<typename From, typename To>
    void dynamic_set_transform<From, To>::handle_inserted(const From &from)
    {
      signal_inserted(f(from));
    }

    template<typename From, typename To>
    void dynamic_set_transform<From, To>::handle_removed(const From &from)
    {
      signal_removed(f(from));
    }

    template<typename From, typename To>
    sigc::connection dynamic_set_transform<From, To>::connect_inserted(const sigc::slot<void, To> &slot)
    {
      return signal_inserted.connect(slot);
    }

    template<typename From, typename To>
    sigc::connection dynamic_set_transform<From, To>::connect_removed(const sigc::slot<void, To> &slot)
    {
      return signal_removed.connect(slot);
    }
  }
}

#endif // DYNAMIC_SET_TRANSFORM_H