~didrocks/unity/fixes-natty-finale

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
// -*- Mode: C++; indent-tabs-mode: nil; tab-width: 2 -*-
/*
 * Copyright (C) 2010 Canonical Ltd
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 *
 * 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.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Authored by: Jason Smith <jason.smith@canonical.com>
 */

#ifndef LAUNCHERMODEL_H
#define LAUNCHERMODEL_H

#include "LauncherIcon.h"

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

class LauncherModel : public sigc::trackable
{

public:
    typedef std::list<LauncherIcon*> Base;
    typedef Base::iterator iterator; 
    typedef Base::reverse_iterator reverse_iterator; 
    
    LauncherModel();
    ~LauncherModel();

    void AddIcon (LauncherIcon *icon);
    void RemoveIcon (LauncherIcon *icon);
    void Sort ();
    int  Size ();

    void OnIconRemove (LauncherIcon *icon);
    
    bool IconHasSister (LauncherIcon *icon);
    
    void ReorderBefore (LauncherIcon *icon, LauncherIcon *other, bool save);

    void ReorderSmart (LauncherIcon *icon, LauncherIcon *other, bool save);
    
    iterator begin ();
    iterator end ();
    reverse_iterator rbegin ();
    reverse_iterator rend ();
    
    iterator main_begin ();
    iterator main_end ();
    reverse_iterator main_rbegin ();
    reverse_iterator main_rend ();
    
    iterator shelf_begin ();
    iterator shelf_end ();
    reverse_iterator shelf_rbegin ();
    reverse_iterator shelf_rend ();
    
    sigc::signal<void, LauncherIcon *> icon_added;
    sigc::signal<void, LauncherIcon *> icon_removed;
    sigc::signal<void> order_changed;
    
private:
    Base _inner;
    Base _inner_shelf;
    Base _inner_main;
    
    void Populate ();
    
    bool IconShouldShelf (LauncherIcon *icon);
    
    static gboolean RemoveCallback (gpointer data);
    
    static bool CompareIcons (LauncherIcon *first, LauncherIcon *second);

/* Template Methods */
public:
    template<class T>
    std::list<T*> GetSublist ()
    {
      std::list<T*> result;
      
      iterator it;
      for (it = begin (); it != end (); it++)
      {
        T *var = dynamic_cast<T*> (*it);
        
        if (var)
          result.push_back (var);
      }
      
      return result;
    }
};

#endif // LAUNCHERMODEL_H