@weng-lab/genomebrowser-ui 0.4.0 → 0.4.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@weng-lab/genomebrowser-ui",
   "private": false,
-  "version": "0.4.0",
+  "version": "0.4.1",
   "license": "MIT",
   "type": "module",
   "publishConfig": {

package/src/TrackSelect/DataGrid/DataGridWrapper.tsx

@@ -5,6 +5,8 @@ import {
   GridAutosizeOptions,
   GridColDef,
   GridColumnVisibilityModel,
+  GridRenderCellParams,
+  GridRowSelectionModel,
   useGridApiRef,
 } from "@mui/x-data-grid-premium";
 import { useEffect, useMemo, useState } from "react";
@@ -88,7 +90,7 @@ export function DataGridWrapper(props: DataGridProps) {
           apiRef={apiRef}
           rows={rows}
           columns={columns}
-          getRowId={(row) => row.id}
+          getRowId={(row: { id: string }) => row.id}
           autosizeOptions={autosizeOptions}
           rowGroupingModel={groupingModel}
           groupingColDef={{
@@ -97,13 +99,15 @@ export function DataGridWrapper(props: DataGridProps) {
             minWidth: 300,
             maxWidth: 500,
             flex: 2,
-            renderCell: (params) => <GroupingCell {...params} />,
+            renderCell: (params: GridRenderCellParams) => (
+              <GroupingCell {...params} />
+            ),
           }}
           columnVisibilityModel={columnVisibilityModel}
           onColumnVisibilityModelChange={setColumnVisibilityModel}
-          onRowSelectionModelChange={(selection) => {
-            const ids = (selection as any)?.ids ?? new Set<string>();
-            onSelectionChange(new Set(ids));
+          onRowSelectionModelChange={(selection: GridRowSelectionModel) => {
+            const ids = selection.ids ?? new Set<string>();
+            onSelectionChange(new Set([...ids].map(String)));
           }}
           rowSelectionPropagation={{ descendants: true, parents: false }}
           disableRowGrouping={false}

package/src/TrackSelect/Folders/biosamples/data/fix-human-collections.mjs

@@ -0,0 +1,119 @@
+import { readFile, writeFile } from "node:fs/promises";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const inputPath = join(__dirname, "human_with_wgbs.json");
+const outputPath = join(__dirname, "humans_fixed.json");
+
+const expectedCounts = {
+  Core: 170,
+  Partial: 1155,
+  Ancillary: 562,
+};
+
+const counts = {
+  Core: 0,
+  Partial: 0,
+  Ancillary: 0,
+  NA: 0,
+};
+
+const specialCases = {
+  GM12866_ENCDO000ABQ: "NA",
+  neural_crest_cell_ENCDO222AAA: "Partial",
+};
+
+const specialCaseFound = Object.fromEntries(
+  Object.keys(specialCases).map((name) => [name, false]),
+);
+
+function classifyTrack(track) {
+  const specialCollection = specialCases[track.name];
+  if (specialCollection) {
+    specialCaseFound[track.name] = true;
+    return specialCollection;
+  }
+
+  const assays = new Set(
+    track.assays.map((assay) => assay.assay.toLowerCase()),
+  );
+
+  if (!assays.has("dnase")) {
+    return "Ancillary";
+  }
+
+  if (
+    assays.has("h3k4me3") &&
+    assays.has("h3k27ac") &&
+    assays.has("ctcf")
+  ) {
+    return "Core";
+  }
+
+  return "Partial";
+}
+
+function isAggregate(track) {
+  return track.name === "aggregate-biosample-data" || track.ontology === "aggregate";
+}
+
+function validateCounts() {
+  const mismatches = Object.entries(expectedCounts).filter(
+    ([collection, expected]) => counts[collection] !== expected,
+  );
+
+  if (mismatches.length === 0) {
+    return;
+  }
+
+  console.error("Classification count validation failed.");
+  console.error("Expected:", expectedCounts);
+  console.error("Actual:", counts);
+  process.exit(1);
+}
+
+const input = JSON.parse(await readFile(inputPath, "utf8"));
+const changes = [];
+
+const fixedTracks = input.tracks.map((track) => {
+  const collection = classifyTrack(track);
+
+  if (!isAggregate(track)) {
+    counts[collection] += 1;
+  }
+
+  if (track.collection !== collection) {
+    changes.push({
+      name: track.name,
+      displayName: track.displayName,
+      from: track.collection,
+      to: collection,
+    });
+  }
+
+  return {
+    ...track,
+    collection,
+  };
+});
+
+validateCounts();
+
+await writeFile(
+  outputPath,
+  `${JSON.stringify({ ...input, tracks: fixedTracks }, null, 2)}\n`,
+);
+
+console.log("Wrote", outputPath);
+console.log("Counts excluding aggregate:", counts);
+console.log("Changed tracks:", changes.length);
+console.log("Special cases found:", specialCaseFound);
+
+const changeCounts = changes.reduce((acc, change) => {
+  const key = `${change.from} -> ${change.to}`;
+  acc[key] = (acc[key] ?? 0) + 1;
+  return acc;
+}, {});
+
+console.log("Change counts:", changeCounts);