94 lines
3.3 KiB
Go
94 lines
3.3 KiB
Go
package app
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// AutoDL's complete resolution enum also controls orientation. The bundled
|
|
// workflow contract supplies fixed choices; client-supplied fixed values are ignored.
|
|
func normalizeAutoDLVideoScreenSpec(profile *VideoCapabilityConfig, modelName string) (*VideoCapabilityConfig, error) {
|
|
dir, err := officialPluginPackageDir()
|
|
if err != nil {
|
|
return nil, BadAuthRequest("无法读取 AutoDL 工作流规格")
|
|
}
|
|
pkg, err := inspectOfficialPluginPackage(filepath.Join(dir, "autodl-comfyui.yingce-plugin"))
|
|
if err != nil {
|
|
return nil, BadAuthRequest("无法读取 AutoDL 工作流规格")
|
|
}
|
|
modelName = strings.TrimPrefix(strings.TrimSpace(modelName), "models/")
|
|
fixed := VideoScreenSpecConfig{Ratios: []string{}, Resolutions: []string{}}
|
|
found := false
|
|
for _, workflow := range pkg.info.Manifest.Contributes.Workflows {
|
|
if workflow.ID != modelName || workflow.ProviderID != "autodl-comfyui" {
|
|
continue
|
|
}
|
|
found = true
|
|
for _, parameter := range workflow.Parameters {
|
|
if parameter.Mapping != "resolution" && parameter.Name != "resolution" {
|
|
continue
|
|
}
|
|
fixed.Resolutions = cleanVideoScreenSpecValues(parameter.Values)
|
|
fixed.DefaultResolution, _ = workflow.Defaults[parameter.Name].(string)
|
|
fixed.DefaultResolution = strings.TrimSpace(fixed.DefaultResolution)
|
|
}
|
|
break
|
|
}
|
|
if !found || len(fixed.Resolutions) == 0 || !containsCapabilityString(fixed.Resolutions, fixed.DefaultResolution) {
|
|
return nil, BadAuthRequest("当前 AutoDL 工作流未声明有效的分辨率列表和默认值")
|
|
}
|
|
value := *profile
|
|
// No AutoDL v2.3 workflow declares an independent aspect-ratio parameter.
|
|
value.Ratios, value.DefaultRatio = []string{}, ""
|
|
value.Resolutions = cleanVideoScreenSpecValues(profile.Resolutions)
|
|
value.DefaultResolution = strings.TrimSpace(profile.DefaultResolution)
|
|
for i, resolution := range value.Resolutions {
|
|
matched := false
|
|
for _, preset := range fixed.Resolutions {
|
|
if strings.EqualFold(resolution, preset) {
|
|
value.Resolutions[i] = preset
|
|
matched = true
|
|
break
|
|
}
|
|
}
|
|
if !matched {
|
|
return nil, BadAuthRequest("分辨率不在当前 AutoDL 工作流的插件预设中：" + resolution)
|
|
}
|
|
if strings.EqualFold(value.Resolutions[i], value.DefaultResolution) {
|
|
value.DefaultResolution = value.Resolutions[i]
|
|
}
|
|
}
|
|
if len(value.Resolutions) == 0 || !containsCapabilityString(value.Resolutions, value.DefaultResolution) {
|
|
return nil, BadAuthRequest("请至少选择一个完整分辨率，并从所选值中设置默认值")
|
|
}
|
|
value.FixedScreenSpec = &fixed
|
|
return &value, nil
|
|
}
|
|
|
|
func cleanVideoScreenSpecValues(values []string) []string {
|
|
result := make([]string, 0, len(values))
|
|
seen := make(map[string]bool)
|
|
for _, value := range values {
|
|
value = strings.TrimSpace(value)
|
|
key := strings.ToLower(value)
|
|
if value != "" && !seen[key] {
|
|
result = append(result, value)
|
|
seen[key] = true
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func applyAutoDLVideoScreenSpec(input *canvasGenerationInput, profile *VideoCapabilityConfig) {
|
|
if input == nil || profile == nil || profile.FixedScreenSpec == nil {
|
|
return
|
|
}
|
|
input.Config.Size = ""
|
|
resolution := videoResolutionNameRequest(profile, input.Config.VQuality)
|
|
if resolution != "" {
|
|
input.Config.VQuality = resolution
|
|
} else if isAutomaticVideoResolution(input.Config.VQuality) {
|
|
input.Config.VQuality = profile.DefaultResolution
|
|
}
|
|
}
|