/*

  * Copyright (c) 2006-2007 David Holroyd

  */

 package uk.co.badgersinfoil.asxsd;

 import java.util.HashMap;

 import java.util.Map;

 import org.eclipse.xsd.XSDComplexTypeDefinition;

 import org.eclipse.xsd.XSDElementDeclaration;

 import org.eclipse.xsd.XSDParticle;

 import org.eclipse.xsd.XSDSimpleTypeDefinition;

 import org.eclipse.xsd.XSDTypeDefinition;

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

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

 public class TypeBuilder {

         private static final String KEY_CURRENT_TYPE_COMPILATION_UNIT = "uk.co.badgersinfoil.asxsd.currentTypeCompilationUnit";

         TypeNameGenerator nameGen;

         private Map representationClassesByQName = new HashMap();

         private CodegenContext context;

         public TypeBuilder(CodegenContext context, TypeNameGenerator nameGen) {

                 this.nameGen = nameGen;

                 this.context = context;

         }

         public String typeNameFor(XSDTypeDefinition typeDef) {

                 return nameGen.typeName(typeDef);

         }

         public static ASCompilationUnit getCurrentTypeCompilationUnit(CodegenContext context) {

                 ASCompilationUnit result = (ASCompilationUnit)context.getAttribute(KEY_CURRENT_TYPE_COMPILATION_UNIT);

                 if (result == null) {

                         throw new IllegalStateException("No current compilation unit");

                 }

                 return result;

         }

         public static ASClassType getCurrentType(CodegenContext context) {

                 return (ASClassType)getCurrentTypeCompilationUnit(context).getType();

         }

         public static void setCurrentTypeCompilationUnit(CodegenContext context, ASCompilationUnit unit) {

                 context.setAttribute(KEY_CURRENT_TYPE_COMPILATION_UNIT, unit);

         }

         /**

          * Handle a top-level complex type definition by creating an AS

          * class.

          */

         private void processComplexType(XSDComplexTypeDefinition typeDef) {

                 context.setCurrentRole(CodegenRole.TYPE);

                 context.generateCode(typeDef);

         }

 }