redocs/org.eclipse.statet.redocs.tex.r/src/org/eclipse/statet/redocs/tex/r/ui/TexRweaveLabelProvider.java

/*=============================================================================#
 # Copyright (c) 2009, 2020 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.redocs.tex.r.ui;

import org.eclipse.jface.viewers.StyledString;
import org.eclipse.swt.graphics.Image;

import org.eclipse.statet.docmlet.tex.core.model.TexSourceElement;
import org.eclipse.statet.docmlet.tex.ui.TexLabelProvider;
import org.eclipse.statet.ltk.model.core.element.EmbeddingForeignElement;
import org.eclipse.statet.ltk.model.core.element.LtkModelElement;
import org.eclipse.statet.ltk.model.core.element.SourceStructElement;
import org.eclipse.statet.r.core.model.RModel;
import org.eclipse.statet.r.ui.RLabelProvider;
import org.eclipse.statet.redocs.r.ui.RedocsRUIResources;


public class TexRweaveLabelProvider extends TexLabelProvider {
	
	
	private final RLabelProvider fRProvider;
	
	
	public TexRweaveLabelProvider(final int rStyle) {
		this.fRProvider= new RLabelProvider(rStyle);
	}
	
	
	@Override
	public Image getImage(final LtkModelElement element) {
		if (element.getModelTypeId() == RModel.R_TYPE_ID) {
			return this.fRProvider.getImage(element);
		}
		if (element.getElementType() == TexSourceElement.C1_EMBEDDED) {
			return RedocsRUIResources.INSTANCE.getImage(RedocsRUIResources.OBJ_RCHUNK_IMAGE_ID);
		}
		return super.getImage(element);
	}
	
	@Override
	public String getText(final LtkModelElement element) {
		if (element.getModelTypeId() == RModel.R_TYPE_ID) {
			return this.fRProvider.getText(element);
		}
		if (element.getElementType() == TexSourceElement.C1_EMBEDDED) {
			final SourceStructElement rElement= ((EmbeddingForeignElement)element).getForeignElement();
			if (rElement != null) {
				return this.fRProvider.getText(rElement);
			}
		}
		return super.getText(element);
	}
	
	@Override
	public StyledString getStyledText(final LtkModelElement element) {
		if (element.getModelTypeId() == RModel.R_TYPE_ID) {
			return this.fRProvider.getStyledText(element);
		}
		if (element.getElementType() == TexSourceElement.C1_EMBEDDED) {
			final SourceStructElement rElement= ((EmbeddingForeignElement)element).getForeignElement();
			if (rElement != null) {
				return this.fRProvider.getStyledText(rElement);
			}
		}
		return super.getStyledText(element);
	}
	
}