HierarchyFeatureExtractor.java

package jetbrains.mps.vcs.changesmanager.tree;

/*Generated by MPS */

import org.jetbrains.annotations.Nullable;
import jetbrains.mps.vcs.changesmanager.tree.features.Feature;
import org.jetbrains.annotations.NotNull;
import jetbrains.mps.ide.ui.tree.MPSTreeNode;
import jetbrains.mps.ide.hierarchy.HierarchyTreeNode;
import jetbrains.mps.vcs.changesmanager.tree.features.NodeFeature;

public class HierarchyFeatureExtractor implements TreeNodeFeatureExtractor {
  public HierarchyFeatureExtractor() {
  }
  @Nullable
  @Override
  public Feature getFeature(@NotNull MPSTreeNode treeNode) {
    if (treeNode instanceof HierarchyTreeNode) {
      return new NodeFeature(((HierarchyTreeNode) treeNode).getNodeReference());
    }
    return null;
  }
}