ohif-viewer/extensions/measurement-tracking/src/panels/PanelMeasurementTableTracking.tsx

import React from 'react';
import { utils } from '@ohif/core';
import { AccordionTrigger, MeasurementTable, useViewportGrid } from '@ohif/ui-next';
import {
  PanelMeasurement,
  StudyMeasurements,
  StudyMeasurementsActions,
  StudySummaryFromMetadata,
  AccordionGroup,
  MeasurementsOrAdditionalFindings,
} from '@ohif/extension-cornerstone';

import { useTrackedMeasurements } from '../getContextModule';
import { UntrackSeriesModal } from './PanelStudyBrowserTracking/untrackSeriesModal';

const { filterAnd, filterPlanarMeasurement, filterMeasurementsBySeriesUID } =
  utils.MeasurementFilters;

function PanelMeasurementTableTracking(props) {
  const [viewportGrid] = useViewportGrid();
  const { measurementService, uiModalService } = props.servicesManager.services;
  const [trackedMeasurements, sendTrackedMeasurementsEvent] = useTrackedMeasurements();
  const { trackedStudy, trackedSeries } = trackedMeasurements.context;
  const measurementFilter = trackedStudy
    ? filterAnd(filterPlanarMeasurement, filterMeasurementsBySeriesUID(trackedSeries))
    : filterPlanarMeasurement;

  const onUntrackConfirm = () => {
    sendTrackedMeasurementsEvent('UNTRACK_ALL', {});
  };

  const onDelete = () => {
    const hasDirtyMeasurements = measurementService
      .getMeasurements()
      .some(measurement => measurement.isDirty);
    hasDirtyMeasurements
      ? uiModalService.show({
          title: 'Untrack Study',
          content: UntrackSeriesModal,
          contentProps: {
            onConfirm: onUntrackConfirm,
            message: 'Are you sure you want to untrack study and delete all measurements?',
          },
        })
      : onUntrackConfirm();
  };

  const EmptyComponent = () => (
    <div data-cy="trackedMeasurements-panel">
      <MeasurementTable
        title="Measurements"
        isExpanded={false}
      >
        <MeasurementTable.Body />
      </MeasurementTable>
    </div>
  );

  const actions = {
    createSR: ({ StudyInstanceUID }) => {
      sendTrackedMeasurementsEvent('SAVE_REPORT', {
        viewportId: viewportGrid.activeViewportId,
        isBackupSave: true,
        StudyInstanceUID,
        measurementFilter,
      });
    },
    onDelete,
  };

  const Header = props => (
    <AccordionTrigger
      asChild={true}
      className="px-0"
    >
      <div data-cy="TrackingHeader">
        <StudySummaryFromMetadata
          {...props}
          actions={actions}
        />
      </div>
    </AccordionTrigger>
  );

  return (
    <div data-cy="trackedMeasurements-panel">
      <PanelMeasurement
        measurementFilter={measurementFilter}
        emptyComponent={EmptyComponent}
        sourceChildren={props.children}
      >
        <StudyMeasurements grouping={props.grouping}>
          <AccordionGroup.Trigger
            key="trackingMeasurementsHeader"
            asChild={true}
          >
            <Header key="trackingHeadChild" />
          </AccordionGroup.Trigger>
          <MeasurementsOrAdditionalFindings
            key="measurementsOrAdditionalFindings"
            activeStudyUID={trackedStudy}
            customHeader={StudyMeasurementsActions}
            measurementFilter={measurementFilter}
            actions={actions}
          />
        </StudyMeasurements>
      </PanelMeasurement>
    </div>
  );
}

export default PanelMeasurementTableTracking;