input/tablet: add tool_mode option to set tablet tools as relative input

Closes #4139.

10 files changed, 149 insertions, 6 deletions

diff --git a/include/sway/commands.h b/include/sway/commands.h
index f549626b..964b3661 100644
--- a/include/sway/commands.h
+++ b/include/sway/commands.h
@@ -264,6 +264,7 @@ sway_cmd input_cmd_scroll_button;
 sway_cmd input_cmd_scroll_method;
 sway_cmd input_cmd_tap;
 sway_cmd input_cmd_tap_button_map;
+sway_cmd input_cmd_tool_mode;
 sway_cmd input_cmd_xkb_capslock;
 sway_cmd input_cmd_xkb_file;
 sway_cmd input_cmd_xkb_layout;
diff --git a/include/sway/config.h b/include/sway/config.h
index ee1852d4..473f723b 100644
--- a/include/sway/config.h
+++ b/include/sway/config.h
@@ -6,11 +6,13 @@
 #include <time.h>
 #include <wlr/interfaces/wlr_switch.h>
 #include <wlr/types/wlr_box.h>
+#include <wlr/types/wlr_tablet_tool.h>
 #include <xkbcommon/xkbcommon.h>
 #include "../include/config.h"
 #include "list.h"
 #include "swaynag.h"
 #include "tree/container.h"
+#include "sway/input/tablet.h"
 #include "sway/tree/root.h"
 #include "wlr-layer-shell-unstable-v1-protocol.h"
 
@@ -116,6 +118,11 @@ enum input_config_mapped_to {
         MAPPED_TO_REGION,
 };
 
+struct input_config_tool {
+        enum wlr_tablet_tool_type type;
+        enum sway_tablet_tool_mode mode;
+};
+
 /**
  * options for input devices
  */
@@ -160,6 +167,8 @@ struct input_config {
         char *mapped_to_output;
         struct wlr_box *mapped_to_region;
 
+        list_t *tools;
+
         bool capturable;
         struct wlr_box region;
 };
diff --git a/include/sway/input/tablet.h b/include/sway/input/tablet.h
index f30e232a..d7e4c242 100644
--- a/include/sway/input/tablet.h
+++ b/include/sway/input/tablet.h
@@ -11,11 +11,17 @@ struct sway_tablet {
         struct wlr_tablet_v2_tablet *tablet_v2;
 };
 
