feat: add Cognite3DModel.setAllNodeColors (#1132)

* feat: add Cognite3DModel.setAllNodeColors

* fix: unused import

* doc: mark setAllNodeColors with version 1.1

* Update viewer/src/public/migration/Cognite3DModel.ts

Co-authored-by: Maksim Nesterenko <maksim.nesterenko@cognite.com>

Co-authored-by: cognite-bulldozer[bot] <51074376+cognite-bulldozer[bot]@users.noreply.github.com>
Co-authored-by: Maksim Nesterenko <maksim.nesterenko@cognite.com>

Author: 3 people

documentation/docs/examples/cad-colors.mdx

@@ -15,17 +15,25 @@ Reveal supports updating the color of individual objects by providing [tree indi
 To update the colors of all nodes, loop over all tree indices in the model and update the colors
 accordingly:
 
-<!-- TODO: implement eficcient coloring fn and update these examples -->
-
 ```jsx runnable
 model.iterateNodesByTreeIndex(treeIndex => {
-  model.setNodeColorByTreeIndex(treeIndex, 127, 127, 127);
+  const value = treeIndex % 200;
+  model.setNodeColorByTreeIndex(treeIndex, 200, value, 200);
 });
 ```
 
+> ** New in 1.1.0 **
+
+Notice how colors are applied in batches using this approach. A more performant choice when a single color is 
+to be applied to all nodes is to use `Cognite3DModel.setAllNodeColors`:
+
+```jsx runnable
+model.setAllNodeColors(127, 127, 127);
+```
+
 ## Reset node colors
 
-Color overrides can be overridden using `Cognite3DModel.resetNodeColorByTreeIndex`, e.g.
+Color overrides can be reset using `Cognite3DModel.resetNodeColorByTreeIndex`, e.g.
 
 ```jsx runnable
 model.iterateNodesByTreeIndex(treeIndex => {

viewer/src/public/migration/Cognite3DModel.ts

@@ -453,6 +453,21 @@ export class Cognite3DModel extends THREE.Object3D implements CogniteModelBase {
     return treeIndices.count;
   }
 
+  /**
+   * Overrrides color for all nodes in the scene.
+   * @param r Red component between 0 and 255.
+   * @param g Green component between 0 and 255.
+   * @param b Blue component between 0 and 255.
+   * @version new in 1.1.0
+   */
+  setAllNodeColors(r: number, g: number, b: number): void {
+    const color: [number, number, number] = [r, g, b];
+    for (let i = 0; i <= this.cadModel.scene.maxTreeIndex; i++) {
+      this.nodeColors.set(i, color);
+    }
+    this.cadNode.requestNodeUpdate([...this.nodeColors.keys()]);
+  }
+
   /**
    * Restore original colors for all nodes.
    */