/* ###
 * IP: GHIDRA
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package docking.widgets.tree;

import java.util.ArrayList;
import java.util.List;

import javax.swing.Icon;

import docking.widgets.filter.*;
import docking.widgets.tree.support.GTreeFilter;
import ghidra.util.FilterTransformer;

public class GTreeFilterFactory {

	private FilterOptions filterOptions;

	public GTreeFilterFactory() {
		this(new FilterOptions());
	}

	public GTreeFilterFactory(FilterOptions filterOptions) {
		this.filterOptions = filterOptions;
	}

	public FilterOptions getFilterOptions() {
		return filterOptions;
	}

	public GTreeFilter getTreeFilter(String text, FilterTransformer<GTreeNode> transformer) {
		GTreeFilter treeFilter = getBaseFilter(text, transformer);

		if (filterOptions.isInverted() && treeFilter != null) {
			treeFilter = new InvertedTreeFilter(treeFilter);
		}
		return treeFilter;
	}

	private GTreeFilter getBaseFilter(String text, FilterTransformer<GTreeNode> transformer) {

		if (filterOptions.isMultiterm() && text.trim().length() > 0) {
			return getMultiWordTableFilter(text, transformer);

		}
		TextFilter textFilter = filterOptions.getTextFilterFactory().getTextFilter(text);
		if (textFilter != null) {
			return new TreeTextFilter(textFilter, transformer);
		}
		return null;
	}

	private GTreeFilter getMultiWordTableFilter(String text,
			FilterTransformer<GTreeNode> transformer) {

		List<TextFilter> filters = new ArrayList<>();
		TermSplitter splitter = filterOptions.getTermSplitter();
		for (String term : splitter.split(text)) {
			TextFilter textFilter = filterOptions.getTextFilterFactory().getTextFilter(term);
			if (textFilter != null) {
				filters.add(textFilter);
			}
		}
		return new MultiTextFilterTreeFilter(filters, transformer,
			filterOptions.getMultitermEvaluationMode());
	}

	public Icon getFilterStateIcon() {
		return filterOptions.getFilterStateIcon();
	}
}
