package graphicalrefactor.views; import java.util.ArrayList; import java.util.List; import javax.swing.JFrame; import com.mxgraph.model.mxCell; import com.mxgraph.model.mxGeometry; import com.mxgraph.model.mxGraphModel; import com.mxgraph.swing.mxGraphComponent; import com.mxgraph.swing.handler.mxRubberband; import com.mxgraph.swing.view.mxICellEditor; import com.mxgraph.util.mxEvent; import com.mxgraph.util.mxEventObject; import com.mxgraph.util.mxEventSource.mxIEventListener; import com.mxgraph.view.mxGraph; import graphicalrefactor.editor.Editor; public class GraphicalRefactor extends JFrame { private static final long serialVersionUID = -8690140317781055614L; private Editor editor; private mxGraph graph; private GraphicalRefactorMenuBar menuBar; private mxGraphComponent graphComponent; public GraphicalRefactor() { setTitle("Graphical Refactor"); setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); graph = new mxGraph() { public boolean isPort(Object cell) { mxGeometry geo = getCellGeometry(cell); return (geo != null) ? geo.isRelative() : false; } public boolean isCellFoldable(Object cell, boolean collapse) { return false; } }; graphComponent = new mxGraphComponent(graph) { protected mxICellEditor createCellEditor() { return new ComboBoxCellEditor(this); } }; graph.getModel().addListener(mxEvent.CHANGE, new mxIEventListener() { public void invoke(Object sender, mxEventObject evt) { List<mxCell> terminals = new ArrayList<>(); for (Object change: ((List) evt.getProperties().get("changes"))) { if (change instanceof mxGraphModel.mxTerminalChange) { mxGraphModel.mxTerminalChange terminalChange = (mxGraphModel.mxTerminalChange) change; mxCell cell = (mxCell) terminalChange.getCell(); mxCell terminal = (mxCell) terminalChange.getTerminal(); terminals.add(terminal); } } if (terminals.size() == 2) { editor.connectEdge(terminals.get(0), terminals.get(1)); } } }); getContentPane().add(graphComponent); new mxRubberband(graphComponent); graph.setAllowDanglingEdges(false); graph.setCellsDisconnectable(true); editor = new Editor(graph); menuBar = new GraphicalRefactorMenuBar(this); setJMenuBar(menuBar); setSize(870, 640); } public mxGraph getGraph() { return graph; } public mxGraphComponent getGraphComponent() { return graphComponent; } public Editor getEditor() { return editor; } public void setEditor(Editor editor) { this.editor = editor; } }