-
- Color Swap Plan
-
- {tooManyColors ? (
-
- Swap instructions are disabled for very large palettes (
- {colorCount} colors). Reduce the image to 64 colors or fewer in
- 2D mode first.
-
- ) : swapPlan.length <= 1 ? (
+
Color Swap Plan
+ {swapPlan.length <= 1 ? (
Only one color configured — no swaps needed.
@@ -203,3 +191,74 @@ export default function PrintInstructions({
);
}
+
+// ---------------------------------------------------------------------------
+// HeadSchedule — multi-head load / swap schedule in layer order
+// ---------------------------------------------------------------------------
+
+function HeadSchedule({ events }: { events: MultiHeadScheduleEvent[] }) {
+ if (events.length === 0) {
+ return (
+
+ No head assignments computed yet.
+
+ );
+ }
+
+ return (
+
+
Head Schedule
+
+ {events.filter(evt => evt.isPrePrint || evt.swapCount > 0).map((evt, evtIdx) => {
+ return (
+
+ {/* Event header */}
+
+ {evt.isPrePrint
+ ? <>Before print — load all heads
+ : <>Layer {evt.startLayer} — swap {evt.swapCount} head{evt.swapCount !== 1 ? 's' : ''}
+ }
+
+
+ {/* Nozzle rows — all heads shown; changed ones highlighted */}
+
+ {evt.nozzles.map((n) => (
+
+
+ Head {n.nozzle}
+
+
+
+ {n.filamentName}
+
+
+ ))}
+
+
+ );
+ })}
+
+
+ );
+}
diff --git a/src/components/ThreeDControls.tsx b/src/components/ThreeDControls.tsx
index 0cd03793..81d6d8df 100644
--- a/src/components/ThreeDControls.tsx
+++ b/src/components/ThreeDControls.tsx
@@ -6,6 +6,7 @@ import { Button } from '@/components/ui/button';
import { Check, RotateCcw, Loader2 } from 'lucide-react';
import { Tabs, TabsList, TabsTrigger, TabsContent } from '@/components/ui/tabs';
import { autoPaintToSliceHeights } from '../lib/autoPaint';
+import { patchedLayersToPlan, patchedLayersToSliceData, buildPerColorLayerColors } from '../lib/patchedLayersToPlan';
import {
loadPrintSettingsFromStorage,
savePrintSettingsToStorage,
@@ -16,6 +17,7 @@ import { useProfileManager } from '../hooks/useProfileManager';
import { useColorSlicing } from '../hooks/useColorSlicing';
import { useSwapPlan } from '../hooks/useSwapPlan';
import { useAutoPaintWorker } from '../hooks/useAutoPaintWorker';
+import { useMultiHeadWorker } from '../hooks/useMultiHeadWorker';
import type {
AutoPaintRepeatLimit,
AutoPaintTransitionOpacity,
@@ -150,6 +152,13 @@ export default function ThreeDControls({
persisted?.regionWeightingMode ?? 'uniform'
);
+ // --- Multi-head mode ---
+ const [multiHeadMode, setMultiHeadMode] = useState(persisted?.multiHeadMode ?? false);
+ const [multiHeadCount, setMultiHeadCount] = useState(persisted?.multiHeadCount ?? 4);
+ const [multiHeadSearchDepth, setMultiHeadSearchDepth] = useState<'fast' | 'balanced' | 'thorough'>(
+ persisted?.multiHeadSearchDepth ?? 'balanced'
+ );
+
const handleEnhancedColorMatchChange = useCallback((v: boolean) => {
setEnhancedColorMatch(v);
if (!v) {
@@ -202,6 +211,9 @@ export default function ThreeDControls({
optimizerSeed,
regionWeightingMode,
smoothMeshing,
+ multiHeadMode,
+ multiHeadCount,
+ multiHeadSearchDepth,
});
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [
@@ -218,6 +230,9 @@ export default function ThreeDControls({
optimizerSeed,
regionWeightingMode,
smoothMeshing,
+ multiHeadMode,
+ multiHeadCount,
+ multiHeadSearchDepth,
]);
useEffect(() => {
@@ -280,6 +295,12 @@ export default function ThreeDControls({
});
const autoPaintProgressPercent = Math.round(Math.max(0, Math.min(1, autoPaintProgress)) * 100);
+ // --- Multi-head analysis (runs in Web Worker to avoid blocking the UI) ---
+ const {
+ isComputing: isMultiHeadComputing,
+ run: runMultiHead,
+ } = useMultiHeadWorker();
+
const autoPaintSliceData = useMemo(() => {
if (!autoPaintResult) return undefined;
return autoPaintToSliceHeights(autoPaintResult, layerHeight, slicerFirstLayerHeight);
@@ -344,7 +365,7 @@ export default function ThreeDControls({
const isInstructionOverLimit = instructionColorCount > 64;
// --- Swap Plan ---
- const { swapPlan, copied, copyToClipboard } = useSwapPlan({
+ const { swapPlan, multiHeadPlan, copied, copyToClipboard } = useSwapPlan({
colorOrder: instructionColorOrder,
colorSliceHeights: instructionColorSliceHeights,
filtered: instructionFiltered,
@@ -352,14 +373,60 @@ export default function ThreeDControls({
slicerFirstLayerHeight: instructionSlicerFirstLayerHeight,
paintMode: instructionPaintMode,
autoPaintResult: instructionAutoPaintResult,
- disabled: isInstructionOverLimit,
+ multiHeadWindows: persisted?.multiHeadWindows,
+ patchedTransitionZones: persisted?.patchedTransitionZones,
+ nozzleAssignments: persisted?.nozzleAssignments,
+ windowRunFilaments: persisted?.windowRunFilaments,
+ preWindowFilaments: persisted?.preWindowFilaments,
+ nonWindowedRanges: persisted?.nonWindowedRanges,
+ filaments,
flatPaint: instructionFlatPaint,
});
// --- Apply handler ---
- const handleApply = useCallback(() => {
+ const handleApply = useCallback(async () => {
if (!onChange) return;
+ // Run the appropriate multi-head optimizer based on the selected mode.
+ let activeResult = null;
+ if (multiHeadMode && paintMode === 'autopaint' && autoPaintResult) {
+ // `filtered` carries SwatchEntry objects at runtime (with pixel-frequency
+ // `count`), even though the prop is typed as the narrower Swatch.
+ const swatches = filtered.map((s) => ({
+ hex: s.hex,
+ count: (s as { count?: number }).count,
+ }));
+ activeResult = await runMultiHead({
+ filaments,
+ autoPaintResult,
+ imageSwatches: swatches,
+ layerHeight,
+ firstLayerHeight: slicerFirstLayerHeight,
+ n: multiHeadCount,
+ searchDepth: multiHeadSearchDepth,
+ });
+ }
+
+ const newMultiHeadWindows = activeResult?.windows ?? [];
+ const patchedTransitionZones = activeResult && activeResult.patchedLayers.length > 0
+ ? patchedLayersToPlan(activeResult.patchedLayers, filaments)
+ : undefined;
+ const patchedSliceData = activeResult && activeResult.patchedLayers.length > 0
+ ? patchedLayersToSliceData(activeResult.patchedLayers, filaments, slicerFirstLayerHeight)
+ : undefined;
+ const perColorLayerColors = activeResult && activeResult.patchedLayers.length > 0
+ ? buildPerColorLayerColors(activeResult.patchedLayers, activeResult.colorLayerFilaments, filaments)
+ : undefined;
+ // Per-colour filament-index-per-layer map. ThreeDView needs this (together with
+ // the window/nozzle data below) to resolve each sub-mesh's physical nozzle; if it
+ // isn't persisted, nozzle tagging silently no-ops and export3mf falls back to
+ // colour-order extruders and all-white filament colours.
+ const colorLayerFilaments = activeResult?.colorLayerFilaments;
+ const windowRunFilaments = activeResult?.windowRunFilaments;
+ const nozzleAssignments = activeResult?.nozzleAssignments;
+ const preWindowFilaments = activeResult?.preWindowFilaments;
+ const nonWindowedRanges = activeResult?.nonWindowedRanges;
+
if (paintMode === 'autopaint' && autoPaintSliceData && autoPaintResult) {
onChange({
layerHeight,
@@ -385,6 +452,18 @@ export default function ThreeDControls({
autoPaintFilamentSwatches: autoPaintSliceData.filamentSwatches,
calibrationLayerHeight,
smoothMeshing,
+ multiHeadMode,
+ multiHeadCount,
+ multiHeadSearchDepth,
+ multiHeadWindows: newMultiHeadWindows,
+ patchedTransitionZones,
+ patchedSliceData,
+ perColorLayerColors,
+ colorLayerFilaments,
+ windowRunFilaments,
+ nozzleAssignments,
+ preWindowFilaments,
+ nonWindowedRanges,
});
} else {
onChange({
@@ -402,6 +481,18 @@ export default function ThreeDControls({
regionWeightingMode,
calibrationLayerHeight,
smoothMeshing,
+ multiHeadMode,
+ multiHeadCount,
+ multiHeadSearchDepth,
+ multiHeadWindows: newMultiHeadWindows,
+ patchedTransitionZones,
+ patchedSliceData,
+ perColorLayerColors,
+ colorLayerFilaments,
+ windowRunFilaments,
+ nozzleAssignments,
+ preWindowFilaments,
+ nonWindowedRanges,
});
}
}, [
@@ -428,6 +519,10 @@ export default function ThreeDControls({
smoothMeshing,
autoPaintResult,
autoPaintSliceData,
+ multiHeadMode,
+ multiHeadCount,
+ multiHeadSearchDepth,
+ runMultiHead,
]);
return (
@@ -437,10 +532,10 @@ export default function ThreeDControls({