134 lines
4.8 KiB
GDScript
134 lines
4.8 KiB
GDScript
extends SceneTree
|
|
|
|
const EntryShell = preload("res://entry/entry_shell.gd")
|
|
const SettingsStore = preload("res://entry/settings_store.gd")
|
|
const DraftService = preload("res://entry/initial_company_draft_service.gd")
|
|
const ContractService = preload("res://entry/guild_contract_service.gd")
|
|
|
|
const DRAFT_PATH := "user://dev4_capture_draft.json"
|
|
const COMPANY_PATH := "user://dev4_capture_company.json"
|
|
const SEED := "ffeeddccbbaa00998877665544332211"
|
|
const CONTRACT_OP := "ababababcdcdcdcdefefefef01010101"
|
|
const HIRE_OP_1 := "11111111333333335555555577777777"
|
|
const HIRE_OP_2 := "22222222444444446666666688888888"
|
|
const VIEWPORT := Vector2i(844, 390)
|
|
|
|
var failures: Array[String] = []
|
|
var output_root := ""
|
|
|
|
|
|
func _initialize() -> void:
|
|
call_deferred("run_capture")
|
|
|
|
|
|
func run_capture() -> void:
|
|
output_root = OS.get_environment("ABG_EVIDENCE_DIR")
|
|
if output_root.is_empty():
|
|
output_root = "/tmp/abg-dev4-party-capture"
|
|
output_root = output_root.path_join("party_line")
|
|
DirAccess.make_dir_recursive_absolute(output_root)
|
|
_cleanup()
|
|
var company := _create_company()
|
|
if company.is_empty():
|
|
_finish(1)
|
|
return
|
|
var viewport := SubViewport.new()
|
|
viewport.size = VIEWPORT
|
|
viewport.render_target_update_mode = SubViewport.UPDATE_ALWAYS
|
|
root.add_child(viewport)
|
|
var shell := EntryShell.new()
|
|
shell.auto_start = false
|
|
shell.initial_company_service.path = DRAFT_PATH
|
|
shell.initial_company_service.company_path = COMPANY_PATH
|
|
viewport.add_child(shell)
|
|
await _frames(3)
|
|
var states := [
|
|
{"name": "01-party-zh", "locale": "zh_CN", "scale": 1.0, "selected": "", "swap": false},
|
|
{"name": "02-selected-zh", "locale": "zh_CN", "scale": 1.0, "select_position_zero": true, "swap": false},
|
|
{"name": "03-swapped-zh", "locale": "zh_CN", "scale": 1.0, "select_position_zero": true, "swap": true},
|
|
{"name": "04-party-en130", "locale": "en", "scale": 1.3, "selected": "", "swap": false},
|
|
]
|
|
for state in states:
|
|
var settings := SettingsStore.new().defaults()
|
|
settings.setup_completed = true
|
|
settings.locale = String(state.locale)
|
|
settings.text_scale = float(state.scale)
|
|
shell.settings = settings
|
|
shell.setup_draft = settings.duplicate(true)
|
|
shell.party_draft_order.clear()
|
|
shell.party_selected_recruit_id = ""
|
|
shell.show_party_line(company)
|
|
await _frames(5)
|
|
var selected_id := String(shell.party_draft_order[0]) if bool(state.get("select_position_zero", false)) else String(state.get("selected", ""))
|
|
shell.party_selected_recruit_id = selected_id
|
|
if bool(state.swap):
|
|
var swapped: Array = shell.party_draft_order.duplicate()
|
|
var temporary = swapped[0]
|
|
swapped[0] = swapped[3]
|
|
swapped[3] = temporary
|
|
shell.party_draft_order.assign(swapped)
|
|
if not selected_id.is_empty() or bool(state.swap):
|
|
shell.show_party_line(company)
|
|
await _frames(5)
|
|
await RenderingServer.frame_post_draw
|
|
var image := viewport.get_texture().get_image()
|
|
var path := output_root.path_join(String(state.name) + ".png")
|
|
if image.save_png(path) != OK:
|
|
failures.append("capture failed: %s" % state.name)
|
|
shell.queue_free()
|
|
viewport.queue_free()
|
|
await _frames(100)
|
|
_cleanup()
|
|
if failures.is_empty():
|
|
print("ABG_DEV4_PARTY_CAPTURE_OK frames=4 viewport=844x390 locales=zh_CN,en text_130=true root=%s" % output_root)
|
|
quit(0)
|
|
else:
|
|
_finish(1)
|
|
|
|
|
|
func _create_company() -> Dictionary:
|
|
var draft: Dictionary = _draft_service().begin_or_load(SEED)
|
|
if not draft.ok:
|
|
return {}
|
|
_draft_service().set_selection([String(draft.draft.offers[0].offer_id), String(draft.draft.offers[2].offer_id)])
|
|
var committed: Dictionary = _draft_service().commit_company()
|
|
if not committed.ok:
|
|
return {}
|
|
var created: Dictionary = _contract_service().create_contract(CONTRACT_OP, "CTR-R1-01", "standard")
|
|
if not created.ok:
|
|
return {}
|
|
var offers: Array = created.company.active_contract.market_board.recruit_offers
|
|
_contract_service().hire_recruit(HIRE_OP_1, String(offers[0].source_offer_id), 14)
|
|
var second: Dictionary = _contract_service().hire_recruit(HIRE_OP_2, String(offers[1].source_offer_id), 14)
|
|
return second.company if second.ok else {}
|
|
|
|
|
|
func _draft_service() -> RefCounted:
|
|
var service := DraftService.new()
|
|
service.path = DRAFT_PATH
|
|
service.company_path = COMPANY_PATH
|
|
return service
|
|
|
|
|
|
func _contract_service() -> RefCounted:
|
|
var service := ContractService.new()
|
|
service.company_path = COMPANY_PATH
|
|
return service
|
|
|
|
|
|
func _frames(count: int) -> void:
|
|
for _index in range(count):
|
|
await process_frame
|
|
|
|
|
|
func _cleanup() -> void:
|
|
for path in [DRAFT_PATH, "%s.bak" % DRAFT_PATH, "%s.candidate" % DRAFT_PATH, "%s.bak.candidate" % DRAFT_PATH, COMPANY_PATH, "%s.bak" % COMPANY_PATH, "%s.candidate" % COMPANY_PATH, "%s.bak.candidate" % COMPANY_PATH, "%s.corrupt" % COMPANY_PATH]:
|
|
if FileAccess.file_exists(path):
|
|
DirAccess.remove_absolute(path)
|
|
|
|
|
|
func _finish(code: int) -> void:
|
|
for failure in failures:
|
|
push_error(failure)
|
|
quit(code)
|