Dogcows Code - chaz/openbox/blob - src/bindings.cc

   1 // -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*-
   2
   3 #ifdef HAVE_CONFIG_H
   4 # include "../config.h"
   5 #endif
   6
   7 #include "bindings.hh"
   8 #include "screen.hh"
   9 #include "openbox.hh"
  10 #include "client.hh"
  11 #include "frame.hh"
  12 #include "python.hh"
  13 #include "otk/display.hh"
  14
  15 extern "C" {
  16 #include <X11/Xlib.h>
  17
  18 #include "gettext.h"
  19 #define _(str) gettext(str)
  20 }
  21
  22 #include <algorithm>
  23
  24 namespace ob {
  25
  26 static bool buttonvalue(const std::string &button, unsigned int *val)
  27 {
  28   if (button == "Left" || button == "1" || button == "Button1") {
  29     *val |= Button1;
  30   } else if (button == "Middle" || button == "2" || button == "Button2") {
  31     *val |= Button2;
  32   } else if (button == "Right" || button == "3" || button == "Button3") {
  33     *val |= Button3;
  34   } else if (button == "Up" || button == "4" || button == "Button4") {
  35     *val |= Button4;
  36   } else if (button == "Down" || button == "5" || button == "Button5") {
  37     *val |= Button5;
  38   } else
  39     return false;
  40   return true;
  41 }
  42
  43 static bool modvalue(const std::string &mod, unsigned int *val)
  44 {
  45   if (mod == "C") {           // control
  46     *val |= ControlMask;
  47   } else if (mod == "S") {    // shift
  48     *val |= ShiftMask;
  49   } else if (mod == "A" ||    // alt/mod1
  50              mod == "M" ||
  51              mod == "Mod1" ||
  52              mod == "M1") {
  53     *val |= Mod1Mask;
  54   } else if (mod == "Mod2" ||   // mod2
  55              mod == "M2") {
  56     *val |= Mod2Mask;
  57   } else if (mod == "Mod3" ||   // mod3
  58              mod == "M3") {
  59     *val |= Mod3Mask;
  60   } else if (mod == "W" ||    // windows/mod4
  61              mod == "Mod4" ||
  62              mod == "M4") {
  63     *val |= Mod4Mask;
  64   } else if (mod == "Mod5" ||   // mod5
  65              mod == "M5") {
  66     *val |= Mod5Mask;
  67   } else {                    // invalid
  68     return false;
  69   }
  70   return true;
  71 }
  72
  73 bool Bindings::translate(const std::string &str, Binding &b,bool askey) const
  74 {
  75   // parse out the base key name
  76   std::string::size_type keybegin = str.find_last_of('-');
  77   keybegin = (keybegin == std::string::npos) ? 0 : keybegin + 1;
  78   std::string key(str, keybegin);
  79
  80   // parse out the requested modifier keys
  81   unsigned int modval = 0;
  82   std::string::size_type begin = 0, end;
  83   while (begin != keybegin) {
  84     end = str.find_first_of('-', begin);
  85
  86     std::string mod(str, begin, end-begin);
  87     if (!modvalue(mod, &modval)) {
  88       printf(_("Invalid modifier element in key binding: %s\n"), mod.c_str());
  89       return false;
  90     }
  91
  92     begin = end + 1;
  93   }
  94
  95   // set the binding
  96   b.modifiers = modval;
  97   if (askey) {
  98     KeySym sym = XStringToKeysym(const_cast<char *>(key.c_str()));
  99     if (sym == NoSymbol) {
 100       printf(_("Invalid Key name in key binding: %s\n"), key.c_str());
 101       return false;
 102     }
 103     if (!(b.key = XKeysymToKeycode(**otk::display, sym)))
 104       printf(_("No valid keycode for Key in key binding: %s\n"), key.c_str());
 105     return b.key != 0;
 106   } else {
 107     return buttonvalue(key, &b.key);
 108   }
 109 }
 110
 111 static void destroytree(KeyBindingTree *tree)
 112 {
 113   while (tree) {
 114     KeyBindingTree *c = tree->first_child;
 115     delete tree;
 116     tree = c;
 117   }
 118 }
 119
 120 KeyBindingTree *Bindings::buildtree(const StringVect &keylist,
 121                                       PyObject *callback) const
 122 {
 123   if (keylist.empty()) return 0; // nothing in the list.. return 0
 124
 125   KeyBindingTree *ret = 0, *p;
 126
 127   StringVect::const_reverse_iterator it, end = keylist.rend();
 128   for (it = keylist.rbegin(); it != end; ++it) {
 129     p = ret;
 130     ret = new KeyBindingTree();
 131     if (!p) {
 132       // this is the first built node, the bottom node of the tree
 133       ret->chain = false;
 134       ret->callbacks.push_back(callback);
 135     }
 136     ret->first_child = p;
 137     if (!translate(*it, ret->binding)) {
 138       destroytree(ret);
 139       ret = 0;
 140       break;
 141     }
 142   }
 143   return ret;
 144 }
 145
 146
 147 Bindings::Bindings()
 148   : _curpos(&_keytree),
 149     _resetkey(0,0),
 150     _timer((otk::Timer *) 0)
 151 {
 152 //  setResetKey("C-g"); // set the default reset key
 153 }
 154
 155
 156 Bindings::~Bindings()
 157 {
 158   if (_timer)
 159     delete _timer;
 160
 161   grabKeys(false);
 162   removeAllKeys();
 163 //  removeAllButtons(); XXX
 164   removeAllEvents();
 165 }
 166
 167
 168 void Bindings::assimilate(KeyBindingTree *node)
 169 {
 170   KeyBindingTree *a, *b, *tmp, *last;
 171
 172   if (!_keytree.first_child) {
 173     // there are no nodes at this level yet
 174     _keytree.first_child = node;
 175   } else {
 176     a = _keytree.first_child;
 177     last = a;
 178     b = node;
 179     while (a) {
 180       last = a;
 181       if (a->binding != b->binding) {
 182         a = a->next_sibling;
 183       } else {
 184         tmp = b;
 185         b = b->first_child;
 186         delete tmp;
 187         a = a->first_child;
 188       }
 189     }
 190     if (last->binding != b->binding)
 191       last->next_sibling = b;
 192     else {
 193       last->first_child = b->first_child;
 194       delete b;
 195     }
 196   }
 197 }
 198
 199
 200 KeyBindingTree *Bindings::find(KeyBindingTree *search,
 201                                  bool *conflict) const {
 202   *conflict = false;
 203   KeyBindingTree *a, *b;
 204   a = _keytree.first_child;
 205   b = search;
 206   while (a && b) {
 207     if (a->binding != b->binding) {
 208       a = a->next_sibling;
 209     } else {
 210       if (a->chain == b->chain) {
 211         if (!a->chain) {
 212           // found it! (return the actual id, not the search's)
 213           return a;
 214         }
 215       } else {
 216         *conflict = true;
 217         return 0; // the chain status' don't match (conflict!)
 218       }
 219       b = b->first_child;
 220       a = a->first_child;
 221     }
 222   }
 223   return 0; // it just isn't in here
 224 }
 225
 226
 227 bool Bindings::addKey(const StringVect &keylist, PyObject *callback)
 228 {
 229   KeyBindingTree *tree, *t;
 230   bool conflict;
 231
 232   if (!(tree = buildtree(keylist, callback)))
 233     return false; // invalid binding requested
 234
 235   t = find(tree, &conflict);
 236   if (conflict) {
 237     // conflicts with another binding
 238     destroytree(tree);
 239     return false;
 240   }
 241
 242   if (t) {
 243     // already bound to something
 244     // XXX: look if callback is already bound to this key?
 245     t->callbacks.push_back(callback);
 246     destroytree(tree);
 247   } else {
 248     // grab the server here to make sure no key pressed go missed
 249     otk::display->grab();
 250     grabKeys(false);
 251
 252     // assimilate this built tree into the main tree
 253     assimilate(tree); // assimilation destroys/uses the tree
 254
 255     grabKeys(true);
 256     otk::display->ungrab();
 257   }
 258
 259   Py_INCREF(callback);
 260
 261   return true;
 262 }
 263
 264
 265 bool Bindings::removeKey(const StringVect &keylist, PyObject *callback)
 266 {
 267   assert(false); // XXX: function not implemented yet
 268
 269   KeyBindingTree *tree;
 270   bool conflict;
 271
 272   if (!(tree = buildtree(keylist, 0)))
 273     return false; // invalid binding requested
 274
 275   KeyBindingTree *t = find(tree, &conflict);
 276   if (t) {
 277     CallbackList::iterator it = std::find(t->callbacks.begin(),
 278                                           t->callbacks.end(),
 279                                           callback);
 280     if (it != t->callbacks.end()) {
 281       // grab the server here to make sure no key pressed go missed
 282       otk::display->grab();
 283       grabKeys(false);
 284
 285       _curpos = &_keytree;
 286
 287       // XXX do shit here ...
 288       Py_XDECREF(*it);
 289
 290       grabKeys(true);
 291       otk::display->ungrab();
 292
 293       return true;
 294     }
 295   }
 296   return false;
 297 }
 298
 299
 300 void Bindings::setResetKey(const std::string &key)
 301 {
 302   Binding b(0, 0);
 303   if (translate(key, b)) {
 304     // grab the server here to make sure no key pressed go missed
 305     otk::display->grab();
 306     grabKeys(false);
 307     _resetkey.key = b.key;
 308     _resetkey.modifiers = b.modifiers;
 309     grabKeys(true);
 310     otk::display->ungrab();
 311   }
 312 }
 313
 314
 315 static void remove_branch(KeyBindingTree *first)
 316 {
 317   KeyBindingTree *p = first;
 318
 319   while (p) {
 320     if (p->first_child)
 321       remove_branch(p->first_child);
 322     KeyBindingTree *s = p->next_sibling;
 323     while(!p->callbacks.empty()) {
 324       Py_XDECREF(p->callbacks.front());
 325       p->callbacks.pop_front();
 326     }
 327     delete p;
 328     p = s;
 329   }
 330 }
 331
 332
 333 void Bindings::removeAllKeys()
 334 {
 335   grabKeys(false);
 336   if (_keytree.first_child) {
 337     remove_branch(_keytree.first_child);
 338     _keytree.first_child = 0;
 339   }
 340   grabKeys(true);
 341 }
 342
 343
 344 void Bindings::grabKeys(bool grab)
 345 {
 346   for (int i = 0; i < openbox->screenCount(); ++i) {
 347     Window root = otk::display->screenInfo(i)->rootWindow();
 348
 349     KeyBindingTree *p = _curpos->first_child;
 350     while (p) {
 351       if (grab) {
 352         otk::display->grabKey(p->binding.key, p->binding.modifiers,
 353                                 root, false, GrabModeAsync, GrabModeAsync,
 354                                 false);
 355       }
 356       else
 357         otk::display->ungrabKey(p->binding.key, p->binding.modifiers,
 358                                   root);
 359       p = p->next_sibling;
 360     }
 361
 362     if (_resetkey.key)
 363       if (grab)
 364         otk::display->grabKey(_resetkey.key, _resetkey.modifiers,
 365                                 root, false, GrabModeAsync, GrabModeAsync,
 366                                 false);
 367       else
 368         otk::display->ungrabKey(_resetkey.key, _resetkey.modifiers,
 369                                   root);
 370   }
 371 }
 372
 373
 374 void Bindings::fireKey(int screen, unsigned int modifiers, unsigned int key,
 375                          Time time)
 376 {
 377   if (key == _resetkey.key && modifiers == _resetkey.modifiers) {
 378     resetChains(this);
 379   } else {
 380     KeyBindingTree *p = _curpos->first_child;
 381     while (p) {
 382       if (p->binding.key == key && p->binding.modifiers == modifiers) {
 383         if (p->chain) {
 384           if (_timer)
 385             delete _timer;
 386           _timer = new otk::Timer(5000, // 5 second timeout
 387                                   (otk::Timer::TimeoutHandler)resetChains,
 388                                   this);
 389           // grab the server here to make sure no key pressed go missed
 390           otk::display->grab();
 391           grabKeys(false);
 392           _curpos = p;
 393           grabKeys(true);
 394           otk::display->ungrab();
 395         } else {
 396           Client *c = openbox->focusedClient();
 397           KeyData data(screen, c, time, modifiers, key);
 398           CallbackList::iterator it, end = p->callbacks.end();
 399           for (it = p->callbacks.begin(); it != end; ++it)
 400             python_callback(*it, &data);
 401           resetChains(this);
 402         }
 403         break;
 404       }
 405       p = p->next_sibling;
 406     }
 407   }
 408 }
 409
 410 void Bindings::resetChains(Bindings *self)
 411 {
 412   if (self->_timer) {
 413     delete self->_timer;
 414     self->_timer = (otk::Timer *) 0;
 415   }
 416   // grab the server here to make sure no key pressed go missed
 417   otk::display->grab();
 418   self->grabKeys(false);
 419   self->_curpos = &self->_keytree;
 420   self->grabKeys(true);
 421   otk::display->ungrab();
 422 }
 423
 424
 425 bool Bindings::addButton(const std::string &but, MouseContext context,
 426                            MouseAction action, PyObject *callback)
 427 {
 428   assert(context >= 0 && context < NUM_MOUSE_CONTEXT);
 429
 430   Binding b(0,0);
 431   if (!translate(but, b, false))
 432     return false;
 433
 434   ButtonBindingList::iterator it, end = _buttons[context].end();
 435
 436   // look for a duplicate binding
 437   for (it = _buttons[context].begin(); it != end; ++it)
 438     if ((*it)->binding.key == b.key &&
 439         (*it)->binding.modifiers == b.modifiers) {
 440       break;
 441     }
 442
 443   ButtonBinding *bind;
 444
 445   // the binding didnt exist yet, add it
 446   if (it == end) {
 447     bind = new ButtonBinding();
 448     bind->binding.key = b.key;
 449     bind->binding.modifiers = b.modifiers;
 450     _buttons[context].push_back(bind);
 451     // grab the button on all clients
 452     for (int sn = 0; sn < openbox->screenCount(); ++sn) {
 453       Screen *s = openbox->screen(sn);
 454       Client::List::iterator c_it, c_end = s->clients.end();
 455       for (c_it = s->clients.begin(); c_it != c_end; ++c_it) {
 456         grabButton(true, bind->binding, context, *c_it);
 457       }
 458     }
 459   } else
 460     bind = *it;
 461   bind->callbacks[action].push_back(callback);
 462   Py_INCREF(callback);
 463   return true;
 464 }
 465
 466 void Bindings::removeAllButtons()
 467 {
 468   for (int i = 0; i < NUM_MOUSE_CONTEXT; ++i) {
 469     ButtonBindingList::iterator it, end = _buttons[i].end();
 470     for (it = _buttons[i].begin(); it != end; ++it) {
 471       for (int a = 0; a < NUM_MOUSE_ACTION; ++a) {
 472         while (!(*it)->callbacks[a].empty()) {
 473           Py_XDECREF((*it)->callbacks[a].front());
 474           (*it)->callbacks[a].pop_front();
 475         }
 476       }
 477       // ungrab the button on all clients
 478       for (int sn = 0; sn < openbox->screenCount(); ++sn) {
 479         Screen *s = openbox->screen(sn);
 480         Client::List::iterator c_it, c_end = s->clients.end();
 481         for (c_it = s->clients.begin(); c_it != c_end; ++c_it) {
 482           grabButton(false, (*it)->binding, (MouseContext)i, *c_it);
 483         }
 484       }
 485     }
 486   }
 487 }
 488
 489 void Bindings::grabButton(bool grab, const Binding &b, MouseContext context,
 490                             Client *client)
 491 {
 492   Window win;
 493   int mode = GrabModeAsync;
 494   unsigned int mask;
 495   switch(context) {
 496   case MC_Frame:
 497     win = client->frame->window();
 498     mask = ButtonPressMask | ButtonMotionMask | ButtonReleaseMask;
 499     break;
 500   case MC_Window:
 501     win = client->frame->plate();
 502     mode = GrabModeSync; // this is handled in fireButton
 503     mask = ButtonPressMask; // can't catch more than this with Sync mode
 504                             // the release event is manufactured by the
 505                             // master buttonPressHandler
 506     break;
 507   default:
 508     // any other elements already get button events, don't grab on them
 509     return;
 510   }
 511   if (grab)
 512     otk::display->grabButton(b.key, b.modifiers, win, false, mask, mode,
 513                              GrabModeAsync, None, None, false);
 514   else
 515     otk::display->ungrabButton(b.key, b.modifiers, win);
 516 }
 517
 518 void Bindings::grabButtons(bool grab, Client *client)
 519 {
 520   for (int i = 0; i < NUM_MOUSE_CONTEXT; ++i) {
 521     ButtonBindingList::iterator it, end = _buttons[i].end();
 522     for (it = _buttons[i].begin(); it != end; ++it)
 523       grabButton(grab, (*it)->binding, (MouseContext)i, client);
 524   }
 525 }
 526
 527 void Bindings::fireButton(MouseData *data)
 528 {
 529   if (data->context == MC_Window) {
 530     // Replay the event, so it goes to the client, and ungrab the device.
 531     XAllowEvents(**otk::display, ReplayPointer, data->time);
 532   }
 533
 534   ButtonBindingList::iterator it, end = _buttons[data->context].end();
 535   for (it = _buttons[data->context].begin(); it != end; ++it)
 536     if ((*it)->binding.key == data->button &&
 537         (*it)->binding.modifiers == data->state) {
 538       CallbackList::iterator c_it,c_end = (*it)->callbacks[data->action].end();
 539       for (c_it = (*it)->callbacks[data->action].begin();
 540            c_it != c_end; ++c_it)
 541         python_callback(*c_it, data);
 542     }
 543 }
 544
 545
 546 bool Bindings::addEvent(EventAction action, PyObject *callback)
 547 {
 548   if (action < 0 || action >= NUM_EVENTS) {
 549     return false;
 550   }
 551 #ifdef    XKB
 552   if (action == EventBell && _eventlist[action].empty())
 553     XkbSelectEvents(**otk::display, XkbUseCoreKbd,
 554                     XkbBellNotifyMask, XkbBellNotifyMask);
 555 #endif // XKB
 556   _eventlist[action].push_back(callback);
 557   Py_INCREF(callback);
 558   return true;
 559 }
 560
 561 bool Bindings::removeEvent(EventAction action, PyObject *callback)
 562 {
 563   if (action < 0 || action >= NUM_EVENTS) {
 564     return false;
 565   }
 566
 567   CallbackList::iterator it = std::find(_eventlist[action].begin(),
 568                                         _eventlist[action].end(),
 569                                         callback);
 570   if (it != _eventlist[action].end()) {
 571     Py_XDECREF(*it);
 572     _eventlist[action].erase(it);
 573 #ifdef    XKB
 574     if (action == EventBell && _eventlist[action].empty())
 575       XkbSelectEvents(**otk::display, XkbUseCoreKbd,
 576                       XkbBellNotifyMask, 0);
 577 #endif // XKB
 578     return true;
 579   }
 580   return false;
 581 }
 582
 583 void Bindings::removeAllEvents()
 584 {
 585   for (int i = 0; i < NUM_EVENTS; ++i) {
 586     while (!_eventlist[i].empty()) {
 587       Py_XDECREF(_eventlist[i].front());
 588       _eventlist[i].pop_front();
 589     }
 590   }
 591 }
 592
 593 void Bindings::fireEvent(EventData *data)
 594 {
 595   CallbackList::iterator c_it, c_end = _eventlist[data->action].end();
 596   for (c_it = _eventlist[data->action].begin(); c_it != c_end; ++c_it)
 597     python_callback(*c_it, data);
 598 }
 599
 600 }