org.hortonmachine.modules.docker.PdalDockerModel Maven / Gradle / Ivy
/*
* This file is part of HortonMachine (http://www.hortonmachine.org)
* (C) HydroloGIS - www.hydrologis.com
*
* The HortonMachine 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 .
*/
package org.hortonmachine.modules.docker;
import java.io.File;
import java.io.IOException;
import org.hortonmachine.gears.libs.modules.HMModel;
import org.hortonmachine.gears.libs.monitor.IHMProgressMonitor;
import org.hortonmachine.gears.utils.files.FileUtilities;
import org.json.JSONArray;
import org.json.JSONObject;
/**
* @author hydrologis
*
*/
public class PdalDockerModel extends HMModel {
private static final String TAG = "2.0";
private static final String PDAL = "pdal/pdal";
private static final String PDAL_WITHTAG = PDAL + ":" + TAG;
protected DockerHandler dockerHandler = new DockerHandler();
protected void startContainer( String volumePath ) throws Exception {
dockerHandler.startContainer(PDAL_WITHTAG, volumePath);
}
/**
* Checks if docker client is initialized and if not, inits it.
*
* If it can't init it: it returns an error string, else null.
*/
protected String checkDockerInstall() {
String error = dockerHandler.initDocker();
if (error != null) {
return error;
}
return null;
}
protected String hasImage() {
return dockerHandler.hasImage(PDAL_WITHTAG);
}
protected void pullImage( IHMProgressMonitor pm ) throws Exception {
dockerHandler.pullImage(PDAL, TAG, pm);
}
protected void removeImage( String id ) {
dockerHandler.removeImage(id);
}
public void execCommand( String command ) throws Exception {
dockerHandler.execCommand(command);
}
public void closeClient() throws Exception {
dockerHandler.closeClient();
}
public File getPipelineFile( File workspaceFile, String filterJson ) throws IOException {
File pipeLineFile = new File(workspaceFile, "pipeline.json");
FileUtilities.writeFile(filterJson, pipeLineFile);
return pipeLineFile;
}
public String getPipelineJson( String inName, String outName, JSONObject... filters ) {
JSONObject root = new JSONObject();
JSONArray pipelineArray = new JSONArray();
root.put("pipeline", pipelineArray);
JSONObject reader = new JSONObject();
reader.put("type", "readers.las");
reader.put("filename", inName);
pipelineArray.put(reader);
for( JSONObject filter : filters ) {
pipelineArray.put(filter);
}
JSONObject writer = new JSONObject();
writer.put("type", "writers.las");
writer.put("minor_version", 1);
writer.put("filename", outName);
pipelineArray.put(writer);
String filterJson = root.toString(2);
return filterJson;
}
}