Select detected ball colors for Shot Lab

src/index.tsx

@@ -6745,6 +6745,10 @@ export type GetTableStateQuery = {
   getTableState: {
     __typename?: "TableStateGQL";
     identifierToPosition: Array<Array<number>>;
+    identifierToColor: Array<{
+      __typename?: "RGBColorGQL";
+      hex: string;
+    } | null>;
   };
 };
 
@@ -6761,6 +6765,10 @@ export type GetLiveTableStateQuery = {
     tableState: {
       __typename?: "TableStateGQL";
       identifierToPosition: Array<Array<number>>;
+      identifierToColor: Array<{
+        __typename?: "RGBColorGQL";
+        hex: string;
+      } | null>;
       homography?: {
         __typename?: "HomographyInfoGQL";
         id: number;
@@ -14426,6 +14434,9 @@ export const GetTableStateDocument = gql`
       useHomography: $useHomography
     ) {
       identifierToPosition
+      identifierToColor {
+        hex
+      }
     }
   }
 `;
@@ -14504,6 +14515,9 @@ export const GetLiveTableStateDocument = gql`
       frameIndex
       tableState {
         identifierToPosition
+        identifierToColor {
+          hex
+        }
         homography {
           ...HomographyInfo
         }

src/operations/shot_simulation.gql

@@ -9,6 +9,9 @@ query GetTableState(
     useHomography: $useHomography
   ) {
     identifierToPosition
+    identifierToColor {
+      hex
+    }
   }
 }
 
@@ -18,6 +21,9 @@ query GetLiveTableState($videoId: Int!) {
     frameIndex
     tableState {
       identifierToPosition
+      identifierToColor {
+        hex
+      }
       homography {
         ...HomographyInfo
       }