001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common Development
008: * and Distribution License("CDDL") (collectively, the "License"). You
009: * may not use this file except in compliance with the License. You can obtain
010: * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
011: * or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
012: * language governing permissions and limitations under the License.
013: *
014: * When distributing the software, include this License Header Notice in each
015: * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
016: * Sun designates this particular file as subject to the "Classpath" exception
017: * as provided by Sun in the GPL Version 2 section of the License file that
018: * accompanied this code. If applicable, add the following below the License
019: * Header, with the fields enclosed by brackets [] replaced by your own
020: * identifying information: "Portions Copyrighted [year]
021: * [name of copyright owner]"
022: *
023: * Contributor(s):
024: *
025: * If you wish your version of this file to be governed by only the CDDL or
026: * only the GPL Version 2, indicate your decision by adding "[Contributor]
027: * elects to include this software in this distribution under the [CDDL or GPL
028: * Version 2] license." If you don't indicate a single choice of license, a
029: * recipient has the option to distribute your version of this file under
030: * either the CDDL, the GPL Version 2 or to extend the choice of license to
031: * its licensees as provided above. However, if you add GPL Version 2 code
032: * and therefore, elected the GPL Version 2 license, then the option applies
033: * only if the new code is made subject to such option by the copyright
034: * holder.
035: */
036: package com.sun.tools.ws.processor.model.jaxb;
037:
038: import com.sun.tools.xjc.api.J2SJAXBModel;
039: import com.sun.tools.xjc.api.Mapping;
040: import com.sun.tools.xjc.api.S2JJAXBModel;
041:
042: import javax.xml.namespace.QName;
043: import java.util.*;
044:
045: /**
046: * Root of the JAXB Model.
047: *
048: * <p>
049: * This is just a wrapper around a list of {@link JAXBMapping}s.
050: *
051: * @author Kohsuke Kawaguchi, Vivek Pandey
052: */
053: public class JAXBModel {
054:
055: /**
056: * All the mappings known to this model.
057: */
058: private List<JAXBMapping> mappings;
059:
060: // index for faster access.
061: private final Map<QName, JAXBMapping> byQName = new HashMap<QName, JAXBMapping>();
062: private final Map<String, JAXBMapping> byClassName = new HashMap<String, JAXBMapping>();
063:
064: private com.sun.tools.xjc.api.JAXBModel rawJAXBModel;
065:
066: public com.sun.tools.xjc.api.JAXBModel getRawJAXBModel() {
067: return rawJAXBModel;
068: }
069:
070: /**
071: * @return Schema to Java model
072: */
073: public S2JJAXBModel getS2JJAXBModel() {
074: if (rawJAXBModel instanceof S2JJAXBModel)
075: return (S2JJAXBModel) rawJAXBModel;
076: return null;
077: }
078:
079: /**
080: * @return Java to Schema JAXBModel
081: */
082: public J2SJAXBModel getJ2SJAXBModel() {
083: if (rawJAXBModel instanceof J2SJAXBModel)
084: return (J2SJAXBModel) rawJAXBModel;
085: return null;
086: }
087:
088: /**
089: * Default constructor for the persistence.
090: */
091: public JAXBModel() {
092: }
093:
094: /**
095: * Constructor that fills in the values from the given raw model
096: */
097: public JAXBModel(com.sun.tools.xjc.api.JAXBModel rawModel) {
098: this .rawJAXBModel = rawModel;
099: if (rawModel instanceof S2JJAXBModel) {
100: S2JJAXBModel model = (S2JJAXBModel) rawModel;
101: List<JAXBMapping> ms = new ArrayList<JAXBMapping>(model
102: .getMappings().size());
103: for (Mapping m : model.getMappings())
104: ms.add(new JAXBMapping(m));
105: setMappings(ms);
106: }
107: }
108:
109: /**
110: */
111: public List<JAXBMapping> getMappings() {
112: return mappings;
113: }
114:
115: //public void setMappings(List<JAXBMapping> mappings) {
116: public void setMappings(List<JAXBMapping> mappings) {
117: this .mappings = mappings;
118: byQName.clear();
119: byClassName.clear();
120: for (JAXBMapping m : mappings) {
121: byQName.put(m.getElementName(), m);
122: byClassName.put(m.getType().getName(), m);
123: }
124: }
125:
126: /**
127: */
128: public JAXBMapping get(QName elementName) {
129: return byQName.get(elementName);
130: }
131:
132: /**
133: */
134: public JAXBMapping get(String className) {
135: return byClassName.get(className);
136: }
137:
138: /**
139: *
140: * @return set of full qualified class names that jaxb will generate
141: */
142: public Set<String> getGeneratedClassNames() {
143: return generatedClassNames;
144: }
145:
146: public void setGeneratedClassNames(Set<String> generatedClassNames) {
147: this .generatedClassNames = generatedClassNames;
148: }
149:
150: private Set<String> generatedClassNames;
151: }
|