react-image-annotate-master-custom 0.0.1

package/RegionLabel/index.js

@@ -0,0 +1,197 @@
+import _objectSpread from "@babel/runtime/helpers/esm/objectSpread";
+import React, { useRef, memo } from "react";
+import Paper from "@mui/material/Paper";
+import { makeStyles } from "@mui/styles";
+import { createTheme, ThemeProvider } from "@mui/material/styles";
+import styles from "./styles";
+import classnames from "classnames";
+import IconButton from "@mui/material/IconButton";
+import Button from "@mui/material/Button";
+import TrashIcon from "@mui/icons-material/Delete";
+import CheckIcon from "@mui/icons-material/Check";
+import TextField from "@mui/material/TextField";
+import Select from "react-select";
+import CreatableSelect from "react-select/creatable";
+import { asMutable } from "seamless-immutable";
+var theme = createTheme();
+var useStyles = makeStyles(function (theme) {
+  return styles;
+});
+export var RegionLabel = function RegionLabel(_ref) {
+  var region = _ref.region,
+      editing = _ref.editing,
+      allowedClasses = _ref.allowedClasses,
+      allowedTags = _ref.allowedTags,
+      onDelete = _ref.onDelete,
+      _onChange = _ref.onChange,
+      onClose = _ref.onClose,
+      onOpen = _ref.onOpen,
+      onRegionClassAdded = _ref.onRegionClassAdded,
+      allowComments = _ref.allowComments;
+  var classes = useStyles();
+  var commentInputRef = useRef(null);
+
+  var onCommentInputClick = function onCommentInputClick(_) {
+    // The TextField wraps the <input> tag with two divs
+    var commentInput = commentInputRef.current.children[0].children[0];
+    if (commentInput) return commentInput.focus();
+  };
+
+  return React.createElement(ThemeProvider, {
+    theme: theme
+  }, React.createElement(Paper, {
+    onClick: function onClick() {
+      return !editing ? onOpen(region) : null;
+    },
+    className: classnames(classes.regionInfo, {
+      highlighted: region.highlighted
+    })
+  }, React.createElement("div", {
+    style: {
+      display: "".concat(!editing ? "unset" : "none")
+    }
+  }, region.cls && React.createElement("div", {
+    className: "name"
+  }, React.createElement("div", {
+    className: "circle",
+    style: {
+      backgroundColor: region.color
+    }
+  }), region.cls), region.tags && React.createElement("div", {
+    className: "tags"
+  }, region.tags.map(function (t) {
+    return React.createElement("div", {
+      key: t,
+      className: "tag"
+    }, t);
+  }))), React.createElement("div", {
+    style: {
+      width: 200,
+      display: "".concat(editing ? "unset" : "none")
+    }
+  }, React.createElement("div", {
+    style: {
+      display: "flex",
+      flexDirection: "row"
+    }
+  }, React.createElement("div", {
+    style: {
+      display: "flex",
+      backgroundColor: region.color || "#888",
+      color: "#fff",
+      padding: 4,
+      paddingLeft: 8,
+      paddingRight: 8,
+      borderRadius: 4,
+      fontWeight: "bold",
+      textShadow: "0px 0px 5px rgba(0,0,0,0.4)"
+    }
+  }, region.type), React.createElement("div", {
+    style: {
+      flexGrow: 1
+    }
+  }), React.createElement(IconButton, {
+    onClick: function onClick() {
+      return onDelete(region);
+    },
+    tabIndex: -1,
+    style: {
+      width: 22,
+      height: 22
+    },
+    size: "small",
+    variant: "outlined",
+    id: "delete-region-".concat(region.id)
+  }, React.createElement(TrashIcon, {
+    style: {
+      marginTop: -8,
+      width: 16,
+      height: 16
+    }
+  }))), (allowedClasses || []).length > 0 && React.createElement("div", {
+    style: {
+      marginTop: 6
+    }
+  }, React.createElement(CreatableSelect, {
+    placeholder: "Classification",
+    onChange: function onChange(o, actionMeta) {
+      if (actionMeta.action == "create-option") {
+        onRegionClassAdded(o.value);
+      }
+
+      return _onChange(_objectSpread({}, region, {
+        cls: o.value
+      }));
+    },
+    value: region.cls ? {
+      label: region.cls,
+      value: region.cls
+    } : null,
+    options: asMutable(allowedClasses.map(function (c) {
+      return {
+        value: c,
+        label: c
+      };
+    }))
+  })), (allowedTags || []).length > 0 && React.createElement("div", {
+    style: {
+      marginTop: 4
+    }
+  }, React.createElement(Select, {
+    onChange: function onChange(newTags) {
+      return _onChange(_objectSpread({}, region, {
+        tags: newTags.map(function (t) {
+          return t.value;
+        })
+      }));
+    },
+    placeholder: "Tags",
+    value: (region.tags || []).map(function (c) {
+      return {
+        label: c,
+        value: c
+      };
+    }),
+    isMulti: true,
+    options: asMutable(allowedTags.map(function (c) {
+      return {
+        value: c,
+        label: c
+      };
+    }))
+  })), allowComments && React.createElement(TextField, {
+    InputProps: {
+      className: classes.commentBox
+    },
+    fullWidth: true,
+    multiline: true,
+    rows: 3,
+    ref: commentInputRef,
+    onClick: onCommentInputClick,
+    value: region.comment || "",
+    onChange: function onChange(event) {
+      return _onChange(_objectSpread({}, region, {
+        comment: event.target.value
+      }));
+    }
+  }), onClose && React.createElement("div", {
+    style: {
+      marginTop: 4,
+      display: "flex"
+    }
+  }, React.createElement("div", {
+    style: {
+      flexGrow: 1
+    }
+  }), React.createElement(Button, {
+    onClick: function onClick() {
+      return onClose(region);
+    },
+    size: "small",
+    variant: "contained",
+    color: "primary"
+  }, React.createElement(CheckIcon, null))))));
+};
+export default memo(RegionLabel, function (prevProps, nextProps) {
+  return prevProps.editing === nextProps.editing && prevProps.region === nextProps.region;
+});

