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.xjc.addon.locator;
037:
038: import java.io.IOException;
039:
040: import javax.xml.bind.annotation.XmlTransient;
041:
042: import com.sun.codemodel.JDefinedClass;
043: import com.sun.codemodel.JMod;
044: import com.sun.codemodel.JVar;
045: import com.sun.codemodel.JMethod;
046: import com.sun.tools.xjc.BadCommandLineException;
047: import com.sun.tools.xjc.Options;
048: import com.sun.tools.xjc.Plugin;
049: import com.sun.tools.xjc.outline.ClassOutline;
050: import com.sun.tools.xjc.outline.Outline;
051: import com.sun.xml.bind.Locatable;
052: import com.sun.xml.bind.annotation.XmlLocation;
053:
054: import org.xml.sax.ErrorHandler;
055: import org.xml.sax.Locator;
056:
057: /**
058: * Generates JAXB objects that implement {@link Locatable}.
059: *
060: * @author
061: * Kohsuke Kawaguchi (kohsuke.kawaguchi@sun.com)
062: */
063: public class SourceLocationAddOn extends Plugin {
064:
065: public String getOptionName() {
066: return "Xlocator";
067: }
068:
069: public String getUsage() {
070: return " -Xlocator : enable source location support for generated code";
071: }
072:
073: public int parseArgument(Options opt, String[] args, int i)
074: throws BadCommandLineException, IOException {
075: return 0; // no option recognized
076: }
077:
078: private static final String fieldName = "locator";
079:
080: public boolean run(Outline outline, Options opt,
081: ErrorHandler errorHandler) {
082:
083: for (ClassOutline ci : outline.getClasses()) {
084: JDefinedClass impl = ci.implClass;
085: if (ci.getSuperClass() == null) {
086: JVar $loc = impl.field(JMod.PROTECTED, Locator.class,
087: fieldName);
088: $loc.annotate(XmlLocation.class);
089: $loc.annotate(XmlTransient.class);
090:
091: impl._implements (Locatable.class);
092:
093: impl.method(JMod.PUBLIC, Locator.class,
094: "sourceLocation").body()._return($loc);
095:
096: JMethod setter = impl.method(JMod.PUBLIC, Void.TYPE,
097: "setSourceLocation");
098: JVar $newLoc = setter
099: .param(Locator.class, "newLocator");
100: setter.body().assign($loc, $newLoc);
101: }
102: }
103:
104: return true;
105: }
106: }
|