GUI.java 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991
  1. package ui.view;
  2. import java.awt.BorderLayout;
  3. import java.awt.Color;
  4. import java.awt.Component;
  5. import java.awt.Cursor;
  6. import java.awt.Dimension;
  7. import java.awt.Image;
  8. import java.awt.Point;
  9. import java.awt.Toolkit;
  10. import java.awt.event.ActionEvent;
  11. import java.awt.event.ActionListener;
  12. import java.awt.event.MouseAdapter;
  13. import java.awt.event.MouseEvent;
  14. import java.beans.PropertyChangeEvent;
  15. import java.beans.PropertyChangeListener;
  16. import java.io.File;
  17. import java.io.IOException;
  18. import java.util.ArrayList;
  19. import javax.swing.BoxLayout;
  20. import javax.swing.DefaultComboBoxModel;
  21. import javax.swing.ImageIcon;
  22. import javax.swing.JButton;
  23. import javax.swing.JCheckBoxMenuItem;
  24. import javax.swing.JComboBox;
  25. import javax.swing.JEditorPane;
  26. import javax.swing.JFileChooser;
  27. import javax.swing.JFrame;
  28. import javax.swing.JLabel;
  29. import javax.swing.JMenu;
  30. import javax.swing.JMenuBar;
  31. import javax.swing.JMenuItem;
  32. import javax.swing.JOptionPane;
  33. import javax.swing.JPanel;
  34. import javax.swing.JPopupMenu;
  35. import javax.swing.JScrollPane;
  36. import javax.swing.JSlider;
  37. import javax.swing.JSplitPane;
  38. import javax.swing.JTabbedPane;
  39. import javax.swing.JTable;
  40. import javax.swing.JToolBar;
  41. import javax.swing.JTree;
  42. import javax.swing.event.ChangeEvent;
  43. import javax.swing.event.ChangeListener;
  44. import javax.swing.table.DefaultTableModel;
  45. import javax.swing.tree.DefaultMutableTreeNode;
  46. import javax.swing.tree.DefaultTreeModel;
  47. import javax.swing.tree.TreeCellRenderer;
  48. import Interfaces.CategoryListener;
  49. import classes.Category;
  50. import classes.CpsEdge;
  51. import classes.CpsNode;
  52. import classes.CpsObject;
  53. import classes.HolonElement;
  54. import classes.HolonObject;
  55. import classes.HolonSwitch;
  56. import classes.HolonTransformer;
  57. import ui.controller.Control;
  58. import ui.model.Model;
  59. public class GUI<E> implements CategoryListener {
  60. private JFrame frmCyberPhysical;
  61. private final JMenuBar menuBar = new JMenuBar();
  62. private final JMenu mnNewMenu = new JMenu("File");
  63. private final JMenu mnNewMenu_1 = new JMenu("Edit");
  64. private final JMenu mnNewMenu_2 = new JMenu("Options");
  65. private final JMenu mnNewMenu_3 = new JMenu("View");
  66. private final JMenu mnHelp = new JMenu("Help");
  67. private final JMenuItem mntmOpen = new JMenuItem("Open");
  68. private final JMenuItem mntmNew = new JMenuItem("New");
  69. private final JMenuItem mntmSave = new JMenuItem("Save");
  70. private final JMenuItem aboutUs = new JMenuItem("About Us");
  71. private final JSplitPane splitPane = new JSplitPane();
  72. private final JSplitPane splitPane_1 = new JSplitPane();
  73. private final JScrollPane scrollPane_1 = new JScrollPane();
  74. private final JTabbedPane tabbedPane = new JTabbedPane(JTabbedPane.TOP);
  75. private final JScrollPane scrollPane_2 = new JScrollPane();
  76. private final JLabel maxGraph = new JLabel("100%");
  77. private final JLabel medGraph = new JLabel("50%");
  78. private final JLabel minGraph = new JLabel("0%");
  79. private final JLabel elementGraph = new JLabel("None ");
  80. private final JTree tree = new JTree();
  81. private final JEditorPane dtrpnHereWillBe = new JEditorPane();
  82. /******************************************
  83. ************* Right Container*************
  84. ******************************************/
  85. // Right Container: here comes the information about the HolonObject, such
  86. // as HolonElements Information, Properties and Consumption/Production graph
  87. private final JSplitPane split_HolonEl_Pro = new JSplitPane(JSplitPane.VERTICAL_SPLIT);
  88. private final JSplitPane split_Graph_HolonEl = new JSplitPane(JSplitPane.VERTICAL_SPLIT);
  89. // In this section are all the Holonelements that correspond to the clicked
  90. // HolonObject with consumption/production, name and amount.
  91. private JTable tableHolonElement = new JTable();
  92. private DefaultTableModel tableModelHolonElement = new DefaultTableModel();
  93. private final JPanel scrollElements = new JPanel();
  94. JScrollPane tableHolonElementScrollPane = new JScrollPane();
  95. // In this section are all the properties that correspond to the clicked
  96. // HolonObject, such as connections, name, Type, etc.
  97. private JTable tableProperties = new JTable();
  98. private JPanel graphLabel = new JPanel();
  99. private DefaulTable tableModelProperties;
  100. private final JScrollPane scrollProperties = new JScrollPane();
  101. // In this section is the graph for the selected HolonElement of the clicked
  102. // HolonObject
  103. private JTable tableGraph = new JTable();
  104. private DefaultTableModel tableModelGraph = new DefaultTableModel();
  105. private final JScrollPane scrollGraph = new JScrollPane();
  106. private final Model model;
  107. private final Control controller;
  108. // Pop up Windows
  109. private AddObjectPopUp addObjectPopUP;
  110. private AboutUsPopUp aboutUsPopUp;
  111. private AddElementPopUp addElementPopUp;
  112. private final JPanel panel = new JPanel();
  113. private final JPanel panel_HolonEl = new JPanel();
  114. private final JComboBox comboBox = new JComboBox();
  115. private final JComboBox comboBoxGraph = new JComboBox();
  116. // Buttons
  117. private final JButton btnAdd = new JButton("+");
  118. private final JButton btnDel = new JButton("-");
  119. private final JButton btnAddHolEL = new JButton("+");
  120. private final JButton btnDelHolEL = new JButton("-");
  121. private final JButton resetGraphBtn = new JButton("Reset");
  122. private final JToolBar toolBar = new JToolBar();
  123. private final JToolBar toolBarHolonEl = new JToolBar();
  124. private final JToolBar toolBarGraph = new JToolBar();
  125. // variables
  126. private boolean dragging = false;
  127. private String actualObjectClicked;
  128. private Image img = null;
  129. private CpsObject tempCps = null;
  130. private HolonElement tempElement = null;
  131. private int yValueElements = 0;
  132. private MyCanvas canvas;
  133. private UnitGraph unitGraph; // for testing, remove later
  134. private final JSplitPane splitPane_3 = new JSplitPane();
  135. private final JSlider sizeSlider = new JSlider();
  136. private final JLabel lblImageSize = new JLabel("Image Size");
  137. // Time Stuff
  138. private TimePanel timePanel;
  139. private final JMenu mnAlgorithm = new JMenu("Algorithm");
  140. private final JCheckBoxMenuItem chckbxmntmUseAlgorithm = new JCheckBoxMenuItem("Use Algorithm");
  141. private final JSplitPane splitPane_2 = new JSplitPane();
  142. private final JLabel label_1 = new JLabel("New label");
  143. private final JLabel lblSelect = new JLabel("Select");
  144. private final JComboBox comboBoxAlgo = new JComboBox();
  145. /**
  146. * Create the application.
  147. */
  148. public GUI(Control control) {
  149. this.controller = control;
  150. this.model = control.getModel();
  151. this.canvas = new MyCanvas(model, control);
  152. this.unitGraph = new UnitGraph(model, control);
  153. control.initListener(this);
  154. initialize();
  155. updateCategories(model.getCategories());
  156. }
  157. /**
  158. * Initialize the contents of the frame.
  159. */
  160. @SuppressWarnings({ "serial", "unchecked" })
  161. private void initialize() {
  162. frmCyberPhysical = new JFrame();
  163. frmCyberPhysical.setTitle("Cyber Physical Systems Model");
  164. frmCyberPhysical.setBounds(100, 100, 1000, 800);
  165. frmCyberPhysical.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  166. frmCyberPhysical.setJMenuBar(menuBar);
  167. frmCyberPhysical.setIconImage(new ImageIcon(this.getClass().getResource("/Images/Dummy_House.png")).getImage()
  168. .getScaledInstance(30, 30, Image.SCALE_SMOOTH));
  169. menuBar.add(mnNewMenu);
  170. mnNewMenu.add(mntmNew);
  171. mnNewMenu.add(mntmOpen);
  172. mnNewMenu.add(mntmSave);
  173. menuBar.add(mnNewMenu_1);
  174. menuBar.add(mnNewMenu_2);
  175. menuBar.add(mnNewMenu_3);
  176. mnNewMenu_3.add(splitPane_3);
  177. sizeSlider.setMinimum(15);
  178. sizeSlider.setMaximum(115);
  179. sizeSlider.addChangeListener(new ChangeListener() {
  180. @Override
  181. public void stateChanged(ChangeEvent e) {
  182. controller.setScale(sizeSlider.getValue());
  183. tree.setRowHeight(model.getScale());
  184. canvas.objectSelectionHighlighting();
  185. canvas.repaint();
  186. }
  187. });
  188. splitPane_3.setRightComponent(sizeSlider);
  189. splitPane_3.setLeftComponent(lblImageSize);
  190. menuBar.add(mnHelp);
  191. mnHelp.add(aboutUs);
  192. menuBar.add(mnAlgorithm);
  193. mnAlgorithm.add(chckbxmntmUseAlgorithm);
  194. mnAlgorithm.add(splitPane_2);
  195. splitPane_2.setLeftComponent(lblSelect);
  196. splitPane_2.setRightComponent(comboBoxAlgo);
  197. unitGraph.setBackground(Color.WHITE);
  198. canvas.setBackground(Color.WHITE);
  199. canvas.setPreferredSize(new Dimension(10000, 10000));
  200. JScrollPane canvasSP = new JScrollPane(canvas);
  201. canvasSP.setBorder(null);
  202. tabbedPane.addTab("Modeling", new ImageIcon(new ImageIcon(this.getClass().getResource("/Images/home.png"))
  203. .getImage().getScaledInstance(30, 30, Image.SCALE_SMOOTH)), canvasSP, "Model a CPS");
  204. tabbedPane.addTab("New tab", null, label_1, null);
  205. tabbedPane.addTab("Simulation",
  206. new ImageIcon(new ImageIcon(this.getClass().getResource("/Images/Dummy_House.png")).getImage()
  207. .getScaledInstance(30, 30, Image.SCALE_SMOOTH)),
  208. scrollPane_2, "Simulate the CPS");
  209. dtrpnHereWillBe.setText("Here will be the Simulation");
  210. scrollPane_2.setViewportView(dtrpnHereWillBe);
  211. /*
  212. * Right Container Set up
  213. */
  214. // Set up of the HolonElements section
  215. Object[] columnNames = { "Device", "Energy", "Quantity" };
  216. tableModelHolonElement.setColumnIdentifiers(columnNames);
  217. tableHolonElement.setBorder(null);
  218. tableHolonElement.setModel(tableModelHolonElement);
  219. tableHolonElement.setFillsViewportHeight(true);
  220. tableHolonElement.setCellSelectionEnabled(true);
  221. tableHolonElement.setColumnSelectionAllowed(true);
  222. // Set up of the Properties section
  223. Object[] colNames = { "Field", "Information" };
  224. tableModelProperties = new DefaulTable(100, colNames.length);
  225. tableModelProperties.setColumnIdentifiers(colNames);
  226. tableProperties.setModel(tableModelProperties);
  227. tableProperties.setFillsViewportHeight(true);
  228. tableProperties.setCellSelectionEnabled(true);
  229. tableProperties.setColumnSelectionAllowed(true);
  230. // Set up of the Graph section
  231. Object[] tempText = { "Here comes the graph for each clicked HolonElement" };
  232. tableModelGraph.setColumnIdentifiers(tempText);
  233. tableGraph.setModel(tableModelGraph);
  234. tableGraph.setFillsViewportHeight(true);
  235. tableGraph.setCellSelectionEnabled(true);
  236. tableGraph.setColumnSelectionAllowed(true);
  237. /*
  238. * End of right container setup
  239. */
  240. scrollProperties.setViewportView(tableProperties);
  241. tableHolonElementScrollPane.setViewportView(tableHolonElement);
  242. scrollGraph.setViewportView(unitGraph);
  243. graphLabel.setLayout(new BorderLayout(0, 10));
  244. graphLabel.add(maxGraph, BorderLayout.NORTH);
  245. graphLabel.add(medGraph, BorderLayout.CENTER);
  246. graphLabel.add(minGraph, BorderLayout.SOUTH);
  247. toolBarGraph.add(elementGraph);
  248. comboBoxGraph.setModel(new DefaultComboBoxModel(new String[] { "Day", "Month", "Year" }));
  249. toolBarGraph.add(comboBoxGraph);
  250. toolBarGraph.add(resetGraphBtn);
  251. toolBarGraph.setFloatable(false);
  252. scrollGraph.setRowHeaderView(graphLabel);
  253. scrollGraph.setColumnHeaderView(toolBarGraph);
  254. scrollElements.setLayout(new BorderLayout(0, 0));
  255. scrollElements.add(panel_HolonEl, BorderLayout.NORTH);
  256. scrollElements.add(tableHolonElementScrollPane);
  257. panel_HolonEl.setLayout(new BoxLayout(panel_HolonEl, BoxLayout.X_AXIS));
  258. toolBarHolonEl.setFloatable(false);
  259. panel_HolonEl.add(toolBarHolonEl);
  260. toolBarHolonEl.add(btnAddHolEL);
  261. btnAddHolEL.addActionListener(new ActionListener() {
  262. public void actionPerformed(ActionEvent arg0) {
  263. CpsObject tempCpsObject = getActualCps();
  264. if (tempCpsObject != null && tempCpsObject.getClass() == HolonObject.class
  265. && tempCpsObject.getID() != 0) {
  266. addElementPopUp = new AddElementPopUp();
  267. addElementPopUp.setVisible(true);
  268. controller.addElementCanvasObject(tempCpsObject.getID(), addElementPopUp.getElement().getEleName(),
  269. addElementPopUp.getElement().getAmount(), addElementPopUp.getElement().getEnergy());
  270. refreshTableHolonElement();
  271. refreshTableProperties();
  272. }
  273. }
  274. });
  275. resetGraphBtn.addActionListener(new ActionListener() {
  276. public void actionPerformed(ActionEvent arg0) {
  277. unitGraph.reset();
  278. }
  279. });
  280. toolBarHolonEl.add(btnDelHolEL);
  281. btnDelHolEL.addActionListener(new ActionListener() {
  282. public void actionPerformed(ActionEvent arg0) {
  283. if (getActualCps().getClass() == HolonObject.class) {
  284. HolonObject obj = (HolonObject) getActualCps();
  285. tempElement = getActualHolonElement(obj, yValueElements);
  286. if (tempElement != null && obj.getClass() == HolonObject.class && obj.getID() != 0) {
  287. controller.deleteElementCanvas(obj.getID(), tempElement.getEleName());
  288. refreshTableHolonElement();
  289. refreshTableProperties();
  290. }
  291. }
  292. }
  293. });
  294. tableHolonElement.addMouseListener(new MouseAdapter() {
  295. public void mousePressed(MouseEvent e) {
  296. HolonObject obj = (HolonObject) getActualCps();
  297. yValueElements = e.getY();
  298. HolonElement ele = getActualHolonElement(obj, yValueElements);
  299. if (ele != null) {
  300. elementGraph.setText(ele.getEleName() + " ");
  301. unitGraph.repaintWithNewElement(ele);
  302. } else {
  303. elementGraph.setText("None ");
  304. }
  305. }
  306. });
  307. tableProperties.addPropertyChangeListener(new PropertyChangeListener() {
  308. @Override
  309. public void propertyChange(PropertyChangeEvent evt) {
  310. try {
  311. Object temp;
  312. if (getActualCps() != null) {
  313. temp = tableModelProperties.getValueAt(0, 1);
  314. getActualCps().setName(temp.toString());
  315. } else {
  316. temp = tableModelProperties.getValueAt(2, 1);
  317. Float ftemp = Float.parseFloat(temp.toString());
  318. System.out.println(ftemp);
  319. model.getSelectedEdge().setCapacity(ftemp);
  320. }
  321. for (CpsEdge c : getActualCps().getConnections()) {
  322. System.out.println(c.getA().getName() + " conected to " + c.getB().getName());
  323. }
  324. } catch (Exception e) {
  325. }
  326. }
  327. });
  328. frmCyberPhysical.getContentPane().setLayout(new BorderLayout(0, 0));
  329. TreeCellRenderer customRenderer = new TreeCellRenderer() {
  330. @Override
  331. public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded,
  332. boolean leaf, int row, boolean hasFocus) {
  333. JLabel label = new JLabel();
  334. Image imgR = null;
  335. if (leaf) {
  336. for (Category cat : model.getCategories()) {
  337. for (CpsObject cps : cat.getObjects()) {
  338. if (value.toString().compareTo(cps.getCompareName()) == 0) {
  339. File checkPath = new File(cps.getImage());
  340. if (checkPath.exists()) {
  341. imgR = new ImageIcon(cps.getImage()).getImage().getScaledInstance(
  342. controller.getScale(), controller.getScale(), java.awt.Image.SCALE_SMOOTH);
  343. } else {
  344. imgR = new ImageIcon(this.getClass().getResource(cps.getImage())).getImage()
  345. .getScaledInstance(controller.getScale(), controller.getScale(),
  346. java.awt.Image.SCALE_SMOOTH);
  347. }
  348. if (imgR != null) {
  349. label.setIcon(new ImageIcon(imgR));
  350. }
  351. label.setText(cps.getName());
  352. }
  353. }
  354. }
  355. }
  356. tree.setRowHeight(controller.getScale());
  357. if (hasFocus) {
  358. label.setForeground(new Color(0, 0, 255));
  359. label.setOpaque(true);
  360. }
  361. if (label.getText().length() == 0) {
  362. label.setText(value.toString());
  363. if (value.toString().compareTo("Categories") != 0) {
  364. label.setIcon(new ImageIcon(this.getClass().getResource("/Images/folder.png")));
  365. }
  366. }
  367. return label;
  368. }
  369. };
  370. tree.setCellRenderer(customRenderer);
  371. tree.addMouseListener(new MouseAdapter() {
  372. public void mouseReleased(MouseEvent e) {
  373. try {
  374. if (dragging) {
  375. int x = (int) canvas.getMousePosition().getX();
  376. int y = (int) canvas.getMousePosition().getY();
  377. CpsObject h = null;
  378. if (tempCps.getClass() == HolonObject.class) {
  379. h = new HolonObject(tempCps);
  380. }
  381. if (tempCps.getClass() == HolonSwitch.class) {
  382. h = new HolonSwitch(tempCps);
  383. }
  384. if (tempCps.getClass() == HolonTransformer.class) {
  385. h = new HolonTransformer(tempCps);
  386. }
  387. h.setPosition(x, y);
  388. controller.addObjectCanvas(h);
  389. for (int i = 0; i < model.getObjectsOnCanvas().size(); i++) {
  390. CpsObject temp = model.getObjectsOnCanvas().get(i);
  391. String output = "Element: " + temp.getObjName() + " with ID: " + temp.getID();
  392. if (temp.getClass() == HolonObject.class) {
  393. output = output + " and Elements: " + ((HolonObject) temp).toStringElements();
  394. }
  395. System.out.println(output);
  396. }
  397. canvas.repaint();
  398. dragging = false;
  399. }
  400. } catch (Exception e2) {
  401. }
  402. frmCyberPhysical.setCursor(new Cursor(Cursor.DEFAULT_CURSOR));
  403. }
  404. });
  405. tree.addMouseListener(new MouseAdapter() {
  406. public void mousePressed(MouseEvent e) {
  407. try {
  408. actualObjectClicked = tree.getPathForLocation(e.getX(), e.getY()).getLastPathComponent().toString();
  409. // if an Object was selected, the porperties are shown in
  410. // the table
  411. DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) tree
  412. .getPathForLocation(e.getX(), e.getY()).getLastPathComponent();
  413. if (selectedNode.getLevel() == 2) {
  414. CpsObject selected = controller.searchObjInCat(selectedNode.toString(),
  415. selectedNode.getParent().toString());
  416. deleteRows();
  417. if (selected instanceof HolonObject && selected != null) {
  418. selected = (HolonObject) selected;
  419. fillElementTable(((HolonObject) selected).getElements());
  420. }
  421. }
  422. for (Category cat : model.getCategories()) {
  423. for (CpsObject cps : cat.getObjects()) {
  424. if (actualObjectClicked.compareTo(cps.getCompareName()) == 0) {
  425. File checkPath = new File(cps.getImage());
  426. if (checkPath.exists()) {
  427. img = new ImageIcon(cps.getImage()).getImage().getScaledInstance(
  428. controller.getScale(), controller.getScale(), java.awt.Image.SCALE_SMOOTH);
  429. } else {
  430. img = new ImageIcon(this.getClass().getResource(cps.getImage())).getImage()
  431. .getScaledInstance(controller.getScale(), controller.getScale(),
  432. java.awt.Image.SCALE_SMOOTH);
  433. }
  434. tempCps = cps;
  435. dragging = true;
  436. Cursor cursor = Toolkit.getDefaultToolkit().createCustomCursor(img, new Point(0, 0),
  437. "Image");
  438. frmCyberPhysical.setCursor(cursor);
  439. }
  440. }
  441. }
  442. } catch (Exception e2) {
  443. }
  444. }
  445. });
  446. scrollPane_1.setViewportView(tree);
  447. scrollPane_1.setColumnHeaderView(panel);
  448. panel.setLayout(new BoxLayout(panel, BoxLayout.Y_AXIS));
  449. toolBar.setFloatable(false);
  450. panel.add(toolBar);
  451. toolBar.add(comboBox);
  452. comboBox.setModel(new DefaultComboBoxModel(new String[] { "Category", "Object", "Transformer", "Switch" }));
  453. // Add Button
  454. btnAdd.addActionListener(new ActionListener() {
  455. public void actionPerformed(ActionEvent arg0) {
  456. Object nodeInfo = tree.getLastSelectedPathComponent();
  457. String selectedOption = comboBox.getSelectedItem().toString();
  458. DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) nodeInfo;
  459. switch (selectedOption) {
  460. case "Category":
  461. String catName = JOptionPane.showInputDialog("Please enter a Name for Category ");
  462. if (catName.length() != 0) {
  463. controller.addCategory(catName);
  464. }
  465. break;
  466. case "Object":
  467. if (selectedNode == null) {
  468. JOptionPane.showMessageDialog(new JFrame(),
  469. "Please select a Category first before adding " + selectedOption + ".");
  470. }
  471. if (selectedNode.getLevel() == 1) {
  472. addObjectPopUP = new AddObjectPopUp();
  473. addObjectPopUP.setVisible(true);
  474. addObjectPopUP.setController(controller);
  475. addObjectPopUP.setCategory(selectedNode.toString());
  476. }
  477. break;
  478. default:
  479. addObjectAction(selectedOption, selectedNode);
  480. break;
  481. }
  482. }
  483. });
  484. /**
  485. * Pop up - About Us with some important information about the
  486. * developers, source and programming stuff
  487. */
  488. aboutUs.addMouseListener(new MouseAdapter() {
  489. @Override
  490. public void mousePressed(MouseEvent e) {
  491. aboutUsPopUp = new AboutUsPopUp();
  492. aboutUsPopUp.setVisible(true);
  493. }
  494. });
  495. /**
  496. * Update of every interaction between the user and the canvas (only on
  497. * the canvas). Basically the update of all the information concerning
  498. * the clicked HolonObject
  499. */
  500. canvas.addMouseListener(new MouseAdapter() {
  501. @Override
  502. public void mousePressed(MouseEvent e) {
  503. // Update of the Information about the HolonElements - only for
  504. refreshTableHolonElement();
  505. // Update of the Information about the Properties - only for
  506. // CpsObjects
  507. // Erase old data
  508. if (tableModelProperties.getRowCount() > 0) {
  509. for (int i = tableModelProperties.getRowCount() - 1; i > -1; i--) {
  510. tableModelProperties.removeRow(i);
  511. }
  512. }
  513. // Write new data
  514. CpsObject temp = getActualCps();
  515. if (temp != null) {
  516. Object[] tempName = { "Name", temp.getName() };
  517. tableModelProperties.addRow(tempName);
  518. Object[] tempId = { "ID", temp.getID() };
  519. tableModelProperties.addRow(tempId);
  520. if (temp.getClass() == HolonObject.class) {
  521. Object[] tempEnergy = { "Total Energy", ((HolonObject) temp).getCurrentEnergy() };
  522. tableModelProperties.addRow(tempEnergy);
  523. }
  524. tableModelProperties.setCellEditable(0, 1, true);
  525. tableModelProperties.setCellEditable(2, 1, false);
  526. ArrayList<CpsEdge> temp_array = temp.getConnectedTo();
  527. if (!temp_array.isEmpty()) {
  528. boolean first = true;
  529. for (CpsEdge temp2 : temp_array) {
  530. if (first) {
  531. first = false;
  532. if (temp.getName().compareTo(temp2.getA().getCompareName()) == 0) {
  533. Object[] tempConnection = { temp.getName() + " is connected to",
  534. temp2.getB().getName() + " with ID: " + temp2.getB().getID() };
  535. tableModelProperties.addRow(tempConnection);
  536. } else {
  537. Object[] tempConnection = { temp.getName() + " is connected to",
  538. temp2.getA().getName() + " with ID: " + temp2.getA().getID() };
  539. tableModelProperties.addRow(tempConnection);
  540. }
  541. } else {
  542. if (temp.getName().compareTo(temp2.getA().getCompareName()) == 0) {
  543. Object[] tempConnection = { "",
  544. temp2.getB().getName() + " with ID: " + temp2.getB().getID() };
  545. tableModelProperties.addRow(tempConnection);
  546. } else {
  547. Object[] tempConnection = { "",
  548. temp2.getA().getName() + " with ID: " + temp2.getA().getID() };
  549. tableModelProperties.addRow(tempConnection);
  550. }
  551. }
  552. }
  553. }
  554. } else if (model.getSelectedEdge() != null) {
  555. Object[] tempName = { "Name", "Edge: " + model.getSelectedEdge().getA().getName() + " to "
  556. + model.getSelectedEdge().getB().getName() };
  557. tableModelProperties.addRow(tempName);
  558. Object[] tempFlow = { "Current flow", model.getSelectedEdge().getFlow() };
  559. tableModelProperties.addRow(tempFlow);
  560. Object[] tempCapacity = { "Max. Capacity", model.getSelectedEdge().getCapacity() };
  561. tableModelProperties.addRow(tempCapacity);
  562. tableModelProperties.setCellEditable(0, 1, false);
  563. tableModelProperties.setCellEditable(2, 1, true);
  564. }
  565. }
  566. });
  567. tableHolonElement.addPropertyChangeListener(new PropertyChangeListener() {
  568. @Override
  569. public void propertyChange(PropertyChangeEvent evt) {
  570. try {
  571. tempCps = controller.searchByID(model.getSelectedObjectID());
  572. if (tempCps != null) {
  573. // System.out.println("change: " + tempCps.toString());
  574. for (int r = 0; r < tableModelHolonElement.getRowCount(); r++) {
  575. ArrayList<Object> temp = new ArrayList<>();
  576. for (int c = 0; c < tableModelHolonElement.getColumnCount(); c++) {
  577. temp.add(tableHolonElement.getValueAt(r, c));
  578. }
  579. // float ch;
  580. // if (temp.get(1).toString().compareTo("-") == 0) {
  581. // ch = -1;
  582. // } else {
  583. // ch = 1;
  584. // }
  585. float fa = Float.parseFloat(temp.get(1).toString());
  586. int i = Integer.parseInt(temp.get(2).toString());
  587. ((HolonObject) tempCps).getElements().get(r).setEleName((String) temp.get(0));
  588. // ((HolonObject)
  589. // tempCps).getElements().get(r).setSign(ch);
  590. ((HolonObject) tempCps).getElements().get(r).setEnergy(fa);
  591. ((HolonObject) tempCps).getElements().get(r).setAmount(i);
  592. // System.out.println(tableHolonElement.getRowCount()
  593. // + " " + temp.get(0) + " " + ch + " " + fa
  594. // + " " + i);
  595. }
  596. if (tableModelHolonElement.getRowCount() > 0) {
  597. for (int i = tableModelHolonElement.getRowCount() - 1; i > -1; i--) {
  598. tableModelHolonElement.removeRow(i);
  599. }
  600. }
  601. for (HolonElement he : canvas.dataSelected) {
  602. Object[] temp = { he.getEleName(), he.getEnergy(), he.getAmount() };
  603. tableModelHolonElement.addRow(temp);
  604. }
  605. }
  606. } catch (Exception e) {
  607. }
  608. }
  609. });
  610. toolBar.add(btnAdd);
  611. // Del Button
  612. btnDel.addActionListener(new ActionListener() {
  613. public void actionPerformed(ActionEvent arg0) {
  614. Object nodeInfo = tree.getLastSelectedPathComponent();
  615. if (nodeInfo != null) {
  616. DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) nodeInfo;
  617. String nodeName = selectedNode.getUserObject().toString();
  618. int depthOfNode = selectedNode.getLevel();
  619. switch (depthOfNode) {
  620. case 1:
  621. int dialogResult = JOptionPane.showConfirmDialog(null,
  622. "Do you realy want to delete the Category " + nodeName + "?", "Warning",
  623. JOptionPane.YES_NO_OPTION);
  624. if (dialogResult == JOptionPane.YES_OPTION) {
  625. controller.deleteCategory(nodeName);
  626. }
  627. break;
  628. case 2:
  629. DefaultMutableTreeNode parent = (DefaultMutableTreeNode) selectedNode.getParent();
  630. controller.delObjectCategory(nodeName, parent.getUserObject().toString());
  631. break;
  632. default:
  633. JOptionPane.showMessageDialog(new JFrame(),
  634. "Please select a Category or an Object in order to delete something.");
  635. }
  636. } else {
  637. JOptionPane.showMessageDialog(new JFrame(),
  638. "Please select a Category or an Object in order to delete something.");
  639. }
  640. }
  641. });
  642. toolBar.add(btnDel);
  643. frmCyberPhysical.getContentPane().add(splitPane);
  644. mntmOpen.addActionListener(new java.awt.event.ActionListener() {
  645. @Override
  646. public void actionPerformed(java.awt.event.ActionEvent evt) {
  647. menuFileExitActionPerformed(evt);
  648. }
  649. private void menuFileExitActionPerformed(java.awt.event.ActionEvent evt) {
  650. JFileChooser fileChooser = new JFileChooser();
  651. JFrame test = new JFrame();
  652. if (fileChooser.showOpenDialog(test) == JFileChooser.APPROVE_OPTION) {
  653. File file = fileChooser.getSelectedFile();
  654. try {
  655. controller.loadFile(file.getAbsolutePath());
  656. canvas.repaint();
  657. tree.repaint();
  658. } catch (IOException e) {
  659. // TODO Auto-generated catch block
  660. e.printStackTrace();
  661. }
  662. }
  663. }
  664. });
  665. mntmSave.addActionListener(new java.awt.event.ActionListener() {
  666. @Override
  667. public void actionPerformed(java.awt.event.ActionEvent evt) {
  668. menuSaveActionPerformed(evt);
  669. }
  670. private void menuSaveActionPerformed(java.awt.event.ActionEvent evt) {
  671. JFileChooser fileChooser = new JFileChooser();
  672. JFrame test = new JFrame();
  673. if (fileChooser.showSaveDialog(test) == JFileChooser.APPROVE_OPTION) {
  674. File file = fileChooser.getSelectedFile();
  675. try {
  676. controller.saveFile(file.getAbsolutePath());
  677. } catch (IOException e) {
  678. // TODO Auto-generated catch block
  679. e.printStackTrace();
  680. }
  681. }
  682. }
  683. });
  684. timePanel = new TimePanel(model, controller);
  685. timePanel.setBorder(null);
  686. ((JSlider) (timePanel.getComponent(1))).addChangeListener(new ChangeListener() {
  687. @Override
  688. public void stateChanged(ChangeEvent e) {
  689. unitGraph.repaint();
  690. }
  691. });
  692. splitPane.setRightComponent(splitPane_1);
  693. splitPane.setDividerLocation(200);
  694. splitPane_1.setDividerLocation(500);
  695. splitPane.setLeftComponent(scrollPane_1);
  696. splitPane_1.setLeftComponent(tabbedPane);
  697. splitPane_1.setRightComponent(split_HolonEl_Pro);
  698. split_HolonEl_Pro.setDividerLocation(600);
  699. split_HolonEl_Pro.setTopComponent(split_Graph_HolonEl);
  700. split_HolonEl_Pro.setBottomComponent(scrollProperties);
  701. split_Graph_HolonEl.setDividerLocation(150);
  702. split_Graph_HolonEl.setTopComponent(scrollGraph);
  703. split_Graph_HolonEl.setBottomComponent(scrollElements);
  704. tabbedPane.setBorder(null);
  705. scrollProperties.setBorder(null);
  706. scrollGraph.setBorder(null);
  707. scrollElements.setBorder(null);
  708. splitPane.setBorder(null);
  709. splitPane_1.setBorder(null);
  710. split_HolonEl_Pro.setBorder(null);
  711. split_Graph_HolonEl.setBorder(null);
  712. scrollPane_2.setBorder(null);
  713. panel_HolonEl.setBorder(null);
  714. tableHolonElementScrollPane.setBorder(null);
  715. frmCyberPhysical.getContentPane().add(timePanel, BorderLayout.SOUTH);
  716. }
  717. /*
  718. * adds a specific object type to selected Category also handles input
  719. * windows and illegal inputs
  720. */
  721. public void addObjectAction(String objType, DefaultMutableTreeNode selectedNode) {
  722. if (selectedNode == null) {
  723. JOptionPane.showMessageDialog(new JFrame(),
  724. "Please select a Category first before adding " + objType + ".");
  725. }
  726. // if selected node is a directory for Categories
  727. else {
  728. if (selectedNode.getLevel() == 1) {
  729. String objname = JOptionPane.showInputDialog("Please enter a Name for the " + objType);
  730. Category cat = controller.searchCategory(selectedNode.getUserObject().toString());
  731. if (objname.length() != 0) {
  732. switch (objType) {
  733. case "Switch":
  734. controller.addSwitch(cat, objname);
  735. break;
  736. case "Transformer":
  737. controller.addTransformer(cat, objname);
  738. break;
  739. }
  740. }
  741. } else {
  742. JOptionPane.showMessageDialog(new JFrame(),
  743. "Objects can not be added to Objects. Please select a Category.");
  744. }
  745. }
  746. }
  747. /**
  748. * reloads the Categories from Model
  749. */
  750. public void updateCategories(final ArrayList<Category> categories) {
  751. tree.setModel(new DefaultTreeModel(new DefaultMutableTreeNode("Categories") {
  752. {
  753. DefaultMutableTreeNode node_1;
  754. for (Category c : categories) {
  755. node_1 = new DefaultMutableTreeNode(c.getName());
  756. // kann eventuell umgeändert werden
  757. for (CpsObject obj : c.getObjects()) {
  758. node_1.add(new DefaultMutableTreeNode(obj.getObjName()));
  759. }
  760. add(node_1);
  761. }
  762. }
  763. }));
  764. }
  765. /**
  766. *
  767. */
  768. public void onChange(ArrayList<Category> categories) {
  769. DefaultTreeModel model = (DefaultTreeModel) tree.getModel();
  770. updateCategories(categories);
  771. model.reload();
  772. }
  773. /**
  774. *
  775. * @return
  776. */
  777. public JFrame getFrmCyberPhysical() {
  778. return frmCyberPhysical;
  779. }
  780. /**
  781. * Getter for selected CpsObject
  782. *
  783. * @return selected CpsObject
  784. */
  785. public CpsObject getActualCps() {
  786. int tempID = model.getSelectedObjectID();
  787. CpsObject tempCps = controller.searchByID(tempID);
  788. return tempCps;
  789. }
  790. /**
  791. * Search for actual selected HolonElement
  792. *
  793. * @param obj
  794. * selected HolonObject
  795. * @param yValue
  796. * Y-Coord in the HolonElementsTable
  797. * @return the selected HolonElement
  798. */
  799. public HolonElement getActualHolonElement(HolonObject obj, int yValue) {
  800. final int yTemp = (int) Math.floor(yValue / 16);
  801. int rowsTotal = tableModelHolonElement.getRowCount();
  802. if (rowsTotal != 0 && rowsTotal > yTemp) {
  803. model.setSelectedHolonElement(obj.getElements().get(yTemp));
  804. return obj.getElements().get(yTemp);
  805. } else {
  806. model.setSelectedHolonElement(null);
  807. return null;
  808. }
  809. }
  810. /*
  811. * Refresh the Table displaying the HolonElements
  812. */
  813. public void refreshTableHolonElement() {
  814. // Update of the Information about the HolonElements - only for
  815. // HolonObjects
  816. deleteRows();
  817. if (canvas.dataSelected != null) {
  818. fillElementTable(canvas.dataSelected);
  819. }
  820. /**
  821. * hinzugef�gt damit man auch nach dem objekt platziert wurde elemente
  822. * von Objekten in Kategorien ansehen kann
  823. */
  824. canvas.dataSelected = null;
  825. }
  826. public void deleteRows() {
  827. if (tableModelHolonElement.getRowCount() > 0) {
  828. for (int i = tableModelHolonElement.getRowCount() - 1; i > -1; i--) {
  829. tableModelHolonElement.removeRow(i);
  830. }
  831. }
  832. }
  833. public void fillElementTable(ArrayList<HolonElement> elements) {
  834. for (HolonElement he : elements) {
  835. Object[] temp = { he.getEleName(), he.getEnergy(), he.getAmount() };
  836. tableModelHolonElement.addRow(temp);
  837. }
  838. }
  839. /**
  840. * Update the information about properties of the selected CpsObject
  841. */
  842. public void refreshTableProperties() {
  843. CpsObject tempCps = getActualCps();
  844. if (tempCps.getClass() == HolonObject.class && tempCps != null) {
  845. tableModelProperties.removeRow(2);
  846. Object[] tempEnergy = { "Total Energy", ((HolonObject) canvas.tempCps).getCurrentEnergy() };
  847. tableModelProperties.insertRow(2, tempEnergy);
  848. }
  849. }
  850. private static void addPopup(Component component, final JPopupMenu popup) {
  851. component.addMouseListener(new MouseAdapter() {
  852. public void mousePressed(MouseEvent e) {
  853. if (e.isPopupTrigger()) {
  854. showMenu(e);
  855. }
  856. }
  857. public void mouseReleased(MouseEvent e) {
  858. if (e.isPopupTrigger()) {
  859. showMenu(e);
  860. }
  861. }
  862. private void showMenu(MouseEvent e) {
  863. popup.show(e.getComponent(), e.getX(), e.getY());
  864. }
  865. });
  866. }
  867. }