add type for colorinfo

simularium · Dec 4, 2023 · f465e52 · f465e52
1 parent 99fd604
commit f465e52
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/visGeometry/ColorHandler.ts b/src/visGeometry/ColorHandler.ts
@@ -1,5 +1,6 @@
 import { map, round, isEqual } from "lodash";
 import { Color } from "three";
+import { AgentColorInfo } from "./types";
 
 export function convertColorStringToNumber(color: number | string): number {
     if (typeof color !== "string") {
@@ -195,10 +196,7 @@ class ColorHandler {
         return this.getColorById(index);
     }
 
-    public getColorInfoForAgentType(agentType: number): {
-        color: Color;
-        colorId: number;
-    } {
+    public getColorInfoForAgentType(agentType: number): AgentColorInfo {
         const color = this.getColorForAgentType(agentType);
         const colorId = this.getColorIdForAgentType(agentType);
         return { color, colorId };

diff --git a/src/visGeometry/types.ts b/src/visGeometry/types.ts
@@ -1,4 +1,4 @@
-import { BufferGeometry, Object3D } from "three";
+import { BufferGeometry, Color, Object3D } from "three";
 import PDBModel from "./PDBModel";
 import { MRTShaders } from "./rendering/MultipassMaterials";
 
@@ -64,3 +64,8 @@ export interface GeometryStoreLoadResponse {
     geometry: MeshLoadRequest | PDBModel;
     errorMessage?: string;
 }
+
+export interface AgentColorInfo {
+    color: Color;
+    colorId: number;
+}