PozzettiAndrea
6bac79935e
Match LiTo's project layout. Net file moves:
spz_route.py -> nodes/spz_route.py
install.py -> root (was briefly in nodes/)
server_tools/ -> web/server_tools/ (Node JS subprocess
tool — non-python
asset, sits next to
browser-served web/)
Root and nodes/ now mirror ComfyUI-LiTo:
Root: __init__.py, install.py, pyproject.toml, comfy-env-root.toml,
comfy-test.toml, README.md, LICENSE, web/, scripts/, nodes/
nodes/: __init__.py, comfy-env.toml, common.py, preview_gaussian.py,
merge_gaussians.py, load_ply.py, spz_route.py
## Root __init__.py — LiTo-style
import logging
log = logging.getLogger("comfyui-gaussianpack")
from comfy_env import register_nodes
NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = register_nodes()
WEB_DIRECTORY = "./web"
Delegates node discovery to `comfy_env.register_nodes()`, which scans
`nodes/__init__.py` for the mappings. WEB_DIRECTORY stays at root
(points at top-level web/) so ComfyUI's static handler doesn't have to
walk into nodes/.
## nodes/__init__.py — clean child package
Imports PreviewGaussians, GaussianMerge, LoadPLY from sibling modules,
calls register_routes() for the SPZ HTTP route, exposes the two
mapping dicts. Dropped `WEB_DIRECTORY` (it belongs at root).
## nodes/comfy-env.toml — pixi env declaration
numpy + plyfile + nanogs (git pip). Same content as before — was
previously at root, where comfy-env's discovery wouldn't pick it up
for isolation purposes. Now lives where comfy-env expects.
## spz_route.py — path fix
NODE_SCRIPT was `THIS_DIR / "server_tools" / "ply_to_spz.mjs"` when the
file was at root. Now at nodes/ with server_tools moved to web/, so:
NODE_SCRIPT = THIS_DIR.parent / "web" / "server_tools" / "ply_to_spz.mjs"
Smoke test (worker pixi env):
PYTHONPATH=.../ComfyUI:. python -c "import nodes; print(nodes.NODE_CLASS_MAPPINGS)"
-> {'PreviewGaussians': ..., 'GaussianMerge': ..., 'LoadPLY': ...}
42 lines
1.3 KiB
JavaScript
42 lines
1.3 KiB
JavaScript
const createStream = (data) => {
|
|
return new ReadableStream({
|
|
async start(controller) {
|
|
controller.enqueue(data);
|
|
controller.close();
|
|
},
|
|
});
|
|
};
|
|
export async function decompressGzipped(data) {
|
|
try {
|
|
const stream = createStream(data);
|
|
if (!stream)
|
|
throw new Error('Failed to create stream from data');
|
|
return await decompressGzipStream(stream);
|
|
}
|
|
catch (error) {
|
|
console.error('Error decompressing gzipped data:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
export async function decompressGzipStream(stream) {
|
|
const decompressor = new DecompressionStream('gzip');
|
|
const decompressedStream = stream.pipeThrough(decompressor);
|
|
const response = new Response(decompressedStream);
|
|
const buffer = await response.arrayBuffer();
|
|
return new Uint8Array(buffer);
|
|
}
|
|
export async function compressGzipped(data) {
|
|
try {
|
|
const stream = createStream(data);
|
|
const compressor = new CompressionStream('gzip');
|
|
const compressedStream = stream.pipeThrough(compressor);
|
|
const response = new Response(compressedStream);
|
|
const buffer = await response.arrayBuffer();
|
|
return new Uint8Array(buffer);
|
|
}
|
|
catch (error) {
|
|
console.error('Error compressing gzipped data:', error);
|
|
throw error;
|
|
}
|
|
}
|