platform/lang-impl/src/com/intellij/ide/todo/nodes/SingleFileToDoNode.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package com.intellij.ide.todo.nodes;
  18
  19 import com.intellij.ide.projectView.PresentationData;
  20 import com.intellij.ide.todo.TodoTreeBuilder;
  21 import com.intellij.ide.util.treeView.AbstractTreeNode;
  22 import com.intellij.openapi.project.Project;
  23 import com.intellij.psi.PsiFile;
  24 import org.jetbrains.annotations.NotNull;
  25
  26 import java.util.ArrayList;
  27 import java.util.Collection;
  28 import java.util.Collections;
  29
  30 public class SingleFileToDoNode extends BaseToDoNode<PsiFile>{
  31   private final TodoFileNode myFileNode = new TodoFileNode(getProject(), getValue(), myBuilder, true);
  32
  33   public SingleFileToDoNode(Project project, PsiFile value, TodoTreeBuilder builder) {
  34     super(project, value, builder);
  35   }
  36
  37   @NotNull
  38   public Collection<AbstractTreeNode> getChildren() {
  39     return new ArrayList<AbstractTreeNode>(Collections.singleton(myFileNode));
  40   }
  41
  42   public void update(PresentationData presentation) {
  43   }
  44
  45   public Object getFileNode() {
  46     return myFileNode;
  47   }
  48
  49   public int getFileCount(final PsiFile val) {
  50     return 1;
  51   }
  52
  53   public int getTodoItemCount(final PsiFile val) {
  54     return getTreeStructure().getTodoItemCount(val);
  55   }
  56 }