lost-sia/src/utils/keyActions.js

Single Image Annotation Tool

export const EDIT_LABEL = 'editLabel'
export const DELETE_ANNO = 'deleteAnno'
export const ENTER_ANNO_ADD_MODE = 'enterAnnoAddMode'
export const LEAVE_ANNO_ADD_MODE = 'leaveAnnoAddMode'
export const UNDO = 'undo'
export const REDO = 'redo'
export const TRAVERSE_ANNOS = 'traverseAnnos'
export const CAM_MOVE_UP = 'camMoveUp'
export const CAM_MOVE_DOWN = 'camMoveDown'
export const CAM_MOVE_LEFT = 'camMoveLeft'
export const CAM_MOVE_RIGHT = 'camMoveRight'
export const CAM_MOVE_STOP = 'camMoveStop'
export const COPY_ANNOTATION = 'copyAnnotation'
export const PASTE_ANNOTATION = 'pasteAnnotation'
export const RECREATE_ANNO = 'recreateAnnotation'
export const DELETE_ANNO_IN_CREATION = 'deleteAnnoInCreation'
export const TOGGLE_ANNO_COMMENT_INPUT = 'toggleAnnoCommentInput'

class KeyMapper {
  constructor(keyActionHandler = undefined) {
    this.controlDown = false
    this.shiftDown = false
    this.keyActionHandler = keyActionHandler
  }

  keyDown(key) {
    switch (key) {
      case 'Enter':
        this.triggerKeyAction(EDIT_LABEL)
        break
      case 'Delete':
        this.triggerKeyAction(DELETE_ANNO)
        break
      case 'Backspace':
        this.triggerKeyAction(DELETE_ANNO)
        break
      case 'Control':
        this.controlDown = true
        this.triggerKeyAction(ENTER_ANNO_ADD_MODE)
        break
      case 'Shift':
        this.shiftDown = true
        break
      case 'z':
        if (this.controlDown) {
          this.triggerKeyAction(UNDO)
        }
        break
      case 'r':
        if (this.controlDown) {
          this.triggerKeyAction(REDO)
        }
        break
      case 'Tab':
        this.triggerKeyAction(TRAVERSE_ANNOS)
        break
      case 'w':
        this.triggerKeyAction(CAM_MOVE_UP)
        break
      case 's':
        this.triggerKeyAction(CAM_MOVE_DOWN)
        break
      case 'a':
        this.triggerKeyAction(CAM_MOVE_LEFT)
        break
      case 'd':
        this.triggerKeyAction(CAM_MOVE_RIGHT)
        break
      case 'e':
        this.triggerKeyAction(RECREATE_ANNO)
        break
      case 'c':
        if (this.controlDown) {
          this.triggerKeyAction(COPY_ANNOTATION)
        } else {
          this.triggerKeyAction(TOGGLE_ANNO_COMMENT_INPUT)
        }
        break
      case 'v':
        if (this.controlDown) {
          this.triggerKeyAction(PASTE_ANNOTATION)
        }
        break
      case 'Escape':
        this.triggerKeyAction(DELETE_ANNO_IN_CREATION)
        break
      default:
        break
    }
  }

  keyUp(key) {
    switch (key) {
      case 'Control':
        this.controlDown = false
        this.triggerKeyAction(LEAVE_ANNO_ADD_MODE)
        break
      case 'Shift':
        this.shiftDown = false
        break
      default:
        break
    }
  }

  triggerKeyAction(keyAction) {
    if (this.keyActionHandler) {
      this.keyActionHandler(keyAction)
    }
  }
}

export default KeyMapper