lang-impl/src/com/intellij/psi/impl/source/tree/AstBufferUtil.java

   1 /*
   2  * @author max
   3  */
   4 package com.intellij.psi.impl.source.tree;
   5
   6 import com.intellij.lang.ASTNode;
   7 import com.intellij.psi.tree.TokenSet;
   8
   9 public class AstBufferUtil {
  10   private AstBufferUtil() {}
  11
  12   public static int toBuffer(ASTNode element, char[] buffer, int offset) {
  13     return toBuffer(element, buffer, offset, null);
  14   }
  15
  16   public static int toBuffer(ASTNode element, char[] buffer, int offset, TokenSet skipTypes) {
  17     if (element instanceof ForeignLeafPsiElement || skipTypes != null && skipTypes.contains(element.getElementType())) return offset;
  18     if (element instanceof LeafElement) {
  19       return ((LeafElement)element).copyTo(buffer, offset);
  20     }
  21
  22     if (element instanceof LazyParseableElement) {
  23       LazyParseableElement lpe = (LazyParseableElement)element;
  24       int lpeResult = lpe.copyTo(buffer, offset);
  25       if (lpeResult > 0) return lpeResult;
  26     }
  27
  28     int curOffset = offset;
  29     for (TreeElement child = (TreeElement)element.getFirstChildNode(); child != null; child = child.getTreeNext()) {
  30       curOffset = toBuffer(child, buffer, curOffset, skipTypes);
  31     }
  32     return curOffset;
  33   }
  34 }