fixed consistency bug in InsertHistory

src/main/java/de/neemann/digital/draw/library/ElementLibrary.java

@@ -201,6 +201,16 @@ public class ElementLibrary implements Iterable<ElementLibrary.ElementContainer>
         }
     }
 
+    /**
+     * Returns the node or null if node not present.
+     *
+     * @param elementName the name
+     * @return the node or null
+     */
+    public LibraryNode getElementNodeOrNull(String elementName) {
+        return map.get(elementName);
+    }
+
     /**
      * Returns a {@link ElementTypeDescription} by a given name.
      * If not found its tried to load it.
@@ -211,9 +221,9 @@ public class ElementLibrary implements Iterable<ElementLibrary.ElementContainer>
      */
     public ElementTypeDescription getElementType(String elementName) throws ElementNotFoundException {
         try {
-            LibraryNode description = map.get(elementName);
+            LibraryNode node = map.get(elementName);
-            if (description != null)
+            if (node != null)
-                return description.getDescription();
+                return node.getDescription();
 
             // effects only some old files!
             elementName = elementName.replace("\\", "/");
@@ -221,18 +231,18 @@ public class ElementLibrary implements Iterable<ElementLibrary.ElementContainer>
                 elementName = new File(elementName).getName();
             }
 
-            description = map.get(elementName);
+            node = map.get(elementName);
-            if (description != null)
+            if (node != null)
-                return description.getDescription();
+                return node.getDescription();
 
             if (rootLibraryPath == null)
                 throw new ElementNotFoundException(Lang.get("err_fileNeedsToBeSaved"));
 
             rescanFolder();
 
-            description = map.get(elementName);
+            node = map.get(elementName);
-            if (description != null)
+            if (node != null)
-                return description.getDescription();
+                return node.getDescription();
         } catch (IOException e) {
             throw new ElementNotFoundException(Lang.get("msg_errorImportingModel"), e);
         }

src/main/java/de/neemann/digital/gui/InsertAction.java

@@ -19,10 +19,10 @@ import java.io.IOException;
  * Created by hneemann on 25.03.17.
  */
 public final class InsertAction extends ToolTipAction {
-    private final LibraryNode node;
     private final InsertHistory insertHistory;
     private final CircuitComponent circuitComponent;
     private final ShapeFactory shapeFactory;
+    private LibraryNode node;
 
     /**
      * Creates a new instance
@@ -69,4 +69,18 @@ public final class InsertAction extends ToolTipAction {
         return node.getName();
     }
 
+    /**
+     * Updates this action to a new node
+     *
+     * @param node the node
+     */
+    public void update(LibraryNode node) {
+        this.node = node;
+        try {
+            final ImageIcon icon = node.getIcon(shapeFactory);
+            setIcon(icon);
+        } catch (IOException ex) {
+            SwingUtilities.invokeLater(new ErrorMessage(Lang.get("msg_errorImportingModel")).addCause(ex));
+        }
+    }
 }

src/main/java/de/neemann/digital/gui/InsertHistory.java

@@ -1,5 +1,6 @@
 package de.neemann.digital.gui;
 
+import de.neemann.digital.draw.library.ElementLibrary;
 import de.neemann.digital.draw.library.LibraryListener;
 import de.neemann.digital.draw.library.LibraryNode;
 
@@ -18,6 +19,7 @@ public class InsertHistory implements LibraryListener {
 
     private static final int MAX_ICONS = 6;
     private final JToolBar bar;
+    private final ElementLibrary library;
     private final ArrayList<WrapperAction> wrappers;
     private int mainTime;
 
@@ -26,8 +28,9 @@ public class InsertHistory implements LibraryListener {
      *
      * @param bar the toolbar to put the elements to
      */
-    public InsertHistory(JToolBar bar) {
+    public InsertHistory(JToolBar bar, ElementLibrary library) {
         this.bar = bar;
+        this.library = library;
         wrappers = new ArrayList<>();
     }
 
@@ -93,8 +96,12 @@ public class InsertHistory implements LibraryListener {
         while (it.hasNext()) {
             WrapperAction w = it.next();
             if (w.action.isCustom()) {
+                LibraryNode n = library.getElementNodeOrNull(w.action.getName());
+                if (n == null) {  // is'nt there, so delete
                     removeWrapperFromBar(w);
                     it.remove();
+                } else
+                    w.update(n);
             }
         }
         bar.revalidate();
@@ -117,5 +124,10 @@ public class InsertHistory implements LibraryListener {
             action.actionPerformed(e);
             time = mainTime++;
         }
+
+        public void update(LibraryNode n) {
+            action.update(n);
+            putValue(Action.SMALL_ICON, action.getValue(Action.SMALL_ICON));
+        }
     }
 }

src/main/java/de/neemann/digital/gui/Main.java

@@ -222,7 +222,7 @@ public class Main extends JFrame implements ClosingWindowListener.ConfirmSave, E
 
         toolBar.addSeparator();
 
-        insertHistory = new InsertHistory(toolBar);
+        insertHistory = new InsertHistory(toolBar, library);
         library.addListener(insertHistory);
         final LibrarySelector librarySelector = new LibrarySelector(library, shapeFactory);
         library.addListener(librarySelector);