Merge pull request 'Add spinType and cueBallSpeed to get shots operation' (#47) from loewy/add-spin-type-to-get-shots into master

Reviewed-on: #47

src/index.tsx

@@ -124,6 +124,7 @@ export type CueObjectFeaturesGql = {
   cueObjectAngle?: Maybe<Scalars["Float"]["output"]>;
   cueObjectDistance?: Maybe<Scalars["Float"]["output"]>;
   shotDirection?: Maybe<ShotDirectionEnum>;
+  spinType?: Maybe<SpinTypeEnum>;
 };
 
 export type DeployedConfigGql = {
@@ -949,7 +950,7 @@ export type FilterInput =
       notFilter?: never;
       orFilters?: never;
       shotDirection?: never;
-      spinType: Array<Scalars["String"]["input"]>;
+      spinType: Array<SpinTypeEnum>;
       tableSize?: never;
       tags?: never;
       targetPocketDistance?: never;
@@ -1458,6 +1459,13 @@ export type ShotGql = {
   videoId: Scalars["Int"]["output"];
 };
 
+export enum SpinTypeEnum {
+  Center = "CENTER",
+  Draw = "DRAW",
+  Follow = "FOLLOW",
+  Unknown = "UNKNOWN",
+}
+
 export type StreamErrorGql = {
   __typename?: "StreamErrorGQL";
   message: Scalars["String"]["output"];
@@ -1777,6 +1785,7 @@ export type GetShotsQueryVariables = Exact<{
   includeCueObjectDistance?: Scalars["Boolean"]["input"];
   includeCueObjectAngle?: Scalars["Boolean"]["input"];
   includeCueBallSpeed?: Scalars["Boolean"]["input"];
+  includeSpinType?: Scalars["Boolean"]["input"];
   includeShotDirection?: Scalars["Boolean"]["input"];
   includeTargetPocketDistance?: Scalars["Boolean"]["input"];
   includeMake?: Scalars["Boolean"]["input"];
@@ -1812,6 +1821,7 @@ export type GetShotsQuery = {
       cueObjectAngle?: number | null;
       cueBallSpeed?: number | null;
       shotDirection?: ShotDirectionEnum | null;
+      spinType?: SpinTypeEnum | null;
     } | null;
     pocketingIntentionFeatures?: {
       __typename?: "PocketingIntentionFeaturesGQL";
@@ -1822,6 +1832,25 @@ export type GetShotsQuery = {
   }>;
 };
 
+export type GetSerializedShotPathsQueryVariables = Exact<{
+  filterInput: FilterInput;
+}>;
+
+export type GetSerializedShotPathsQuery = {
+  __typename?: "Query";
+  getShots: Array<{
+    __typename?: "ShotGQL";
+    id: number;
+    videoId: number;
+    startFrame: number;
+    endFrame: number;
+    serializedShotPaths?: {
+      __typename?: "SerializedShotPathsGQL";
+      b64EncodedBuffer?: string | null;
+    } | null;
+  }>;
+};
+
 export type GetShotAnnotationTypesQueryVariables = Exact<{
   [key: string]: never;
 }>;
@@ -2715,6 +2744,7 @@ export const GetShotsDocument = gql`
     $includeCueObjectDistance: Boolean! = false
     $includeCueObjectAngle: Boolean! = false
     $includeCueBallSpeed: Boolean! = false
+    $includeSpinType: Boolean! = false
     $includeShotDirection: Boolean! = false
     $includeTargetPocketDistance: Boolean! = false
     $includeMake: Boolean! = false
@@ -2748,6 +2778,7 @@ export const GetShotsDocument = gql`
         cueObjectAngle @include(if: $includeCueObjectAngle)
         cueBallSpeed @include(if: $includeCueBallSpeed)
         shotDirection @include(if: $includeShotDirection)
+        spinType @include(if: $includeSpinType)
       }
       pocketingIntentionFeatures
         @include(if: $includePocketingIntentionFeatures) {
@@ -2781,6 +2812,7 @@ export const GetShotsDocument = gql`
  *      includeCueObjectDistance: // value for 'includeCueObjectDistance'
  *      includeCueObjectAngle: // value for 'includeCueObjectAngle'
  *      includeCueBallSpeed: // value for 'includeCueBallSpeed'
+ *      includeSpinType: // value for 'includeSpinType'
  *      includeShotDirection: // value for 'includeShotDirection'
  *      includeTargetPocketDistance: // value for 'includeTargetPocketDistance'
  *      includeMake: // value for 'includeMake'
@@ -2832,6 +2864,85 @@ export type GetShotsQueryResult = Apollo.QueryResult<
   GetShotsQuery,
   GetShotsQueryVariables
 >;
+export const GetSerializedShotPathsDocument = gql`
+  query GetSerializedShotPaths($filterInput: FilterInput!) {
+    getShots(filterInput: $filterInput) {
+      id
+      videoId
+      startFrame
+      endFrame
+      serializedShotPaths {
+        b64EncodedBuffer
+      }
+    }
+  }
+`;
+
+/**
+ * __useGetSerializedShotPathsQuery__
+ *
+ * To run a query within a React component, call `useGetSerializedShotPathsQuery` and pass it any options that fit your needs.
+ * When your component renders, `useGetSerializedShotPathsQuery` returns an object from Apollo Client that contains loading, error, and data properties
+ * you can use to render your UI.
+ *
+ * @param baseOptions options that will be passed into the query, supported options are listed on: https://www.apollographql.com/docs/react/api/react-hooks/#options;
+ *
+ * @example
+ * const { data, loading, error } = useGetSerializedShotPathsQuery({
+ *   variables: {
+ *      filterInput: // value for 'filterInput'
+ *   },
+ * });
+ */
+export function useGetSerializedShotPathsQuery(
+  baseOptions: Apollo.QueryHookOptions<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useQuery<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >(GetSerializedShotPathsDocument, options);
+}
+export function useGetSerializedShotPathsLazyQuery(
+  baseOptions?: Apollo.LazyQueryHookOptions<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useLazyQuery<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >(GetSerializedShotPathsDocument, options);
+}
+export function useGetSerializedShotPathsSuspenseQuery(
+  baseOptions?: Apollo.SuspenseQueryHookOptions<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useSuspenseQuery<
+    GetSerializedShotPathsQuery,
+    GetSerializedShotPathsQueryVariables
+  >(GetSerializedShotPathsDocument, options);
+}
+export type GetSerializedShotPathsQueryHookResult = ReturnType<
+  typeof useGetSerializedShotPathsQuery
+>;
+export type GetSerializedShotPathsLazyQueryHookResult = ReturnType<
+  typeof useGetSerializedShotPathsLazyQuery
+>;
+export type GetSerializedShotPathsSuspenseQueryHookResult = ReturnType<
+  typeof useGetSerializedShotPathsSuspenseQuery
+>;
+export type GetSerializedShotPathsQueryResult = Apollo.QueryResult<
+  GetSerializedShotPathsQuery,
+  GetSerializedShotPathsQueryVariables
+>;
 export const GetShotAnnotationTypesDocument = gql`
   query GetShotAnnotationTypes {
     getShotAnnotationTypes {

src/operations/shots.gql

@@ -9,6 +9,7 @@ query GetShots(
   $includeCueObjectDistance: Boolean! = false
   $includeCueObjectAngle: Boolean! = false
   $includeCueBallSpeed: Boolean! = false
+  $includeSpinType: Boolean! = false
   $includeShotDirection: Boolean! = false
   $includeTargetPocketDistance: Boolean! = false
   $includeMake: Boolean! = false
@@ -42,6 +43,7 @@ query GetShots(
       cueObjectAngle @include(if: $includeCueObjectAngle)
       cueBallSpeed @include(if: $includeCueBallSpeed)
       shotDirection @include(if: $includeShotDirection)
+      spinType @include(if: $includeSpinType)
     }
     pocketingIntentionFeatures
       @include(if: $includePocketingIntentionFeatures) {
@@ -52,6 +54,18 @@ query GetShots(
   }
 }
 
+query GetSerializedShotPaths($filterInput: FilterInput!) {
+  getShots(filterInput: $filterInput) {
+    id
+    videoId
+    startFrame
+    endFrame
+    serializedShotPaths {
+      b64EncodedBuffer
+    }
+  }
+}
+
 query GetShotAnnotationTypes {
   getShotAnnotationTypes {
     id

src/schema.gql

@@ -102,8 +102,8 @@ input FilterInput @oneOf {
   kickAngle: RangeFilter
   kickDistance: RangeFilter
   cueAngleAfterObject: RangeFilter
+  spinType: [SpinTypeEnum!]
   cueSpeedAfterObject: RangeFilter
-  spinType: [String!]
   falsePositiveScore: RangeFilter
 }
 
@@ -140,6 +140,13 @@ input ShotAnnotationInput {
   name: String!
 }
 
+enum SpinTypeEnum {
+  DRAW
+  FOLLOW
+  CENTER
+  UNKNOWN
+}
+
 type BucketSetGQL {
   keyName: String!
   feature: String!
@@ -191,6 +198,7 @@ type CueObjectFeaturesGQL {
   cueObjectAngle: Float
   cueBallSpeed: Float
   shotDirection: ShotDirectionEnum
+  spinType: SpinTypeEnum
 }
 
 type PocketingIntentionFeaturesGQL {