React UI: Changing color for a group (#1205)

cvat-core/src/annotations-collection.js

@@ -127,6 +127,7 @@
                 groups: this.groups,
                 frameMeta: this.frameMeta,
                 history: this.history,
+                groupColors: {},
             };
         }
 
@@ -139,7 +140,8 @@
 
             for (const tag of data.tags) {
                 const clientID = ++this.count;
-                const tagModel = new Tag(tag, clientID, this.injection);
+                const color = colors[clientID % colors.length];
+                const tagModel = new Tag(tag, clientID, color, this.injection);
                 this.tags[tagModel.frame] = this.tags[tagModel.frame] || [];
                 this.tags[tagModel.frame].push(tagModel);
                 this.objects[clientID] = tagModel;

cvat-core/src/annotations-objects.js

@@ -7,6 +7,7 @@
     require:false
 */
 
+
 (() => {
     const ObjectState = require('./object-state');
     const {
@@ -136,9 +137,10 @@
     }
 
     class Annotation {
-        constructor(data, clientID, injection) {
+        constructor(data, clientID, color, injection) {
             this.taskLabels = injection.labels;
             this.history = injection.history;
+            this.groupColors = injection.groupColors;
             this.clientID = clientID;
             this.serverID = data.id;
             this.group = data.group;
@@ -146,11 +148,31 @@
             this.frame = data.frame;
             this.removed = false;
             this.lock = false;
+            this.color = color;
             this.updated = Date.now();
             this.attributes = data.attributes.reduce((attributeAccumulator, attr) => {
                 attributeAccumulator[attr.spec_id] = attr.value;
                 return attributeAccumulator;
             }, {});
+            this.groupObject = Object.defineProperties({}, {
+                color: {
+                    get: () => {
+                        if (this.group) {
+                            return this.groupColors[this.group]
+                                || colors[this.group % colors.length];
+                        }
+                        return defaultGroupColor;
+                    },
+                    set: (newColor) => {
+                        if (this.group && typeof (newColor) === 'string' && /^#[0-9A-F]{6}$/i.test(newColor)) {
+                            this.groupColors[this.group] = newColor;
+                        }
+                    },
+                },
+                id: {
+                    get: () => this.group,
+                },
+            });
             this.appendDefaultAttributes(this.label);
 
             injection.groups.max = Math.max(injection.groups.max, this.group);
@@ -229,51 +251,6 @@
             }, [this.clientID]);
         }
 
-        appendDefaultAttributes(label) {
-            const labelAttributes = label.attributes;
-            for (const attribute of labelAttributes) {
-                if (!(attribute.id in this.attributes)) {
-                    this.attributes[attribute.id] = attribute.defaultValue;
-                }
-            }
-        }
-
-        updateTimestamp(updated) {
-            const anyChanges = updated.label || updated.attributes || updated.points
-                || updated.outside || updated.occluded || updated.keyframe
-                || updated.zOrder;
-
-            if (anyChanges) {
-                this.updated = Date.now();
-            }
-        }
-
-        delete(force) {
-            if (!this.lock || force) {
-                this.removed = true;
-
-                this.history.do(HistoryActions.REMOVED_OBJECT, () => {
-                    this.removed = false;
-                }, () => {
-                    this.removed = true;
-                }, [this.clientID]);
-            }
-
-            return this.removed;
-        }
-    }
-
-    class Drawn extends Annotation {
-        constructor(data, clientID, color, injection) {
-            super(data, clientID, injection);
-
-            this.frameMeta = injection.frameMeta;
-            this.hidden = false;
-
-            this.color = color;
-            this.shapeType = null;
-        }
-
         _validateStateBeforeSave(frame, data, updated) {
             let fittedPoints = [];
 
@@ -363,6 +340,48 @@
             return fittedPoints;
         }
 
+        appendDefaultAttributes(label) {
+            const labelAttributes = label.attributes;
+            for (const attribute of labelAttributes) {
+                if (!(attribute.id in this.attributes)) {
+                    this.attributes[attribute.id] = attribute.defaultValue;
+                }
+            }
+        }
+
+        updateTimestamp(updated) {
+            const anyChanges = updated.label || updated.attributes || updated.points
+                || updated.outside || updated.occluded || updated.keyframe
+                || updated.zOrder;
+
+            if (anyChanges) {
+                this.updated = Date.now();
+            }
+        }
+
+        delete(force) {
+            if (!this.lock || force) {
+                this.removed = true;
+
+                this.history.do(HistoryActions.REMOVED_OBJECT, () => {
+                    this.removed = false;
+                }, () => {
+                    this.removed = true;
+                }, [this.clientID]);
+            }
+
+            return this.removed;
+        }
+    }
+
+    class Drawn extends Annotation {
+        constructor(data, clientID, color, injection) {
+            super(data, clientID, color, injection);
+            this.frameMeta = injection.frameMeta;
+            this.hidden = false;
+            this.shapeType = null;
+        }
+
         save() {
             throw new ScriptingError(
                 'Is not implemented',
@@ -432,10 +451,7 @@
                 points: [...this.points],
                 attributes: { ...this.attributes },
                 label: this.label,
-                group: {
-                    color: this.group ? colors[this.group % colors.length] : defaultGroupColor,
-                    id: this.group,
-                },
+                group: this.groupObject,
                 color: this.color,
                 hidden: this.hidden,
                 updated: this.updated,
@@ -618,10 +634,7 @@
             return {
                 ...this.getPosition(frame, prev, next),
                 attributes: this.getAttributes(frame),
-                group: {
-                    color: this.group ? colors[this.group % colors.length] : defaultGroupColor,
-                    id: this.group,
-                },
+                group: this.groupObject,
                 objectType: ObjectType.TRACK,
                 shapeType: this.shapeType,
                 clientID: this.clientID,
@@ -1053,8 +1066,8 @@
     }
 
     class Tag extends Annotation {
-        constructor(data, clientID, injection) {
-            super(data, clientID, injection);
+        constructor(data, clientID, color, injection) {
+            super(data, clientID, color, injection);
         }
 
         // Method is used to export data to the server
@@ -1091,7 +1104,7 @@
                 lock: this.lock,
                 attributes: { ...this.attributes },
                 label: this.label,
-                group: this.group,
+                group: this.groupObject,
                 updated: this.updated,
                 frame,
             };

cvat-ui/src/actions/annotation-actions.ts

@@ -1040,3 +1040,22 @@ export function changeLabelColorAsync(
         }
     };
 }
+
+export function changeGroupColorAsync(
+    sessionInstance: any,
+    frameNumber: number,
+    group: number,
+    color: string,
+): ThunkAction<Promise<void>, {}, {}, AnyAction> {
+    return async (dispatch: ActionCreator<Dispatch>): Promise<void> => {
+        const state: CombinedState = getStore().getState();
+        const groupStates = state.annotation.annotations.states
+            .filter((_state: any): boolean => _state.group.id === group);
+        if (groupStates.length) {
+            groupStates[0].group.color = color;
+            dispatch(updateAnnotationsAsync(sessionInstance, frameNumber, groupStates));
+        } else {
+            dispatch(updateAnnotationsAsync(sessionInstance, frameNumber, []));
+        }
+    };
+}

cvat-ui/src/components/task-page/job-list.tsx

@@ -50,8 +50,8 @@ function JobListComponent(props: Props & RouteComponentProps): JSX.Element {
                 <Button type='link' href={`${baseURL}/?id=${id}`}>{`Job #${id}`}</Button>
                 |
                 <Tooltip title='Beta version of new UI written in React. It is to get
-                                acquainted only, we do not recommend use it to annotations
-                                process because it is lack of some features and can be unstable.'
+                                acquainted only, we do not recommend use it to annotation
+                                process because it lacks of some features and can be unstable.'
                 >
                     <Button
                         type='link'

cvat-ui/src/containers/annotation-page/standard-workspace/objects-side-bar/object-item.tsx

@@ -8,9 +8,11 @@ import {
 } from 'reducers/interfaces';
 import {
     collapseObjectItems,
+    changeLabelColorAsync,
     updateAnnotationsAsync,
     changeFrameAsync,
     removeObjectAsync,
+    changeGroupColorAsync,
     copyShape as copyShapeAction,
     activateObject as activateObjectAction,
     propagateObject as propagateObjectAction,
@@ -46,6 +48,8 @@ interface DispatchToProps {
     removeObject: (sessionInstance: any, objectState: any) => void;
     copyShape: (objectState: any) => void;
     propagateObject: (objectState: any) => void;
+    changeLabelColor(sessionInstance: any, frameNumber: number, label: any, color: string): void;
+    changeGroupColor(sessionInstance: any, frameNumber: number, group: number, color: string): void;
 }
 
 function mapStateToProps(state: CombinedState, own: OwnProps): StateToProps {
@@ -130,6 +134,22 @@ function mapDispatchToProps(dispatch: any): DispatchToProps {
         propagateObject(objectState: any): void {
             dispatch(propagateObjectAction(objectState));
         },
+        changeLabelColor(
+            sessionInstance: any,
+            frameNumber: number,
+            label: any,
+            color: string,
+        ): void {
+            dispatch(changeLabelColorAsync(sessionInstance, frameNumber, label, color));
+        },
+        changeGroupColor(
+            sessionInstance: any,
+            frameNumber: number,
+            group: number,
+            color: string,
+        ): void {
+            dispatch(changeGroupColorAsync(sessionInstance, frameNumber, group, color));
+        },
     };
 }
 
@@ -336,11 +356,22 @@ class ObjectItemContainer extends React.PureComponent<Props> {
 
     private changeColor = (color: string): void => {
         const {
+            jobInstance,
             objectState,
+            colorBy,
+            changeLabelColor,
+            changeGroupColor,
+            frameNumber,
         } = this.props;
 
-        objectState.color = color;
-        this.commit();
+        if (colorBy === ColorBy.INSTANCE) {
+            objectState.color = color;
+            this.commit();
+        } else if (colorBy === ColorBy.GROUP) {
+            changeGroupColor(jobInstance, frameNumber, objectState.group.id, color);
+        } else if (colorBy === ColorBy.LABEL) {
+            changeLabelColor(jobInstance, frameNumber, objectState.label, color);
+        }
     };
 
     private changeLabel = (labelID: string): void => {