001: /*
002: * Portions Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
003: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
004: *
005: * This code is free software; you can redistribute it and/or modify it
006: * under the terms of the GNU General Public License version 2 only, as
007: * published by the Free Software Foundation. Sun designates this
008: * particular file as subject to the "Classpath" exception as provided
009: * by Sun in the LICENSE file that accompanied this code.
010: *
011: * This code is distributed in the hope that it will be useful, but WITHOUT
012: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
013: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
014: * version 2 for more details (a copy is included in the LICENSE file that
015: * accompanied this code).
016: *
017: * You should have received a copy of the GNU General Public License version
018: * 2 along with this work; if not, write to the Free Software Foundation,
019: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
020: *
021: * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
022: * CA 95054 USA or visit www.sun.com if you need additional information or
023: * have any questions.
024: */
025: package com.sun.tools.internal.ws.processor.model.jaxb;
026:
027: import com.sun.tools.internal.xjc.api.*;
028:
029: import javax.xml.namespace.QName;
030:
031: import java.util.ArrayList;
032: import java.util.HashMap;
033: import java.util.HashSet;
034: import java.util.List;
035: import java.util.Map;
036: import java.util.Set;
037:
038: import com.sun.codemodel.internal.JType;
039:
040: /**
041: * Root of the JAXB Model.
042: *
043: * <p>
044: * This is just a wrapper around a list of {@link JAXBMapping}s.
045: *
046: * @author Kohsuke Kawaguchi, Vivek Pandey
047: */
048: public class JAXBModel {
049:
050: /**
051: * All the mappings known to this model.
052: */
053: private List<JAXBMapping> mappings;
054:
055: // index for faster access.
056: private final Map<QName, JAXBMapping> byQName = new HashMap<QName, JAXBMapping>();
057: private final Map<String, JAXBMapping> byClassName = new HashMap<String, JAXBMapping>();
058:
059: private com.sun.tools.internal.xjc.api.JAXBModel rawJAXBModel;
060:
061: public com.sun.tools.internal.xjc.api.JAXBModel getRawJAXBModel() {
062: return rawJAXBModel;
063: }
064:
065: /**
066: * @return Schema to Java model
067: */
068: public S2JJAXBModel getS2JJAXBModel() {
069: if (rawJAXBModel instanceof S2JJAXBModel)
070: return (S2JJAXBModel) rawJAXBModel;
071: return null;
072: }
073:
074: /**
075: * @return Java to Schema JAXBModel
076: */
077: public J2SJAXBModel getJ2SJAXBModel() {
078: if (rawJAXBModel instanceof J2SJAXBModel)
079: return (J2SJAXBModel) rawJAXBModel;
080: return null;
081: }
082:
083: /**
084: * Default constructor for the persistence.
085: */
086: public JAXBModel() {
087: }
088:
089: /**
090: * Constructor that fills in the values from the given raw model
091: */
092: public JAXBModel(com.sun.tools.internal.xjc.api.JAXBModel rawModel) {
093: this .rawJAXBModel = rawModel;
094: if (rawModel instanceof S2JJAXBModel) {
095: S2JJAXBModel model = (S2JJAXBModel) rawModel;
096: List<JAXBMapping> ms = new ArrayList<JAXBMapping>(model
097: .getMappings().size());
098: for (Mapping m : model.getMappings())
099: ms.add(new JAXBMapping(m));
100: setMappings(ms);
101: }
102: }
103:
104: /**
105: */
106: public List<JAXBMapping> getMappings() {
107: return mappings;
108: }
109:
110: //public void setMappings(List<JAXBMapping> mappings) {
111: public void setMappings(List<JAXBMapping> mappings) {
112: this .mappings = mappings;
113: byQName.clear();
114: byClassName.clear();
115: for (JAXBMapping m : mappings) {
116: byQName.put(m.getElementName(), m);
117: byClassName.put(m.getType().getName(), m);
118: }
119: }
120:
121: /**
122: */
123: public JAXBMapping get(QName elementName) {
124: return byQName.get(elementName);
125: }
126:
127: /**
128: */
129: public JAXBMapping get(String className) {
130: return byClassName.get(className);
131: }
132:
133: /**
134: *
135: * @return set of full qualified class names that jaxb will generate
136: */
137: public Set<String> getGeneratedClassNames() {
138: return generatedClassNames;
139: }
140:
141: public void setGeneratedClassNames(Set<String> generatedClassNames) {
142: this .generatedClassNames = generatedClassNames;
143: }
144:
145: private Set<String> generatedClassNames;
146: }
|