~sethj/ubuntu/wily/unity/fix-for-1445595

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
/*
 * Copyright 2012 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 warranties of
 * MERCHANTABILITY, SATISFACTORY QUALITY 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
 * version 3 along with this program.  If not, see
 * <http://www.gnu.org/licenses/>
 *
 * Authored by: Marco Trevisan (TreviƱo) <marco.trevisan@canonical.com>

 */

#ifndef TEST_USCREEN_MOCK_H
#define TEST_USCREEN_MOCK_H

#include "MultiMonitor.h"
#include "UScreen.h"

namespace unity
{

struct MockUScreen : UScreen
{
  MockUScreen()
  {
    Reset(false);
  }

  ~MockUScreen()
  {
    if (default_screen_ == this)
      default_screen_ = nullptr;
  }

  void Reset(bool emit_change = true)
  {
    default_screen_ = this;
    primary_ = 0;
    monitors_ = {nux::Geometry(0, 0, MONITOR_WIDTH, MONITOR_HEIGHT)};

    if (emit_change)
      changed.emit(primary_, monitors_);
  }

  void SetupFakeMultiMonitor(int primary = 0, bool emit_change = true)
  {
    SetPrimary(primary, false);
    monitors_.clear();

    for (unsigned i = 0, total_width = 0; i < monitors::MAX; ++i)
    {
      monitors_.push_back(nux::Geometry(total_width, 0, MONITOR_WIDTH, MONITOR_HEIGHT));
      total_width += MONITOR_WIDTH;

      if (emit_change)
        changed.emit(GetPrimaryMonitor(), GetMonitors());
    }
  }

  void SetPrimary(int primary, bool emit = true)
  {
    if (primary_ != primary)
    {
      primary_ = primary;

      if (emit)
        changed.emit(primary_, monitors_);
    }
  }

  void SetMonitors(std::vector<nux::Geometry> const& monitors)
  {
    if (!std::equal(monitors_.begin(), monitors_.end(), monitors.begin()))
    {
      monitors_ = monitors;
      changed.emit(primary_, monitors_);
    }
  }

  static const unsigned MONITOR_WIDTH = 1024;
  static const unsigned MONITOR_HEIGHT = 768;
};

}

#endif