Show multiple sensitivities in TimeSeriesSensitivities module (#301)

Co-authored-by: jorgenherje <[email protected]>
equinor · Sep 27, 2023 · 1c2144d · 1c2144d
1 parent 814781d
commit 1c2144d
Show file tree

Hide file tree

Showing 22 changed files with 390 additions and 299 deletions.
diff --git a/frontend/src/modules/Sensitivity/queryHooks.tsx b/frontend/src/modules/Sensitivity/queryHooks.tsx
diff --git a/frontend/src/modules/SimulationTimeSeriesSensitivity/loadModule.tsx b/frontend/src/modules/SimulationTimeSeriesSensitivity/loadModule.tsx
@@ -8,10 +8,11 @@ import { view } from "./view";
 const defaultState: State = {
     vectorSpec: null,
     resamplingFrequency: Frequency_api.MONTHLY,
-    selectedSensitivity: null,
+    selectedSensitivities: null,
     showStatistics: true,
     showRealizations: false,
     realizationsToInclude: null,
+    showHistorical: true,
 };
 
 const module = ModuleRegistry.initModule<State>("SimulationTimeSeriesSensitivity", defaultState);

diff --git a/frontend/src/modules/SimulationTimeSeriesSensitivity/queryHooks.tsx b/frontend/src/modules/SimulationTimeSeriesSensitivity/queryHooks.tsx
@@ -1,12 +1,17 @@
-import { Frequency_api, VectorDescription_api, VectorStatisticSensitivityData_api } from "@api";
+import {
+    Frequency_api,
+    VectorDescription_api,
+    VectorHistoricalData_api,
+    VectorStatisticSensitivityData_api,
+} from "@api";
 import { VectorRealizationData_api } from "@api";
 import { apiService } from "@framework/ApiService";
 import { UseQueryResult, useQuery } from "@tanstack/react-query";
 
 const STALE_TIME = 60 * 1000;
 const CACHE_TIME = 60 * 1000;
 
-export function useVectorsQuery(
+export function useVectorListQuery(
     caseUuid: string | undefined,
     ensembleName: string | undefined
 ): UseQueryResult<Array<VectorDescription_api>> {
@@ -15,7 +20,7 @@ export function useVectorsQuery(
         queryFn: () => apiService.timeseries.getVectorList(caseUuid ?? "", ensembleName ?? ""),
         staleTime: STALE_TIME,
         cacheTime: CACHE_TIME,
-        enabled: caseUuid && ensembleName ? true : false,
+        enabled: !!(caseUuid && ensembleName),
     });
 }
 
@@ -46,7 +51,7 @@ export function useVectorDataQuery(
             ),
         staleTime: STALE_TIME,
         cacheTime: CACHE_TIME,
-        enabled: caseUuid && ensembleName && vectorName && allOrNonEmptyRealArr ? true : false,
+        enabled: !!(caseUuid && ensembleName && vectorName && allOrNonEmptyRealArr),
     });
 }
 
@@ -69,6 +74,28 @@ export function useStatisticalVectorSensitivityDataQuery(
             ),
         staleTime: STALE_TIME,
         cacheTime: CACHE_TIME,
-        enabled: allowEnable && caseUuid && ensembleName && vectorName && resampleFrequency ? true : false,
+        enabled: !!(allowEnable && caseUuid && ensembleName && vectorName && resampleFrequency),
+    });
+}
+
+export function useHistoricalVectorDataQuery(
+    caseUuid: string | undefined,
+    ensembleName: string | undefined,
+    vectorName: string | undefined,
+    resampleFrequency: Frequency_api | null,
+    allowEnable: boolean
+): UseQueryResult<VectorHistoricalData_api> {
+    return useQuery({
+        queryKey: ["getHistoricalVectorData", caseUuid, ensembleName, vectorName, resampleFrequency],
+        queryFn: () =>
+            apiService.timeseries.getHistoricalVectorData(
+                caseUuid ?? "",
+                ensembleName ?? "",
+                vectorName ?? "",
+                resampleFrequency ?? Frequency_api.MONTHLY
+            ),
+        staleTime: STALE_TIME,
+        cacheTime: CACHE_TIME,
+        enabled: !!(allowEnable && caseUuid && ensembleName && vectorName && resampleFrequency),
     });
 }
diff --git a/frontend/src/modules/SimulationTimeSeriesSensitivity/registerModule.ts b/frontend/src/modules/SimulationTimeSeriesSensitivity/registerModule.ts
@@ -6,7 +6,7 @@ import { State } from "./state";
 
 ModuleRegistry.registerModule<State>({
     moduleName: "SimulationTimeSeriesSensitivity",
-    defaultTitle: "Simulation time series sensitivity",
+    defaultTitle: "Simulation time series per sensitivity",
     syncableSettingKeys: [SyncSettingKey.ENSEMBLE, SyncSettingKey.TIME_SERIES],
     broadcastChannelsDef,
 });