added HashingCollectionsNode.SetDictStorageNode

Author: tomasstupka

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/common/HashingCollectionNodes.java

@@ -42,6 +42,7 @@
 
 import com.oracle.graal.python.builtins.objects.common.HashingCollectionNodesFactory.GetDictStorageNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingCollectionNodesFactory.LenNodeGen;
+import com.oracle.graal.python.builtins.objects.common.HashingCollectionNodesFactory.SetDictStorageNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingCollectionNodesFactory.SetItemNodeGen;
 import com.oracle.graal.python.nodes.PGuards;
 import com.oracle.graal.python.nodes.PNodeWithContext;
@@ -131,4 +132,30 @@ public static GetDictStorageNode getUncached() {
             return GetDictStorageNodeGen.getUncached();
         }
     }
+
+    @ImportStatic({PGuards.class})
+    @GenerateUncached
+    public abstract static class SetDictStorageNode extends PNodeWithContext {
+
+        public abstract void execute(PHashingCollection c, HashingStorage storage);
+
+        @Specialization(limit = "4", guards = {"c.getClass() == cachedClass"})
+        void setStorageCached(PHashingCollection c, HashingStorage storage,
+                        @Cached("c.getClass()") Class<? extends PHashingCollection> cachedClass) {
+            cachedClass.cast(c).setDictStorage(storage);
+        }
+
+        @Specialization(replaces = "setStorageCached")
+        static void getStorageGeneric(PHashingCollection c, HashingStorage storage) {
+            c.setDictStorage(storage);
+        }
+
+        public static SetDictStorageNode create() {
+            return SetDictStorageNodeGen.create();
+        }
+
+        public static SetDictStorageNode getUncached() {
+            return SetDictStorageNodeGen.getUncached();
+        }
+    }
 }