org.jnario.feature.compiler.FeatureBatchCompiler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.jnario.standalone Show documentation

The required libraries to execute Jnario specifications without Eclipse.

The newest version!

/*******************************************************************************
 * Copyright (c) 2012 BMW Car IT 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
 *******************************************************************************/
package org.jnario.feature.compiler;

import static com.google.common.collect.Iterables.filter;

import java.util.List;

import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.emf.common.util.TreeIterator;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.jnario.compiler.JnarioBatchCompiler;
import org.jnario.feature.feature.Feature;
import org.jnario.feature.feature.Scenario;
import org.jnario.feature.naming.FeatureClassNameProvider;

import com.google.inject.Inject;

/**
 * @author Sebastian Benz - Initial contribution and API
 */
public class FeatureBatchCompiler extends JnarioBatchCompiler {
	
	@Inject
	public FeatureClassNameProvider nameProvider;

	@Override
	protected String getClassName(EObject eObject) {
		return nameProvider.toJavaClassName(eObject);
	}
	
	@Override
	protected void addObjectsWithClasses(Resource resource, List result) {
		TreeIterator allContents = resource.getAllContents();
		while (allContents.hasNext()) {
			Notifier notifier = allContents.next();
			if (notifier instanceof Feature) {
				Feature feature = (Feature) notifier;
				result.add(feature);
				if (feature.getBackground() != null) {
					result.add(feature.getBackground());
				}
				Iterable scenarios = filter(feature.getMembers(), Scenario.class);
				for (Scenario scenario : scenarios) {
					result.add(scenario);
				}
				allContents.prune();
			}
		}
	}

}