+enum sway_tablet_tool_mode {
+        SWAY_TABLET_TOOL_MODE_ABSOLUTE,
+        SWAY_TABLET_TOOL_MODE_RELATIVE,
+};
+
 struct sway_tablet_tool {
         struct sway_seat *seat;
         struct sway_tablet *tablet;
         struct wlr_tablet_v2_tablet_tool *tablet_v2_tool;
 
+        enum sway_tablet_tool_mode mode;
         double tilt_x, tilt_y;
 
         struct wl_listener set_cursor;
diff --git a/sway/commands/input.c b/sway/commands/input.c
index 53db9a16..c9bb8e06 100644
--- a/sway/commands/input.c
+++ b/sway/commands/input.c
@@ -29,6 +29,7 @@ static struct cmd_handler input_handlers[] = {
         { "scroll_method", input_cmd_scroll_method },
         { "tap", input_cmd_tap },
         { "tap_button_map", input_cmd_tap_button_map },
+        { "tool_mode", input_cmd_tool_mode },
         { "xkb_file", input_cmd_xkb_file },
         { "xkb_layout", input_cmd_xkb_layout },
         { "xkb_model", input_cmd_xkb_model },
diff --git a/sway/commands/input/tool_mode.c b/sway/commands/input/tool_mode.c
new file mode 100644
index 00000000..04316857
--- /dev/null
+++ b/sway/commands/input/tool_mode.c
@@ -0,0 +1,73 @@
+#include <strings.h>
+#include "sway/commands.h"
+#include "sway/config.h"
+
+static void set_tool_mode(struct input_config *ic,
+                enum wlr_tablet_tool_type type, enum sway_tablet_tool_mode mode) {
+        for (int i = 0; i < ic->tools->length; i++) {
+                struct input_config_tool *tool = ic->tools->items[i];
+                if (tool->type == type) {
+                        tool->mode = mode;
+                        return;
+                }
+        }
+
+        struct input_config_tool *tool = calloc(1, sizeof(*tool));
+        if (tool) {
+                tool->type = type;
+                tool->mode = mode;
+                list_add(ic->tools, tool);
+        }
+}
+
+struct cmd_results *input_cmd_tool_mode(int argc, char **argv) {
+        struct cmd_results *error;
+        if ((error = checkarg(argc, "tool_mode", EXPECTED_AT_LEAST, 2))) {
+                return error;
+        }
+
+        struct input_config *ic = config->handler_context.input_config;
+        if (!ic) {
+                return cmd_results_new(CMD_FAILURE, "No input device defined.");
+        }
+
+        enum sway_tablet_tool_mode tool_mode;
+        if (!strcasecmp(argv[1], "absolute")) {
+                tool_mode = SWAY_TABLET_TOOL_MODE_ABSOLUTE;
+        } else if (!strcasecmp(argv[1], "relative")) {
+                tool_mode = SWAY_TABLET_TOOL_MODE_RELATIVE;
+        } else {
+                goto invalid_command;
+        }
+
+        if (!strcasecmp(argv[0], "*")) {
+                set_tool_mode(ic, WLR_TABLET_TOOL_TYPE_PEN, tool_mode);
+                set_tool_mode(ic, WLR_TABLET_TOOL_TYPE_ERASER, tool_mode);
+                set_tool_mode(ic, WLR_TABLET_TOOL_TYPE_BRUSH, tool_mode);
+                set_tool_mode(ic, WLR_TABLET_TOOL_TYPE_PENCIL, tool_mode);
+                set_tool_mode(ic, WLR_TABLET_TOOL_TYPE_AIRBRUSH, tool_mode);
+        } else {
+                enum wlr_tablet_tool_type tool_type;
+                if (!strcasecmp(argv[0], "pen")) {
+                        tool_type = WLR_TABLET_TOOL_TYPE_PEN;
+                } else if (!strcasecmp(argv[0], "eraser")) {
+                        tool_type = WLR_TABLET_TOOL_TYPE_ERASER;
+                } else if (!strcasecmp(argv[0], "brush")) {
+                        tool_type = WLR_TABLET_TOOL_TYPE_BRUSH;
+                } else if (!strcasecmp(argv[0], "pencil")) {
+                        tool_type = WLR_TABLET_TOOL_TYPE_PENCIL;
+                } else if (!strcasecmp(argv[0], "airbrush")) {
+                        tool_type = WLR_TABLET_TOOL_TYPE_AIRBRUSH;
+                } else {
+                        goto invalid_command;
+                }
+
+                set_tool_mode(ic, tool_type, tool_mode);
+        }
+
+        return cmd_results_new(CMD_SUCCESS, NULL);
+
+invalid_command:
+        return cmd_results_new(CMD_INVALID,
+                "Expected 'tool_mode <tool> <absolute|relative>'");
+}
diff --git a/sway/config/input.c b/sway/config/input.c
index 2ed9c016..a998e170 100644
--- a/sway/config/input.c
+++ b/sway/config/input.c
@@ -40,6 +40,7 @@ struct input_config *new_input_config(const char* identifier) {
         input->xkb_numlock = INT_MIN;
         input->xkb_capslock = INT_MIN;
         input->xkb_file_is_set = false;
+        input->tools = create_list();
 
         return input;
 }
@@ -153,6 +154,22 @@ void merge_input_config(struct input_config *dst, struct input_config *src) {
                 memcpy(dst->calibration_matrix.matrix, src->calibration_matrix.matrix,
                         sizeof(src->calibration_matrix.matrix));
         }
+        for (int i = 0; i < src->tools->length; i++) {
+                struct input_config_tool *src_tool = src->tools->items[i];
+                for (int j = 0; j < dst->tools->length; j++) {
+                        struct input_config_tool *dst_tool = dst->tools->items[j];
+                        if (src_tool->type == dst_tool->type) {
+                                dst_tool->mode = src_tool->mode;
+                                goto tool_merge_outer;
+                        }
+                }
+
+                struct input_config_tool *dst_tool = malloc(sizeof(*dst_tool));
+                memcpy(dst_tool, src_tool, sizeof(*dst_tool));
+                list_add(dst->tools, dst_tool);
+
+                tool_merge_outer:;
+        }
 }
 
 static bool validate_xkb_merge(struct input_config *dest,
@@ -358,6 +375,7 @@ void free_input_config(struct input_config *ic) {
         free(ic->mapped_from_region);
         free(ic->mapped_to_output);
         free(ic->mapped_to_region);
+        list_free_items_and_destroy(ic->tools);
         free(ic);
 }
 
diff --git a/sway/input/cursor.c b/sway/input/cursor.c
index 61d75b8a..e47410a5 100644
--- a/sway/input/cursor.c
+++ b/sway/input/cursor.c
@@ -567,14 +567,14 @@ static void handle_tablet_tool_position(struct sway_cursor *cursor,
                         ic->mapped_from_region, &x, &y);
         }
 
-        switch (tool->tablet_v2_tool->wlr_tool->type) {
-        case WLR_TABLET_TOOL_TYPE_LENS:
-        case WLR_TABLET_TOOL_TYPE_MOUSE:
-                wlr_cursor_move(cursor->cursor, input_device->wlr_device, dx, dy);
-                break;
-        default:
+        switch (tool->mode) {
+        case SWAY_TABLET_TOOL_MODE_ABSOLUTE:
                 wlr_cursor_warp_absolute(cursor->cursor, input_device->wlr_device,
                         change_x ? x : NAN, change_y ? y : NAN);
+                break;
+        case SWAY_TABLET_TOOL_MODE_RELATIVE:
+                wlr_cursor_move(cursor->cursor, input_device->wlr_device, dx, dy);
+                break;
         }
 
         double sx, sy;
diff --git a/sway/input/tablet.c b/sway/input/tablet.c
index b74347aa..5f81f772 100644
--- a/sway/input/tablet.c
+++ b/sway/input/tablet.c
@@ -140,6 +140,29 @@ void sway_tablet_tool_configure(struct sway_tablet *tablet,
                 return;
         }
 
+        switch (wlr_tool->type) {
+        case WLR_TABLET_TOOL_TYPE_LENS:
+        case WLR_TABLET_TOOL_TYPE_MOUSE:
+                tool->mode = SWAY_TABLET_TOOL_MODE_RELATIVE;
+                break;
+        default:
+                tool->mode = SWAY_TABLET_TOOL_MODE_ABSOLUTE;
+
+                struct input_config *ic = input_device_get_config(
+                        tablet->seat_device->input_device);
+                if (!ic) {
+                        break;
+                }
+
+                for (int i = 0; i < ic->tools->length; i++) {
+                        struct input_config_tool *tool_config = ic->tools->items[i];
+                        if (tool_config->type == wlr_tool->type) {
+                                tool->mode = tool_config->mode;
+                                break;
+                        }
+                }
+        }
+
         tool->seat = tablet->seat_device->sway_seat;
         tool->tablet = tablet;
         tool->tablet_v2_tool =
diff --git a/sway/meson.build b/sway/meson.build
index b65a5211..6e138101 100644
--- a/sway/meson.build
+++ b/sway/meson.build
@@ -168,6 +168,7 @@ sway_sources = files(
         'commands/input/scroll_method.c',
         'commands/input/tap.c',
         'commands/input/tap_button_map.c',
+        'commands/input/tool_mode.c',
         'commands/input/xkb_capslock.c',
         'commands/input/xkb_file.c',
         'commands/input/xkb_layout.c',
diff --git a/sway/sway-input.5.scd b/sway/sway-input.5.scd
index 9ec5eeae..d8180c1c 100644
--- a/sway/sway-input.5.scd
+++ b/sway/sway-input.5.scd
@@ -94,6 +94,17 @@ The following commands may only be used in the configuration file.
 *input* <identifier> xkb_numlock enabled|disabled
         Initially enables or disables NumLock on startup, the default is disabled.
 
+## TABLET CONFIGURATION
+
+*input* <identifier> tool_mode <tool> <absolute|relative>
+        Sets whether movement of a tablet tool should be treated as absolute or
+        relative; the default is absolute.
+
+        Valid values for _\<tool\>_ are currently "pen", "eraser", "brush",
+        "pencil", "airbrush", and the wildcard _\*_, which matches all tools.
+
+        Mouse and lens tools ignore this setting and are always treated as relative.
+
 ## MAPPING CONFIGURATION
 
 *input* <identifier> map_to_output <identifier>