Add node helper script for generating mission file from spice kernels

2026-01-08 04:31:08 -06:00 · 2016-08-23 12:47:41 -04:00
parent f539d5b0b0
commit cd16e73f56
1 changed files with 123 additions and 0 deletions
--- a/support/mission/ckbrief2mission.js
+++ b/support/mission/ckbrief2mission.js
@@ -0,0 +1,123 @@
+var USAGE_TEXT = '\n\
+USAGE:\n\
+node spice2mission.js <input directory> [,<ckbrief arguments...>] \n\
+\n\
+PURPOSE:\n\
+This script takes a root directory containing SPICE files (*.bc) \n\
+and prints lua code representing the contents for a .mission-file\n\
+to std out. The directory structure is used to create the hierarchy\n\
+of mission phases.\
+';
+
+const exec = require('child_process').exec;
+
+var KEY_START = 'start';
+var KEY_END = 'end';
+var FILE_START = 'Summary for: ';
+var TIME_START = 'Begin ET: ';
+var SPACES = '                                                                   ';
+
+var inputDir = process.argv[2];
+var additionalArguments = process.argv.splice(3);
+
+
+run();
+
+function run(){
+	if(process.argv.length == 2){
+		console.log('Error: expected at least one (1) argument');
+		console.log(USAGE_TEXT);
+		return;
+	}
+
+	execute('find ' + inputDir + ' -type f -name *.bc*', (files) => {
+		files = files.split('\n').join(' ');
+		//console.log(files);
+		var cmd = "ckbrief.exe " + additionalArguments.join(" ") + " " + files + '-g';
+		execute(cmd, handleCkbriefOutput);
+	});
+}
+
+function onSuccess(cb){
+	return (error, stdout, stderr) => {
+		if(error) return console.error('exec error: ' + error);
+		if(stderr) return console.error(stderr);
+		cb(stdout);
+	}
+}
+
+function execute(cmd, cb){
+	//console.log('executing: ' + cmd);
+	exec(cmd, onSuccess(cb));
+}
+
+/**
+* Handles the output from the shell command
+* $ find <inputDit> -type f -name *.bc | xargs ckbrief.exe <additional arguments...> -g
+*/
+function handleCkbriefOutput(output){
+	var mission = {};
+	var currentLeaf = null;
+	var lines = output.split('\n');
+	for (var i = 0; i < lines.length; i++) {
+		var line = lines[i].trim();
+		if(line.startsWith(FILE_START)){
+			var pathToFile = line.substr(FILE_START.length);
+			//console.log(currentLeaf);
+			currentLeaf = addPathToTree(mission, pathToFile);
+		}
+		else if(line.startsWith(TIME_START)){
+			var startTime = line.substr(TIME_START.length + 0, TIME_START.length + 14);
+			var endTime = line.substr(TIME_START.length + 34).trim();
+			if(currentLeaf[KEY_START] === undefined) {
+				currentLeaf[KEY_START] = startTime;
+			}
+			currentLeaf[KEY_END] = endTime;
+		}
+	}
+	//console.log(JSON.stringify(mission, null, 2));
+
+	var luaStr = toLuaTable(mission, 0);
+	console.log(luaStr);
+}
+
+function toLuaTable(mission, d){
+	if(d === undefined) d = 0;
+	if(mission === undefined) return '';
+	//if(d > 2) return '';
+	var str = d ? "" : "return ";
+
+	var indent = SPACES.substr(0, d*2);
+	for (var phaseName in mission) {
+		if (mission.hasOwnProperty(phaseName)) {
+			var phase = mission[phaseName];	
+			str += '\n' + indent + '{ Name = "' + phaseName + '", ';
+			if(phase[KEY_START] !== undefined && phase[KEY_END] !== undefined){
+				str += 'TimeRange = { Start = "' + phase[KEY_START] + '", End = "' + phase[KEY_END] + '" }';
+			}
+			else {
+				str += 'Phases = {' + toLuaTable(phase, d + 1) + '\n' + indent + '}';
+			}
+			str += '},';
+		}
+	}
+
+	// strip out last ','.
+	return str.substr(0, str.length-1);
+}
+
+function addPathToTree(obj, components){
+	if(typeof components === 'string') {
+		components = components.split('/');
+	}
+	
+	var o = obj;
+	var key;
+	for (var i = 0; i < components.length; i++) {
+		key = components[i];
+		if(o[key] === undefined) o[key] = {};
+		o = o[key];
+	}
+
+	return o; // returning the leaf component
+}