OpenDroneMap-NodeODM/libs/odmRunner.js

128 wiersze
3.7 KiB
JavaScript

/*
Node-OpenDroneMap Node.js App and REST API to access OpenDroneMap.
Copyright (C) 2016 Node-OpenDroneMap Contributors
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
"use strict";
let fs = require('fs');
let path = require('path');
let assert = require('assert');
let spawn = require('child_process').spawn;
let config = require('../config.js');
let logger = require('./logger');
module.exports = {
run: function(options, projectName, done, outputReceived){
assert(projectName !== undefined, "projectName must be specified");
assert(options["project-path"] !== undefined, "project-path must be defined");
let command = [path.join(config.odm_path, "run.py")];
for (var name in options){
let value = options[name];
// Skip false booleans
if (value === false) continue;
command.push("--" + name);
// We don't specify "--time true" (just "--time")
if (typeof value !== 'boolean'){
command.push(value);
}
}
command.push(projectName);
logger.info(`About to run: python ${command.join(" ")}`);
if (config.test){
logger.info("Test mode is on, command will not execute");
let outputTestFile = path.join("..", "tests", "odm_output.txt");
fs.readFile(path.resolve(__dirname, outputTestFile), 'utf8', (err, text) => {
if (!err){
let lines = text.split("\n");
lines.forEach(line => outputReceived(line));
done(null, 0, null);
}else{
logger.warn(`Error: ${err.message}`);
done(err);
}
});
return; // Skip rest
}
// Launch
let childProcess = spawn("python", command, {cwd: config.odm_path});
childProcess
.on('exit', (code, signal) => done(null, code, signal))
.on('error', done);
childProcess.stdout.on('data', chunk => outputReceived(chunk.toString()));
childProcess.stderr.on('data', chunk => outputReceived(chunk.toString()));
return childProcess;
},
getJsonOptions: function(done){
// In test mode, we don't call ODM,
// instead we return a mock
if (config.test){
let optionsTestFile = path.join("..", "tests", "odm_options.json");
fs.readFile(path.resolve(__dirname, optionsTestFile), 'utf8', (err, json) => {
if (!err){
try{
let options = JSON.parse(json);
done(null, options);
}catch(e){
logger.warn(`Invalid test options ${optionsTestFile}: ${err.message}`);
done(e);
}
}else{
logger.warn(`Error: ${err.message}`);
done(err);
}
});
return; // Skip rest
}
// Launch
let childProcess = spawn("python", [path.join(__dirname, "..", "helpers", "odmOptionsToJson.py"),
"--project-path", config.odm_path, "bogusname"]);
let output = [];
childProcess
.on('exit', (code, signal) => {
try{
let json = JSON.parse(output.join(""));
done(null, json);
}catch(err){
done(new Error(`Could not load list of options from OpenDroneMap. Is OpenDroneMap installed in ${config.odm_path}? Make sure that OpenDroneMap is installed and that --odm_path is set properly: ${err.message}`));
}
})
.on('error', done);
let processOutput = chunk => output.push(chunk.toString());
childProcess.stdout.on('data', processOutput);
childProcess.stderr.on('data', processOutput);
}
};