✨ close #138

anucvml · Mar 6, 2022 · 25b4f0e · 25b4f0e
1 parent e3e9819
commit 25b4f0e
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 5 deletions.
diff --git a/src/App.vue b/src/App.vue
@@ -284,6 +284,11 @@ if (showPopup) {
     preferenceStore.showPopup = false
   }
 }
+window.addEventListener('beforeunload', event => {
+  if (annotationStore.hasVideo && !mainStore.isSaved) {
+    event.returnValue = 'The annotations are not saved!'
+  }
+})
 </script>
 
 <style lang="sass">

diff --git a/src/store/annotation.js b/src/store/annotation.js
@@ -2,6 +2,7 @@ import deepClone from 'lodash.clonedeep'
 import { defineStore } from 'pinia'
 import { computed, reactive, toRefs, watch } from 'vue'
 import utils from '~/libs/utils.js'
+import { useMainStore } from '~/store/index.js'
 import { usePreferenceStore } from '~/store/preference.js'
 import { ActionAnnotation, ObjectAnnotation, RegionAnnotation, SkeletonAnnotation } from '../libs/annotationlib.js'
 
@@ -35,13 +36,12 @@ const DEFAULT_ANNOTATION = {
   copyMode: false,
   addPointMode: false,
   delPointMode: false,
-  indicatingMode: false,
-
-  isSaved: true
+  indicatingMode: false
 }
 
 export const useAnnotationStore = defineStore('annotation', () => {
   const preferenceStore = usePreferenceStore()
+  const mainStore = useMainStore()
   let defaultAnnotation = deepClone(DEFAULT_ANNOTATION)
   const state = reactive(DEFAULT_ANNOTATION)
   watch(
@@ -77,6 +77,15 @@ export const useAnnotationStore = defineStore('annotation', () => {
       state.leftCurrentFrame = 0
     }
   })
+  watch(() => [
+    state.objectAnnotationListMap,
+    state.regionAnnotationListMap,
+    state.skeletonAnnotationListMap,
+    state.actionAnnotationList,
+    state.keyframeList
+  ], () => {
+    mainStore.isSaved = false
+  }, { deep: true })
   return {
     ...toRefs(state),
     hasVideo: computed(() => {
@@ -108,6 +117,7 @@ export const useAnnotationStore = defineStore('annotation', () => {
             }
           })
       }
+      mainStore.isSaved = true
       return {
         video: state.video,
         keyframeList: state.keyframeList,
@@ -200,6 +210,7 @@ export const useAnnotationStore = defineStore('annotation', () => {
         )
       }
       state.actionAnnotationList = actionAnnotationList
+      mainStore.isSaved = true
     }
   }
 })
diff --git a/src/store/configuration.js b/src/store/configuration.js
@@ -382,7 +382,7 @@ export const useConfigurationStore = defineStore('configuration', () => {
       return {
         objectLabelData: state.objectLabelData,
         actionLabelData: state.actionLabelData,
-        skeletonTypeData: state.skeletonTypeData,
+        skeletonTypeData: state.skeletonTypeData
       }
     },
     importObjectLabelData,

diff --git a/src/store/index.js b/src/store/index.js
@@ -6,6 +6,7 @@ export const useMainStore = defineStore('main', {
   state: () => ({
     drawer: false,
     zoom: false,
-    submitURL: null
+    submitURL: null,
+    isSaved: true
   })
 })