Java ArrayFactory类代码示例

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
/**
 * @return children of specified type, taken from stubs (if this element is currently stub-based) or AST (otherwise).
 */
public <S extends StubElement, Psi extends PsiElement> Psi[] getStubOrPsiChildren(final IStubElementType<S, Psi> elementType, ArrayFactory<Psi> f) {
  T stub = myStub;
  if (stub != null) {
    //noinspection unchecked
    return (Psi[])stub.getChildrenByType(elementType, f);
  }
  else {
    final ASTNode[] nodes = SharedImplUtil.getChildrenOfType(getNode(), elementType);
    Psi[] psiElements = f.create(nodes.length);
    for (int i = 0; i < nodes.length; i++) {
      //noinspection unchecked
      psiElements[i] = (Psi)nodes[i].getPsi();
    }
    return psiElements;
  }
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@NotNull
public <T extends PsiElement> T[] getChildrenAsPsiElements(@Nullable TokenSet filter, ArrayFactory<T> constructor) {
  ApplicationManager.getApplication().assertReadAccessAllowed();
  int count = countChildren(filter);
  T[] result = constructor.create(count);
  if (count == 0) {
    return result;
  }
  int idx = 0;
  for (ASTNode child = getFirstChildNode(); child != null && idx < count; child = child.getTreeNext()) {
    if (filter == null || filter.contains(child.getElementType())) {
      @SuppressWarnings("unchecked") T element = (T)child.getPsi();
      LOG.assertTrue(element != null, child);
      result[idx++] = element;
    }
  }
  return result;
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@NotNull
public PyImportElement[] getImportElements() {
  final PyFromImportStatementStub stub = getStub();
  if (stub != null) {
    return stub.getChildrenByType(PyElementTypes.IMPORT_ELEMENT, new ArrayFactory<PyImportElement>() {
      @NotNull
      public PyImportElement[] create(int count) {
        return new PyImportElement[count];
      }
    });
  }
  List<PyImportElement> result = new ArrayList<PyImportElement>();
  final ASTNode importKeyword = getNode().findChildByType(PyTokenTypes.IMPORT_KEYWORD);
  if (importKeyword != null) {
    for (ASTNode node = importKeyword.getTreeNext(); node != null; node = node.getTreeNext()) {
      if (node.getElementType() == PyElementTypes.IMPORT_ELEMENT) {
        result.add((PyImportElement)node.getPsi());
      }
    }
  }
  return result.toArray(new PyImportElement[result.size()]);
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
/**
 * @return children of specified type, taken from stubs (if this element is currently stub-based) or AST (otherwise).
 */
@Nonnull
public <S extends StubElement, Psi extends PsiElement> Psi[] getStubOrPsiChildren(@Nonnull IStubElementType<S, ? extends Psi> elementType, @Nonnull ArrayFactory<Psi> f) {
  T stub = getGreenStub();
  if (stub != null) {
    //noinspection unchecked
    return (Psi[])stub.getChildrenByType(elementType, f);
  }
  else {
    final ASTNode[] nodes = SharedImplUtil.getChildrenOfType(getNode(), elementType);
    Psi[] psiElements = f.create(nodes.length);
    for (int i = 0; i < nodes.length; i++) {
      //noinspection unchecked
      psiElements[i] = (Psi)nodes[i].getPsi();
    }
    return psiElements;
  }
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Nonnull
public <T extends PsiElement> T[] getChildrenAsPsiElements(@Nullable TokenSet filter, ArrayFactory<T> constructor) {
  ApplicationManager.getApplication().assertReadAccessAllowed();
  int count = countChildren(filter);
  T[] result = constructor.create(count);
  if (count == 0) {
    return result;
  }
  int idx = 0;
  for (ASTNode child = getFirstChildNode(); child != null && idx < count; child = child.getTreeNext()) {
    if (filter == null || filter.contains(child.getElementType())) {
      @SuppressWarnings("unchecked") T element = (T)child.getPsi();
      LOG.assertTrue(element != null, child);
      result[idx++] = element;
    }
  }
  return result;
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Nonnull
@Override
public PsiPackage createPackage(@Nonnull PsiManager psiManager,
                                @Nonnull PsiPackageManager packageManager,
                                @Nonnull Class<? extends ModuleExtension> extensionClass,
                                @Nonnull String packageName) {
  return new PsiPackageBase(psiManager, packageManager, extensionClass, packageName) {
    @Override
    protected ArrayFactory<? extends PsiPackage> getPackageArrayFactory() {
      return PsiPackage.ARRAY_FACTORY;
    }

    @RequiredReadAction
    @Nonnull
    @Override
    public Language getLanguage() {
      return SandLanguage.INSTANCE;
    }
  };
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
private static <E extends PsiElement> E[] getStubChildrenByType(JSClassStub stub, final IElementType elementType, ArrayFactory<E> f)
{
	assert JSElementTypes.INCLUDE_DIRECTIVE != elementType;

	ArrayList<E> result = new ArrayList<E>(Arrays.asList(stub.getChildrenByType(elementType, f)));
	JSIncludeDirective[] includes = stub.getChildrenByType(JSElementTypes.INCLUDE_DIRECTIVE, new ArrayFactory<JSIncludeDirective>()
	{
		@Override
		public JSIncludeDirective[] create(final int count)
		{
			return new JSIncludeDirective[count];
		}
	});
	Collection<JSFile> visited = new THashSet<JSFile>();
	TokenSet filter = TokenSet.create(JSElementTypes.INCLUDE_DIRECTIVE, elementType);
	for(JSIncludeDirective include : includes)
	{
		PsiFile file = include.resolveFile();
		if(file instanceof JSFile)
		{
			process(filter, (JSFile) file, result, visited);
		}
	}
	return result.toArray(f.create(result.size()));
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Override
public <E extends PsiElement> E[] getChildrenByType(final IElementType elementType, final ArrayFactory<E> f) {
  int count = countChildren(elementType);

  E[] result = f.create(count);
  if (count > 0) fillFilteredChildren(elementType, result);

  return result;
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@NotNull
public PyImportElement[] getImportElements() {
  final PyImportStatementStub stub = getStub();
  if (stub != null) {
    return stub.getChildrenByType(PyElementTypes.IMPORT_ELEMENT, new ArrayFactory<PyImportElement>() {
      @NotNull
      public PyImportElement[] create(int count) {
        return new PyImportElement[count];
      }
    });
  }
  return childrenToPsi(TokenSet.create(PyElementTypes.IMPORT_ELEMENT), new PyImportElement[0]);
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
public <Stub extends StubElement, Psi extends PsiElement> Psi[] getStubOrPsiChildren(final IStubElementType<Stub, Psi> elementType, ArrayFactory<Psi> f) {
  T stub = myStub;
  if (stub != null) {
    //noinspection unchecked
    return (Psi[])stub.getChildrenByType(elementType, f);
  }
  else {
    final ASTNode[] nodes = SharedImplUtil.getChildrenOfType(getNode(), elementType);
    Psi[] psiElements = f.create(nodes.length);
    for (int i = 0; i < nodes.length; i++) {
      psiElements[i] = (Psi)nodes[i].getPsi();
    }
    return psiElements;
  }
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Nonnull
@Override
public <E extends PsiElement> E[] getChildrenByType(@Nonnull final IElementType elementType, @Nonnull final ArrayFactory<E> f) {
  int count = countChildren(elementType);

  E[] result = f.create(count);
  if (count > 0) fillFilteredChildren(elementType, result);

  return result;
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Override
protected ArrayFactory<? extends PsiPackage> getPackageArrayFactory()
{
	return HaxePackage.ARRAY_FACTORY;
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Nullable
public <T extends PsiElement> T getFirstStubOrPsiChild(TokenSet tokenSet, ArrayFactory<T> arrayFactory)
{
	T[] stubOrPsiChildren = getStubOrPsiChildren(tokenSet, arrayFactory);
	return ArrayUtil2.safeGet(stubOrPsiChildren, 0);
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Nullable
public <T extends PsiElement> T getStubOrPsiChildByIndex(TokenSet tokenSet, ArrayFactory<T> arrayFactory, int index)
{
	T[] stubOrPsiChildren = getStubOrPsiChildren(tokenSet, arrayFactory);
	return ArrayUtil2.safeGet(stubOrPsiChildren, index);
}
 

import com.intellij.util.ArrayFactory; //导入依赖的package包/类
@Override
protected ArrayFactory<? extends PsiPackage> getPackageArrayFactory()
{
	return DotNetPackage.ARRAY_FACTORY;
}