examples/org.eclipse.ocl.examples.autogen/src/org/eclipse/ocl/examples/autogen/lookup/LookupFilterCG2JavaVisitor.java

/*******************************************************************************
 * Copyright (c) 2014, 2016 Willink Transformations 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
 *   Adolfo Sanchez-Barbudo Herrera (University of York) - Lookup Environment/Visitor
 *******************************************************************************/
package org.eclipse.ocl.examples.autogen.lookup;

import java.util.List;

import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.eclipse.ocl.examples.autogen.java.AutoCG2JavaVisitor;
import org.eclipse.ocl.examples.codegen.cgmodel.CGClass;
import org.eclipse.ocl.examples.codegen.cgmodel.CGOperation;
import org.eclipse.ocl.examples.codegen.cgmodel.CGPackage;
import org.eclipse.ocl.examples.codegen.cgmodel.CGParameter;
import org.eclipse.ocl.examples.codegen.cgmodel.CGProperty;
import org.eclipse.ocl.examples.codegen.cgmodel.CGValuedElement;
import org.eclipse.ocl.examples.codegen.java.JavaConstants;
import org.eclipse.ocl.pivot.evaluation.Executor;

/**
 * LookupCG2JavaVisitor refines the regular generation of Java code from an optimized Auto CG transformation tree
 * to add contributions that are inadequately represented by the CG model.
 */
public class LookupFilterCG2JavaVisitor extends AutoCG2JavaVisitor<@NonNull LookupFilterGenerator>
{
	public LookupFilterCG2JavaVisitor(@NonNull LookupFilterGenerator codeGenerator, @NonNull CGPackage cgPackage,
			@Nullable List<CGValuedElement> sortedGlobals) {
		super(codeGenerator, cgPackage, sortedGlobals);
	}

	@Override
	protected void doConstructor(@NonNull CGClass cgClass) {
		js.append("public " + cgClass.getName() + "(");
		js.appendIsRequired(true);
		js.append(" ");
		js.appendClassReference(Executor.class);
		js.append(" "+JavaConstants.EXECUTOR_NAME);
		// We add the original operation parameters
		addFilterParameters(cgClass); // The first and unique OP will have the filtering operation
		js.append(") {\n");
		js.pushIndentation(null);
		js.append("super(");
		js.appendClassReference(null, getFilteredType(cgClass));
		js.append(".class");
		js.append(");\n");
	
		addFilterPropsInit(cgClass);
		js.append("this." + JavaConstants.EXECUTOR_NAME + " = " +JavaConstants.EXECUTOR_NAME + ";\n");
		js.append("this." + JavaConstants.ID_RESOLVER_NAME + " = " + JavaConstants.EXECUTOR_NAME + ".getIdResolver();\n");
	
		js.popIndentation();
		js.append("}\n");
	}
	
	
	private void addFilterParameters(CGClass cgClass) {
		for (CGProperty filteringVar : context.getFilteringVars(cgClass)) {
			js.append(",");
			js.appendDeclaration(filteringVar);
		}
	}
	
	private void addFilterPropsInit(CGClass cgClass) {
		for (CGProperty filteringVar : context.getFilteringVars(cgClass)) {
			String varName = filteringVar.getValueName();
			js.append("this.");
			js.append(varName);
			js.append(" =  ");
			js.append(varName);
			js.append(";\n");
		}
	}
	
	private CGValuedElement getFilteredType(CGClass cgClass) {
		
		//We get the first CGoperation, and the parameter will refer to the filteredType
		CGOperation cgFilteringOp = cgClass.getOperations().get(0);
		CGParameter param = cgFilteringOp.getParameters().get(0);
		return param.getTypedValue();
	}
  	

	
}