archive/org.eclipse.ocl.examples.parser.ocl/src/org/eclipse/ocl/examples/parser/ocl/OCLFileAnalyzer.java

/**
 * <copyright>
 * 
 * Copyright (c) 2007,2010 E.D.Willink 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:
 * E.D.Willink - initial API and implementation
 * 
 * </copyright>
 *
 * $Id: OCLFileAnalyzer.java,v 1.3 2010/04/08 06:26:38 ewillink Exp $
 */
package org.eclipse.ocl.examples.parser.ocl;

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

import lpg.runtime.Monitor;

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EEnumLiteral;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EOperation;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EParameter;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.EcoreFactory;
import org.eclipse.emf.ecore.xmi.XMIResource;
import org.eclipse.ocl.Environment;
import org.eclipse.ocl.cst.PackageDeclarationCS;
import org.eclipse.ocl.cst.SimpleNameCS;
import org.eclipse.ocl.ecore.CallOperationAction;
import org.eclipse.ocl.ecore.Constraint;
import org.eclipse.ocl.ecore.SendSignalAction;
import org.eclipse.ocl.examples.parser.environment.ecore.ICSTFileEcoreAnalyzer;
import org.eclipse.ocl.examples.parser.ocl.environment.OCLFileEnvironment;
import org.eclipse.ocl.examples.parser.ocl.environment.OCLTopLevelEnvironment;
import org.eclipse.ocl.lpg.AbstractLexer;
import org.eclipse.ocl.parser.OCLAnalyzer;
import org.eclipse.ocl.parser.backtracking.OCLBacktrackingLexer;
import org.eclipse.ocl.parser.backtracking.OCLBacktrackingParser;

public class OCLFileAnalyzer extends
		OCLAnalyzer<EPackage, EClassifier, EOperation, EStructuralFeature, 
		EEnumLiteral, EParameter, EObject, CallOperationAction, SendSignalAction, Constraint, EClass, EObject>
implements ICSTFileEcoreAnalyzer<OCLTopLevelEnvironment>
{
	private Monitor monitor;
	private OCLTopLevelEnvironment topLevelEnvironment = null;
	private EClass iteratorsClass = null;

	public OCLFileAnalyzer(OCLFileEnvironment environment, Monitor monitor) {
		super(new OCLBacktrackingParser(new OCLBacktrackingLexer(environment)));
		this.monitor = monitor;
	}

	public boolean analyze(OCLTopLevelEnvironment environment) {
		this.topLevelEnvironment = environment;
		List<Constraint> constraints = new ArrayList<Constraint>();
		for (PackageDeclarationCS pkgdecl : environment.getCSTNode().getPackageDeclarations())
			packageDeclarationCS(pkgdecl, constraints);
		installConstraints(topLevelEnvironment, constraints);
		return true;
	}

	@Override
	protected Object createDummyPackage(
			Environment<EPackage, EClassifier, EOperation, EStructuralFeature, EEnumLiteral, EParameter, EObject, CallOperationAction, SendSignalAction, Constraint, EClass, EObject> env,
			PackageDeclarationCS packageDeclarationCS) {
		return EcoreFactory.eINSTANCE.createEPackage();
	}

    @Override
	public AbstractLexer getLexer() {
    	return getAbstractParser().getLexer();
    }

	@Override
	public Environment<EPackage, EClassifier, EOperation, EStructuralFeature, EEnumLiteral, EParameter, EObject, CallOperationAction, SendSignalAction, Constraint, EClass, EObject> getOCLEnvironment() {
		return topLevelEnvironment != null ? topLevelEnvironment : super.getOCLEnvironment();
	}

	protected void installConstraints(OCLTopLevelEnvironment topLevelEnvironment, List<Constraint> constraints) {
		if (!constraints.isEmpty()) {
			XMIResource astNode = topLevelEnvironment.getASTNode();
			if (astNode != null) {
				astNode.getContents().addAll(constraints);
			}
		}
	}

	protected boolean isCancelled() {
		return monitor != null ? monitor.isCancelled() : false;
	}

	@Override
	protected void resolveIteratorOperation(SimpleNameCS simpleNameCS,
			Environment<EPackage, EClassifier, EOperation, EStructuralFeature, EEnumLiteral, EParameter, EObject, CallOperationAction, SendSignalAction, Constraint, EClass, EObject> env) {
		if (iteratorsClass == null) {
			EPackage iteratorsPackage = EcoreFactory.eINSTANCE.createEPackage();
			iteratorsPackage.setName("iterators");
			iteratorsClass = EcoreFactory.eINSTANCE.createEClass();
			iteratorsClass.setName("Iterators");
			iteratorsPackage.getEClassifiers().add(iteratorsClass);
		}
		EOperation oper = EcoreFactory.eINSTANCE.createEOperation();
		oper.setName(simpleNameCS.getValue());
		iteratorsClass.getEOperations().add(oper);
		simpleNameCS.setAst(oper);
	}
}