source: trunk/src/org/expeditee/agents/SearchTree.java@ 291

package org.expeditee.agents;

import java.util.Collection;
import java.util.HashSet;

import org.expeditee.gui.Frame;
import org.expeditee.gui.FrameGraphics;
import org.expeditee.gui.FrameIO;
import org.expeditee.items.Text;

public class SearchTree extends SearchAgent {
        public SearchTree(String searchText) {
                super(searchText);
        }
        
        @Override
        protected Frame process(Frame frame) {
                try {
                        searchTree(new HashSet<String>(), _startName);
                } catch (Exception e) {
                        e.printStackTrace();
                }
                _results.save();

                String resultFrameName = _results.getName();
                if (_clicked != null)
                        _clicked.setLink(resultFrameName);

                return _results.getFirstFrame();
        }

        public boolean searchTree(Collection<String> visitedFrames, String frameName)
                        throws Exception {
                if (_stop) {
                        return false;
                }
                // Avoid infinate loops
                if (visitedFrames.contains(frameName))
                        return false;
                visitedFrames.add(frameName);

                Frame frameToSearch = FrameIO.LoadFrame(frameName);
                if (frameToSearch == null)
                        return false;

                overwriteMessage("Searching " + frameName);

                for (Text itemToSearch : frameToSearch.getTextItems()) {
                        // Search for the item and add it to the results page if
                        // it is found
                        if (searchItem(itemToSearch, _pattern,
                                        _replacementString)) {
                                // Add a linked item to the results frame
                                _results.addText(frameName, null, frameName, null, false);
                        }
                        if (!itemToSearch.isAnnotation()) {
                                String link = itemToSearch.getAbsoluteLink();
                                if (link != null) {
                                        searchTree(visitedFrames, link);
                                }
                        }
                }
                FrameGraphics.requestRefresh(true);
                FrameIO.SaveFrame(frameToSearch, false);
                return true;
        }
}