GUI.java 33 KB

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