@platforma-open/milaboratories.repertoire-mutation-heatmap.block 1.0.0 → 1.1.0

package/dist/index.js

@@ -8905,6 +8905,8 @@ const dataModel = new DataModelBuilder().from("v1").init(() => ({
 	valueMode: "abundance",
 	normalize: true,
 	filter: { membership: "all" },
+	roundFrequencyRefs: [],
+	compositionEpsilon: 1e-6,
 	stateHeatmapState: {
 		title: "Deep Mutational Scanning",
 		template: "heatmap",
@@ -8930,6 +8932,19 @@ const dataModel = new DataModelBuilder().from("v1").init(() => ({
 			hideAxisLabels: false,
 			cellSize: 20
 		} }
+	},
+	compositionHeatmapState: {
+		title: "Composition Enrichment",
+		template: "heatmap",
+		currentTab: null,
+		layersSettings: { heatmap: {
+			normalizationDirection: null,
+			transform: null
+		} },
+		axesSettings: { axisY: {
+			hideAxisLabels: false,
+			cellSize: 20
+		} }
 	}
 }));
 const platforma$1 = BlockModelV3.create(dataModel).args((data) => {
@@ -8944,18 +8959,63 @@ const platforma$1 = BlockModelV3.create(dataModel).args((data) => {
 		level: data.level,
 		valueMode: data.valueMode,
 		normalize: data.valueMode === "abundance" ? data.normalize : false,
-		filter: canonicalizeFilter(data.filter)
+		filter: canonicalizeFilter(data.filter),
+		roundFrequencyRefs: data.roundFrequencyRefs ?? [],
+		compositionEpsilon: data.compositionEpsilon ?? 1e-6
 	};
 }).output("stateMatrixOptions", (ctx) => ctx.resultPool.getOptions([{ name: STATE_MATRIX }], { label: { includeNativeLabel: true } })).output("abundanceOptions", (ctx) => ctx.resultPool.getOptions([{
 	axes: [{ name: SAMPLE_ID_AXIS }, { name: VARIANT_KEY_AXIS }],
 	annotations: { "pl7.app/isAbundance": "true" }
 }], { label: { includeNativeLabel: true } })).output("propertyOptions", (ctx) => {
-	const anchor = ctx.data.stateMatrixRef;
-	if (anchor === void 0) return void 0;
-	return ctx.resultPool.getCanonicalOptions({ main: anchor }, [{ axes: [{
-		anchor: "main",
-		idx: 0
-	}] }]);
+	const stateMatrixRef = ctx.data.stateMatrixRef;
+	if (stateMatrixRef === void 0) return void 0;
+	const stateSpec = ctx.resultPool.getPColumnSpecByRef(stateMatrixRef);
+	if (!stateSpec) return void 0;
+	const anchors = { main: stateSpec };
+	const abundanceRef = ctx.data.abundanceRef;
+	const abundanceSpec = abundanceRef ? ctx.resultPool.getPColumnSpecByRef(abundanceRef) : void 0;
+	if (abundanceSpec) anchors.freqAnchor = abundanceSpec;
+	const sourceColumns = ctx.resultPool.selectColumns((spec) => spec.valueType !== "File" && !(spec.annotations?.["pl7.app/isLinkerColumn"] === "true" && spec.axesSpec.length > 2));
+	const collection = new ColumnCollectionBuilder(ctx.getService("pframeSpec")).addSource(sourceColumns).build({ anchors });
+	if (!collection) return void 0;
+	const numeric = /* @__PURE__ */ new Set([
+		"Int",
+		"Long",
+		"Float",
+		"Double"
+	]);
+	const seen = /* @__PURE__ */ new Set();
+	const options = [];
+	for (const m of collection.findColumns({
+		mode: "enrichment",
+		maxHops: 0
+	})) {
+		if (!numeric.has(m.column.spec.valueType)) continue;
+		const dedupKey = m.column.spec.name + "|" + JSON.stringify(m.column.spec.domain ?? {});
+		if (seen.has(dedupKey)) continue;
+		seen.add(dedupKey);
+		options.push({
+			label: m.column.spec.annotations?.["pl7.app/label"] ?? m.column.id,
+			value: m.column.id
+		});
+	}
+	return options;
+}).output("roundFrequencyOptions", (ctx) => {
+	const abundanceRef = ctx.data.abundanceRef;
+	if (abundanceRef === void 0) return void 0;
+	const anchorSpec = ctx.resultPool.getPColumnSpecByRef(abundanceRef);
+	if (!anchorSpec) return void 0;
+	const sourceColumns = ctx.resultPool.selectColumns((spec) => spec.valueType !== "File" && !(spec.annotations?.["pl7.app/isLinkerColumn"] === "true" && spec.axesSpec.length > 2));
+	const collection = new ColumnCollectionBuilder(ctx.getService("pframeSpec")).addSource(sourceColumns).build({ anchors: { freqAnchor: anchorSpec } });
+	if (!collection) return void 0;
+	return collection.findColumns({
+		include: { name: "pl7.app/frequency" },
+		mode: "enrichment",
+		maxHops: 0
+	}).map((m) => ({
+		label: m.column.spec.annotations?.["pl7.app/label"] ?? m.column.id,
+		value: m.column.id
+	}));
 }).output("knownAbundanceOptions", (ctx) => ctx.resultPool.getOptions([{
 	axes: [{ name: SAMPLE_ID_AXIS }, { name: KNOWN_VARIANT_KEY_AXIS }],
 	annotations: {
@@ -8998,6 +9058,20 @@ const platforma$1 = BlockModelV3.create(dataModel).args((data) => {
 	} catch {
 		return;
 	}
+}).outputWithStatus("compositionHeatmapPf", (ctx) => {
+	try {
+		const pCols = ctx.outputs?.resolve("compositionHeatmapPf")?.getPColumns();
+		if (pCols === void 0) return void 0;
+		return createPFrameForGraphs(ctx, pCols);
+	} catch {
+		return;
+	}
+}).output("compositionHeatmapPCols", (ctx) => {
+	try {
+		return ctx.outputs?.resolve("compositionHeatmapPf")?.getPColumns();
+	} catch {
+		return;
+	}
 }).outputWithStatus("knownHeatmapPf", (ctx) => {
 	const { knownAbundanceRef } = ctx.data;
 	if (knownAbundanceRef === void 0) return void 0;
@@ -9046,6 +9120,11 @@ const platforma$1 = BlockModelV3.create(dataModel).args((data) => {
 		href: "/",
 		label: "Mutation Heatmap"
 	}];
+	if (ctx.data.roundFrequencyRefs.length >= 2) sections.push({
+		type: "link",
+		href: "/composition",
+		label: "Composition Enrichment"
+	});
 	if (ctx.data.knownAbundanceRef !== void 0) sections.push({
 		type: "link",
 		href: "/known-heatmap",