public void test1() throws Exception {
// Creates graph with model
mxGraph graph = new mxGraph();
Object parent = graph.getDefaultParent();
Object v1, v2;
graph.getModel().beginUpdate();
try {
v1 = graph.insertVertex(parent, null, "Hello", 20, 20, 80, 30);
v2 = graph.insertVertex(parent, null, "World!", 200, 150, 80, 30);
graph.insertEdge(parent, null, "e1", v1, v2);
} finally {
graph.getModel().endUpdate();
}
mxCodec codec = new mxCodec();
Node node = codec.encode(graph.getModel());
String xml1 = mxXmlUtils.getXml(node);
System.out.println("xml=" + mxUtils.getPrettyXml(node));
Document doc = mxDomUtils.createDocument();
doc.appendChild(doc.importNode(node, true));
codec = new mxCodec(doc);
Object model = codec.decode(node);
codec = new mxCodec();
node = codec.encode(model);
String xml2 = mxXmlUtils.getXml(node);
assertEquals(xml1, xml2);
}
/** Saves XML+PNG format. */
protected void saveXmlPng(TrackSchemeFrame frame, String filename, Color bg) throws IOException {
final mxGraphComponent graphComponent = trackScheme.getGUI().graphComponent;
final mxGraph graph = trackScheme.getGraph();
// Creates the image for the PNG file
BufferedImage image =
mxCellRenderer.createBufferedImage(
graph, null, 1, bg, graphComponent.isAntiAlias(), null, graphComponent.getCanvas());
// Creates the URL-encoded XML data
mxCodec codec = new mxCodec();
String xml = URLEncoder.encode(mxXmlUtils.getXml(codec.encode(graph.getModel())), "UTF-8");
mxPngEncodeParam param = mxPngEncodeParam.getDefaultEncodeParam(image);
param.setCompressedText(new String[] {"mxGraphModel", xml});
// Saves as a PNG file
FileOutputStream outputStream = new FileOutputStream(new File(filename));
try {
mxPngImageEncoder encoder = new mxPngImageEncoder(outputStream, param);
if (image != null) {
encoder.encode(image);
} else {
JOptionPane.showMessageDialog(graphComponent, "No Image Data");
}
} finally {
outputStream.close();
}
}
@SuppressWarnings("unchecked")
public void test2() throws Exception {
Map<String, Object> map = new Hashtable<String, Object>();
map.put("a", "b");
map.put("b", new mxCell("Hello, World!"));
mxCodec codec = new mxCodec();
Node node = codec.encode(map);
String xml1 = mxXmlUtils.getXml(node);
codec = new mxCodec();
Map<String, Object> map2 = (Hashtable<String, Object>) codec.decode(node);
codec = new mxCodec();
node = codec.encode(map2);
String xml2 = mxXmlUtils.getXml(node);
assertEquals(xml1, xml2);
assertEquals(map.size(), map2.size());
assertEquals(map.get("a"), map2.get("a"));
}
public void actionPerformed(ActionEvent e) {
final mxGraphComponent graphComponent = trackScheme.getGUI().graphComponent;
final mxGraph graph = trackScheme.getGraph();
FileFilter selectedFilter = null;
DefaultFileFilter xmlPngFilter = new DefaultFileFilter(".png", "PNG+XML file (.png)");
FileFilter vmlFileFilter = new DefaultFileFilter(".html", "VML file (.html)");
String filename = null;
boolean dialogShown = false;
String wd;
if (lastDir != null) {
wd = lastDir;
} else {
wd = System.getProperty("user.dir");
}
JFileChooser fc = new JFileChooser(wd);
// Adds the default file format
FileFilter defaultFilter = xmlPngFilter;
fc.addChoosableFileFilter(defaultFilter);
// Adds special vector graphics formats and HTML
fc.addChoosableFileFilter(new DefaultFileFilter(".pdf", "PDF file (.pdf)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".svg", "SVG file (.svg)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".html", "HTML file (.html)"));
fc.addChoosableFileFilter(vmlFileFilter);
fc.addChoosableFileFilter(new DefaultFileFilter(".txt", "Graph Drawing file (.txt)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".mxe", "mxGraph Editor file (.mxe)"));
// Adds a filter for each supported image format
Object[] imageFormats = ImageIO.getReaderFormatNames();
// Finds all distinct extensions
HashSet<String> formats = new HashSet<String>();
for (int i = 0; i < imageFormats.length; i++) {
String ext = imageFormats[i].toString().toLowerCase();
formats.add(ext);
}
imageFormats = formats.toArray();
for (int i = 0; i < imageFormats.length; i++) {
String ext = imageFormats[i].toString();
fc.addChoosableFileFilter(
new DefaultFileFilter("." + ext, ext.toUpperCase() + " File (." + ext + ")"));
}
// Adds filter that accepts all supported image formats
fc.addChoosableFileFilter(new DefaultFileFilter.ImageFileFilter("All Images"));
fc.setFileFilter(defaultFilter);
int rc = fc.showDialog(null, "Save");
dialogShown = true;
if (rc != JFileChooser.APPROVE_OPTION) {
return;
} else {
lastDir = fc.getSelectedFile().getParent();
}
filename = fc.getSelectedFile().getAbsolutePath();
selectedFilter = fc.getFileFilter();
if (selectedFilter instanceof DefaultFileFilter) {
String ext = ((DefaultFileFilter) selectedFilter).getExtension();
if (!filename.toLowerCase().endsWith(ext)) {
filename += ext;
}
}
if (new File(filename).exists()
&& JOptionPane.showConfirmDialog(graphComponent, "Overwrite existing file?")
!= JOptionPane.YES_OPTION) {
return;
}
try {
String ext = filename.substring(filename.lastIndexOf('.') + 1);
if (ext.equalsIgnoreCase("svg")) {
mxSvgCanvas canvas =
(mxSvgCanvas)
mxCellRenderer.drawCells(
graph,
null,
1,
null,
new CanvasFactory() {
public mxICanvas createCanvas(int width, int height) {
TrackSchemeSvgCanvas canvas =
new TrackSchemeSvgCanvas(mxDomUtils.createSvgDocument(width, height));
canvas.setEmbedded(true);
return canvas;
}
});
mxUtils.writeFile(mxXmlUtils.getXml(canvas.getDocument()), filename);
} else if (selectedFilter == vmlFileFilter) {
mxUtils.writeFile(
mxXmlUtils.getXml(
mxCellRenderer.createVmlDocument(graph, null, 1, null, null).getDocumentElement()),
filename);
} else if (ext.equalsIgnoreCase("html")) {
mxUtils.writeFile(
mxXmlUtils.getXml(
mxCellRenderer.createHtmlDocument(graph, null, 1, null, null).getDocumentElement()),
filename);
} else if (ext.equalsIgnoreCase("mxe") || ext.equalsIgnoreCase("xml")) {
mxCodec codec = new mxCodec();
String xml = mxXmlUtils.getXml(codec.encode(graph.getModel()));
mxUtils.writeFile(xml, filename);
} else if (ext.equalsIgnoreCase("txt")) {
String content = mxGdCodec.encode(graph); // .getDocumentString();
mxUtils.writeFile(content, filename);
} else if (ext.equalsIgnoreCase("pdf")) {
exportGraphToPdf(graph, filename);
} else {
Color bg = null;
if ((!ext.equalsIgnoreCase("gif") && !ext.equalsIgnoreCase("png"))
|| JOptionPane.showConfirmDialog(graphComponent, "Transparent Background?")
!= JOptionPane.YES_OPTION) {
bg = graphComponent.getBackground();
}
if (selectedFilter == xmlPngFilter || (ext.equalsIgnoreCase("png") && !dialogShown)) {
saveXmlPng(trackScheme.getGUI(), filename, bg);
} else {
BufferedImage image =
mxCellRenderer.createBufferedImage(
graph,
null,
1,
bg,
graphComponent.isAntiAlias(),
null,
graphComponent.getCanvas());
if (image != null) {
ImageIO.write(image, ext, new File(filename));
} else {
JOptionPane.showMessageDialog(graphComponent, "No Image Data");
}
}
}
} catch (Throwable ex) {
ex.printStackTrace();
JOptionPane.showMessageDialog(
graphComponent, ex.toString(), "Error", JOptionPane.ERROR_MESSAGE);
}
}
@Override
public String encode(AbstractMeta meta) throws Exception {
TransMeta transMeta = (TransMeta) meta;
mxGraph graph = new mxGraph();
graph.getModel().beginUpdate();
mxCell parent = (mxCell) graph.getDefaultParent();
Document doc = mxUtils.createDocument();
try {
Element e = super.encodeCommRootAttr(transMeta, doc);
e.setAttribute("trans_version", transMeta.getTransversion());
e.setAttribute("trans_type", transMeta.getTransformationType().getCode());
e.setAttribute("trans_status", String.valueOf(transMeta.getTransstatus()));
// variables
Properties sp = new Properties();
JSONArray jsonArray = new JSONArray();
String[] keys = Variables.getADefaultVariableSpace().listVariables();
for (int i = 0; i < keys.length; i++) {
sp.put(keys[i], Variables.getADefaultVariableSpace().getVariable(keys[i]));
}
List<String> vars = transMeta.getUsedVariables();
for (int i = 0; i < vars.size(); i++) {
String varname = vars.get(i);
if (!varname.startsWith(Const.INTERNAL_VARIABLE_PREFIX)
&& Const.indexOfString(varname, transMeta.listParameters()) < 0) {
JSONObject param = new JSONObject();
param.put("var_name", varname);
param.put("var_value", sp.getProperty(varname, ""));
jsonArray.add(param);
}
}
for (String varname : Const.INTERNAL_JOB_VARIABLES) {
String value = transMeta.getVariable(varname);
if (!Const.isEmpty(value)) {
JSONObject param = new JSONObject();
param.put("var_name", varname);
param.put("var_value", value);
jsonArray.add(param);
}
}
e.setAttribute("variables", jsonArray.toString());
TransLogTable transLogTable = transMeta.getTransLogTable();
JSONObject jsonObject = new JSONObject();
jsonObject.put("connection", transLogTable.getConnectionName());
jsonObject.put("schema", transLogTable.getSchemaName());
jsonObject.put("table", transLogTable.getTableName());
jsonObject.put("size_limit_lines", transLogTable.getLogSizeLimit());
jsonObject.put("interval", transLogTable.getLogInterval());
jsonObject.put("timeout_days", transLogTable.getTimeoutInDays());
JSONArray fields = new JSONArray();
for (LogTableField field : transLogTable.getFields()) {
JSONObject jsonField = new JSONObject();
jsonField.put("id", field.getId());
jsonField.put("enabled", field.isEnabled());
jsonField.put("name", field.getFieldName());
jsonField.put("subjectAllowed", field.isSubjectAllowed());
if (field.isSubjectAllowed()) {
jsonField.put("subject", field.getSubject() == null ? "" : field.getSubject().toString());
} else {
jsonField.put("subject", "-");
}
jsonField.put("description", StringEscapeHelper.encode(field.getDescription()));
fields.add(jsonField);
}
jsonObject.put("fields", fields);
e.setAttribute("transLogTable", jsonObject.toString());
StepLogTable stepLogTable = transMeta.getStepLogTable();
jsonObject = new JSONObject();
jsonObject.put("connection", stepLogTable.getConnectionName());
jsonObject.put("schema", stepLogTable.getSchemaName());
jsonObject.put("table", stepLogTable.getTableName());
jsonObject.put("timeout_days", stepLogTable.getTimeoutInDays());
fields = new JSONArray();
for (LogTableField field : stepLogTable.getFields()) {
JSONObject jsonField = new JSONObject();
jsonField.put("id", field.getId());
jsonField.put("enabled", field.isEnabled());
jsonField.put("name", field.getFieldName());
jsonField.put("description", StringEscapeHelper.encode(field.getDescription()));
fields.add(jsonField);
}
jsonObject.put("fields", fields);
e.setAttribute("stepLogTable", jsonObject.toString());
PerformanceLogTable performanceLogTable = transMeta.getPerformanceLogTable();
jsonObject = new JSONObject();
jsonObject.put("connection", performanceLogTable.getConnectionName());
jsonObject.put("schema", performanceLogTable.getSchemaName());
jsonObject.put("table", performanceLogTable.getTableName());
jsonObject.put("interval", performanceLogTable.getLogInterval());
jsonObject.put("timeout_days", performanceLogTable.getTimeoutInDays());
fields = new JSONArray();
for (LogTableField field : performanceLogTable.getFields()) {
JSONObject jsonField = new JSONObject();
jsonField.put("id", field.getId());
jsonField.put("enabled", field.isEnabled());
jsonField.put("name", field.getFieldName());
jsonField.put("description", StringEscapeHelper.encode(field.getDescription()));
fields.add(jsonField);
}
jsonObject.put("fields", fields);
e.setAttribute("performanceLogTable", jsonObject.toString());
MetricsLogTable metricsLogTable = transMeta.getMetricsLogTable();
jsonObject = new JSONObject();
jsonObject.put("connection", metricsLogTable.getConnectionName());
jsonObject.put("schema", metricsLogTable.getSchemaName());
jsonObject.put("table", metricsLogTable.getTableName());
jsonObject.put("timeout_days", metricsLogTable.getTimeoutInDays());
fields = new JSONArray();
for (LogTableField field : metricsLogTable.getFields()) {
JSONObject jsonField = new JSONObject();
jsonField.put("id", field.getId());
jsonField.put("enabled", field.isEnabled());
jsonField.put("name", field.getFieldName());
jsonField.put("description", StringEscapeHelper.encode(field.getDescription()));
fields.add(jsonField);
}
jsonObject.put("fields", fields);
e.setAttribute("metricsLogTable", jsonObject.toString());
jsonObject = new JSONObject();
jsonObject.put(
"connection",
transMeta.getMaxDateConnection() == null
? ""
: transMeta.getMaxDateConnection().getName());
jsonObject.put("table", transMeta.getMaxDateTable());
jsonObject.put("field", transMeta.getMaxDateField());
jsonObject.put("offset", transMeta.getMaxDateOffset());
jsonObject.put("maxdiff", transMeta.getMaxDateDifference());
e.setAttribute("maxdate", jsonObject.toString());
e.setAttribute("size_rowset", String.valueOf(transMeta.getSizeRowset()));
e.setAttribute("sleep_time_empty", String.valueOf(transMeta.getSleepTimeEmpty()));
e.setAttribute("sleep_time_full", String.valueOf(transMeta.getSleepTimeFull()));
e.setAttribute("unique_connections", transMeta.isUsingUniqueConnections() ? "Y" : "N");
e.setAttribute("feedback_shown", transMeta.isFeedbackShown() ? "Y" : "N");
e.setAttribute("feedback_size", String.valueOf(transMeta.getFeedbackSize()));
e.setAttribute(
"using_thread_priorities", transMeta.isUsingThreadPriorityManagment() ? "Y" : "N");
e.setAttribute(
"capture_step_performance", transMeta.isCapturingStepPerformanceSnapShots() ? "Y" : "N");
e.setAttribute(
"step_performance_capturing_delay",
String.valueOf(transMeta.getStepPerformanceCapturingDelay()));
e.setAttribute(
"step_performance_capturing_size_limit",
transMeta.getStepPerformanceCapturingSizeLimit());
super.encodeSlaveServers(e, transMeta);
encodeClusterSchema(e, transMeta);
encodePartitionSchema(e, transMeta);
try {
if (transMeta.getKey() != null) {
e.setAttribute("key_for_session_key", XMLHandler.encodeBinaryData(transMeta.getKey()));
} else {
e.setAttribute("key_for_session_key", "");
}
} catch (Exception e1) {
e1.printStackTrace();
e.setAttribute("key_for_session_key", "");
}
e.setAttribute("is_key_private", transMeta.isPrivateKey() ? "Y" : "N");
super.encodeNote(doc, graph, transMeta);
super.encodeDatabases(e, transMeta);
parent.setValue(e);
// encode steps and hops
HashMap<StepMeta, Object> cells = new HashMap<StepMeta, Object>();
List<StepMeta> list = transMeta.getSteps();
for (int i = 0; i < list.size(); i++) {
StepMeta step = (StepMeta) list.get(i);
Point p = step.getLocation();
StepEncoder stepEncoder = (StepEncoder) PluginFactory.getBean(step.getStepID());
PluginInterface plugin =
PluginRegistry.getInstance().getPlugin(StepPluginType.class, step.getStepID());
Object cell =
graph.insertVertex(
parent,
null,
stepEncoder.encodeStep(step),
p.x,
p.y,
40,
40,
"icon;image=" + SvgImageUrl.getUrl(plugin));
cells.put(step, cell);
}
for (int i = 0; i < transMeta.nrTransHops(); i++) {
TransHopMeta transHopMeta = transMeta.getTransHop(i);
Object v1 = cells.get(transHopMeta.getFromStep());
Object v2 = cells.get(transHopMeta.getToStep());
graph.insertEdge(parent, null, TransHopMetaCodec.encode(transHopMeta), v1, v2);
}
} finally {
graph.getModel().endUpdate();
}
mxCodec codec = new mxCodec();
return mxUtils.getPrettyXml(codec.encode(graph.getModel()));
}