lang-api/src/com/intellij/patterns/IElementTypePattern.java

   1 /*
   2  * Copyright (c) 2000-2007 JetBrains s.r.o. All Rights Reserved.
   3  */
   4 package com.intellij.patterns;
   5
   6 import com.intellij.psi.tree.IElementType;
   7 import com.intellij.psi.tree.TokenSet;
   8 import com.intellij.patterns.ObjectPattern;
   9 import com.intellij.patterns.PatternCondition;
  10 import com.intellij.patterns.MatchingContext;
  11 import com.intellij.patterns.TraverseContext;
  12 import org.jetbrains.annotations.NotNull;
  13
  14 /**
  15  * @author peter
  16  */
  17 public class IElementTypePattern extends ObjectPattern<IElementType, IElementTypePattern> {
  18   protected IElementTypePattern() {
  19     super(IElementType.class);
  20   }
  21
  22   public IElementTypePattern or(@NotNull final IElementType... types){
  23     return tokenSet(TokenSet.create(types));
  24   }
  25
  26   public IElementTypePattern tokenSet(@NotNull final TokenSet tokenSet){
  27     return with(new PatternCondition<IElementType>() {
  28       public boolean accepts(@NotNull final IElementType type, final MatchingContext matchingContext, @NotNull final TraverseContext traverseContext) {
  29         return tokenSet.contains(type);
  30       }
  31     });
  32   }
  33
  34 }