import type {
  ArrayShape,
  ArrayValue,
  AxisMapping,
  ComplexType,
  Dataset,
  NumArray,
  NumArrayDataset,
  NumericType,
  ScalarShape,
  ScaleType,
  StringType,
} from '@h5web/shared';

export type NxAttribute =
  | 'NX_class'
  | 'signal'
  | 'interpretation'
  | 'axes'
  | 'default'
  | 'long_name'
  | 'units'
  | 'SILX_style'
  | 'auxiliary_signals';

export interface DatasetInfo {
  label: string;
  unit: string | undefined;
}

interface DatasetDef<
  T extends NumericType | ComplexType = NumericType | ComplexType
> extends DatasetInfo {
  dataset: Dataset<ArrayShape, T>;
}

type WithError<T extends DatasetDef> = T & {
  errorDataset?: NumArrayDataset;
};

export type AuxDef = WithError<DatasetDef<NumericType>>;
export type AxisDef = DatasetDef<NumericType>;

export interface SilxStyle {
  signalScaleType?: ScaleType;
  axisScaleTypes?: AxisMapping<ScaleType>;
}

export interface NxData<
  T extends NumericType | ComplexType = NumericType | ComplexType
> {
  titleDataset?: Dataset<ScalarShape, StringType>;
  signalDef: WithError<DatasetDef<T>>;
  auxDefs: AuxDef[];
  axisDefs: AxisMapping<AxisDef>;
  silxStyle: SilxStyle;
}

export interface NxValues<T extends NumericType | ComplexType> {
  title: string;
  signal: ArrayValue<T>;
  errors?: NumArray;
  auxValues: NumArray[];
  auxErrors: (NumArray | undefined)[];
  axisValues: AxisMapping<NumArray>;
}