95 lines
2.3 KiB
TypeScript
95 lines
2.3 KiB
TypeScript
import Bun from 'bun';
|
|
import path from 'node:path';
|
|
import { parseArgs } from 'node:util';
|
|
import fs from 'node:fs';
|
|
import { featurize } from '../preprocess/featurize';
|
|
import { parseTja } from '../preprocess/parse';
|
|
|
|
const { values } = parseArgs({
|
|
args: Bun.argv,
|
|
options: {
|
|
tja: {
|
|
type: "string"
|
|
},
|
|
workingDir: {
|
|
type: "string"
|
|
},
|
|
script: {
|
|
type: "string"
|
|
}
|
|
},
|
|
allowPositionals: true,
|
|
});
|
|
|
|
if (!values.tja || !values.workingDir || !values.script) {
|
|
console.error("Usage: bun script/predict_tja.ts --tja <path_to_tja> [--workingDir <dir>] [--script <python_script>]");
|
|
process.exit(1);
|
|
}
|
|
|
|
const tjaPath = values.tja;
|
|
if (!fs.existsSync(tjaPath)) {
|
|
console.error(`File not found: ${tjaPath}`);
|
|
process.exit(1);
|
|
}
|
|
|
|
const workingDir = values.workingDir!;
|
|
const pythonScript = values.script!;
|
|
|
|
if (!fs.existsSync(pythonScript)) {
|
|
console.error(`Python script not found: ${pythonScript}`);
|
|
process.exit(1);
|
|
}
|
|
|
|
// 1. TJA 파싱 및 피처 추출
|
|
const tjaContent = fs.readFileSync(tjaPath, 'utf-8');
|
|
const songno = path.basename(tjaPath, '.tja');
|
|
const parsed = parseTja(tjaContent);
|
|
|
|
const features: any[] = [];
|
|
if (parsed.oni) {
|
|
features.push({
|
|
songno,
|
|
difficulty: 'oni',
|
|
...featurize(parsed.oni)
|
|
});
|
|
}
|
|
if (parsed.edit) {
|
|
features.push({
|
|
songno,
|
|
difficulty: 'ura',
|
|
...featurize(parsed.edit)
|
|
});
|
|
}
|
|
|
|
if (features.length === 0) {
|
|
console.error("No Oni or Ura difficulty found in the TJA file.");
|
|
process.exit(1);
|
|
}
|
|
|
|
// 2. 임시 피처 파일 저장
|
|
const tempFeaturePath = path.join(workingDir, `temp_predict_${Date.now()}.json`);
|
|
fs.writeFileSync(tempFeaturePath, JSON.stringify(features, null, 2), 'utf-8');
|
|
|
|
try {
|
|
// 3. Python 예측 스크립트 실행
|
|
const proc = Bun.spawnSync([
|
|
"python3",
|
|
pythonScript,
|
|
"--workingDir", workingDir,
|
|
"--feature", tempFeaturePath,
|
|
"--songno", songno
|
|
]);
|
|
|
|
if (proc.success) {
|
|
console.log(proc.stdout.toString());
|
|
} else {
|
|
console.error("Prediction failed:");
|
|
console.error(proc.stderr.toString());
|
|
}
|
|
} finally {
|
|
// 4. 임시 파일 삭제
|
|
if (fs.existsSync(tempFeaturePath)) {
|
|
fs.unlinkSync(tempFeaturePath);
|
|
}
|
|
}
|