retiam 1.0.0 → 1.0.1

package/bin/render-config.js

@@ -0,0 +1,87 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { pathToFileURL } from 'node:url';
+
+export const CLI_RENDER_DEFAULTS = Object.freeze({
+    fps: 60,
+    width: 1920,
+    height: 1080,
+    videoBitrate: '20M',
+    outputFormat: 'mp4',
+    outputDir: './outputs/video',
+    maxDuration: 0,
+    devServerPort: 5173,
+    wsPort: 5001,
+    replayFile: null,
+});
+
+// Serializable RenderConfig fields forwarded to SceneComposer before GPU setup.
+export const RENDER_CONFIG_KEYS = Object.freeze([
+    'AA_Function', 'AA', 'sample_count', 'PPLL', 'max_layers',
+    'shadowsEnabled', 'shadowMethod', 'lightSize', 'frustumCulling',
+    'shadingStyle', 'celBands', 'celRimStrength', 'celRimPower', 'goochWarm', 'goochCool',
+    'raytraceEnabled', 'rtSamples', 'rtMaxAccumulation', 'bvhMethod', 'gamma', 'bounce_count',
+    'denoiser', 'oidnModel',
+]);
+
+export async function loadRenderConfig(projectRoot) {
+    const preferred = path.resolve(projectRoot, 'retiam.config.js');
+    const configPath = fs.existsSync(preferred) ? preferred : path.resolve(projectRoot, 'config.js');
+    if (!fs.existsSync(configPath)) return { defaultConfig: {}, configSource: 'built-in defaults' };
+
+    try {
+        const module = await import(pathToFileURL(configPath).href);
+        return { defaultConfig: module.default || {}, configSource: path.basename(configPath) };
+    } catch (error) {
+        console.warn(`⚠️  Failed to import ${path.basename(configPath)}, falling back to default values.`, error);
+        return { defaultConfig: {}, configSource: 'built-in defaults' };
+    }
+}
+
+export function resolveRenderOptions(defaultConfig = {}, args = [], projectRoot = process.cwd()) {
+    const defaults = Object.fromEntries(Object.entries(CLI_RENDER_DEFAULTS)
+        .map(([key, value]) => [key, defaultConfig[key] ?? value]));
+    const config = {
+        fps: defaults.fps,
+        width: defaults.width,
+        height: defaults.height,
+        bitrate: defaults.videoBitrate,
+        maxDuration: defaults.maxDuration,
+        wsPort: defaults.wsPort,
+        devServerPort: defaults.devServerPort,
+    };
+    const renderConfigOverrides = {};
+    for (const key of RENDER_CONFIG_KEYS) {
+        if (defaultConfig[key] !== undefined) renderConfigOverrides[key] = defaultConfig[key];
+    }
+
+    const targetFile = args[0] && !args[0].startsWith('-') ? args[0] : null;
+    let replayFile = defaults.replayFile;
+    let outputDir = path.resolve(projectRoot, defaults.outputDir);
+
+    for (let i = targetFile ? 1 : 0; i < args.length; i++) {
+        const flag = args[i];
+        if (flag === '-w' || flag === '--width') config.width = parseInt(args[++i], 10);
+        else if (flag === '-h' || flag === '--height') config.height = parseInt(args[++i], 10);
+        else if (flag === '-f' || flag === '--fps') config.fps = parseInt(args[++i], 10);
+        else if (flag === '-d' || flag === '--duration') config.maxDuration = parseInt(args[++i], 10);
+        else if (flag === '-o' || flag === '--out') outputDir = path.resolve(projectRoot, args[++i]);
+        else if (flag === '--raytrace') {
+            const value = args[i + 1];
+            if (value === 'true' || value === 'false') {
+                renderConfigOverrides.raytraceEnabled = value === 'true';
+                i++;
+            } else renderConfigOverrides.raytraceEnabled = true;
+        } else if (flag === '--no-raytrace') renderConfigOverrides.raytraceEnabled = false;
+        else if (flag === '--denoiser') renderConfigOverrides.denoiser = args[++i];
+        else if (flag === '--raw') renderConfigOverrides.denoiser = 'native';
+        else if (flag === '--replay') replayFile = args[++i];
+        else if (flag.endsWith('.json') && !replayFile) replayFile = flag;
+    }
+
+    // The browser sends raw buffers and auxiliary passes only for native raytracing.
+    config.useRawFrames = renderConfigOverrides.denoiser === 'native' && renderConfigOverrides.raytraceEnabled === true;
+    config.exportAlbedoNormal = config.useRawFrames;
+
+    return { config, renderConfigOverrides, targetFile, replayFile, outputDir, outputFormat: defaults.outputFormat };
+}