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
| import { types } from "mobx-state-tree";
|
| import BaseTool from "./Base";
| import ToolMixin from "../mixins/Tool";
| import { NodeViews } from "../components/Node/Node";
| import { DrawingTool } from "../mixins/DrawingTool";
| import { FF_DEV_3793, isFF } from "../utils/feature-flags";
|
| const _Tool = types
| .model("KeyPointTool", {
| default: types.optional(types.boolean, true),
| group: "segmentation",
| shortcut: "tool:key-point",
| smart: true,
| })
| .views(() => ({
| get tagTypes() {
| return {
| stateTypes: "keypointlabels",
| controlTagTypes: ["keypointlabels", "keypoint"],
| };
| },
| get viewTooltip() {
| return "Key Point";
| },
| get iconComponent() {
| return self.dynamic ? NodeViews.KeyPointRegionModel.altIcon : NodeViews.KeyPointRegionModel.icon;
| },
| }))
| .actions((self) => ({
| clickEv(ev, [x, y]) {
| if (!self.canStartDrawing()) return;
| if (!self.isAllowedInteraction(ev)) return;
|
| const c = self.control;
|
| if (c.type === "keypointlabels" && !c.isSelected) return;
| if (self.annotation.isReadOnly()) return;
|
| const keyPoint = self.createRegion({
| ...self.control?.getSnappedPoint({
| x,
| y,
| }),
| ...(isFF(FF_DEV_3793)
| ? {
| // strokeWidth is visual only, so it's in screen dimensions in config
| width: self.obj.canvasToInternalX(Number(c.strokewidth)),
| }
| : {
| width: Number(c.strokewidth),
| coordstype: "px",
| }),
| dynamic: self.dynamic,
| negative: self.dynamic && ev.altKey,
| });
|
| keyPoint.setDrawing(false);
| keyPoint.notifyDrawingFinished();
| },
| }));
|
| const KeyPoint = types.compose(_Tool.name, ToolMixin, BaseTool, DrawingTool, _Tool);
|
| export { KeyPoint };
|
|
