org.jpmml.xjc.OutlineUtil 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) 2021 Villu Ruusmann
 */
package org.jpmml.xjc;

import java.math.BigInteger;
import java.util.Map;

import com.sun.codemodel.JDefinedClass;
import com.sun.codemodel.JFieldVar;
import com.sun.codemodel.JMod;
import com.sun.codemodel.JMods;
import com.sun.codemodel.JType;
import com.sun.tools.xjc.model.CAttributePropertyInfo;
import com.sun.tools.xjc.model.CElementPropertyInfo;
import com.sun.tools.xjc.model.CPropertyInfo;
import com.sun.tools.xjc.model.CReferencePropertyInfo;
import com.sun.tools.xjc.model.CValuePropertyInfo;
import com.sun.xml.xsom.XSComponent;
import com.sun.xml.xsom.XSParticle;

public class OutlineUtil {

	private OutlineUtil(){
	}

	static
	public boolean isRequired(JDefinedClass beanClazz, CPropertyInfo propertyInfo){
		Map fieldVars = beanClazz.fields();

		JFieldVar fieldVar = fieldVars.get(propertyInfo.getName(false));

		JMods modifiers = fieldVar.mods();
		if((modifiers.getValue() & JMod.STATIC) == JMod.STATIC){
			return false;
		}

		JType type = fieldVar.type();

		if(propertyInfo instanceof CAttributePropertyInfo){
			CAttributePropertyInfo attributePropertyInfo = (CAttributePropertyInfo)propertyInfo;

			return attributePropertyInfo.isRequired();
		} else

		if(propertyInfo instanceof CElementPropertyInfo){
			CElementPropertyInfo elementPropertyInfo = (CElementPropertyInfo)propertyInfo;

			if(propertyInfo.isCollection()){
				type = CodeModelUtil.getElementType(type);
			}

			switch(type.fullName()){
				case "org.dmg.pmml.EmbeddedModel":
				case "org.dmg.pmml.TableLocator":
					return false;
				default:
					break;
			}

			boolean required = elementPropertyInfo.isRequired();

			XSComponent xsComponent = propertyInfo.getSchemaComponent();
			if(xsComponent instanceof XSParticle){
				XSParticle xsParticle = (XSParticle)xsComponent;

				BigInteger minOccurs = xsParticle.getMinOccurs();
				BigInteger maxOccurs = xsParticle.getMaxOccurs();

				required |= (minOccurs.intValue() >= 1);
			}

			return required;
		} else

		if(propertyInfo instanceof CReferencePropertyInfo){
			CReferencePropertyInfo referencePropertyInfo = (CReferencePropertyInfo)propertyInfo;

			return referencePropertyInfo.isRequired();
		} else

		if(propertyInfo instanceof CValuePropertyInfo){
			CValuePropertyInfo valuePropertyInfo = (CValuePropertyInfo)propertyInfo;

			return true;
		} else

		{
			throw new IllegalArgumentException();
		}
	}
}