gronodandDevin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
32d2184d2e
Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
517 lines
20 KiB
Swift
517 lines
20 KiB
Swift
import Combine
|
||
import Foundation
|
||
import ICCeryCore
|
||
|
||
/// Stage 1/2 form state, runner orchestration, resume flow, and preset
|
||
/// application (issues #7–#11).
|
||
///
|
||
/// `wizard` stays authoritative for persisted identity + disk gating;
|
||
/// this model owns the editable form, logs, gallery, and preset state.
|
||
/// All process work runs through `ArgyllRunner` off `@MainActor`; only
|
||
/// coalesced log batches and completion hop back.
|
||
@MainActor
|
||
final class TargetWorkflowViewModel: ObservableObject {
|
||
|
||
let wizard: WizardViewModel
|
||
let environment: AppEnvironment
|
||
private let fileDialogs = FileDialogService.shared
|
||
|
||
// MARK: - Stage 1 form (targen)
|
||
|
||
@Published var colourSpace: ColourSpace = .rgb {
|
||
didSet {
|
||
guard colourSpace != oldValue else { return }
|
||
// CMYK black patches default to 0, RGB to 4 (docs/08).
|
||
blackPatches = colourSpace == .cmyk ? 0 : 4
|
||
}
|
||
}
|
||
@Published var patchPreset: PatchCountPreset = .standard800
|
||
/// `#patchCountCustom` — used when `patchPreset == .custom`.
|
||
@Published var customPatchCount = 2500
|
||
@Published var whitePatches = 4
|
||
@Published var blackPatches = 4
|
||
|
||
// Advanced — each optional flag is enabled + value, so an untouched
|
||
// control emits nothing (#advanced fields are opt-in).
|
||
@Published var greyStepsEnabled = false
|
||
@Published var greySteps = 5
|
||
@Published var singleChannelEnabled = false
|
||
@Published var singleChannelSteps = 5
|
||
@Published var neutralStepsEnabled = false
|
||
@Published var neutralSteps = 3
|
||
@Published var neutralConcEnabled = false
|
||
@Published var neutralConcentration = 0.50
|
||
@Published var preconditioningProfile: String?
|
||
@Published var highQuality = false
|
||
@Published var adaptationEnabled = false
|
||
@Published var adaptation = 0.10
|
||
@Published var algorithm: FullSpreadAlgorithm = .ofps
|
||
@Published var inkLimitEnabled = false
|
||
@Published var totalInkLimit = 320
|
||
@Published var darkEmphasisEnabled = false
|
||
@Published var darkEmphasis = 1.0
|
||
@Published var devicePowerEnabled = false
|
||
@Published var devicePower = 1.0
|
||
|
||
/// `#targetBasename` — no placeholder is ever invented (#60).
|
||
@Published var targetBasename = ""
|
||
/// `#selectedPathDisplay` / resolved cwd.
|
||
@Published var targetDirectory: URL?
|
||
|
||
// MARK: - Stage 2 form (printtarg)
|
||
|
||
@Published var instrument: PrintInstrument = .i1
|
||
@Published var pageSize: PageSize = .a4
|
||
@Published var customPageW = 210.0
|
||
@Published var customPageH = 297.0
|
||
@Published var bitDepth: TiffBitDepth = .eight
|
||
/// `#tiffDpi` — two-way bound; presets can change it (150-DPI draft
|
||
/// regression must be visible here).
|
||
@Published var tiffDpi = 300
|
||
@Published var layoutOrder: LayoutOrder = .deterministic
|
||
@Published var customSeed = 1
|
||
@Published var labelIsCustom = false
|
||
@Published var customLabel = ""
|
||
@Published var metaPrinter = ""
|
||
@Published var metaInkSet = ""
|
||
@Published var metaDriverPaper = ""
|
||
@Published var metaActualPaper = ""
|
||
|
||
// MARK: - Run state
|
||
|
||
@Published var targenRunning = false
|
||
@Published var targenLog: [String] = []
|
||
@Published var printtargRunning = false
|
||
@Published var printtargLog: [String] = []
|
||
@Published var printtargResult: PrinttargResult?
|
||
/// Sticky until the target changes: `.ti2` resume landed us on
|
||
/// Stage 3 (`#stage3LoadedTargetBanner` data).
|
||
@Published var resumedFromTi2 = false
|
||
|
||
// MARK: - Presets
|
||
|
||
@Published var presets: [ProfilingPreset] = []
|
||
@Published var selectedPresetID = "none"
|
||
@Published var showingSavePreset = false
|
||
@Published var showingManagePresets = false
|
||
@Published var savePresetName = ""
|
||
@Published var savePresetDesc = ""
|
||
|
||
/// Stage 3 measurement workflow, owned at the app level so it persists
|
||
/// across stage switches and can observe settings changes.
|
||
@Published var measurement: MeasurementWorkflowViewModel
|
||
/// Stage 4/5 profile workflow, owned at the app level so it persists
|
||
/// across stage switches and can observe preset values.
|
||
@Published var profile: ProfileWorkflowViewModel
|
||
/// Stage 0 calibration workflow.
|
||
@Published var calibration: CalibrationViewModel!
|
||
/// Stage 2 unmanaged print session.
|
||
@Published var print: PrintSessionViewModel!
|
||
|
||
init(environment: AppEnvironment = .live()) {
|
||
self.environment = environment
|
||
self.wizard = WizardViewModel(stateStore: environment.stateStore)
|
||
self.measurement = MeasurementWorkflowViewModel(
|
||
wizard: wizard,
|
||
environment: environment
|
||
)
|
||
self.profile = ProfileWorkflowViewModel(
|
||
wizard: wizard,
|
||
environment: environment
|
||
)
|
||
self.print = PrintSessionViewModel(wizard: wizard, environment: environment)
|
||
self.calibration = nil
|
||
self.calibration = CalibrationViewModel(
|
||
workflow: self,
|
||
profile: self.profile,
|
||
environment: environment
|
||
)
|
||
reloadPresets()
|
||
}
|
||
|
||
// MARK: - Derived
|
||
|
||
var effectivePatchCount: Int {
|
||
patchPreset.patchCount ?? customPatchCount
|
||
}
|
||
|
||
var canGenerate: Bool {
|
||
PathSecurity.isValidBasename(targetBasename) && targetDirectory != nil
|
||
}
|
||
|
||
var labelMetadata: TargetLabelMetadata {
|
||
TargetLabelMetadata(
|
||
printer: metaPrinter, inkSet: metaInkSet,
|
||
driverPaper: metaDriverPaper, actualPaper: metaActualPaper)
|
||
}
|
||
|
||
/// `#targetLabelPreview` — live preview of the automatic label.
|
||
var automaticLabel: String {
|
||
PrinttargLabel.automatic(
|
||
basename: wizard.basename.isEmpty ? "target" : wizard.basename,
|
||
metadata: labelMetadata)
|
||
}
|
||
|
||
var selectedPreset: ProfilingPreset? {
|
||
presets.first { $0.id == selectedPresetID }
|
||
}
|
||
|
||
// MARK: - Stage 1: generate
|
||
|
||
func buildTargenConfig() -> TargenConfig {
|
||
TargenConfig(
|
||
colourSpace: colourSpace,
|
||
patchCount: effectivePatchCount,
|
||
whitePatches: whitePatches,
|
||
blackPatches: blackPatches,
|
||
greySteps: greyStepsEnabled ? greySteps : nil,
|
||
singleChannelSteps: singleChannelEnabled ? singleChannelSteps : nil,
|
||
neutralSteps: neutralStepsEnabled ? neutralSteps : nil,
|
||
neutralConcentration: neutralConcEnabled ? neutralConcentration : nil,
|
||
preconditioningProfile: preconditioningProfile,
|
||
ofpsHighQuality: highQuality ? true : nil,
|
||
ofpsAdaptation: adaptationEnabled ? adaptation : nil,
|
||
fullSpreadAlgorithm: algorithm == .ofps ? nil : algorithm,
|
||
totalInkLimit: inkLimitEnabled ? totalInkLimit : nil,
|
||
darkEmphasis: darkEmphasisEnabled ? darkEmphasis : nil,
|
||
devicePower: devicePowerEnabled ? devicePower : nil,
|
||
basename: targetBasename,
|
||
workingDirectory: targetDirectory
|
||
)
|
||
}
|
||
|
||
func browseForTargetFile() {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.saveTargetURL
|
||
: fileDialogs.selectTargetFile()
|
||
guard let url else { return }
|
||
targetBasename = url.deletingPathExtension().lastPathComponent
|
||
targetDirectory = url.deletingLastPathComponent()
|
||
}
|
||
|
||
func browseForWorkingDirectory() {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.workDirURL
|
||
: fileDialogs.selectDirectory()
|
||
if let url { targetDirectory = url }
|
||
}
|
||
|
||
func browseForPreconditioningProfile() {
|
||
if let url = fileDialogs.selectProfileFile() {
|
||
preconditioningProfile = url.path
|
||
}
|
||
}
|
||
|
||
func generateTarget() {
|
||
guard canGenerate, !targenRunning else { return }
|
||
let config = buildTargenConfig()
|
||
resumedFromTi2 = false
|
||
let runner = environment.runner
|
||
Task { @MainActor in
|
||
do {
|
||
let url = try await ProcessRunSupport.runLogged(
|
||
setRunning: { self.targenRunning = $0 },
|
||
resetLog: { self.targenLog = [] },
|
||
onLog: { self.targenLog.append(contentsOf: $0) }
|
||
) { onLog in
|
||
try await runner.runTargen(config: config, onLogBatch: onLog)
|
||
}
|
||
wizard.setTarget(
|
||
basename: config.basename,
|
||
workingDirectory: config.workingDirectory)
|
||
wizard.refreshGating()
|
||
wizard.showNotice("Target generated: \(url.lastPathComponent)")
|
||
wizard.go(to: .layOutPrint)
|
||
} catch {
|
||
wizard.showNotice(
|
||
"targen failed: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
}
|
||
|
||
// MARK: - Issue 8: resume an existing target
|
||
|
||
/// `#btn-import-dataset` — open a measured dataset, write a canonical
|
||
/// `.ti3` to the working directory, and set the target (issue #30).
|
||
func importMeasurementDataset() {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.datasetImportURL
|
||
: fileDialogs.selectDatasetFile()
|
||
guard let url else { return }
|
||
importMeasurementDataset(from: url)
|
||
}
|
||
|
||
/// Test seam (issue #80): unit tests pass missing or malformed URLs
|
||
/// directly instead of mutating the global environment.
|
||
func importMeasurementDataset(from url: URL) {
|
||
do {
|
||
let dataset = try CGATSParser.parse(url: url)
|
||
guard let directory = targetDirectory ?? wizard.effectiveWorkingDirectory else {
|
||
wizard.showNotice("Choose a working directory before importing.", kind: .warning)
|
||
return
|
||
}
|
||
|
||
let stem = url.deletingPathExtension().lastPathComponent
|
||
let output = directory.appendingPathComponent("\(stem).ti3")
|
||
try CGATSWriter.write(dataset, to: output)
|
||
|
||
wizard.setTarget(basename: stem, workingDirectory: directory)
|
||
wizard.refreshGating()
|
||
wizard.showNotice("Imported \(dataset.samples.count) patches from \(url.lastPathComponent)")
|
||
|
||
if wizard.isUnlocked(.verifyInstall) {
|
||
wizard.go(to: .verifyInstall)
|
||
} else if wizard.isUnlocked(.buildProfile) {
|
||
wizard.go(to: .buildProfile)
|
||
} else {
|
||
wizard.showNotice("Imported dataset is not ready for profiling.", kind: .warning)
|
||
}
|
||
} catch {
|
||
wizard.showNotice("Import failed: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
|
||
/// `#btnOpenExisting` — open `.ti1`/`.ti2` (open dialog, #103).
|
||
/// `.ti1` → Stage 2; `.ti2` → Stage 3 with the resume notice, but
|
||
/// only when the sibling `.ti1` exists so the artefact gate holds.
|
||
func openExistingTarget() {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.existingTargetURL
|
||
: fileDialogs.selectExistingTarget()
|
||
guard let url else { return }
|
||
|
||
let stem = url.deletingPathExtension().lastPathComponent
|
||
let dir = url.deletingLastPathComponent()
|
||
guard PathSecurity.isValidBasename(stem) else {
|
||
wizard.showNotice("Invalid target name.", kind: .error)
|
||
return
|
||
}
|
||
|
||
switch url.pathExtension.lowercased() {
|
||
case "ti1":
|
||
wizard.setTarget(basename: stem, workingDirectory: dir)
|
||
wizard.refreshGating()
|
||
resumedFromTi2 = false
|
||
measurement.resumedFromTi2 = false
|
||
wizard.go(to: .layOutPrint)
|
||
case "ti2":
|
||
let header = Ti2Header.parse(url)
|
||
guard header.hasSiblingTi1 else {
|
||
wizard.showNotice(
|
||
"Cannot resume \(stem).ti2 — the sibling \(stem).ti1 is missing.",
|
||
kind: .error)
|
||
return
|
||
}
|
||
wizard.setTarget(basename: stem, workingDirectory: dir)
|
||
wizard.refreshGating()
|
||
resumedFromTi2 = true
|
||
measurement.resumedFromTi2 = true
|
||
wizard.showNotice("Resumed from .ti2", kind: .info, autoHideAfter: nil)
|
||
wizard.go(to: .measure)
|
||
default:
|
||
wizard.showNotice(
|
||
"Not a target file — choose a .ti1 or .ti2.", kind: .error)
|
||
}
|
||
}
|
||
|
||
// MARK: - Stage 2: create layout
|
||
|
||
func buildPrinttargConfig() -> PrinttargConfig {
|
||
PrinttargConfig(
|
||
instrument: instrument,
|
||
pageSize: pageSize,
|
||
customPageWidth: customPageW,
|
||
customPageHeight: customPageH,
|
||
bitDepth: bitDepth,
|
||
dpi: tiffDpi,
|
||
layoutOrder: layoutOrder,
|
||
customSeed: customSeed,
|
||
label: PrinttargLabel.resolved(
|
||
customLabel: labelIsCustom ? customLabel : nil,
|
||
basename: wizard.basename,
|
||
metadata: labelMetadata),
|
||
calibrationFile: profile.applyCalibration ? profile.calibrationFile : nil,
|
||
calibrationEmbedOnly: false,
|
||
basename: wizard.basename,
|
||
workingDirectory: wizard.effectiveWorkingDirectory
|
||
)
|
||
}
|
||
|
||
func createLayout() {
|
||
guard wizard.isUnlocked(.layOutPrint), !printtargRunning else { return }
|
||
let config = buildPrinttargConfig()
|
||
printtargResult = nil
|
||
let runner = environment.runner
|
||
Task { @MainActor in
|
||
do {
|
||
let result = try await ProcessRunSupport.runLogged(
|
||
setRunning: { self.printtargRunning = $0 },
|
||
resetLog: { self.printtargLog = [] },
|
||
onLog: { self.printtargLog.append(contentsOf: $0) }
|
||
) { onLog in
|
||
try await runner.runPrinttarg(config: config, onLogBatch: onLog)
|
||
}
|
||
printtargResult = result
|
||
wizard.refreshGating()
|
||
wizard.showNotice(
|
||
"Layout created — \(result.manifest.pages.count) page(s) ready.")
|
||
} catch {
|
||
wizard.showNotice(
|
||
"printtarg failed: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
}
|
||
|
||
/// `#btnAdvanceToStage3` — manual advance once `.ti2` exists.
|
||
func advanceToStage3() {
|
||
wizard.refreshGating()
|
||
wizard.go(to: .measure)
|
||
}
|
||
|
||
// MARK: - Presets
|
||
|
||
func reloadPresets() {
|
||
presets = environment.presetStore.all()
|
||
}
|
||
|
||
/// Applies every Stage 1/2 field of the preset to the live form
|
||
/// (bidirectional — the draft preset's dpi=150 must be visible).
|
||
func applyPreset(_ preset: ProfilingPreset) {
|
||
let targen = TargenConfig(preset: preset, basename: targetBasename, workingDirectory: targetDirectory)
|
||
applyTargenForm(targen)
|
||
// Stage 4 state (incl. calibration) is applied before Stage 2 so
|
||
// the layout config receives the preset's calibration path, not
|
||
// stale live state (#82).
|
||
profile.applyPreset(preset)
|
||
let printtarg = PrinttargConfig(
|
||
preset: preset,
|
||
basename: wizard.basename,
|
||
workingDirectory: wizard.effectiveWorkingDirectory,
|
||
calibrationFile: profile.applyCalibration && !profile.calibrationFile.isEmpty
|
||
? profile.calibrationFile : nil
|
||
)
|
||
applyPrinttargForm(printtarg)
|
||
selectedPresetID = preset.id
|
||
}
|
||
|
||
private func applyTargenForm(_ config: TargenConfig) {
|
||
colourSpace = config.colourSpace
|
||
patchPreset = PatchCountPreset(rawValue: "\(config.patchCount)") ?? .custom
|
||
customPatchCount = config.patchCount
|
||
whitePatches = config.whitePatches
|
||
blackPatches = config.blackPatches
|
||
greySteps = config.greySteps ?? 5
|
||
greyStepsEnabled = config.greySteps != nil
|
||
singleChannelSteps = config.singleChannelSteps ?? 5
|
||
singleChannelEnabled = config.singleChannelSteps != nil
|
||
neutralSteps = config.neutralSteps ?? 3
|
||
neutralStepsEnabled = config.neutralSteps != nil
|
||
neutralConcentration = config.neutralConcentration ?? 0.50
|
||
neutralConcEnabled = config.neutralConcentration != nil
|
||
preconditioningProfile = config.preconditioningProfile
|
||
highQuality = config.ofpsHighQuality == true
|
||
adaptation = config.ofpsAdaptation ?? 0.10
|
||
adaptationEnabled = config.ofpsAdaptation != nil
|
||
algorithm = config.fullSpreadAlgorithm ?? .ofps
|
||
totalInkLimit = config.totalInkLimit ?? 320
|
||
inkLimitEnabled = config.totalInkLimit != nil
|
||
darkEmphasis = config.darkEmphasis ?? 1.0
|
||
darkEmphasisEnabled = config.darkEmphasis != nil
|
||
devicePower = config.devicePower ?? 1.0
|
||
devicePowerEnabled = config.devicePower != nil
|
||
}
|
||
|
||
private func applyPrinttargForm(_ config: PrinttargConfig) {
|
||
instrument = config.instrument
|
||
pageSize = config.pageSize
|
||
customPageW = config.customPageWidth
|
||
customPageH = config.customPageHeight
|
||
bitDepth = config.bitDepth
|
||
tiffDpi = config.dpi
|
||
layoutOrder = config.layoutOrder
|
||
customSeed = config.customSeed
|
||
}
|
||
|
||
/// Snapshot of the live Stage 1/2 form as a custom preset.
|
||
func saveCurrentAsPreset() {
|
||
let name = savePresetName.trimmingCharacters(in: .whitespacesAndNewlines)
|
||
guard !name.isEmpty else {
|
||
wizard.showNotice("Preset needs a name.", kind: .warning)
|
||
return
|
||
}
|
||
let preset = ProfilingPreset(
|
||
id: "custom-\(UUID().uuidString.lowercased())",
|
||
name: name,
|
||
description: savePresetDesc.trimmingCharacters(in: .whitespacesAndNewlines),
|
||
targen: buildTargenConfig(),
|
||
printtarg: buildPrinttargConfig(),
|
||
colprof: profile.buildColprofConfig(),
|
||
calibrationFile: profile.calibrationFile.isEmpty ? nil : profile.calibrationFile,
|
||
applyCalibration: profile.applyCalibration ? true : nil
|
||
)
|
||
do {
|
||
try environment.presetStore.saveCustom(preset)
|
||
reloadPresets()
|
||
selectedPresetID = preset.id
|
||
showingSavePreset = false
|
||
savePresetName = ""
|
||
savePresetDesc = ""
|
||
wizard.showNotice("Preset saved: \(preset.name)")
|
||
} catch {
|
||
wizard.showNotice(
|
||
"Could not save preset: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
|
||
func deletePreset(_ preset: ProfilingPreset) {
|
||
do {
|
||
if try environment.presetStore.deleteCustom(id: preset.id) {
|
||
if selectedPresetID == preset.id { selectedPresetID = "none" }
|
||
reloadPresets()
|
||
} else {
|
||
wizard.showNotice("Built-in presets cannot be deleted.", kind: .warning)
|
||
}
|
||
} catch {
|
||
wizard.showNotice(
|
||
"Could not delete preset: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
|
||
func importPreset() {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.presetImportURL
|
||
: fileDialogs.selectPresetFile()
|
||
guard let url else { return }
|
||
do {
|
||
let data = try Data(contentsOf: url)
|
||
let preset = try environment.presetStore.import(data)
|
||
try environment.presetStore.saveCustom(preset)
|
||
reloadPresets()
|
||
selectedPresetID = preset.id
|
||
wizard.showNotice("Preset imported: \(preset.name)")
|
||
} catch {
|
||
wizard.showNotice(
|
||
"Import failed: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
|
||
func exportPreset(_ preset: ProfilingPreset) {
|
||
let url = UITestHooks.isEnabled
|
||
? UITestHooks.presetExportURL
|
||
: fileDialogs.selectPresetSavePath(name: preset.id)
|
||
guard let url else { return }
|
||
do {
|
||
try environment.presetStore.export(preset)
|
||
.write(to: url, options: .atomic)
|
||
wizard.showNotice("Preset exported: \(url.lastPathComponent)")
|
||
} catch {
|
||
wizard.showNotice(
|
||
"Export failed: \(error.localizedDescription)", kind: .error)
|
||
}
|
||
}
|
||
|
||
static func parseCustomPage(_ raw: String) -> (Double, Double)? {
|
||
PageSize.parseCustom(raw)
|
||
}
|
||
}
|