@macrostrat/column-views 3.3.5 → 3.4.1

package/src/units/index.ts

@@ -13,7 +13,8 @@ import { TrackedLabeledUnit } from "./composite";
 import { useEnvironments, useLithologies } from "@macrostrat/data-provider";
 import { useMemo } from "react";
 import { resolveID } from "./resolvers";
-import { Lithology } from "@macrostrat/api-types";
+import { BaseUnit, Lithology } from "@macrostrat/api-types";
+import { UnitWithDefinedOverlap } from "../prepare-units/helpers.ts";
 
 export * from "./composite";
 export * from "./types";
@@ -56,15 +57,34 @@ export function BasicUnitComponent({ division, ...rest }) {
   });
 }
 
-export function UnitComponent({ division, nColumns = 2, ...rest }) {
-  const width = rest.width ?? useColumnLayout()?.width;
+interface UnitComponentProps<T extends BaseUnit> {
+  division: UnitWithDefinedOverlap<T> | T;
+  nColumns: number;
+  width?: number;
+}
 
-  const nOverlappingUnits = division.overlappingUnits?.length ?? 0;
-  const _nColumns = nOverlappingUnits == 0 ? 1 : nColumns;
-  const columnIx = (division.column ?? 0) % _nColumns;
-  const reducedWidth = width / _nColumns;
-  const x = (columnIx * width) / _nColumns;
+export function UnitComponent<T extends BaseUnit>({
+  division,
+  nColumns = 2,
+  ...rest
+}: UnitComponentProps<T>) {
+  const width = rest.width ?? useColumnLayout()?.width;
 
+  let _nColumns = nColumns;
+  if (isNaN(_nColumns)) {
+    _nColumns = 2;
+  }
+  let columnIx = 0;
+  let x = 0;
+
+  let reducedWidth = _nColumns;
+  if ("overlap" in division) {
+    const overlap = division.overlap;
+    _nColumns = Math.min(_nColumns, overlap.nColumns);
+    columnIx = overlap.column;
+    reducedWidth = width / _nColumns;
+    x = (columnIx * width) / _nColumns;
+  }
   return h(TrackedLabeledUnit, {
     division,
     ...rest,