r/org.eclipse.statet.r.core/src/org/eclipse/statet/internal/r/core/sourcemodel/CompositeSourceElement.java

/*=============================================================================#
 # Copyright (c) 2011, 2021 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.internal.r.core.sourcemodel;

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

import org.eclipse.statet.jcommons.collections.ImCollections;
import org.eclipse.statet.jcommons.collections.ImList;
import org.eclipse.statet.jcommons.lang.NonNull;
import org.eclipse.statet.jcommons.lang.NonNullByDefault;
import org.eclipse.statet.jcommons.lang.Nullable;
import org.eclipse.statet.jcommons.text.core.TextRegion;

import org.eclipse.statet.ltk.model.core.element.LtkModelElementFilter;
import org.eclipse.statet.r.core.model.RCompositeSourceElement;
import org.eclipse.statet.r.core.model.RLangSourceElement;
import org.eclipse.statet.r.core.model.RSourceUnit;


@NonNullByDefault
public class CompositeSourceElement extends RSourceFileElement
		implements RCompositeSourceElement {
	
	
	private final ImList<? extends EmbeddedRBuildElement> compositeElements;
	
	private final TextRegion sourceRange;
	
	private volatile @Nullable List<RLangSourceElement> allSourceChildren;
	
	
	public CompositeSourceElement(final RSourceUnit su, final BuildSourceFrame envir,
			final List<? extends EmbeddedRBuildElement> elements, final TextRegion sourceRange) {
		super(su, envir);
		
		this.compositeElements= ImCollections.toList(elements);
		this.sourceRange= sourceRange;
	}
	
	
//	@Override
//	public int getElementType() {
//		return RElement.C2_SOURCE_FILE | 0x1;
//	}
	
	@Override
	public ImList<? extends RLangSourceElement> getCompositeElements() {
		return this.compositeElements;
	}
	
	
	@Override
	public TextRegion getSourceRange() {
		return this.sourceRange;
	}
	
	@Override
	public boolean hasSourceChildren(final @Nullable LtkModelElementFilter<? super RLangSourceElement> filter) {
		for (final EmbeddedRBuildElement element : this.compositeElements) {
			if (element.hasSourceChildren(filter)) {
				return true;
			}
		}
		return false;
	}
	
	@Override
	public List<RLangSourceElement> getSourceChildren(final @Nullable LtkModelElementFilter<? super RLangSourceElement> filter) {
		if (filter == null) {
			List<RLangSourceElement> children= this.allSourceChildren;
			if (children == null) {
				final List<? extends RLangSourceElement>[] compositeLists= new @NonNull List[this.compositeElements.size()];
				for (int i= 0; i < compositeLists.length; i++) {
					compositeLists[i]= this.compositeElements.get(i).getSourceChildren(null);
				}
				children= ImCollections.concatList(compositeLists);
				this.allSourceChildren= children;
			}
			return children;
		}
		else {
			final List<RLangSourceElement> children= new ArrayList<>();
			for (final EmbeddedRBuildElement element : this.compositeElements) {
				final List<? extends RLangSourceElement> list= element.getSourceChildren(null);
				for (final RLangSourceElement child : list) {
					if (filter.include(child)) {
						children.add(child);
					}
				}
			}
			return children;
		}
	}
	
}