remoting/protocol/input_event_tracker.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "remoting/protocol/input_event_tracker.h"
   6
   7 #include "base/logging.h"
   8 #include "remoting/proto/event.pb.h"
   9
  10 namespace remoting {
  11 namespace protocol {
  12
  13 InputEventTracker::InputEventTracker(InputStub* input_stub)
  14     : input_stub_(input_stub),
  15       mouse_button_state_(0) {
  16 }
  17
  18 InputEventTracker::~InputEventTracker() {
  19 }
  20
  21 bool InputEventTracker::IsKeyPressed(uint32 usb_keycode) const {
  22   return pressed_keys_.find(usb_keycode) != pressed_keys_.end();
  23 }
  24
  25 int InputEventTracker::PressedKeyCount() const {
  26   return pressed_keys_.size();
  27 }
  28
  29 void InputEventTracker::ReleaseAll() {
  30   std::set<uint32>::iterator i;
  31   for (i = pressed_keys_.begin(); i != pressed_keys_.end(); ++i) {
  32     KeyEvent event;
  33     event.set_pressed(false);
  34     event.set_usb_keycode(*i);
  35     input_stub_->InjectKeyEvent(event);
  36   }
  37   pressed_keys_.clear();
  38
  39   for (int i = MouseEvent::BUTTON_UNDEFINED + 1;
  40        i < MouseEvent::BUTTON_MAX; ++i) {
  41     if (mouse_button_state_ & (1 << (i - 1))) {
  42       MouseEvent mouse;
  43
  44       // TODO(wez): EventInjectors should cope with positionless events by
  45       // using the current cursor position, and we wouldn't set position here.
  46       mouse.set_x(mouse_pos_.x());
  47       mouse.set_y(mouse_pos_.y());
  48
  49       mouse.set_button((MouseEvent::MouseButton)i);
  50       mouse.set_button_down(false);
  51       input_stub_->InjectMouseEvent(mouse);
  52     }
  53   }
  54   mouse_button_state_ = 0;
  55 }
  56
  57 void InputEventTracker::InjectKeyEvent(const KeyEvent& event) {
  58   // We don't need to track the keyboard lock states of key down events.
  59   // Pressed keys will be released with |lock_states| set to 0.
  60   // The lock states of auto generated key up events don't matter as long as
  61   // we release all the pressed keys at blurring/disconnection time.
  62   if (event.has_pressed()) {
  63     if (event.has_usb_keycode()) {
  64       if (event.pressed()) {
  65         pressed_keys_.insert(event.usb_keycode());
  66       } else {
  67         pressed_keys_.erase(event.usb_keycode());
  68       }
  69     }
  70   }
  71   input_stub_->InjectKeyEvent(event);
  72 }
  73
  74 void InputEventTracker::InjectMouseEvent(const MouseEvent& event) {
  75   if (event.has_x() && event.has_y()) {
  76     mouse_pos_ = webrtc::DesktopVector(event.x(), event.y());
  77   }
  78   if (event.has_button() && event.has_button_down()) {
  79     // Button values are defined in remoting/proto/event.proto.
  80     if (event.button() >= 1 && event.button() < MouseEvent::BUTTON_MAX) {
  81       uint32 button_change = 1 << (event.button() - 1);
  82       if (event.button_down()) {
  83         mouse_button_state_ |= button_change;
  84       } else {
  85         mouse_button_state_ &= ~button_change;
  86       }
  87     }
  88   }
  89   input_stub_->InjectMouseEvent(event);
  90 }
  91
  92 }  // namespace protocol
  93 }  // namespace remoting