package/RegionLabel/styles.js

@@ -0,0 +1,48 @@
+import { grey } from "@mui/material/colors";
+export default {
+  regionInfo: {
+    fontSize: 12,
+    cursor: "default",
+    transition: "opacity 200ms",
+    opacity: 0.5,
+    "&:hover": {
+      opacity: 0.9,
+      cursor: "pointer"
+    },
+    "&.highlighted": {
+      opacity: 0.9,
+      "&:hover": {
+        opacity: 1
+      }
+    },
+    // pointerEvents: "none",
+    fontWeight: 600,
+    color: grey[900],
+    padding: 8,
+    "& .name": {
+      display: "flex",
+      flexDirection: "row",
+      alignItems: "center",
+      "& .circle": {
+        marginRight: 4,
+        boxShadow: "0px 0px 2px rgba(0,0,0,0.4)",
+        width: 10,
+        height: 10,
+        borderRadius: 5
+      }
+    },
+    "& .tags": {
+      "& .tag": {
+        color: grey[700],
+        display: "inline-block",
+        margin: 1,
+        fontSize: 10,
+        textDecoration: "underline"
+      }
+    }
+  },
+  commentBox: {
+    fontWeight: 400,
+    fontSize: 13
+  }
+};

package/RegionSelectAndTransformBoxes/index.js

