@zzalai/leafer-point-annotation 1.0.0

package/src/utils/PointCommands.ts

@@ -0,0 +1,74 @@
+import { Group } from 'leafer-ui';
+import { ICommand } from '@zzalai/leafer-undo-redo';
+import type { PointAnnotation } from '@/types';
+
+export interface PointAnnotationElement extends Group {
+  data: PointAnnotation;
+}
+
+export class AddPointCommand implements ICommand {
+  private container: Group;
+  private element: PointAnnotationElement;
+  private dataArray: PointAnnotation[];
+  private pointData: PointAnnotation;
+
+  constructor(
+    container: Group,
+    element: PointAnnotationElement,
+    dataArray: PointAnnotation[],
+    pointData: PointAnnotation
+  ) {
+    this.container = container;
+    this.element = element;
+    this.dataArray = dataArray;
+    this.pointData = pointData;
+  }
+
+  execute(): void {
+    this.container.add(this.element);
+    this.dataArray.push(this.pointData);
+  }
+
+  undo(): void {
+    this.container.remove(this.element);
+    const index = this.dataArray.findIndex(p => p.id === this.pointData.id);
+    if (index > -1) {
+      this.dataArray.splice(index, 1);
+    }
+  }
+}
+
+export class RemovePointCommand implements ICommand {
+  private container: Group;
+  private element: PointAnnotationElement;
+  private dataArray: PointAnnotation[];
+  private insertIndex: number;
+
+  constructor(
+    container: Group,
+    element: PointAnnotationElement,
+    dataArray: PointAnnotation[]
+  ) {
+    this.container = container;
+    this.element = element;
+    this.dataArray = dataArray;
+    this.insertIndex = dataArray.findIndex(p => p.id === element.data.id);
+  }
+
+  execute(): void {
+    this.container.remove(this.element);
+    const index = this.dataArray.findIndex(p => p.id === this.element.data.id);
+    if (index > -1) {
+      this.dataArray.splice(index, 1);
+    }
+  }
+
+  undo(): void {
+    this.container.add(this.element);
+    if (this.insertIndex > -1 && this.insertIndex <= this.dataArray.length) {
+      this.dataArray.splice(this.insertIndex, 0, this.element.data);
+    } else {
+      this.dataArray.push(this.element.data);
+    }
+  }
+}

package/src/utils/YOLOExporter.ts

@@ -0,0 +1,39 @@
+import { PointAnnotation } from '@/types';
+
+export interface YOLOExportResult {
+  annotations: string;
+  classNames: string;
+  imageWidth: number;
+  imageHeight: number;
+}
+
+export function exportYOLOFormat(
+  pointAnnotations: PointAnnotation[],
+  imageWidth: number,
+  imageHeight: number,
+  options?: {
+    className?: string;
+    pointSize?: number;
+  }
+): YOLOExportResult {
+  const className = options?.className || 'point';
+  const pointSize = options?.pointSize || 20;
+
+  const classId = 0;
+
+  const annotationsLines = pointAnnotations.map(point => {
+    const xCenter = point.normalized.x;
+    const yCenter = point.normalized.y;
+    const width = pointSize / imageWidth;
+    const height = pointSize / imageHeight;
+
+    return `${classId} ${xCenter.toFixed(6)} ${yCenter.toFixed(6)} ${width.toFixed(6)} ${height.toFixed(6)}`;
+  });
+
+  return {
+    annotations: annotationsLines.join('\n'),
+    classNames: className,
+    imageWidth,
+    imageHeight,
+  };
+}

package/src/vite-env.d.ts

@@ -0,0 +1,7 @@
+/// <reference types="vite/client" />
+
+declare module '*.vue' {
+  import type { DefineComponent } from 'vue'
+  const component: DefineComponent<{}, {}, any>
+  export default component
+}

package/tsconfig.json

@@ -0,0 +1,24 @@
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "useDefineForClassFields": true,
+    "module": "ESNext",
+    "lib": ["ES2020", "DOM", "DOM.Iterable"],
+    "skipLibCheck": true,
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "resolveJsonModule": true,
+    "isolatedModules": true,
+    "noEmit": true,
+    "jsx": "preserve",
+    "strict": true,
+    "noUnusedLocals": true,
+    "noUnusedParameters": true,
+    "noFallthroughCasesInSwitch": true,
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  },
+  "include": ["src/**/*.ts", "src/**/*.d.ts", "src/**/*.tsx", "src/**/*.vue"],
+  "references": [{ "path": "./tsconfig.node.json" }]
+}

package/tsconfig.node.json

@@ -0,0 +1,10 @@
+{
+  "compilerOptions": {
+    "composite": true,
+    "skipLibCheck": true,
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "allowSyntheticDefaultImports": true
+  },
+  "include": ["vite.config.ts"]
+}

package/vite.config.ts

@@ -0,0 +1,42 @@
+import { defineConfig } from 'vite'
+import vue from '@vitejs/plugin-vue'
+import dts from 'vite-plugin-dts'
+import path from 'path'
+
+export default defineConfig({
+  plugins: [
+    vue(),
+    process.env.NODE_ENV === 'production' ? dts() : null
+  ].filter(Boolean),
+  resolve: {
+    alias: {
+      '@': path.resolve(__dirname, 'src')
+    }
+  },
+  build: {
+    lib: {
+      entry: './src/index.ts',
+      name: 'LeaferPointAnnotation',
+      formats: ['es', 'umd'],
+      fileName: (format) => `leafer-point-annotation.${format}.js`
+    },
+    rollupOptions: {
+      external: ['vue'],
+      output: {
+        globals: {
+          vue: 'Vue'
+        },
+        exports: "named"
+      }
+    },
+    minify: 'terser',
+    terserOptions: {
+      compress: {
+        drop_console: true,
+        drop_debugger: true,
+        pure_funcs: ['console.log'],
+        passes: 1
+      }
+    }
+  }
+})

package/vite.docs.config.ts

@@ -0,0 +1,28 @@
+import { defineConfig } from 'vite'
+import vue from '@vitejs/plugin-vue'
+import { resolve } from 'path'
+
+export default defineConfig({
+  plugins: [vue()],
+  base: './',
+  resolve: {
+    alias: {
+      '@': resolve(__dirname, './src')
+    }
+  },
+  build: {
+    outDir: 'docs',
+    assetsDir: 'assets',
+    minify: 'terser',
+    terserOptions: {
+      compress: {
+        drop_console: true,
+        drop_debugger: true
+      }
+    }
+  },
+  server: {
+    port: 3000,
+    open: true
+  }
+})