import { createColumnHelper } from '@tanstack/react-table';
import { BenchmarkResult } from '../../../types';
import { formatNumber, formatSize } from '../utils/formatters';

const columnHelper = createColumnHelper<BenchmarkResult>();

export const columns = [
  columnHelper.accessor('dataset', {
    header: 'Dataset',
    cell: info => info.getValue(),
  }),
  columnHelper.accessor('algorithm', {
    header: 'Algorithm',
    cell: info => info.getValue(),
  }),
  columnHelper.accessor('compression_ratio', {
    header: 'Compression Ratio',
    cell: info => `${formatNumber(info.getValue())}x`,
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.compression_ratio;
      const b = rowB.original.compression_ratio;
      return a - b;
    },
  }),
  columnHelper.accessor('encode_time', {
    header: 'Encode Time (s)',
    cell: info => formatNumber(info.getValue(), 4),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.encode_time;
      const b = rowB.original.encode_time;
      return a - b;
    },
  }),
  columnHelper.accessor('decode_time', {
    header: 'Decode Time (s)',
    cell: info => formatNumber(info.getValue(), 4),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.decode_time;
      const b = rowB.original.decode_time;
      return a - b;
    },
  }),
  columnHelper.accessor('encode_mb_per_sec', {
    header: 'Encode Speed (MB/s)',
    cell: info => formatNumber(info.getValue()),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.encode_mb_per_sec;
      const b = rowB.original.encode_mb_per_sec;
      return a - b;
    },
  }),
  columnHelper.accessor('decode_mb_per_sec', {
    header: 'Decode Speed (MB/s)',
    cell: info => formatNumber(info.getValue()),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.decode_mb_per_sec;
      const b = rowB.original.decode_mb_per_sec;
      return a - b;
    },
  }),
  columnHelper.accessor('original_size', {
    header: 'Original Size',
    cell: info => formatSize(info.getValue()),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.original_size;
      const b = rowB.original.original_size;
      return a - b;
    },
  }),
  columnHelper.accessor('compressed_size', {
    header: 'Compressed Size',
    cell: info => formatSize(info.getValue()),
    sortingFn: (rowA, rowB) => {
      const a = rowA.original.compressed_size;
      const b = rowB.original.compressed_size;
      return a - b;
    },
  }),
];