@@ -0,0 +1,166 @@
+import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
+import React, { Fragment, memo } from "react";
+import HighlightBox from "../HighlightBox";
+import { styled } from "@mui/material/styles";
+import { createTheme, ThemeProvider } from "@mui/material/styles";
+import PreventScrollToParents from "../PreventScrollToParents";
+import Tooltip from "@mui/material/Tooltip";
+var theme = createTheme();
+var TransformGrabber = styled("div")(function (_ref) {
+  var theme = _ref.theme;
+  return {
+    width: 8,
+    height: 8,
+    zIndex: 2,
+    border: "2px solid #FFF",
+    position: "absolute"
+  };
+});
+var boxCursorMap = [["nw-resize", "n-resize", "ne-resize"], ["w-resize", "grab", "e-resize"], ["sw-resize", "s-resize", "se-resize"]];
+
+var arePropsEqual = function arePropsEqual(prev, next) {
+  return prev.region === next.region && prev.dragWithPrimary === next.dragWithPrimary && prev.createWithPrimary === next.createWithPrimary && prev.zoomWithPrimary === next.zoomWithPrimary && prev.mat === next.mat;
+};
+
+export var RegionSelectAndTransformBox = memo(function (_ref2) {
+  var r = _ref2.region,
+      mouseEvents = _ref2.mouseEvents,
+      projectRegionBox = _ref2.projectRegionBox,
+      dragWithPrimary = _ref2.dragWithPrimary,
+      createWithPrimary = _ref2.createWithPrimary,
+      zoomWithPrimary = _ref2.zoomWithPrimary,
+      onBeginMovePoint = _ref2.onBeginMovePoint,
+      onSelectRegion = _ref2.onSelectRegion,
+      layoutParams = _ref2.layoutParams,
+      _ref2$fullImageSegmen = _ref2.fullImageSegmentationMode,
+      fullImageSegmentationMode = _ref2$fullImageSegmen === void 0 ? false : _ref2$fullImageSegmen,
+      mat = _ref2.mat,
+      onBeginBoxTransform = _ref2.onBeginBoxTransform,
+      onBeginMovePolygonPoint = _ref2.onBeginMovePolygonPoint,
+      onBeginMoveKeypoint = _ref2.onBeginMoveKeypoint,
+      onAddPolygonPoint = _ref2.onAddPolygonPoint,
+      showHighlightBox = _ref2.showHighlightBox;
+  var pbox = projectRegionBox(r);
+  var _layoutParams$current = layoutParams.current,
+      iw = _layoutParams$current.iw,
+      ih = _layoutParams$current.ih;
+  return React.createElement(ThemeProvider, {
+    theme: theme
+  }, React.createElement(Fragment, null, React.createElement(PreventScrollToParents, null, showHighlightBox && r.type !== "polygon" && React.createElement(HighlightBox, {
+    region: r,
+    mouseEvents: mouseEvents,
+    dragWithPrimary: dragWithPrimary,
+    createWithPrimary: createWithPrimary,
+    zoomWithPrimary: zoomWithPrimary,
+    onBeginMovePoint: onBeginMovePoint,
+    onSelectRegion: onSelectRegion,
+    pbox: pbox
+  }), r.type === "box" && !dragWithPrimary && !zoomWithPrimary && !r.locked && r.highlighted && mat.a < 1.2 && [[0, 0], [0.5, 0], [1, 0], [1, 0.5], [1, 1], [0.5, 1], [0, 1], [0, 0.5], [0.5, 0.5]].map(function (_ref3, i) {
+    var _ref4 = _slicedToArray(_ref3, 2),
+        px = _ref4[0],
+        py = _ref4[1];
+
+    return React.createElement(TransformGrabber, Object.assign({
+      key: i
+    }, mouseEvents, {
+      onMouseDown: function onMouseDown(e) {
+        if (e.button === 0) return onBeginBoxTransform(r, [px * 2 - 1, py * 2 - 1]);
+        mouseEvents.onMouseDown(e);
+      },
+      style: {
+        left: pbox.x - 4 - 2 + pbox.w * px,
+        top: pbox.y - 4 - 2 + pbox.h * py,
+        cursor: boxCursorMap[py * 2][px * 2],
+        borderRadius: px === 0.5 && py === 0.5 ? 4 : undefined
+      }
+    }));
+  }), r.type === "polygon" && !dragWithPrimary && !zoomWithPrimary && !r.locked && r.highlighted && r.points.map(function (_ref5, i) {
+    var _ref6 = _slicedToArray(_ref5, 2),
+        px = _ref6[0],
+        py = _ref6[1];
+
+    var proj = mat.clone().inverse().applyToPoint(px * iw, py * ih);
+    return React.createElement(TransformGrabber, Object.assign({
+      key: i
+    }, mouseEvents, {
+      onMouseDown: function onMouseDown(e) {
+        if (e.button === 0 && (!r.open || i === 0)) return onBeginMovePolygonPoint(r, i);
+        mouseEvents.onMouseDown(e);
+      },
+      style: {
+        cursor: !r.open ? "move" : i === 0 ? "pointer" : undefined,
+        zIndex: 10,
+        pointerEvents: r.open && i === r.points.length - 1 ? "none" : undefined,
+        left: proj.x - 4,
+        top: proj.y - 4
+      }
+    }));
+  }), r.type === "polygon" && r.highlighted && !dragWithPrimary && !zoomWithPrimary && !r.locked && !r.open && r.points.length > 1 && r.points.map(function (p1, i) {
+    return [p1, r.points[(i + 1) % r.points.length]];
+  }).map(function (_ref7) {
+    var _ref8 = _slicedToArray(_ref7, 2),
+        p1 = _ref8[0],
+        p2 = _ref8[1];
+
+    return [(p1[0] + p2[0]) / 2, (p1[1] + p2[1]) / 2];
+  }).map(function (pa, i) {
+    var proj = mat.clone().inverse().applyToPoint(pa[0] * iw, pa[1] * ih);
+    return React.createElement(TransformGrabber, Object.assign({
+      key: i
+    }, mouseEvents, {
+      onMouseDown: function onMouseDown(e) {
+        if (e.button === 0) return onAddPolygonPoint(r, pa, i + 1);
+        mouseEvents.onMouseDown(e);
+      },
+      style: {
+        cursor: "copy",
+        zIndex: 10,
+        left: proj.x - 4,
+        top: proj.y - 4,
+        border: "2px dotted #fff",
+        opacity: 0.5
+      }
+    }));
+  }), r.type === "keypoints" && !dragWithPrimary && !zoomWithPrimary && !r.locked && r.highlighted && Object.entries(r.points).map(function (_ref9, i) {
+    var _ref10 = _slicedToArray(_ref9, 2),
+        keypointId = _ref10[0],
+        _ref10$ = _ref10[1],
+        px = _ref10$.x,
+        py = _ref10$.y;
+
+    var proj = mat.clone().inverse().applyToPoint(px * iw, py * ih);
+    return React.createElement(Tooltip, {
+      title: keypointId,
+      key: i
+    }, React.createElement(TransformGrabber, Object.assign({
+      key: i
+    }, mouseEvents, {
+      onMouseDown: function onMouseDown(e) {
+        if (e.button === 0 && (!r.open || i === 0)) return onBeginMoveKeypoint(r, keypointId);
+        mouseEvents.onMouseDown(e);
+      },
+      style: {
+        cursor: !r.open ? "move" : i === 0 ? "pointer" : undefined,
+        zIndex: 10,
+        pointerEvents: r.open && i === r.points.length - 1 ? "none" : undefined,
+        left: proj.x - 4,
+        top: proj.y - 4
+      }
+    })));
+  }))));
+}, arePropsEqual);
+export var RegionSelectAndTransformBoxes = memo(function (props) {
+  return props.regions.filter(function (r) {
+    return r.visible || r.visible === undefined;
+  }) //.filter((r) => !r.locked)
+  .map(function (r, i) {
+    return React.createElement(RegionSelectAndTransformBox, Object.assign({
+      key: r.id
+    }, props, {
+      region: r
+    }));
+  });
+}, function (n, p) {
+  return n.regions === p.regions && n.mat === p.mat;
+});
+export default RegionSelectAndTransformBoxes;

