core/org.eclipse.cdt.ui/src/org/eclipse/cdt/internal/ui/text/correction/CCorrectionProcessor.java - cdt/org.eclipse.cdt - Git at Google

 /*******************************************************************************
  * Copyright (c) 2000, 2012 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
  * 	   Sergey Prigogin (Google)
  *     Anton Leherbauer (Wind River Systems)
  *******************************************************************************/

 package org.eclipse.cdt.internal.ui.text.correction;

 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;

 import org.eclipse.core.resources.IMarker;
 import org.eclipse.core.runtime.IConfigurationElement;
 import org.eclipse.core.runtime.ISafeRunnable;
 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.core.runtime.MultiStatus;
 import org.eclipse.core.runtime.Platform;
 import org.eclipse.core.runtime.SafeRunner;
 import org.eclipse.core.runtime.Status;
 import org.eclipse.jface.text.Position;
 import org.eclipse.jface.text.contentassist.ContentAssistEvent;
 import org.eclipse.jface.text.contentassist.ICompletionListener;
 import org.eclipse.jface.text.contentassist.ICompletionProposal;
 import org.eclipse.jface.text.quickassist.IQuickAssistInvocationContext;
 import org.eclipse.jface.text.source.Annotation;
 import org.eclipse.jface.text.source.IAnnotationModel;
 import org.eclipse.jface.text.source.ISourceViewer;
 import org.eclipse.ltk.core.refactoring.NullChange;
 import org.eclipse.ui.IEditorPart;
 import org.eclipse.ui.IMarkerHelpRegistry;
 import org.eclipse.ui.IMarkerResolution;
 import org.eclipse.ui.ide.IDE;
 import org.eclipse.ui.texteditor.SimpleMarkerAnnotation;

 import org.eclipse.cdt.core.model.ITranslationUnit;
 import org.eclipse.cdt.ui.CUIPlugin;
 import org.eclipse.cdt.ui.text.ICCompletionProposal;
 import org.eclipse.cdt.ui.text.IInvocationContext;
 import org.eclipse.cdt.ui.text.IProblemLocation;
 import org.eclipse.cdt.ui.text.IQuickAssistProcessor;
 import org.eclipse.cdt.ui.text.IQuickFixProcessor;

 import org.eclipse.cdt.internal.ui.editor.ICAnnotation;
 import org.eclipse.cdt.internal.ui.text.contentassist.CCompletionProposalComparator;
 import org.eclipse.cdt.internal.ui.text.correction.proposals.ChangeCorrectionProposal;

 public class CCorrectionProcessor implements org.eclipse.jface.text.quickassist.IQuickAssistProcessor {
 	private static final String QUICKFIX_PROCESSOR_CONTRIBUTION_ID= "quickFixProcessors"; //$NON-NLS-1$
 	private static final String QUICKASSIST_PROCESSOR_CONTRIBUTION_ID= "quickAssistProcessors"; //$NON-NLS-1$

 	private static ContributedProcessorDescriptor[] fgContributedAssistProcessors= null;
 	private static ContributedProcessorDescriptor[] fgContributedCorrectionProcessors= null;

 	private static ContributedProcessorDescriptor[] getProcessorDescriptors(String contributionId, boolean testMarkerTypes) {
 		IConfigurationElement[] elements= Platform.getExtensionRegistry().getConfigurationElementsFor(CUIPlugin.PLUGIN_ID, contributionId);
 		ArrayList<ContributedProcessorDescriptor> res= new ArrayList<ContributedProcessorDescriptor>(elements.length);

 		for (int i= 0; i < elements.length; i++) {
 			ContributedProcessorDescriptor desc= new ContributedProcessorDescriptor(elements[i], testMarkerTypes);
 			IStatus status= desc.checkSyntax();
 			if (status.isOK()) {
 				res.add(desc);
 			} else {
 				CUIPlugin.log(status);
 			}
 		}
 		return res.toArray(new ContributedProcessorDescriptor[res.size()]);
 	}

 	private static ContributedProcessorDescriptor[] getCorrectionProcessors() {
 		if (fgContributedCorrectionProcessors == null) {
 			fgContributedCorrectionProcessors= getProcessorDescriptors(QUICKFIX_PROCESSOR_CONTRIBUTION_ID, true);
 		}
 		return fgContributedCorrectionProcessors;
 	}

 	private static ContributedProcessorDescriptor[] getAssistProcessors() {
 		if (fgContributedAssistProcessors == null) {
 			fgContributedAssistProcessors= getProcessorDescriptors(QUICKASSIST_PROCESSOR_CONTRIBUTION_ID, false);
 		}
 		return fgContributedAssistProcessors;
 	}

 	public static boolean hasCorrections(ITranslationUnit tu, int problemId, String markerType) {
 		ContributedProcessorDescriptor[] processors= getCorrectionProcessors();
 		SafeHasCorrections collector= new SafeHasCorrections(tu, problemId);
 		for (int i= 0; i < processors.length; i++) {
 			if (processors[i].canHandleMarkerType(markerType)) {
 				collector.process(processors[i]);
 				if (collector.hasCorrections()) {
 					return true;
 				}
 			}
 		}
 		return false;
 	}

 	public static boolean isQuickFixableType(Annotation annotation) {
 		return (annotation instanceof ICAnnotation || annotation instanceof SimpleMarkerAnnotation) && !annotation.isMarkedDeleted();
 	}

 	public static boolean hasCorrections(Annotation annotation) {
 		if (annotation instanceof ICAnnotation) {
 			ICAnnotation cAnnotation= (ICAnnotation) annotation;
 			int problemId= cAnnotation.getId();
 			if (problemId != -1) {
 				ITranslationUnit tu= cAnnotation.getTranslationUnit();
 				if (tu != null) {
 					return hasCorrections(tu, problemId, cAnnotation.getMarkerType());
 				}
 			}
 		}
 		if (annotation instanceof SimpleMarkerAnnotation) {
 			return hasCorrections(((SimpleMarkerAnnotation) annotation).getMarker());
 		}
 		return false;
 	}

 	private static boolean hasCorrections(IMarker marker) {
 		if (marker == null || !marker.exists())
 			return false;

 		IMarkerHelpRegistry registry= IDE.getMarkerHelpRegistry();
 		return registry != null && registry.hasResolutions(marker);
 	}

 	public static boolean hasAssists(CorrectionContext context) {
 		ContributedProcessorDescriptor[] processors= getAssistProcessors();
 		SafeHasAssist collector= new SafeHasAssist(context);

 		for (int i= 0; i < processors.length; i++) {
 			collector.process(processors[i]);
 			if (collector.hasAssists()) {
 				return true;
 			}
 		}
 		return false;
 	}

 	private CCorrectionAssistant fAssistant;
 	private String fErrorMessage;

 	/*
 	 * Constructor for CCorrectionProcessor.
 	 */
 	public CCorrectionProcessor(CCorrectionAssistant assistant) {
 		fAssistant= assistant;
 		fAssistant.addCompletionListener(new ICompletionListener() {

 			@Override
 			public void assistSessionEnded(ContentAssistEvent event) {
 				fAssistant.setStatusLineVisible(false);
 			}

 			@Override
 			public void assistSessionStarted(ContentAssistEvent event) {
 				fAssistant.setStatusLineVisible(true);
 			}

 			@Override
 			public void selectionChanged(ICompletionProposal proposal, boolean smartToggle) {
 				if (proposal instanceof IStatusLineProposal) {
 					IStatusLineProposal statusLineProposal= (IStatusLineProposal)proposal;
 					String message= statusLineProposal.getStatusMessage();
 					if (message != null) {
 						fAssistant.setStatusMessage(message);
 					} else {
 						fAssistant.setStatusMessage(""); //$NON-NLS-1$
 					}
 				} else {
 					fAssistant.setStatusMessage(""); //$NON-NLS-1$
 				}
 			}
 		});
 	}

 	/*
 	 * @see IContentAssistProcessor#computeCompletionProposals(ITextViewer, int)
 	 */
 	@Override
 	public ICompletionProposal[] computeQuickAssistProposals(IQuickAssistInvocationContext quickAssistContext) {
 		ISourceViewer viewer= quickAssistContext.getSourceViewer();
 		int documentOffset= quickAssistContext.getOffset();

 		IEditorPart part= fAssistant.getEditor();

 		ITranslationUnit tu= CUIPlugin.getDefault().getWorkingCopyManager().getWorkingCopy(part.getEditorInput());
 		IAnnotationModel model= CUIPlugin.getDefault().getDocumentProvider().getAnnotationModel(part.getEditorInput());

 		int length= viewer != null ? viewer.getSelectedRange().y : 0;
 		CorrectionContext context= new CorrectionContext(tu, viewer, documentOffset, length);

 		Annotation[] annotations= fAssistant.getAnnotationsAtOffset();

 		fErrorMessage= null;

 		ICCompletionProposal[] res= null;
 		if (model != null && annotations != null) {
 			ArrayList<ICCompletionProposal> proposals= new ArrayList<ICCompletionProposal>(10);
 			IStatus status= collectProposals(context, model, annotations, true, !fAssistant.isUpdatedOffset(), proposals);
 			res= proposals.toArray(new ICCompletionProposal[proposals.size()]);
 			if (!status.isOK()) {
 				fErrorMessage= status.getMessage();
 				CUIPlugin.log(status);
 			}
 		}

 		if (res == null || res.length == 0) {
 			return new ICCompletionProposal[]
 					{ new ChangeCorrectionProposal(CorrectionMessages.NoCorrectionProposal_description, new NullChange(""), 0, null) }; //$NON-NLS-1$
 		}
 		if (res.length > 1) {
 			Arrays.sort(res, new CCompletionProposalComparator());
 		}
 		return res;
 	}

 	public static IStatus collectProposals(CorrectionContext context, IAnnotationModel model, Annotation[] annotations, boolean addQuickFixes, boolean addQuickAssists, Collection<ICCompletionProposal> proposals) {
 		ArrayList<ProblemLocation> problems= new ArrayList<ProblemLocation>();

 		// collect problem locations and corrections from marker annotations
 		for (int i= 0; i < annotations.length; i++) {
 			Annotation curr= annotations[i];
 			if (curr instanceof ICAnnotation) {
 				ProblemLocation problemLocation= getProblemLocation((ICAnnotation) curr, model);
 				if (problemLocation != null) {
 					problems.add(problemLocation);
 				}
 			}
 			if (addQuickFixes && curr instanceof SimpleMarkerAnnotation) {
 				// collect marker proposals
 				collectMarkerProposals((SimpleMarkerAnnotation) curr, proposals);
 			}
 		}
 		MultiStatus resStatus= null;

 		IProblemLocation[] problemLocations= problems.toArray(new IProblemLocation[problems.size()]);
 		if (addQuickFixes) {
 			IStatus status= collectCorrections(context, problemLocations, proposals);
 			if (!status.isOK()) {
 				resStatus= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_quickfix_message, null);
 				resStatus.add(status);
 			}
 		}
 		if (addQuickAssists) {
 			IStatus status= collectAssists(context, problemLocations, proposals);
 			if (!status.isOK()) {
 				if (resStatus == null) {
 					resStatus= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_quickassist_message, null);
 				}
 				resStatus.add(status);
 			}
 		}
 		if (resStatus != null) {
 			return resStatus;
 		}
 		return Status.OK_STATUS;
 	}

 	private static ProblemLocation getProblemLocation(ICAnnotation cAnnotation, IAnnotationModel model) {
 		int problemId= cAnnotation.getId();
 		if (problemId != -1) {
 			Position pos= model.getPosition((Annotation) cAnnotation);
 			if (pos != null) {
 				return new ProblemLocation(pos.getOffset(), pos.getLength(), cAnnotation); // java problems all handled by the quick assist processors
 			}
 		}
 		return null;
 	}

 	private static void collectMarkerProposals(SimpleMarkerAnnotation annotation, Collection<ICCompletionProposal> proposals) {
 		IMarker marker= annotation.getMarker();
 		IMarkerResolution[] res= IDE.getMarkerHelpRegistry().getResolutions(marker);
 		if (res.length > 0) {
 			for (int i= 0; i < res.length; i++) {
 				proposals.add(new MarkerResolutionProposal(res[i], marker));
 			}
 		}
 	}

 	private static abstract class SafeCorrectionProcessorAccess implements ISafeRunnable {
 		private MultiStatus fMulti= null;
 		private ContributedProcessorDescriptor fDescriptor;

 		public void process(ContributedProcessorDescriptor[] desc) {
 			for (int i= 0; i < desc.length; i++) {
 				fDescriptor= desc[i];
 				SafeRunner.run(this);
 			}
 		}

 		public void process(ContributedProcessorDescriptor desc) {
 			fDescriptor= desc;
 			SafeRunner.run(this);
 		}

 		@Override
 		public void run() throws Exception {
 			safeRun(fDescriptor);
 		}

 		protected abstract void safeRun(ContributedProcessorDescriptor processor) throws Exception;

 		@Override
 		public void handleException(Throwable exception) {
 			if (fMulti == null) {
 				fMulti= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.OK, CorrectionMessages.CCorrectionProcessor_error_status, null);
 			}
 			fMulti.merge(new Status(IStatus.ERROR, CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_status, exception));
 		}

 		public IStatus getStatus() {
 			if (fMulti == null) {
 				return Status.OK_STATUS;
 			}
 			return fMulti;
 		}
 	}

 	private static class SafeCorrectionCollector extends SafeCorrectionProcessorAccess {
 		private final CorrectionContext fContext;
 		private final Collection<ICCompletionProposal> fProposals;
 		private IProblemLocation[] fLocations;

 		public SafeCorrectionCollector(CorrectionContext context, Collection<ICCompletionProposal> proposals) {
 			fContext= context;
 			fProposals= proposals;
 		}

 		public void setProblemLocations(IProblemLocation[] locations) {
 			fLocations= locations;
 		}

 		@Override
 		public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
 			IQuickFixProcessor curr= (IQuickFixProcessor) desc.getProcessor(fContext.getTranslationUnit());
 			if (curr != null) {
 				ICCompletionProposal[] res= curr.getCorrections(fContext, fLocations);
 				if (res != null) {
 					for (int k= 0; k < res.length; k++) {
 						fProposals.add(res[k]);
 					}
 				}
 			}
 		}
 	}

 	private static class SafeAssistCollector extends SafeCorrectionProcessorAccess {
 		private final IInvocationContext fContext;
 		private final IProblemLocation[] fLocations;
 		private final Collection<ICCompletionProposal> fProposals;

 		public SafeAssistCollector(IInvocationContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
 			fContext= context;
 			fLocations= locations;
 			fProposals= proposals;
 		}

 		@Override
 		public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
 			IQuickAssistProcessor curr= (IQuickAssistProcessor) desc.getProcessor(fContext.getTranslationUnit());
 			if (curr != null) {
 				ICCompletionProposal[] res= curr.getAssists(fContext, fLocations);
 				if (res != null) {
 					for (int k= 0; k < res.length; k++) {
 						fProposals.add(res[k]);
 					}
 				}
 			}
 		}
 	}

 	private static class SafeHasAssist extends SafeCorrectionProcessorAccess {
 		private final CorrectionContext fContext;
 		private boolean fHasAssists;

 		public SafeHasAssist(CorrectionContext context) {
 			fContext= context;
 			fHasAssists= false;
 		}

 		public boolean hasAssists() {
 			return fHasAssists;
 		}

 		@Override
 		public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
 			IQuickAssistProcessor processor= (IQuickAssistProcessor) desc.getProcessor(fContext.getTranslationUnit());
 			if (processor != null && processor.hasAssists(fContext)) {
 				fHasAssists= true;
 			}
 		}
 	}

 	private static class SafeHasCorrections extends SafeCorrectionProcessorAccess {
 		private final ITranslationUnit fCu;
 		private final int fProblemId;
 		private boolean fHasCorrections;

 		public SafeHasCorrections(ITranslationUnit tu, int problemId) {
 			fCu= tu;
 			fProblemId= problemId;
 			fHasCorrections= false;
 		}

 		public boolean hasCorrections() {
 			return fHasCorrections;
 		}

 		@Override
 		public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
 			IQuickFixProcessor processor= (IQuickFixProcessor) desc.getProcessor(fCu);
 			if (processor != null && processor.hasCorrections(fCu, fProblemId)) {
 				fHasCorrections= true;
 			}
 		}
 	}

 	public static IStatus collectCorrections(CorrectionContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
 		ContributedProcessorDescriptor[] processors= getCorrectionProcessors();
 		SafeCorrectionCollector collector= new SafeCorrectionCollector(context, proposals);
 		for (int i= 0; i < processors.length; i++) {
 			ContributedProcessorDescriptor curr= processors[i];
 			IProblemLocation[] handled= getHandledProblems(locations, curr);
 			if (handled != null) {
 				collector.setProblemLocations(handled);
 				collector.process(curr);
 			}
 		}
 		return collector.getStatus();
 	}

 	private static IProblemLocation[] getHandledProblems(IProblemLocation[] locations, ContributedProcessorDescriptor processor) {
 		// implementation tries to avoid creating a new array
 		boolean allHandled= true;
 		ArrayList<IProblemLocation> res= null;
 		for (int i= 0; i < locations.length; i++) {
 			IProblemLocation curr= locations[i];
 			if (processor.canHandleMarkerType(curr.getMarkerType())) {
 				if (!allHandled) { // first handled problem
 					if (res == null) {
 						res= new ArrayList<IProblemLocation>(locations.length - i);
 					}
 					res.add(curr);
 				}
 			} else if (allHandled) {
 				if (i > 0) { // first non handled problem
 					res= new ArrayList<IProblemLocation>(locations.length - i);
 					for (int k= 0; k < i; k++) {
 						res.add(locations[k]);
 					}
 				}
 				allHandled= false;
 			}
 		}
 		if (allHandled) {
 			return locations;
 		}
 		if (res == null) {
 			return null;
 		}
 		return res.toArray(new IProblemLocation[res.size()]);
 	}

 	public static IStatus collectAssists(CorrectionContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
 		ContributedProcessorDescriptor[] processors= getAssistProcessors();
 		SafeAssistCollector collector= new SafeAssistCollector(context, locations, proposals);
 		collector.process(processors);

 		return collector.getStatus();
 	}

 	/*
 	 * @see IContentAssistProcessor#getErrorMessage()
 	 */
 	@Override
 	public String getErrorMessage() {
 		return fErrorMessage;
 	}

 	/*
 	 * @see org.eclipse.jface.text.quickassist.IQuickAssistProcessor#canFix(org.eclipse.jface.text.source.Annotation)
 	 */
 	@Override
 	public boolean canFix(Annotation annotation) {
 		return hasCorrections(annotation);
 	}

 	/*
 	 * @see org.eclipse.jface.text.quickassist.IQuickAssistProcessor#canAssist(org.eclipse.jface.text.quickassist.IQuickAssistInvocationContext)
 	 */
 	@Override
 	public boolean canAssist(IQuickAssistInvocationContext invocationContext) {
 		if (invocationContext instanceof CorrectionContext)
 			return hasAssists((CorrectionContext) invocationContext);
 		return false;
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2000, 2012 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
	* Sergey Prigogin (Google)
	* Anton Leherbauer (Wind River Systems)
	*******************************************************************************/

	package org.eclipse.cdt.internal.ui.text.correction;

	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.Collection;

	import org.eclipse.core.resources.IMarker;
	import org.eclipse.core.runtime.IConfigurationElement;
	import org.eclipse.core.runtime.ISafeRunnable;
	import org.eclipse.core.runtime.IStatus;
	import org.eclipse.core.runtime.MultiStatus;
	import org.eclipse.core.runtime.Platform;
	import org.eclipse.core.runtime.SafeRunner;
	import org.eclipse.core.runtime.Status;
	import org.eclipse.jface.text.Position;
	import org.eclipse.jface.text.contentassist.ContentAssistEvent;
	import org.eclipse.jface.text.contentassist.ICompletionListener;
	import org.eclipse.jface.text.contentassist.ICompletionProposal;
	import org.eclipse.jface.text.quickassist.IQuickAssistInvocationContext;
	import org.eclipse.jface.text.source.Annotation;
	import org.eclipse.jface.text.source.IAnnotationModel;
	import org.eclipse.jface.text.source.ISourceViewer;
	import org.eclipse.ltk.core.refactoring.NullChange;
	import org.eclipse.ui.IEditorPart;
	import org.eclipse.ui.IMarkerHelpRegistry;
	import org.eclipse.ui.IMarkerResolution;
	import org.eclipse.ui.ide.IDE;
	import org.eclipse.ui.texteditor.SimpleMarkerAnnotation;

	import org.eclipse.cdt.core.model.ITranslationUnit;
	import org.eclipse.cdt.ui.CUIPlugin;
	import org.eclipse.cdt.ui.text.ICCompletionProposal;
	import org.eclipse.cdt.ui.text.IInvocationContext;
	import org.eclipse.cdt.ui.text.IProblemLocation;
	import org.eclipse.cdt.ui.text.IQuickAssistProcessor;
	import org.eclipse.cdt.ui.text.IQuickFixProcessor;

	import org.eclipse.cdt.internal.ui.editor.ICAnnotation;
	import org.eclipse.cdt.internal.ui.text.contentassist.CCompletionProposalComparator;
	import org.eclipse.cdt.internal.ui.text.correction.proposals.ChangeCorrectionProposal;

	public class CCorrectionProcessor implements org.eclipse.jface.text.quickassist.IQuickAssistProcessor {
	private static final String QUICKFIX_PROCESSOR_CONTRIBUTION_ID= "quickFixProcessors"; //$NON-NLS-1$
	private static final String QUICKASSIST_PROCESSOR_CONTRIBUTION_ID= "quickAssistProcessors"; //$NON-NLS-1$

	private static ContributedProcessorDescriptor[] fgContributedAssistProcessors= null;
	private static ContributedProcessorDescriptor[] fgContributedCorrectionProcessors= null;

	private static ContributedProcessorDescriptor[] getProcessorDescriptors(String contributionId, boolean testMarkerTypes) {
	IConfigurationElement[] elements= Platform.getExtensionRegistry().getConfigurationElementsFor(CUIPlugin.PLUGIN_ID, contributionId);
	ArrayList<ContributedProcessorDescriptor> res= new ArrayList<ContributedProcessorDescriptor>(elements.length);

	for (int i= 0; i < elements.length; i++) {
	ContributedProcessorDescriptor desc= new ContributedProcessorDescriptor(elements[i], testMarkerTypes);
	IStatus status= desc.checkSyntax();
	if (status.isOK()) {
	res.add(desc);
	} else {
	CUIPlugin.log(status);
	}
	}
	return res.toArray(new ContributedProcessorDescriptor[res.size()]);
	}

	private static ContributedProcessorDescriptor[] getCorrectionProcessors() {
	if (fgContributedCorrectionProcessors == null) {
	fgContributedCorrectionProcessors= getProcessorDescriptors(QUICKFIX_PROCESSOR_CONTRIBUTION_ID, true);
	}
	return fgContributedCorrectionProcessors;
	}

	private static ContributedProcessorDescriptor[] getAssistProcessors() {
	if (fgContributedAssistProcessors == null) {
	fgContributedAssistProcessors= getProcessorDescriptors(QUICKASSIST_PROCESSOR_CONTRIBUTION_ID, false);
	}
	return fgContributedAssistProcessors;
	}

	public static boolean hasCorrections(ITranslationUnit tu, int problemId, String markerType) {
	ContributedProcessorDescriptor[] processors= getCorrectionProcessors();
	SafeHasCorrections collector= new SafeHasCorrections(tu, problemId);
	for (int i= 0; i < processors.length; i++) {
	if (processors[i].canHandleMarkerType(markerType)) {
	collector.process(processors[i]);
	if (collector.hasCorrections()) {
	return true;
	}
	}
	}
	return false;
	}

	public static boolean isQuickFixableType(Annotation annotation) {
	return (annotation instanceof ICAnnotation \|\| annotation instanceof SimpleMarkerAnnotation) && !annotation.isMarkedDeleted();
	}

	public static boolean hasCorrections(Annotation annotation) {
	if (annotation instanceof ICAnnotation) {
	ICAnnotation cAnnotation= (ICAnnotation) annotation;
	int problemId= cAnnotation.getId();
	if (problemId != -1) {
	ITranslationUnit tu= cAnnotation.getTranslationUnit();
	if (tu != null) {
	return hasCorrections(tu, problemId, cAnnotation.getMarkerType());
	}
	}
	}
	if (annotation instanceof SimpleMarkerAnnotation) {
	return hasCorrections(((SimpleMarkerAnnotation) annotation).getMarker());
	}
	return false;
	}

	private static boolean hasCorrections(IMarker marker) {
	if (marker == null \|\| !marker.exists())
	return false;

	IMarkerHelpRegistry registry= IDE.getMarkerHelpRegistry();
	return registry != null && registry.hasResolutions(marker);
	}

	public static boolean hasAssists(CorrectionContext context) {
	ContributedProcessorDescriptor[] processors= getAssistProcessors();
	SafeHasAssist collector= new SafeHasAssist(context);

	for (int i= 0; i < processors.length; i++) {
	collector.process(processors[i]);
	if (collector.hasAssists()) {
	return true;
	}
	}
	return false;
	}

	private CCorrectionAssistant fAssistant;
	private String fErrorMessage;

	/*
	* Constructor for CCorrectionProcessor.
	*/
	public CCorrectionProcessor(CCorrectionAssistant assistant) {
	fAssistant= assistant;
	fAssistant.addCompletionListener(new ICompletionListener() {

	@Override
	public void assistSessionEnded(ContentAssistEvent event) {
	fAssistant.setStatusLineVisible(false);
	}

	@Override
	public void assistSessionStarted(ContentAssistEvent event) {
	fAssistant.setStatusLineVisible(true);
	}

	@Override
	public void selectionChanged(ICompletionProposal proposal, boolean smartToggle) {
	if (proposal instanceof IStatusLineProposal) {
	IStatusLineProposal statusLineProposal= (IStatusLineProposal)proposal;
	String message= statusLineProposal.getStatusMessage();
	if (message != null) {
	fAssistant.setStatusMessage(message);
	} else {
	fAssistant.setStatusMessage(""); //$NON-NLS-1$
	}
	} else {
	fAssistant.setStatusMessage(""); //$NON-NLS-1$
	}
	}
	});
	}

	/*
	* @see IContentAssistProcessor#computeCompletionProposals(ITextViewer, int)
	*/
	@Override
	public ICompletionProposal[] computeQuickAssistProposals(IQuickAssistInvocationContext quickAssistContext) {
	ISourceViewer viewer= quickAssistContext.getSourceViewer();
	int documentOffset= quickAssistContext.getOffset();

	IEditorPart part= fAssistant.getEditor();

	ITranslationUnit tu= CUIPlugin.getDefault().getWorkingCopyManager().getWorkingCopy(part.getEditorInput());
	IAnnotationModel model= CUIPlugin.getDefault().getDocumentProvider().getAnnotationModel(part.getEditorInput());

	int length= viewer != null ? viewer.getSelectedRange().y : 0;
	CorrectionContext context= new CorrectionContext(tu, viewer, documentOffset, length);

	Annotation[] annotations= fAssistant.getAnnotationsAtOffset();

	fErrorMessage= null;

	ICCompletionProposal[] res= null;
	if (model != null && annotations != null) {
	ArrayList<ICCompletionProposal> proposals= new ArrayList<ICCompletionProposal>(10);
	IStatus status= collectProposals(context, model, annotations, true, !fAssistant.isUpdatedOffset(), proposals);
	res= proposals.toArray(new ICCompletionProposal[proposals.size()]);
	if (!status.isOK()) {
	fErrorMessage= status.getMessage();
	CUIPlugin.log(status);
	}
	}

	if (res == null \|\| res.length == 0) {
	return new ICCompletionProposal[]
	{ new ChangeCorrectionProposal(CorrectionMessages.NoCorrectionProposal_description, new NullChange(""), 0, null) }; //$NON-NLS-1$
	}
	if (res.length > 1) {
	Arrays.sort(res, new CCompletionProposalComparator());
	}
	return res;
	}

	public static IStatus collectProposals(CorrectionContext context, IAnnotationModel model, Annotation[] annotations, boolean addQuickFixes, boolean addQuickAssists, Collection<ICCompletionProposal> proposals) {
	ArrayList<ProblemLocation> problems= new ArrayList<ProblemLocation>();

	// collect problem locations and corrections from marker annotations
	for (int i= 0; i < annotations.length; i++) {
	Annotation curr= annotations[i];
	if (curr instanceof ICAnnotation) {
	ProblemLocation problemLocation= getProblemLocation((ICAnnotation) curr, model);
	if (problemLocation != null) {
	problems.add(problemLocation);
	}
	}
	if (addQuickFixes && curr instanceof SimpleMarkerAnnotation) {
	// collect marker proposals
	collectMarkerProposals((SimpleMarkerAnnotation) curr, proposals);
	}
	}
	MultiStatus resStatus= null;

	IProblemLocation[] problemLocations= problems.toArray(new IProblemLocation[problems.size()]);
	if (addQuickFixes) {
	IStatus status= collectCorrections(context, problemLocations, proposals);
	if (!status.isOK()) {
	resStatus= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_quickfix_message, null);
	resStatus.add(status);
	}
	}
	if (addQuickAssists) {
	IStatus status= collectAssists(context, problemLocations, proposals);
	if (!status.isOK()) {
	if (resStatus == null) {
	resStatus= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_quickassist_message, null);
	}
	resStatus.add(status);
	}
	}
	if (resStatus != null) {
	return resStatus;
	}
	return Status.OK_STATUS;
	}

	private static ProblemLocation getProblemLocation(ICAnnotation cAnnotation, IAnnotationModel model) {
	int problemId= cAnnotation.getId();
	if (problemId != -1) {
	Position pos= model.getPosition((Annotation) cAnnotation);
	if (pos != null) {
	return new ProblemLocation(pos.getOffset(), pos.getLength(), cAnnotation); // java problems all handled by the quick assist processors
	}
	}
	return null;
	}

	private static void collectMarkerProposals(SimpleMarkerAnnotation annotation, Collection<ICCompletionProposal> proposals) {
	IMarker marker= annotation.getMarker();
	IMarkerResolution[] res= IDE.getMarkerHelpRegistry().getResolutions(marker);
	if (res.length > 0) {
	for (int i= 0; i < res.length; i++) {
	proposals.add(new MarkerResolutionProposal(res[i], marker));
	}
	}
	}

	private static abstract class SafeCorrectionProcessorAccess implements ISafeRunnable {
	private MultiStatus fMulti= null;
	private ContributedProcessorDescriptor fDescriptor;

	public void process(ContributedProcessorDescriptor[] desc) {
	for (int i= 0; i < desc.length; i++) {
	fDescriptor= desc[i];
	SafeRunner.run(this);
	}
	}

	public void process(ContributedProcessorDescriptor desc) {
	fDescriptor= desc;
	SafeRunner.run(this);
	}

	@Override
	public void run() throws Exception {
	safeRun(fDescriptor);
	}

	protected abstract void safeRun(ContributedProcessorDescriptor processor) throws Exception;

	@Override
	public void handleException(Throwable exception) {
	if (fMulti == null) {
	fMulti= new MultiStatus(CUIPlugin.PLUGIN_ID, IStatus.OK, CorrectionMessages.CCorrectionProcessor_error_status, null);
	}
	fMulti.merge(new Status(IStatus.ERROR, CUIPlugin.PLUGIN_ID, IStatus.ERROR, CorrectionMessages.CCorrectionProcessor_error_status, exception));
	}

	public IStatus getStatus() {
	if (fMulti == null) {
	return Status.OK_STATUS;
	}
	return fMulti;
	}
	}

	private static class SafeCorrectionCollector extends SafeCorrectionProcessorAccess {
	private final CorrectionContext fContext;
	private final Collection<ICCompletionProposal> fProposals;
	private IProblemLocation[] fLocations;

	public SafeCorrectionCollector(CorrectionContext context, Collection<ICCompletionProposal> proposals) {
	fContext= context;
	fProposals= proposals;
	}

	public void setProblemLocations(IProblemLocation[] locations) {
	fLocations= locations;
	}

	@Override
	public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
	IQuickFixProcessor curr= (IQuickFixProcessor) desc.getProcessor(fContext.getTranslationUnit());
	if (curr != null) {
	ICCompletionProposal[] res= curr.getCorrections(fContext, fLocations);
	if (res != null) {
	for (int k= 0; k < res.length; k++) {
	fProposals.add(res[k]);
	}
	}
	}
	}
	}

	private static class SafeAssistCollector extends SafeCorrectionProcessorAccess {
	private final IInvocationContext fContext;
	private final IProblemLocation[] fLocations;
	private final Collection<ICCompletionProposal> fProposals;

	public SafeAssistCollector(IInvocationContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
	fContext= context;
	fLocations= locations;
	fProposals= proposals;
	}

	@Override
	public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
	IQuickAssistProcessor curr= (IQuickAssistProcessor) desc.getProcessor(fContext.getTranslationUnit());
	if (curr != null) {
	ICCompletionProposal[] res= curr.getAssists(fContext, fLocations);
	if (res != null) {
	for (int k= 0; k < res.length; k++) {
	fProposals.add(res[k]);
	}
	}
	}
	}
	}

	private static class SafeHasAssist extends SafeCorrectionProcessorAccess {
	private final CorrectionContext fContext;
	private boolean fHasAssists;

	public SafeHasAssist(CorrectionContext context) {
	fContext= context;
	fHasAssists= false;
	}

	public boolean hasAssists() {
	return fHasAssists;
	}

	@Override
	public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
	IQuickAssistProcessor processor= (IQuickAssistProcessor) desc.getProcessor(fContext.getTranslationUnit());
	if (processor != null && processor.hasAssists(fContext)) {
	fHasAssists= true;
	}
	}
	}

	private static class SafeHasCorrections extends SafeCorrectionProcessorAccess {
	private final ITranslationUnit fCu;
	private final int fProblemId;
	private boolean fHasCorrections;

	public SafeHasCorrections(ITranslationUnit tu, int problemId) {
	fCu= tu;
	fProblemId= problemId;
	fHasCorrections= false;
	}

	public boolean hasCorrections() {
	return fHasCorrections;
	}

	@Override
	public void safeRun(ContributedProcessorDescriptor desc) throws Exception {
	IQuickFixProcessor processor= (IQuickFixProcessor) desc.getProcessor(fCu);
	if (processor != null && processor.hasCorrections(fCu, fProblemId)) {
	fHasCorrections= true;
	}
	}
	}

	public static IStatus collectCorrections(CorrectionContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
	ContributedProcessorDescriptor[] processors= getCorrectionProcessors();
	SafeCorrectionCollector collector= new SafeCorrectionCollector(context, proposals);
	for (int i= 0; i < processors.length; i++) {
	ContributedProcessorDescriptor curr= processors[i];
	IProblemLocation[] handled= getHandledProblems(locations, curr);
	if (handled != null) {
	collector.setProblemLocations(handled);
	collector.process(curr);
	}
	}
	return collector.getStatus();
	}

	private static IProblemLocation[] getHandledProblems(IProblemLocation[] locations, ContributedProcessorDescriptor processor) {
	// implementation tries to avoid creating a new array
	boolean allHandled= true;
	ArrayList<IProblemLocation> res= null;
	for (int i= 0; i < locations.length; i++) {
	IProblemLocation curr= locations[i];
	if (processor.canHandleMarkerType(curr.getMarkerType())) {
	if (!allHandled) { // first handled problem
	if (res == null) {
	res= new ArrayList<IProblemLocation>(locations.length - i);
	}
	res.add(curr);
	}
	} else if (allHandled) {
	if (i > 0) { // first non handled problem
	res= new ArrayList<IProblemLocation>(locations.length - i);
	for (int k= 0; k < i; k++) {
	res.add(locations[k]);
	}
	}
	allHandled= false;
	}
	}
	if (allHandled) {
	return locations;
	}
	if (res == null) {
	return null;
	}
	return res.toArray(new IProblemLocation[res.size()]);
	}

	public static IStatus collectAssists(CorrectionContext context, IProblemLocation[] locations, Collection<ICCompletionProposal> proposals) {
	ContributedProcessorDescriptor[] processors= getAssistProcessors();
	SafeAssistCollector collector= new SafeAssistCollector(context, locations, proposals);
	collector.process(processors);

	return collector.getStatus();
	}

	/*
	* @see IContentAssistProcessor#getErrorMessage()
	*/
	@Override
	public String getErrorMessage() {
	return fErrorMessage;
	}

	/*
	* @see org.eclipse.jface.text.quickassist.IQuickAssistProcessor#canFix(org.eclipse.jface.text.source.Annotation)
	*/
	@Override
	public boolean canFix(Annotation annotation) {
	return hasCorrections(annotation);
	}

	/*
	* @see org.eclipse.jface.text.quickassist.IQuickAssistProcessor#canAssist(org.eclipse.jface.text.quickassist.IQuickAssistInvocationContext)
	*/
	@Override
	public boolean canAssist(IQuickAssistInvocationContext invocationContext) {
	if (invocationContext instanceof CorrectionContext)
	return hasAssists((CorrectionContext) invocationContext);
	return false;
	}
	}