org.jpmml.xjc.CopyConstructorPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pmml-xjc Show documentation
JPMML XJC plugins for programmatic class model customization
There is a newer version: 1.6.5
/*
 * Copyright (c) 2019 Villu Ruusmann
 */
package org.jpmml.xjc;

import java.util.Collection;

import com.sun.codemodel.JBlock;
import com.sun.codemodel.JClass;
import com.sun.codemodel.JCodeModel;
import com.sun.codemodel.JDefinedClass;
import com.sun.codemodel.JExpr;
import com.sun.codemodel.JMethod;
import com.sun.codemodel.JMod;
import com.sun.codemodel.JType;
import com.sun.codemodel.JVar;
import com.sun.tools.xjc.Options;
import com.sun.tools.xjc.Plugin;
import com.sun.tools.xjc.model.CPropertyInfo;
import com.sun.tools.xjc.model.Model;
import com.sun.tools.xjc.outline.ClassOutline;
import com.sun.tools.xjc.outline.FieldOutline;
import com.sun.tools.xjc.outline.Outline;
import org.xml.sax.ErrorHandler;

public class CopyConstructorPlugin extends Plugin {

	@Override
	public String getOptionName(){
		return "XcopyConstructor";
	}

	@Override
	public String getUsage(){
		return null;
	}

	@Override
	public boolean run(Outline outline, Options options, ErrorHandler errorHandler){
		Model model = outline.getModel();

		JCodeModel codeModel = model.codeModel;

		JClass nodeClass = codeModel.ref("org.dmg.pmml.tree.Node");
		JClass scoreDistributionClass = codeModel.ref("org.dmg.pmml.ScoreDistribution");
		JClass copyConstructorAnnotation = codeModel.ref("org.jpmml.model.annotations.CopyConstructor");

		Collection classOutlines = outline.getClasses();
		for(ClassOutline classOutline : classOutlines){
			JDefinedClass beanClazz = classOutline.implClass;

			JClass beanSuperClazz;
			String objectParamName;

			if(checkType(beanClazz, "org.dmg.pmml.tree.ComplexNode")){
				beanSuperClazz = nodeClass;
				objectParamName = "node";
			} else

			if(checkType(beanClazz, "org.dmg.pmml.ComplexScoreDistribution")){
				beanSuperClazz = scoreDistributionClass;
				objectParamName = "scoreDistribution";
			} else

			{
				continue;
			}

			JMethod defaultConstructor = beanClazz.getConstructor(new JType[0]);
			if(defaultConstructor == null){
				defaultConstructor = beanClazz.constructor(JMod.PUBLIC);
			}

			JMethod copyConstructor = beanClazz.constructor(JMod.PUBLIC);
			copyConstructor.annotate(copyConstructorAnnotation);

			JVar objectParam = copyConstructor.param(beanSuperClazz, objectParamName);

			JBlock body = copyConstructor.body();

			FieldOutline[] fieldOutlines = classOutline.getDeclaredFields();
			for(FieldOutline fieldOutline : fieldOutlines){
				CPropertyInfo propertyInfo = fieldOutline.getPropertyInfo();

				String getterName = "get" + propertyInfo.getName(true);
				String setterName = "set" + propertyInfo.getName(true);

				if(propertyInfo.isCollection()){
					JBlock block = body._if(objectParam.invoke("has" + propertyInfo.getName(true)))._then();

					block.add(JExpr.invoke(getterName).invoke("addAll").arg(objectParam.invoke(getterName)));
				} else

				{
					JBlock block = body;

					block.add(JExpr.invoke(setterName).arg(objectParam.invoke(getterName)));
				}
			}

			JMethod toComplexObjectMethod = beanClazz.method(JMod.PUBLIC, beanClazz, "to" + beanClazz.name());
			toComplexObjectMethod.annotate(Override.class);

			toComplexObjectMethod.body()._return(JExpr._this());
		}

		return true;
	}

	static
	private boolean checkType(JType type, String fullName){
		return (type.fullName()).equals(fullName);
	}
}