Select only relavant variables for the chart (#4881)

DARPA-ASKEM · Sep 20, 2024 · de0a46b · de0a46b
1 parent 39213a1
commit de0a46b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/...hmi-client/src/components/workflow/ops/simulate-ciemss/tera-simulate-ciemss-drilldown.vue b/...hmi-client/src/components/workflow/ops/simulate-ciemss/tera-simulate-ciemss-drilldown.vue
@@ -354,7 +354,6 @@ const preparedCharts = computed(() => {
 
  const result = runResults.value[selectedRunId.value];
  const resultSummary = runResultsSummary.value[selectedRunId.value];
-
  const reverseMap: Record<string, string> = {};
  Object.keys(pyciemssMap).forEach((key) => {
  reverseMap[`${pyciemssMap[key]}_mean`] = key;

diff --git a/packages/client/hmi-client/src/services/charts.ts b/packages/client/hmi-client/src/services/charts.ts
@@ -1,5 +1,5 @@
 import { percentile } from '@/utils/math';
-import { isEmpty } from 'lodash';
+import { isEmpty, pick } from 'lodash';
 import { VisualizationSpec } from 'vega-embed';
 import { v4 as uuidv4 } from 'uuid';
 import { ChartAnnotation, Intervention } from '@/types/Types';
@@ -398,8 +398,12 @@ export function createForecastChart(
 
  // Helper function to capture common layer structure
  const newLayer = (layer: ForecastChartLayer, markType: string) => {
+ const selectedFields = layer.variables.concat([layer.timeField]);
+ if (layer.groupField) selectedFields.push(layer.groupField);
+
+ const data = Array.isArray(layer.data) ? { values: layer.data.map((d) => pick(d, selectedFields)) } : layer.data;
  const header = {
- data: Array.isArray(layer.data) ? { values: layer.data } : layer.data,
+ data,
  transform: [
  {
  fold: layer.variables,