source/com/intellij/slicer/DuplicateMap.java

   1 package com.intellij.slicer;
   2
   3 import gnu.trove.THashMap;
   4 import gnu.trove.TObjectHashingStrategy;
   5
   6 import java.util.Map;
   7
   8 /**
   9 * @author cdr
  10 */ // rehash map on each PSI modification since SmartPsiPointer's hashCode() and equals() are changed
  11 class DuplicateMap {
  12   private final Map<SliceUsage, SliceNode> myDuplicates = new THashMap<SliceUsage, SliceNode>(USAGEINFO_EQUALITY);
  13   private static final TObjectHashingStrategy<SliceUsage> USAGEINFO_EQUALITY = new TObjectHashingStrategy<SliceUsage>() {
  14     public int computeHashCode(SliceUsage object) {
  15       return object.getUsageInfo().hashCode();
  16     }
  17
  18     public boolean equals(SliceUsage o1, SliceUsage o2) {
  19       return o1.getUsageInfo().equals(o2.getUsageInfo());
  20     }
  21   };
  22   private static final TObjectHashingStrategy<SliceNode> SLICE_NODE_EQUALITY = new TObjectHashingStrategy<SliceNode>() {
  23     public int computeHashCode(SliceNode object) {
  24       return object.getValue().getUsageInfo().hashCode();
  25     }
  26
  27     public boolean equals(SliceNode o1, SliceNode o2) {
  28       return o1.getValue().getUsageInfo().equals(o2.getValue().getUsageInfo());
  29     }
  30   };
  31
  32   public SliceNode putNodeCheckDupe(SliceNode node) {
  33     SliceUsage usage = node.getValue();
  34     SliceNode eq = myDuplicates.get(usage);
  35     if (eq == null) {
  36       myDuplicates.put(usage, node);
  37     }
  38     return eq;
  39   }
  40
  41   public void clear() {
  42     myDuplicates.clear();
  43   }
  44 }