nan
b5a2f9690c
- Added depth.py for MoGe-2 depth sampling and balloon deformation functionality. - Introduced validate_balloon_asset.py to validate balloon shape modifiers and material properties in Blender. - Enhanced test_server.py with new unit tests for balloon depth analysis and job progress tracking. - Updated existing tests to improve code clarity and maintainability.
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
"""SnapForm Job Server without Blender dependencies."""
|
|
|
|
from blendjob import JobServer
|
|
|
|
from .jobs import generate_depth as _generate_depth
|
|
from .jobs import generate_surface as _generate_surface
|
|
from .jobs.background import run as _remove_background
|
|
from .jobs.balloon import run as _analyze_balloon
|
|
from .jobs.debug_ben2 import run as _debug_ben2
|
|
from .jobs.debug_da3 import run as _debug_da3
|
|
from .jobs.debug_moge2 import run as _debug_moge2
|
|
from .jobs.frames import run as _extract_frames
|
|
from .model_manager import ModelManager, ModelSessionCache
|
|
|
|
|
|
server = JobServer("SnapForm Job Server")
|
|
|
|
|
|
@server.resource("model_manager")
|
|
def create_model_manager(job_server):
|
|
return ModelManager(job_server.storage_root / "models")
|
|
|
|
|
|
@server.job("download-model")
|
|
def download_model(context, parameters):
|
|
manager = context.resource("model_manager")
|
|
return manager.download(context, parameters["model"])
|
|
|
|
|
|
@server.job("download-required-models")
|
|
def download_required_models(context, parameters):
|
|
manager = context.resource("model_manager")
|
|
return manager.download_missing(context, parameters["models"])
|
|
|
|
|
|
@server.job("analyze-balloon")
|
|
def analyze_balloon(context, parameters):
|
|
return _analyze_balloon(context, parameters)
|
|
|
|
|
|
@server.job("remove-background")
|
|
def remove_background(context, parameters):
|
|
return _remove_background(context, parameters)
|
|
|
|
|
|
@server.job("extract-frames")
|
|
def extract_frames(context, parameters):
|
|
return _extract_frames(context, parameters)
|
|
|
|
|
|
@server.job("generate-depth")
|
|
def generate_depth(context, parameters):
|
|
return _generate_depth(context, parameters)
|
|
|
|
|
|
@server.job("generate-surface")
|
|
def generate_surface(context, parameters):
|
|
return _generate_surface(context, parameters)
|
|
|
|
|
|
@server.job("debug-da3")
|
|
def debug_da3(context, parameters):
|
|
return _debug_da3(context, parameters)
|
|
|
|
|
|
@server.job("debug-moge2")
|
|
def debug_moge2(context, parameters):
|
|
return _debug_moge2(context, parameters)
|
|
|
|
|
|
@server.job("debug-ben2")
|
|
def debug_ben2(context, parameters):
|
|
return _debug_ben2(context, parameters)
|
|
|
|
|
|
__all__ = ("ModelManager", "ModelSessionCache", "server")
|