mirror of
https://github.com/storytold/lightcraft.git
synced 2026-10-25 18:47:42 -06:00
Brandon ThomasandClaude Opus 5.5
c65065de2f
One config-dir helper (engine::config::config_dir; camera_profiles re-exports it) with both the face-models and denoise-models defaults; hosts set up both model folders; Settings has Faces and AI Denoise tabs; both command sets registered; both model dialogs; locale entries from both; one [features] table per app manifest; Cargo.lock regenerated by cargo; the DecodedSource cache cost counts both segmentation mattes and the denoised twin. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
59 lines
2.5 KiB
Rust
59 lines
2.5 KiB
Rust
//! Where LightCraft keeps per-user data on this machine: the config folder (app settings, camera
|
|
//! profiles, face and denoise models).
|
|
//!
|
|
//! Hosts that have a file system (the desktop app, the CLI, the MCP server) share these defaults so a model
|
|
//! installed from one is there for the others. Nothing here is applied automatically: a [`Session`] has no
|
|
//! face- or denoise-models folder until a host asks for one ([`Session::with_default_face_models`],
|
|
//! [`Session::with_default_denoise_models`]), so tests stay hermetic.
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use crate::Session;
|
|
|
|
/// LightCraft's per-user config folder (`…/LightCraft`), if the platform tells us where.
|
|
pub fn config_dir() -> Option<PathBuf> {
|
|
if cfg!(target_os = "macos") {
|
|
std::env::var_os("HOME").map(|h| PathBuf::from(h).join("Library/Application Support/LightCraft"))
|
|
} else if cfg!(windows) {
|
|
std::env::var_os("APPDATA").map(|a| PathBuf::from(a).join("LightCraft"))
|
|
} else {
|
|
std::env::var_os("XDG_CONFIG_HOME")
|
|
.map(PathBuf::from)
|
|
.or_else(|| std::env::var_os("HOME").map(|h| PathBuf::from(h).join(".config")))
|
|
.map(|c| c.join("lightcraft"))
|
|
}
|
|
}
|
|
|
|
/// Where face models are kept: `$LIGHTCRAFT_FACE_MODELS` if set, else `<config>/models`.
|
|
pub fn default_face_models_dir() -> Option<PathBuf> {
|
|
std::env::var_os("LIGHTCRAFT_FACE_MODELS").filter(|v| !v.is_empty()).map(PathBuf::from).or_else(|| config_dir().map(|d| d.join("models")))
|
|
}
|
|
|
|
/// Where opt-in denoise models are kept: `$LIGHTCRAFT_DENOISE_MODELS` if set, else `<config>/denoise-models`.
|
|
pub fn default_denoise_models_dir() -> Option<PathBuf> {
|
|
std::env::var_os("LIGHTCRAFT_DENOISE_MODELS")
|
|
.filter(|v| !v.is_empty())
|
|
.map(PathBuf::from)
|
|
.or_else(|| config_dir().map(|d| d.join("denoise-models")))
|
|
}
|
|
|
|
impl Session {
|
|
/// Keep face models in the shared default folder ([`default_face_models_dir`]).
|
|
pub fn with_default_face_models(mut self) -> Self {
|
|
self.face_models_dir = default_face_models_dir();
|
|
self
|
|
}
|
|
|
|
/// Keep denoise models in the shared default folder ([`default_denoise_models_dir`]).
|
|
pub fn with_default_denoise_models(mut self) -> Self {
|
|
self.set_denoise_models_dir(default_denoise_models_dir());
|
|
self
|
|
}
|
|
|
|
/// Point AI denoise at `dir` (or at no folder), dropping what was learned about the old one.
|
|
pub fn set_denoise_models_dir(&mut self, dir: Option<PathBuf>) {
|
|
self.denoise.models_dir = dir;
|
|
self.denoise.touch();
|
|
}
|
|
}
|