| /******************************************************************************* |
| * Copyright (c) 2000, 2015 IBM Corporation and others. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * IBM Corporation - initial API and implementation |
| * Juerg Billeter, juergbi@ethz.ch - 47136 Search view should show match objects |
| * Ulrich Etter, etteru@ethz.ch - 47136 Search view should show match objects |
| * Roman Fuchs, fuchsro@ethz.ch - 47136 Search view should show match objects |
| *******************************************************************************/ |
| package org.eclipse.search.internal.ui.text; |
| |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.Map; |
| import java.util.Set; |
| |
| import org.eclipse.core.resources.IProject; |
| import org.eclipse.core.resources.IResource; |
| |
| import org.eclipse.jface.viewers.AbstractTreeViewer; |
| import org.eclipse.jface.viewers.ITreeContentProvider; |
| import org.eclipse.jface.viewers.Viewer; |
| |
| import org.eclipse.search.ui.text.AbstractTextSearchResult; |
| import org.eclipse.search.ui.text.Match; |
| |
| |
| public class FileTreeContentProvider implements ITreeContentProvider, IFileSearchContentProvider { |
| |
| private final Object[] EMPTY_ARR= new Object[0]; |
| |
| private AbstractTextSearchResult fResult; |
| private FileSearchPage fPage; |
| private AbstractTreeViewer fTreeViewer; |
| private Map<Object, Set<Object>> fChildrenMap; |
| |
| FileTreeContentProvider(FileSearchPage page, AbstractTreeViewer viewer) { |
| fPage= page; |
| fTreeViewer= viewer; |
| } |
| |
| @Override |
| public Object[] getElements(Object inputElement) { |
| Object[] children= getChildren(inputElement); |
| int elementLimit= getElementLimit(); |
| if (elementLimit != -1 && elementLimit < children.length) { |
| Object[] limitedChildren= new Object[elementLimit]; |
| System.arraycopy(children, 0, limitedChildren, 0, elementLimit); |
| return limitedChildren; |
| } |
| return children; |
| } |
| |
| private int getElementLimit() { |
| return fPage.getElementLimit().intValue(); |
| } |
| |
| @Override |
| public void dispose() { |
| // nothing to do |
| } |
| |
| @Override |
| public void inputChanged(Viewer viewer, Object oldInput, Object newInput) { |
| if (newInput instanceof FileSearchResult) { |
| initialize((FileSearchResult) newInput); |
| } |
| } |
| |
| private synchronized void initialize(AbstractTextSearchResult result) { |
| fResult= result; |
| fChildrenMap= new HashMap<>(); |
| boolean showLineMatches= !((FileSearchQuery) fResult.getQuery()).isFileNameSearch(); |
| |
| if (result != null) { |
| Object[] elements= result.getElements(); |
| for (int i= 0; i < elements.length; i++) { |
| if (showLineMatches) { |
| Match[] matches= result.getMatches(elements[i]); |
| for (int j= 0; j < matches.length; j++) { |
| insert(((FileMatch) matches[j]).getLineElement(), false); |
| } |
| } else { |
| insert(elements[i], false); |
| } |
| } |
| } |
| } |
| |
| private void insert(Object child, boolean refreshViewer) { |
| Object parent= getParent(child); |
| while (parent != null) { |
| if (insertChild(parent, child)) { |
| if (refreshViewer) |
| fTreeViewer.add(parent, child); |
| } else { |
| if (refreshViewer) |
| fTreeViewer.refresh(parent); |
| return; |
| } |
| child= parent; |
| parent= getParent(child); |
| } |
| if (insertChild(fResult, child)) { |
| if (refreshViewer) |
| fTreeViewer.add(fResult, child); |
| } |
| } |
| |
| /** |
| * Adds the child to the parent. |
| * |
| * @param parent the parent |
| * @param child the child |
| * @return <code>true</code> if this set did not already contain the specified element |
| |
| */ |
| private boolean insertChild(Object parent, Object child) { |
| Set<Object> children= fChildrenMap.get(parent); |
| if (children == null) { |
| children= new HashSet<>(); |
| fChildrenMap.put(parent, children); |
| } |
| return children.add(child); |
| } |
| |
| private boolean hasChild(Object parent, Object child) { |
| Set<Object> children= fChildrenMap.get(parent); |
| return children != null && children.contains(child); |
| } |
| |
| |
| private void remove(Object element, boolean refreshViewer) { |
| // precondition here: fResult.getMatchCount(child) <= 0 |
| |
| if (hasChildren(element)) { |
| if (refreshViewer) |
| fTreeViewer.refresh(element); |
| } else { |
| if (!hasMatches(element)) { |
| fChildrenMap.remove(element); |
| Object parent= getParent(element); |
| if (parent != null) { |
| removeFromSiblings(element, parent); |
| remove(parent, refreshViewer); |
| } else { |
| removeFromSiblings(element, fResult); |
| if (refreshViewer) |
| fTreeViewer.refresh(); |
| } |
| } else { |
| if (refreshViewer) { |
| fTreeViewer.refresh(element); |
| } |
| } |
| } |
| } |
| |
| private boolean hasMatches(Object element) { |
| if (element instanceof LineElement) { |
| LineElement lineElement= (LineElement) element; |
| return lineElement.getNumberOfMatches(fResult) > 0; |
| } |
| return fResult.getMatchCount(element) > 0; |
| } |
| |
| |
| private void removeFromSiblings(Object element, Object parent) { |
| Set<Object> siblings= fChildrenMap.get(parent); |
| if (siblings != null) { |
| siblings.remove(element); |
| } |
| } |
| |
| @Override |
| public Object[] getChildren(Object parentElement) { |
| Set<Object> children= fChildrenMap.get(parentElement); |
| if (children == null) |
| return EMPTY_ARR; |
| return children.toArray(); |
| } |
| |
| @Override |
| public boolean hasChildren(Object element) { |
| return getChildren(element).length > 0; |
| } |
| |
| @Override |
| public synchronized void elementsChanged(Object[] updatedElements) { |
| for (int i= 0; i < updatedElements.length; i++) { |
| if (!(updatedElements[i] instanceof LineElement)) { |
| // change events to elements are reported in file search |
| if (fResult.getMatchCount(updatedElements[i]) > 0) |
| insert(updatedElements[i], true); |
| else |
| remove(updatedElements[i], true); |
| } else { |
| // change events to line elements are reported in text search |
| LineElement lineElement= (LineElement) updatedElements[i]; |
| int nMatches= lineElement.getNumberOfMatches(fResult); |
| if (nMatches > 0) { |
| if (hasChild(lineElement.getParent(), lineElement)) { |
| fTreeViewer.update(new Object[] { lineElement, lineElement.getParent() }, null); |
| } else { |
| insert(lineElement, true); |
| } |
| } else { |
| remove(lineElement, true); |
| } |
| } |
| } |
| } |
| |
| @Override |
| public void clear() { |
| initialize(fResult); |
| fTreeViewer.refresh(); |
| } |
| |
| @Override |
| public Object getParent(Object element) { |
| if (element instanceof IProject) |
| return null; |
| if (element instanceof IResource) { |
| IResource resource = (IResource) element; |
| return resource.getParent(); |
| } |
| if (element instanceof LineElement) { |
| return ((LineElement) element).getParent(); |
| } |
| |
| if (element instanceof FileMatch) { |
| FileMatch match= (FileMatch) element; |
| return match.getLineElement(); |
| } |
| return null; |
| } |
| } |