// reloads the element on all open panels and adds it to the custom element panel if not already // there. private void updateElement(Entity element) { // if a new element has been created add it to current diagram if (this.originalElement == null) { DiagramHandler current = null; DrawPanel c = Umlet.getInstance().getGUI().getCurrentDiagram(); if (c == null) { Umlet.getInstance().doNew(); current = Umlet.getInstance().getGUI().getCurrentDiagram().getHandler(); } else current = c.getHandler(); Vector<Entity> ents = current.getDrawPanel().getAllEntities(); // set location for element int x = 10, y = 10; for (Entity e : ents) { if (e.getY() + e.getHeight() + 10 > y) y = e.getY() + e.getHeight() + 10; } Rectangle bounds = new Rectangle(x, y, element.getWidth(), element.getHeight()); this.addElementToDiagram(element, current, true, bounds, element.getState()); } else { // replace edited element (and ONLY edited element) this.originalElement.getHandler().getDrawPanel().remove(this.originalElement); this.addElementToDiagram( element, this.originalElement.getHandler(), true, this.originalElement.getBounds(), this.originalElement.getState()); } }