refactor(es modules): move all files to src, try vite 3.0

src/modules/ui/rivers-editor.js

@@ -0,0 +1,281 @@
+import {findCell, getPackPolygon} from "/src/utils/graphUtils";
+import {tip, clearMainTip} from "/src/scripts/tooltips";
+import {getSegmentId} from "/src/utils/lineUtils";
+import {rn} from "/src/utils/numberUtils";
+import {rand} from "/src/utils/probabilityUtils";
+
+export function editRiver(id) {
+  if (customization) return;
+  if (elSelected && id === elSelected.attr("id")) return;
+  closeDialogs(".stable");
+  if (!layerIsOn("toggleRivers")) toggleRivers();
+
+  document.getElementById("toggleCells").dataset.forced = +!layerIsOn("toggleCells");
+  if (!layerIsOn("toggleCells")) toggleCells();
+
+  elSelected = d3.select("#" + id).on("click", addControlPoint);
+
+  tip(
+    "Drag control points to change the river course. Click on point to remove it. Click on river to add additional control point. For major changes please create a new river instead",
+    true
+  );
+  debug.append("g").attr("id", "controlCells");
+  debug.append("g").attr("id", "controlPoints");
+
+  updateRiverData();
+
+  const river = getRiver();
+  const {cells, points} = river;
+  const riverPoints = Rivers.getRiverPoints(cells, points);
+  drawControlPoints(riverPoints);
+  drawCells(cells);
+
+  $("#riverEditor").dialog({
+    title: "Edit River",
+    resizable: false,
+    position: {my: "left top", at: "left+10 top+10", of: "#map"},
+    close: closeRiverEditor
+  });
+
+  if (fmg.modules.editRiver) return;
+  fmg.modules.editRiver = true;
+
+  // add listeners
+  document.getElementById("riverCreateSelectingCells").addEventListener("click", createRiver);
+  document.getElementById("riverEditStyle").addEventListener("click", () => editStyle("rivers"));
+  document.getElementById("riverElevationProfile").addEventListener("click", showElevationProfile);
+  document.getElementById("riverLegend").addEventListener("click", editRiverLegend);
+  document.getElementById("riverRemove").addEventListener("click", removeRiver);
+  document.getElementById("riverName").addEventListener("input", changeName);
+  document.getElementById("riverType").addEventListener("input", changeType);
+  document.getElementById("riverNameCulture").addEventListener("click", generateNameCulture);
+  document.getElementById("riverNameRandom").addEventListener("click", generateNameRandom);
+  document.getElementById("riverMainstem").addEventListener("change", changeParent);
+  document.getElementById("riverSourceWidth").addEventListener("input", changeSourceWidth);
+  document.getElementById("riverWidthFactor").addEventListener("input", changeWidthFactor);
+
+  function getRiver() {
+    const riverId = +elSelected.attr("id").slice(5);
+    const river = pack.rivers.find(r => r.i === riverId);
+    return river;
+  }
+
+  function updateRiverData() {
+    const r = getRiver();
+
+    document.getElementById("riverName").value = r.name;
+    document.getElementById("riverType").value = r.type;
+
+    const parentSelect = document.getElementById("riverMainstem");
+    parentSelect.options.length = 0;
+    const parent = r.parent || r.i;
+    const sortedRivers = pack.rivers.slice().sort((a, b) => (a.name > b.name ? 1 : -1));
+    sortedRivers.forEach(river => {
+      const opt = new Option(river.name, river.i, false, river.i === parent);
+      parentSelect.options.add(opt);
+    });
+    document.getElementById("riverBasin").value = pack.rivers.find(river => river.i === r.basin).name;
+
+    document.getElementById("riverDischarge").value = r.discharge + " m³/s";
+    document.getElementById("riverSourceWidth").value = r.sourceWidth;
+    document.getElementById("riverWidthFactor").value = r.widthFactor;
+
+    updateRiverLength(r);
+    updateRiverWidth(r);
+  }
+
+  function updateRiverLength(river) {
+    river.length = rn(elSelected.node().getTotalLength() / 2, 2);
+    const lengthUI = `${rn(river.length * distanceScaleInput.value)} ${distanceUnitInput.value}`;
+    document.getElementById("riverLength").value = lengthUI;
+  }
+
+  function updateRiverWidth(river) {
+    const {addMeandering, getWidth, getOffset} = Rivers;
+    const {cells, discharge, widthFactor, sourceWidth} = river;
+    const meanderedPoints = addMeandering(cells);
+    river.width = getWidth(getOffset(discharge, meanderedPoints.length, widthFactor, sourceWidth));
+
+    const width = `${rn(river.width * distanceScaleInput.value, 3)} ${distanceUnitInput.value}`;
+    document.getElementById("riverWidth").value = width;
+  }
+
+  function drawControlPoints(points) {
+    debug
+      .select("#controlPoints")
+      .selectAll("circle")
+      .data(points)
+      .join("circle")
+      .attr("cx", d => d[0])
+      .attr("cy", d => d[1])
+      .attr("r", 0.6)
+      .call(d3.drag().on("start", dragControlPoint))
+      .on("click", removeControlPoint);
+  }
+
+  function drawCells(cells) {
+    const validCells = [...new Set(cells)].filter(i => pack.cells.i[i]);
+    debug
+      .select("#controlCells")
+      .selectAll(`polygon`)
+      .data(validCells)
+      .join("polygon")
+      .attr("points", d => getPackPolygon(d));
+  }
+
+  function dragControlPoint() {
+    const {r, fl} = pack.cells;
+    const river = getRiver();
+
+    const {x: x0, y: y0} = d3.event;
+    const initCell = findCell(x0, y0);
+
+    let movedToCell = null;
+
+    d3.event.on("drag", function () {
+      const {x, y} = d3.event;
+      const currentCell = findCell(x, y);
+
+      movedToCell = initCell !== currentCell ? currentCell : null;
+
+      this.setAttribute("cx", x);
+      this.setAttribute("cy", y);
+      this.__data__ = [rn(x, 1), rn(y, 1)];
+      redrawRiver();
+      drawCells(river.cells);
+    });
+
+    d3.event.on("end", () => {
+      if (movedToCell && !r[movedToCell]) {
+        // swap river data
+        r[initCell] = 0;
+        r[movedToCell] = river.i;
+        const sourceFlux = fl[initCell];
+        fl[initCell] = fl[movedToCell];
+        fl[movedToCell] = sourceFlux;
+        redrawRiver();
+      }
+    });
+  }
+
+  function redrawRiver() {
+    const river = getRiver();
+    river.points = debug.selectAll("#controlPoints > *").data();
+    river.cells = river.points.map(([x, y]) => findCell(x, y));
+
+    const {widthFactor, sourceWidth} = river;
+    const meanderedPoints = Rivers.addMeandering(river.cells, river.points);
+
+    lineGen.curve(d3.curveCatmullRom.alpha(0.1));
+    const path = Rivers.getRiverPath(meanderedPoints, widthFactor, sourceWidth);
+    elSelected.attr("d", path);
+
+    updateRiverLength(river);
+    if (fmg.modules.elevation) showEPForRiver(elSelected.node());
+  }
+
+  function addControlPoint() {
+    const [x, y] = d3.mouse(this);
+    const point = [rn(x, 1), rn(y, 1)];
+
+    const river = getRiver();
+    if (!river.points) river.points = debug.selectAll("#controlPoints > *").data();
+
+    const index = getSegmentId(river.points, point, 2);
+    river.points.splice(index, 0, point);
+    drawControlPoints(river.points);
+    redrawRiver();
+  }
+
+  function removeControlPoint() {
+    this.remove();
+    redrawRiver();
+
+    const {cells} = getRiver();
+    drawCells(cells);
+  }
+
+  function changeName() {
+    getRiver().name = this.value;
+  }
+
+  function changeType() {
+    getRiver().type = this.value;
+  }
+
+  function generateNameCulture() {
+    const r = getRiver();
+    r.name = riverName.value = Rivers.getName(r.mouth);
+  }
+
+  function generateNameRandom() {
+    const r = getRiver();
+    if (r) r.name = riverName.value = Names.getBase(rand(nameBases.length - 1));
+  }
+
+  function changeParent() {
+    const r = getRiver();
+    r.parent = +this.value;
+    r.basin = pack.rivers.find(river => river.i === r.parent).basin;
+    document.getElementById("riverBasin").value = pack.rivers.find(river => river.i === r.basin).name;
+  }
+
+  function changeSourceWidth() {
+    const river = getRiver();
+    river.sourceWidth = +this.value;
+    updateRiverWidth(river);
+    redrawRiver();
+  }
+
+  function changeWidthFactor() {
+    const river = getRiver();
+    river.widthFactor = +this.value;
+    updateRiverWidth(river);
+    redrawRiver();
+  }
+
+  function showElevationProfile() {
+    fmg.modules.elevation = true;
+    showEPForRiver(elSelected.node());
+  }
+
+  function editRiverLegend() {
+    const id = elSelected.attr("id");
+    const river = getRiver();
+    editNotes(id, river.name + " " + river.type);
+  }
+
+  function removeRiver() {
+    alertMessage.innerHTML = "Are you sure you want to remove the river and all its tributaries";
+    $("#alert").dialog({
+      resizable: false,
+      width: "22em",
+      title: "Remove river and tributaries",
+      buttons: {
+        Remove: function () {
+          $(this).dialog("close");
+          const river = +elSelected.attr("id").slice(5);
+          Rivers.remove(river);
+          elSelected.remove();
+          $("#riverEditor").dialog("close");
+        },
+        Cancel: function () {
+          $(this).dialog("close");
+        }
+      }
+    });
+  }
+
+  function closeRiverEditor() {
+    debug.select("#controlPoints").remove();
+    debug.select("#controlCells").remove();
+
+    elSelected.on("click", null);
+    unselect();
+    clearMainTip();
+
+    const forced = +document.getElementById("toggleCells").dataset.forced;
+    document.getElementById("toggleCells").dataset.forced = 0;
+    if (forced && layerIsOn("toggleCells")) toggleCells();
+  }
+}