package/RegionSelectorSidebarBox/index.js

@@ -0,0 +1,248 @@
+import _objectSpread from "@babel/runtime/helpers/esm/objectSpread";
+import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
+import React, { Fragment, useState, memo } from "react";
+import SidebarBoxContainer from "../SidebarBoxContainer";
+import { makeStyles } from "@mui/styles";
+import { createTheme, ThemeProvider } from "@mui/material/styles";
+import { styled } from "@mui/material/styles";
+import { grey } from "@mui/material/colors";
+import RegionIcon from "@mui/icons-material/PictureInPicture";
+import Grid from "@mui/material/Grid";
+import ReorderIcon from "@mui/icons-material/SwapVert";
+import PieChartIcon from "@mui/icons-material/PieChart";
+import TrashIcon from "@mui/icons-material/Delete";
+import LockIcon from "@mui/icons-material/Lock";
+import UnlockIcon from "@mui/icons-material/LockOpen";
+import VisibleIcon from "@mui/icons-material/Visibility";
+import VisibleOffIcon from "@mui/icons-material/VisibilityOff";
+import styles from "./styles";
+import classnames from "classnames";
+import isEqual from "lodash/isEqual";
+var theme = createTheme();
+var useStyles = makeStyles(function (theme) {
+  return styles;
+});
+var HeaderSep = styled("div")(function (_ref) {
+  var theme = _ref.theme;
+  return {
+    borderTop: "1px solid ".concat(grey[200]),
+    marginTop: 2,
+    marginBottom: 2
+  };
+});
+
+var Chip = function Chip(_ref2) {
+  var color = _ref2.color,
+      text = _ref2.text;
+  var classes = useStyles();
+  return React.createElement("span", {
+    className: classes.chip
+  }, React.createElement("div", {
+    className: "color",
+    style: {
+      backgroundColor: color
+    }
+  }), React.createElement("div", {
+    className: "text"
+  }, text));
+};
+
+var RowLayout = function RowLayout(_ref3) {
+  var header = _ref3.header,
+      highlighted = _ref3.highlighted,
+      order = _ref3.order,
+      classification = _ref3.classification,
+      area = _ref3.area,
+      tags = _ref3.tags,
+      trash = _ref3.trash,
+      lock = _ref3.lock,
+      visible = _ref3.visible,
+      onClick = _ref3.onClick;
+  var classes = useStyles();
+
+  var _useState = useState(false),
+      _useState2 = _slicedToArray(_useState, 2),
+      mouseOver = _useState2[0],
+      changeMouseOver = _useState2[1];
+
+  return React.createElement("div", {
+    onClick: onClick,
+    onMouseEnter: function onMouseEnter() {
+      return changeMouseOver(true);
+    },
+    onMouseLeave: function onMouseLeave() {
+      return changeMouseOver(false);
+    },
+    className: classnames(classes.row, {
+      header: header,
+      highlighted: highlighted
+    })
+  }, React.createElement(Grid, {
+    container: true,
+    alignItems: "center"
+  }, React.createElement(Grid, {
+    item: true,
+    xs: 2
+  }, React.createElement("div", {
+    style: {
+      textAlign: "right",
+      paddingRight: 10
+    }
+  }, order)), React.createElement(Grid, {
+    item: true,
+    xs: 5
+  }, classification), React.createElement(Grid, {
+    item: true,
+    xs: 2
+  }, React.createElement("div", {
+    style: {
+      textAlign: "right",
+      paddingRight: 6
+    }
+  }, area)), React.createElement(Grid, {
+    item: true,
+    xs: 1
+  }, trash), React.createElement(Grid, {
+    item: true,
+    xs: 1
+  }, lock), React.createElement(Grid, {
+    item: true,
+    xs: 1
+  }, visible)));
+};
+
+var RowHeader = function RowHeader() {
+  return React.createElement(RowLayout, {
+    header: true,
+    highlighted: false,
+    order: React.createElement(ReorderIcon, {
+      className: "icon"
+    }),
+    classification: React.createElement("div", {
+      style: {
+        paddingLeft: 10
+      }
+    }, "Class"),
+    area: React.createElement(PieChartIcon, {
+      className: "icon"
+    }),
+    trash: React.createElement(TrashIcon, {
+      className: "icon"
+    }),
+    lock: React.createElement(LockIcon, {
+      className: "icon"
+    }),
+    visible: React.createElement(VisibleIcon, {
+      className: "icon"
+    })
+  });
+};
+
+var MemoRowHeader = memo(RowHeader);
+
+var Row = function Row(_ref4) {
+  var r = _ref4.region,
+      highlighted = _ref4.highlighted,
+      onSelectRegion = _ref4.onSelectRegion,
+      onDeleteRegion = _ref4.onDeleteRegion,
+      onChangeRegion = _ref4.onChangeRegion,
+      visible = _ref4.visible,
+      locked = _ref4.locked,
+      color = _ref4.color,
+      cls = _ref4.cls,
+      index = _ref4.index;
+  return React.createElement(RowLayout, {
+    header: false,
+    highlighted: highlighted,
+    onClick: function onClick() {
+      return onSelectRegion(r);
+    },
+    order: "#".concat(index + 1),
+    classification: React.createElement(Chip, {
+      text: cls || "",
+      color: color || "#ddd"
+    }),
+    area: "",
+    trash: React.createElement(TrashIcon, {
+      onClick: function onClick() {
+        return onDeleteRegion(r);
+      },
+      className: "icon2"
+    }),
+    lock: r.locked ? React.createElement(LockIcon, {
+      onClick: function onClick() {
+        return onChangeRegion(_objectSpread({}, r, {
+          locked: false
+        }));
+      },
+      className: "icon2"
+    }) : React.createElement(UnlockIcon, {
+      onClick: function onClick() {
+        return onChangeRegion(_objectSpread({}, r, {
+          locked: true
+        }));
+      },
+      className: "icon2"
+    }),
+    visible: r.visible || r.visible === undefined ? React.createElement(VisibleIcon, {
+      onClick: function onClick() {
+        return onChangeRegion(_objectSpread({}, r, {
+          visible: false
+        }));
+      },
+      className: "icon2"
+    }) : React.createElement(VisibleOffIcon, {
+      onClick: function onClick() {
+        return onChangeRegion(_objectSpread({}, r, {
+          visible: true
+        }));
+      },
+      className: "icon2"
+    })
+  });
+};
+
+var MemoRow = memo(Row, function (prevProps, nextProps) {
+  return prevProps.highlighted === nextProps.highlighted && prevProps.visible === nextProps.visible && prevProps.locked === nextProps.locked && prevProps.id === nextProps.id && prevProps.index === nextProps.index && prevProps.cls === nextProps.cls && prevProps.color === nextProps.color;
+});
+var emptyArr = [];
+export var RegionSelectorSidebarBox = function RegionSelectorSidebarBox(_ref5) {
+  var _ref5$regions = _ref5.regions,
+      regions = _ref5$regions === void 0 ? emptyArr : _ref5$regions,
+      onDeleteRegion = _ref5.onDeleteRegion,
+      onChangeRegion = _ref5.onChangeRegion,
+      onSelectRegion = _ref5.onSelectRegion;
+  var classes = useStyles();
+  return React.createElement(ThemeProvider, {
+    theme: theme
+  }, React.createElement(SidebarBoxContainer, {
+    title: "Regions",
+    subTitle: "",
+    icon: React.createElement(RegionIcon, {
+      style: {
+        color: grey[700]
+      }
+    }),
+    expandedByDefault: true
+  }, React.createElement("div", {
+    className: classes.container
+  }, React.createElement(MemoRowHeader, null), React.createElement(HeaderSep, null), regions.map(function (r, i) {
+    return React.createElement(MemoRow, Object.assign({
+      key: r.id
+    }, r, {
+      region: r,
+      index: i,
+      onSelectRegion: onSelectRegion,
+      onDeleteRegion: onDeleteRegion,
+      onChangeRegion: onChangeRegion
+    }));
+  }))));
+};
+
+var mapUsedRegionProperties = function mapUsedRegionProperties(r) {
+  return [r.id, r.color, r.locked, r.visible, r.highlighted];
+};
+
+export default memo(RegionSelectorSidebarBox, function (prevProps, nextProps) {
+  return isEqual((prevProps.regions || emptyArr).map(mapUsedRegionProperties), (nextProps.regions || emptyArr).map(mapUsedRegionProperties));
+});