Dedup stackmaps.

runtime/vm/precompiler.h

 // Copyright (c) 2015, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #ifndef VM_PRECOMPILER_H_
 #define VM_PRECOMPILER_H_
 
 #include "vm/allocation.h"
 #include "vm/hash_map.h"
 #include "vm/object.h"
 
 namespace dart {
 
 // Forward declarations.
 class Class;
 class Error;
 class Field;
 class Function;
 class GrowableObjectArray;
 class RawError;
 class String;
 
 class SymbolPair {
  public:
   // Typedefs needed for the DirectChainedHashMap template.
   typedef const String* Key;
-  typedef bool Value;
+  typedef const String* Value;
   typedef SymbolPair Pair;
 
-  SymbolPair() : key_(NULL), value_(false) {}
-  SymbolPair(Key key, Value value) : key_(key), value_(value) {
+  SymbolPair() : key_(NULL) {}
+  SymbolPair(Key key, Value value) : key_(key) {
     ASSERT(key->IsNotTemporaryScopedHandle());
+    ASSERT(key == value);
   }
 
   static Key KeyOf(Pair kv) { return kv.key_; }
 
-  static Value ValueOf(Pair kv) { return kv.value_; }
+  static Value ValueOf(Pair kv) { return kv.key_; }
 
   static inline intptr_t Hashcode(Key key) {
     return key->Hash();
   }
 
   static inline bool IsKeyEqual(Pair pair, Key key) {
     return pair.key_->raw() == key->raw();
   }
 
  private:
   Key key_;

[Florian Schneider, 2015/10/07 10:40:07]

No need for key_ either.

This is basicaly the same as PointerKeyValueTrait - modulo IsKeyEqual, which
invokes Equals, instead of comparing the raw-ptr. Actually, PointerKeyValueTrait
may just work as a drop-in replacement.

[rmacnak, 2015/10/07 22:19:06]

Oh, I misunderstood this to be used as the backing storage of the map.

-  Value value_;
 };
 
 
 class SymbolSet : public ValueObject {
  public:
   explicit SymbolSet(Zone* zone) : zone_(zone), map_() {}
 
   void Add(const String& symbol) {
     ASSERT(symbol.IsSymbol());
     if (symbol.IsNotTemporaryScopedHandle()) {
-      SymbolPair pair(&symbol, true);
+      SymbolPair pair(&symbol, &symbol);
       map_.Insert(pair);
     } else {
-      SymbolPair pair(&String::ZoneHandle(zone_, symbol.raw()), true);
+      const String* zone_symbol = &String::ZoneHandle(zone_, symbol.raw());
+      SymbolPair pair(zone_symbol, zone_symbol);
       map_.Insert(pair);
     }
   }
 
   bool Includes(const String& symbol) {
     ASSERT(symbol.IsSymbol());
-    return map_.Lookup(&symbol);
+    return map_.Lookup(&symbol) != NULL;
   }
 
  private:
   Zone* zone_;
   DirectChainedHashMap<SymbolPair> map_;
 };
 
 
+class StackmapPair {
+ public:
+  // Typedefs needed for the DirectChainedHashMap template.
+  typedef const Stackmap* Key;
+  typedef const Stackmap* Value;
+  typedef StackmapPair Pair;
+
+  StackmapPair() : key_(NULL) {}
+  StackmapPair(Key key, Value value) : key_(key) {
+    ASSERT(key->IsNotTemporaryScopedHandle());
+    ASSERT(key == value);
+  }
+
+  static Key KeyOf(Pair kv) { return kv.key_; }
+
+  static Value ValueOf(Pair kv) { return kv.key_; }
+
+  static inline intptr_t Hashcode(Key key) {

[Florian Schneider, 2015/10/07 10:40:07]

Can you avoid this class and use PointerKeyValueTrait instead, if you add a
corresponding Hashcode method to Stackmap?

[rmacnak, 2015/10/07 22:19:06]

Not quite. It wants Stackmap::Equals(Stackmap*&) instead of
Stackmap::Equals(Stackmap&)

+    return key->PcOffset();
+  }
+
+  static inline bool IsKeyEqual(Pair pair, Key key) {
+    return pair.key_->Equals(*key);
+  }
+
+ private:
+  Key key_;
+};
+
+
+class StackmapSet : public ValueObject {

[Florian Schneider, 2015/10/07 10:40:07]

typedef DirectChainedHashmap<PointerKeyValueTrait<Stackmap> > StackmapSet;

[rmacnak, 2015/10/07 22:19:06]

Dropped StackmapSet and SymbolsSet for typedefs.

+ public:
+  explicit StackmapSet(Zone* zone) : zone_(zone), map_() {}
+
+  void Add(const Stackmap& stackmap) {
+    StackmapPair pair(&stackmap, &stackmap);
+    map_.Insert(pair);
+  }
+
+  const Stackmap* Lookup(const Stackmap& stackmap) {
+    return map_.Lookup(&stackmap);
+  }
+
+ private:
+  Zone* zone_;
+  DirectChainedHashMap<StackmapPair> map_;
+};
+
+
 class Precompiler : public ValueObject {
  public:
   static RawError* CompileAll(
       Dart_QualifiedFunctionName embedder_entry_points[],
       bool reset_fields);
 
  private:
   Precompiler(Thread* thread, bool reset_fields);
 
   void DoCompileAll(Dart_QualifiedFunctionName embedder_entry_points[]);
@@ skipping 10 matching lines @@
   void AddClass(const Class& cls);
   void AddSelector(const String& selector);
   bool IsSent(const String& selector);
 
   void ProcessFunction(const Function& function);
   void CheckForNewDynamicFunctions();
 
   void DropUncompiledFunctions();
   void BindStaticCalls();
   void BindStaticCalls(const Function& function);
+  void DedupStackmaps();
+  void DedupStackmaps(const Function& function);
+  RawStackmap* DedupStackmap(const Stackmap& stackmap);
 
   Thread* thread() const { return thread_; }
   Zone* zone() const { return zone_; }
   Isolate* isolate() const { return isolate_; }
 
   Thread* thread_;
   Zone* zone_;
   Isolate* isolate_;
 
   const bool reset_fields_;
 
   bool changed_;
   intptr_t function_count_;
   intptr_t class_count_;
   intptr_t selector_count_;
   intptr_t dropped_function_count_;
 
   const GrowableObjectArray& libraries_;
   const GrowableObjectArray& pending_functions_;
   const GrowableObjectArray& collected_closures_;
   SymbolSet sent_selectors_;
+  StackmapSet stackmaps_;
   Error& error_;
 };
 
 }  // namespace dart
 
 #endif  // VM_PRECOMPILER_H_