// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import { useRef, useState } from "preact/hooks";

const enablePanning = false;
const altKeyPans = true;

const menuItems = [
  {
    category: "itemCount",
    options: ["Show all", "Top 10", "Top 25"],
  },
  {
    category: "sortOrder",
    options: ["Sort a-z", "High to low", "Low to high"],
  },
  {
    category: "labels",
    options: ["Raw labels", "Ket labels", "No labels"],
  },
];
const maxMenuOptions = 3;
const defaultMenuSelection: { [idx: string]: number } = {
  itemCount: 0,
  sortOrder: 0,
  labels: 0,
};

const reKetResult = /^\[(?:(Zero|One), *)*(Zero|One)\]$/;
function resultToKet(result: string): string {
  if (typeof result !== "string") return "ERROR";

  if (reKetResult.test(result)) {
    // The result is a simple array of Zero and One
    // The below will return an array of "Zero" or "One" in the order found
    const matches = result.match(/(One|Zero)/g);
    let ket = "|";
    matches?.forEach((digit) => (ket += digit == "One" ? "1" : "0"));
    ket += "⟩";
    return ket;
  } else {
    return result;
  }
}

export function Histogram(props: {
  shotCount: number;
  data: Map<string, number>;
  filter: string;
  onFilter: (filter: string) => void;
  shotsHeader: boolean;
}) {
  const [hoverLabel, setHoverLabel] = useState("");
  const [scale, setScale] = useState({ zoom: 1.0, offset: 1.0 });
  const [menuSelection, setMenuSelection] = useState(defaultMenuSelection);

  const gMenu = useRef<SVGGElement>(null);
  const gInfo = useRef<SVGGElement>(null);

  let maxItemsToShow = 0; // All
  switch (menuSelection["itemCount"]) {
    case 1:
      maxItemsToShow = 10;
      break;
    case 2:
      maxItemsToShow = 25;
      break;
  }
  const showKetLabels = menuSelection["labels"] === 1;

  const bucketArray = [...props.data];

  // Calculate bucket percentages before truncating for display
  let totalAllBuckets = 0;
  let sizeBiggestBucket = 0;
  bucketArray.forEach((x) => {
    totalAllBuckets += x[1];
    sizeBiggestBucket = Math.max(x[1], sizeBiggestBucket);
  });

  let histogramLabel = `${bucketArray.length} unique results`;
  if (maxItemsToShow > 0) {
    // Sort from high to low then take the first n
    bucketArray.sort((a, b) => (a[1] < b[1] ? 1 : -1));
    if (bucketArray.length > maxItemsToShow) {
      histogramLabel = `Top ${maxItemsToShow} of ${histogramLabel}`;
      bucketArray.length = maxItemsToShow;
    }
  }
  if (props.filter) {
    histogramLabel += `. Shot filter: ${
      showKetLabels ? resultToKet(props.filter) : props.filter
    }`;
  }

  bucketArray.sort((a, b) => {
    // If they can be converted to numbers, then sort as numbers, else lexically
    const ax = Number(a[0]);
    const bx = Number(b[0]);
    switch (menuSelection["sortOrder"]) {
      case 1: // high-to-low
        return a[1] < b[1] ? 1 : -1;
        break;
      case 2: // low-to-high
        return a[1] > b[1] ? 1 : -1;
        break;
      default: // a-z
        if (!isNaN(ax) && !isNaN(bx)) return ax < bx ? -1 : 1;
        return a[0] < b[0] ? -1 : 1;
        break;
    }
  });

  function onMouseOverRect(evt: MouseEvent) {
    const target = evt.target as SVGRectElement;
    const title = target.querySelector("title")?.textContent;
    setHoverLabel(title || "");
  }

  function onMouseOutRect() {
    setHoverLabel("");
  }

  function onClickRect(evt: MouseEvent) {
    const targetElem = evt.target as SVGRectElement;
    const rawLabel = targetElem.getAttribute("data-raw-label");

    if (rawLabel === props.filter) {
      // Clicked the already selected bar. Clear the filter
      props.onFilter("");
    } else {
      props.onFilter(rawLabel || "");
    }
  }

  function toggleMenu() {
    if (!gMenu.current) return;
    if (gMenu.current.style.display === "inline") {
      gMenu.current.style.display = "none";
    } else {
      gMenu.current.style.display = "inline";
      if (gInfo.current) gInfo.current.style.display = "none";
    }
  }

  function menuClicked(category: string, idx: number) {
    if (!gMenu.current) return;
    const newMenuSelection = { ...menuSelection };
    newMenuSelection[category] = idx;
    setMenuSelection(newMenuSelection);
    if (category === "itemCount") {
      setScale({ zoom: 1, offset: 1 });
    }
    gMenu.current.style.display = "none";
  }

  function toggleInfo() {
    if (!gInfo.current) return;

    gInfo.current.style.display =
      gInfo.current.style.display === "inline" ? "none" : "inline";
  }

  // Each menu item has a width of 32px and a height of 10px
  // Menu items are 38px apart on the x-axis, and 11px on the y-axis.
  const menuItemWidth = 38;
  const menuItemHeight = 11;
  const menuBoxWidth = menuItems.length * menuItemWidth - 2;
  const menuBoxHeight = maxMenuOptions * menuItemHeight + 3;

  const barAreaWidth = 163;
  const barAreaHeight = 72;
  const fontOffset = 1.2;

  // Scale the below for when zoomed
  const barBoxWidth = (barAreaWidth * scale.zoom) / bucketArray.length;
  const barPaddingPercent = 0.1; // 10%
  const barPaddingSize = barBoxWidth * barPaddingPercent;
  const barFillWidth = barBoxWidth - 2 * barPaddingSize;
  const showLabels = barBoxWidth > 5 && menuSelection["labels"] !== 2;

  function onWheel(e: WheelEvent): void {
    e.preventDefault();

    // currentTarget is the element the listener is attached to, the main svg
    // element in this case.
    const svgElem = e.currentTarget as SVGSVGElement;

    // Below gets the mouse location in the svg element coordinates. This stays
    // consistent while the scroll is occuring (i.e. it is the point the mouse
    // was at when scrolling started).
    const mousePoint = new DOMPoint(e.clientX, e.clientY).matrixTransform(
      svgElem.getScreenCTM()?.inverse(),
    );

    /*
    While zooming, we want is to track the point the mouse is at when scrolling, and pin
    that location on the screen. That means adjusting the scroll offset.

    SVG translation is used to pan left and right, but zooming is done manually (making the
    bars wider or thinner) to keep the fonts from getting streched, which occurs with scaling.

    deltaX and deltaY do not accumulate across events, they are a new delta each time.
    */

    let newScrollOffset = scale.offset;
    let newZoom = scale.zoom;

    // *** First handle any zooming ***
    if (!altKeyPans || !e.altKey) {
      newZoom = scale.zoom + e.deltaY * 0.05;
      newZoom = Math.min(Math.max(1, newZoom), 50);

      // On zooming in, need to shift left to maintain mouse point, and vice verca.
      const oldChartWidth = barAreaWidth * scale.zoom;
      const mousePointOnChart = 0 - scale.offset + mousePoint.x;
      const percentRightOnChart = mousePointOnChart / oldChartWidth;
      const chartWidthGrowth =
        newZoom * barAreaWidth - scale.zoom * barAreaWidth;
      const shiftLeftAdjust = percentRightOnChart * chartWidthGrowth;
      newScrollOffset = scale.offset - shiftLeftAdjust;
    }

    // *** Then handle any panning ***
    if (enablePanning) {
      newScrollOffset -= e.deltaX;
    }
    if (!enablePanning && altKeyPans && e.altKey) {
      newScrollOffset -= e.deltaY;
    }

    // Don't allow offset > 1 (scrolls the first bar right of the left edge of the area)
    // Don't allow for less than 0 - barwidths + screen width (scrolls last bar left of the right edge)
    const maxScrollRight = 1 - (barAreaWidth * newZoom - barAreaWidth);
    const boundScrollOffset = Math.min(
      Math.max(newScrollOffset, maxScrollRight),
      1,
    );

    setScale({ zoom: newZoom, offset: boundScrollOffset });
  }

  return (
    <>
      {props.shotsHeader ? (
        <h4 style="margin: 8px 0px">Total shots: {props.shotCount}</h4>
      ) : null}
      <svg class="histogram" viewBox="0 0 165 100" onWheel={onWheel}>
        <g transform={`translate(${scale.offset},4)`}>
          {bucketArray.map((entry, idx) => {
            const label = showKetLabels ? resultToKet(entry[0]) : entry[0];

            const height = barAreaHeight * (entry[1] / sizeBiggestBucket);
            const x = barBoxWidth * idx + barPaddingSize;
            const labelX = barBoxWidth * idx + barBoxWidth / 2 - fontOffset;
            const y = barAreaHeight + 15 - height;
            const barLabel = `${label} at ${(
              (entry[1] / totalAllBuckets) *
              100
            ).toFixed(2)}%`;
            let barClass = "bar";

            if (entry[0] === props.filter) {
              barClass += " bar-selected";
            }

            return (
              <>
                <rect
                  class={barClass}
                  x={x}
                  y={y}
                  width={barFillWidth}
                  height={height}
                  onMouseOver={onMouseOverRect}
                  onMouseOut={onMouseOutRect}
                  onClick={onClickRect}
                  data-raw-label={entry[0]}
                >
                  <title>{barLabel}</title>
                </rect>
                {
                  <text
                    class="bar-label"
                    x={labelX}
                    y="85"
                    visibility={showLabels ? "visible" : "hidden"}
                    transform={`rotate(90, ${labelX}, 85)`}
                  >
                    {label}
                  </text>
                }
              </>
            );
          })}
        </g>

        <text class="histo-label" x="2" y="97">
          {histogramLabel}
        </text>
        <text class="hover-text" x="85" y="6">
          {hoverLabel}
        </text>

        {/* The settings icon */}
        <g
          class="menu-icon"
          transform="translate(2, 2) scale(0.3 0.3)"
          onClick={toggleMenu}
        >
          <rect width="24" height="24" fill="white" stroke-widths="0.5"></rect>
          <path
            d="M3 5 H21 M3 12 H21 M3 19 H21"
            stroke-width="1.75"
            stroke-linecap="round"
          />
          <rect x="6" y="3" width="4" height="4" rx="1" stroke-width="1.5" />
          <rect x="15" y="10" width="4" height="4" rx="1" stroke-width="1.5" />
          <rect x="9" y="17" width="4" height="4" rx="1" stroke-width="1.5" />
        </g>

        {/* The info icon */}
        <g
          class="menu-icon"
          transform="translate(156, 2) scale(0.3 0.3)"
          onClick={toggleInfo}
        >
          <rect width="24" height="24" stroke-width="0"></rect>
          <circle cx="12" cy="13" r="10" stroke-width="1.5" />
          <path
            stroke-width="2.5"
            stroke-linecap="round"
            d="M12 8 V8 M12 12.5 V18"
          />
        </g>

        {/* The menu box */}
        <g
          id="menu"
          ref={gMenu}
          transform="translate(8, 2)"
          style="display: none;"
        >
          <rect
            x="0"
            y="0"
            rx="2"
            width={menuBoxWidth}
            height={menuBoxHeight}
            class="menu-box"
          ></rect>

          {
            // Menu items
            menuItems.map((item, col) => {
              return item.options.map((option, row) => {
                let classList = "menu-item";
                if (menuSelection[item.category] === row)
                  classList += " menu-selected";
                return (
                  <>
                    <rect
                      x={2 + col * menuItemWidth}
                      y={2 + row * menuItemHeight}
                      rx="1"
                      class={classList}
                      onClick={() => menuClicked(item.category, row)}
                    ></rect>
                    <text
                      x={5 + col * menuItemWidth}
                      y={9 + row * menuItemHeight}
                      class="menu-text"
                    >
                      {option}
                    </text>
                  </>
                );
              });
            })
          }
          {
            // Column separators
            menuItems.map((item, idx) => {
              return idx >= menuItems.length - 1 ? null : (
                <line
                  class="menu-separator"
                  x1={37 + idx * menuItemWidth}
                  y1="2"
                  x2={37 + idx * menuItemWidth}
                  y2={maxMenuOptions * menuItemHeight + 1}
                ></line>
              );
            })
          }
        </g>

        {/* The info box */}
        <g ref={gInfo} style="display: none;">
          <rect
            width="155"
            height="76"
            rx="5"
            x="5"
            y="6"
            class="help-info"
            onClick={toggleInfo}
          />
          <text y="6" class="help-info-text">
            <tspan x="10" dy="10">
              This histogram shows the frequency of unique 'shot' results.
            </tspan>
            <tspan x="10" dy="10">
              Click the top-left 'settings' icon for display options.
            </tspan>
            <tspan x="10" dy="10">
              You can zoom the chart using the mouse scroll wheel.
            </tspan>
            <tspan x="10" dy="7">
              (Or using a trackpad gesture).
            </tspan>
            <tspan x="10" dy="10">
              When zoomed, to pan left &amp; right, press 'Alt' while scrolling.
            </tspan>
            <tspan x="10" dy="10">
              Click on a bar to filter the shot details to that result.
            </tspan>
            <tspan x="10" dy="12">
              Click anywhere in this box to dismiss it.
            </tspan>
          </text>
        </g>
      </svg>
    </>
  );
}
