68 lines
2.7 KiB
Java
68 lines
2.7 KiB
Java
package _VisualDVM.Passes.All;
|
||
import Common.Passes.Pass;
|
||
import Common.Passes.PassException;
|
||
import Common.Utils.Utils_;
|
||
import Common.Visual.UI;
|
||
import _VisualDVM.Global;
|
||
import _VisualDVM.ProjectData.SapforData.Functions.UI.Graph.FunctionsGraphForm;
|
||
import com.mxgraph.io.mxCodec;
|
||
import com.mxgraph.util.mxCellRenderer;
|
||
import com.mxgraph.util.mxXmlUtils;
|
||
import com.mxgraph.util.png.mxPngEncodeParam;
|
||
import com.mxgraph.util.png.mxPngImageEncoder;
|
||
import com.mxgraph.view.mxGraph;
|
||
|
||
import java.awt.*;
|
||
import java.awt.image.BufferedImage;
|
||
import java.io.File;
|
||
import java.io.FileOutputStream;
|
||
import java.net.URLEncoder;
|
||
import java.nio.file.Paths;
|
||
public class SaveGraph extends Pass<File> {
|
||
FunctionsGraphForm graphForm = null;
|
||
@Override
|
||
protected boolean needsAnimation() {
|
||
return true;
|
||
}
|
||
@Override
|
||
public String getIconPath() {
|
||
return "/icons/ScreenShot.png";
|
||
}
|
||
@Override
|
||
protected boolean canStart(Object... args) {
|
||
graphForm = Global.mainModule.getUI().getMainWindow().getProjectWindow().getFunctionsWindow().getFunctionsGraphWindow();
|
||
if (!graphForm.isShown())
|
||
Log.Writeln_("Сначала отобразите граф");
|
||
if (Global.mainModule.getProject().functionsGraph.isEmpty())
|
||
Log.Writeln_("Граф процедур пуст");
|
||
return Log.isEmpty();
|
||
}
|
||
@Override
|
||
protected void body() throws Exception {
|
||
target = Paths.get(Global.visualiser.getWorkspace().getAbsolutePath(), "Graph.png").toFile();
|
||
if (target.exists() && !target.delete())
|
||
throw new PassException("Не удается удалить предыдущий файл " + Utils_.Brackets(target.getAbsolutePath()));
|
||
//-
|
||
mxGraph graph = graphForm.getControl().getGraph();
|
||
BufferedImage image = mxCellRenderer.createBufferedImage(graph,
|
||
null, 1, Color.WHITE,
|
||
graphForm.getControl().isAntiAlias(),
|
||
null,
|
||
graphForm.getControl().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[]{"graph", xml});
|
||
// Saves as a PNG file
|
||
FileOutputStream outputStream = new FileOutputStream(target);
|
||
mxPngImageEncoder encoder = new mxPngImageEncoder(outputStream, param);
|
||
encoder.encode(image);
|
||
outputStream.close();
|
||
}
|
||
@Override
|
||
protected void showDone() throws Exception {
|
||
UI.Info("Граф успешно сохранен в файл: " + Utils_.Brackets(target.getAbsolutePath()));
|
||
}
|
||
}
|