CVAT 3D Annotation - Added initial cuboid placement in all views

cvat-canvas3d/src/typescript/canvas3d.ts

@@ -4,7 +4,9 @@
 
 import pjson from '../../package.json';
 import { Canvas3dController, Canvas3dControllerImpl } from './canvas3dController';
-import { Canvas3dModel, Canvas3dModelImpl, Mode } from './canvas3dModel';
+import {
+    Canvas3dModel, Canvas3dModelImpl, Mode, DrawData,
+} from './canvas3dModel';
 import { Canvas3dView, Canvas3dViewImpl, ViewsDOM } from './canvas3dView';
 import { Master } from './master';
 
@@ -17,6 +19,9 @@ interface Canvas3d {
     mode(): Mode;
     render(): void;
     keyControls(keys: KeyboardEvent): void;
+    mouseControls(type: string, event: MouseEvent): void;
+    draw(drawData: DrawData): void;
+    cancel(): void;
 }
 
 class Canvas3dImpl implements Canvas3d {
@@ -38,10 +43,18 @@ class Canvas3dImpl implements Canvas3d {
         this.view.keyControls(keys);
     }
 
+    public mouseControls(type: string, event: MouseEvent): void {
+        this.view.mouseControls(type, event);
+    }
+
     public render(): void {
         this.view.render();
     }
 
+    public draw(drawData: DrawData): void {
+        this.model.draw(drawData);
+    }
+
     public setup(frameData: any): void {
         this.model.setup(frameData);
     }
@@ -53,6 +66,10 @@ class Canvas3dImpl implements Canvas3d {
     public isAbleToChangeFrame(): boolean {
         return this.model.isAbleToChangeFrame();
     }
+
+    public cancel(): void {
+        this.model.cancel();
+    }
 }
 
 export { Canvas3dImpl as Canvas3d, Canvas3dVersion };

cvat-canvas3d/src/typescript/canvas3dController.ts

@@ -3,8 +3,10 @@
 // SPDX-License-Identifier: MIT
 
 import { Canvas3dModel, Mode } from './canvas3dModel';
+import { DrawData } from '../../../cvat-canvas/src/typescript/canvasModel';
 
 export interface Canvas3dController {
+    readonly drawData: DrawData;
     mode: Mode;
 }
 
@@ -22,4 +24,8 @@ export class Canvas3dControllerImpl implements Canvas3dController {
     public get mode(): Mode {
         return this.model.mode;
     }
+
+    public get drawData(): DrawData {
+        return this.model.data.drawData;
+    }
 }

cvat-canvas3d/src/typescript/canvas3dModel.ts

@@ -26,6 +26,13 @@ export enum FrameZoom {
     MAX = 10,
 }
 
+export enum ViewType {
+    PERSPECTIVE = 'perspective',
+    TOP = 'top',
+    SIDE = 'side',
+    FRONT = 'front',
+}
+
 export enum UpdateReasons {
     IMAGE_CHANGED = 'image_changed',
     OBJECTS_UPDATED = 'objects_updated',
@@ -39,6 +46,7 @@ export enum UpdateReasons {
 export enum Mode {
     IDLE = 'idle',
     DRAG = 'drag',
+    CREATE = 'create',
     RESIZE = 'resize',
     DRAW = 'draw',
     EDIT = 'edit',
@@ -61,6 +69,8 @@ export interface Canvas3dModel {
     data: Canvas3dDataModel;
     setup(frameData: any): void;
     isAbleToChangeFrame(): boolean;
+    draw(drawData: DrawData): void;
+    cancel(): void;
 }
 
 export class Canvas3dModelImpl extends MasterImpl implements Canvas3dModel {
@@ -133,4 +143,18 @@ export class Canvas3dModelImpl extends MasterImpl implements Canvas3dModel {
 
         return !isUnable;
     }
+
+    public draw(drawData: DrawData): void {
+        if (drawData.enabled && this.data.drawData.enabled) {
+            throw new Error('Drawing has been already started');
+        }
+        this.data.drawData.enabled = drawData.enabled;
+        this.data.mode = Mode.DRAW;
+
+        this.notify(UpdateReasons.DRAW);
+    }
+
+    public cancel(): void {
+        this.notify(UpdateReasons.CANCEL);
+    }
 }