/*

  * Copyright (c) 2007 David Holroyd

  */

 package uk.co.badgersinfoil.asxsd.components;

 import java.util.Iterator;

 import java.util.List;

 import org.eclipse.xsd.XSDCompositor;

 import org.eclipse.xsd.XSDConcreteComponent;

 import org.eclipse.xsd.XSDModelGroup;

 import org.eclipse.xsd.XSDParticle;

 import uk.co.badgersinfoil.asxsd.CodegenContext;

 import uk.co.badgersinfoil.asxsd.CodegenRole;

 import uk.co.badgersinfoil.asxsd.XSDUtils;

 import uk.co.badgersinfoil.metaas.ActionScriptFactory;

 import uk.co.badgersinfoil.metaas.dom.ASExpression;

 import uk.co.badgersinfoil.metaas.dom.StatementContainer;

 public class SequenceMappingComponent extends AbstractMappingComponent {

         public void generateCode(CodegenContext context,

                                  XSDConcreteComponent component)

         {

                 CodegenRole role = context.getCurrentRole();

                 if (role == CodegenRole.UNMARSHAL) {

                         XSDModelGroup modelGroup = (XSDModelGroup)component;

                         List particles = modelGroup.getParticles();

                         StatementContainer code = context.getCurrentMethodCode();

                         code.addComment(" process sequence,");

                         for (Iterator i=particles.iterator(); i.hasNext(); ) {

                                 XSDParticle part = (XSDParticle)i.next();

                                 context.generateCode(part);

                         }

                 } else if (role == CodegenRole.MARSHAL) {

                         XSDModelGroup modelGroup = (XSDModelGroup)component;

                         List particles = modelGroup.getParticles();

                         StatementContainer code = context.getCurrentMethodCode();

                         code.addComment(" process sequence,");

                         for (Iterator i=particles.iterator(); i.hasNext(); ) {

                                 XSDParticle part = (XSDParticle)i.next();

                                 context.generateCode(part);

                         }

                 } else if (role == CodegenRole.TYPE) {

                         XSDModelGroup modelGroup = (XSDModelGroup)component;

                         List particles = modelGroup.getParticles();

                         for (Iterator i=particles.iterator(); i.hasNext(); ) {

                                 XSDParticle part = (XSDParticle)i.next();

                                 context.generateCode(part);

                         }

                 } else {

                         throw new IllegalArgumentException(getClass().getName() + " does not support role "+role);

                 }

         }

         public ASExpression createExpression(CodegenContext context,

                                              XSDConcreteComponent component)

         {

                 CodegenRole role = context.getCurrentRole();

                 XSDModelGroup modelGroup = (XSDModelGroup)component;

                 ActionScriptFactory fact = context.getFactory();

                 List particles = modelGroup.getParticles();

                 if (role == CodegenRole.UNMARSHAL_DETECT) {

                         if (particles.isEmpty()) {

                                 return fact.newExpression("false");

                         }

                         Iterator i=particles.iterator();

                         ASExpression detect = context.createExpression((XSDParticle)i.next());

                         while (i.hasNext()) {

                                 XSDParticle part = (XSDParticle)i.next();

                                 ASExpression rhs = context.createExpression(part);

                                 detect = fact.newOrExpression(detect, rhs);

                                 if (XSDUtils.isRequiredToAppear(part)) {

                                         // Don't need to check for other

                                         // elements if this one is mandatory,

                                         break;

                                 }

                         }

                         return detect;

                 }

                 if (role == CodegenRole.MARSHAL_DETECT) {

                         if (particles.isEmpty()) {

                                 return fact.newExpression("false");

                         }

                         Iterator i = particles.iterator();

                         ASExpression detect = context.createExpression((XSDParticle)i.next());

                         while (i.hasNext()) {

                                 XSDParticle part = (XSDParticle)i.next();

                                 ASExpression rhs = context.createExpression(part);

                                 detect = fact.newOrExpression(detect, rhs);

                                 if (XSDUtils.isRequiredToAppear(part)) {

                                         // Don't need to check for other

                                         // elements if this one is mandatory,

                                         break;

                                 }

                         }

                         return detect;

                 }

                 return super.createExpression(context, component);

         }

         public String createTypeNameFor(CodegenContext context,

                                         XSDConcreteComponent component)

         {

                 return context.createTypeNameFor(component.getContainer()) + "Sequence";

         }

         public boolean willAccept(XSDConcreteComponent component) {

                 if (component instanceof XSDModelGroup) {

                         XSDModelGroup modelGroup = (XSDModelGroup)component;

                         return modelGroup.getCompositor() == XSDCompositor.SEQUENCE_LITERAL;

                 }

                 return false;

         }

 }