ltk/org.eclipse.statet.ltk.ui/src/org/eclipse/statet/ltk/ui/sourceediting/assist/AssistProposalCollector.java

/*=============================================================================#
 # Copyright (c) 2011, 2019 Stephan Wahlbrink and others.
 # 
 # This program and the accompanying materials are made available under the
 # terms of the Eclipse Public License 2.0 which is available at
 # https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 # which is available at https://www.apache.org/licenses/LICENSE-2.0.
 # 
 # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 # 
 # Contributors:
 #     Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
 #=============================================================================*/

package org.eclipse.statet.ltk.ui.sourceediting.assist;

import java.util.HashMap;
import java.util.Map;


public class AssistProposalCollector {
	
	
	private final Map<IAssistCompletionProposal, IAssistCompletionProposal> proposals;
	
	
	public AssistProposalCollector() {
		this.proposals= new HashMap<>();
	}
	
	
	public void add(final IAssistCompletionProposal proposal) {
		final IAssistCompletionProposal existing= this.proposals.put(proposal, proposal);
		if (existing != null && existing.getRelevance() > proposal.getRelevance()) {
			this.proposals.put(existing, existing);
		}
	}
	
	public int getCount() {
		return this.proposals.size();
	}
	
	public IAssistCompletionProposal[] toArray() {
		return this.proposals.values().toArray(
				new IAssistCompletionProposal[this.proposals.size()] );
	}
	
	
}