001 //$HeadURL: svn+ssh://jwilden@svn.wald.intevation.org/deegree/base/branches/2.5_testing/src/org/deegree/tools/datastore/SchemaAnnotator.java $ 002 /*---------------------------------------------------------------------------- 003 This file is part of deegree, http://deegree.org/ 004 Copyright (C) 2001-2009 by: 005 Department of Geography, University of Bonn 006 and 007 lat/lon GmbH 008 009 This library is free software; you can redistribute it and/or modify it under 010 the terms of the GNU Lesser General Public License as published by the Free 011 Software Foundation; either version 2.1 of the License, or (at your option) 012 any later version. 013 This library is distributed in the hope that it will be useful, but WITHOUT 014 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS 015 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more 016 details. 017 You should have received a copy of the GNU Lesser General Public License 018 along with this library; if not, write to the Free Software Foundation, Inc., 019 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 020 021 Contact information: 022 023 lat/lon GmbH 024 Aennchenstr. 19, 53177 Bonn 025 Germany 026 http://lat-lon.de/ 027 028 Department of Geography, University of Bonn 029 Prof. Dr. Klaus Greve 030 Postfach 1147, 53001 Bonn 031 Germany 032 http://www.geographie.uni-bonn.de/deegree/ 033 034 e-mail: info@deegree.org 035 ----------------------------------------------------------------------------*/ 036 package org.deegree.tools.datastore; 037 038 import java.io.File; 039 import java.io.FileInputStream; 040 import java.io.FileWriter; 041 import java.io.IOException; 042 import java.net.URL; 043 import java.util.Properties; 044 045 import javax.xml.transform.OutputKeys; 046 import javax.xml.transform.TransformerException; 047 048 import org.deegree.framework.xml.XMLFragment; 049 import org.deegree.framework.xml.XSLTDocument; 050 import org.xml.sax.SAXException; 051 052 /** 053 * Annotates a plain GML application schema with (default) mapping information. 054 * 055 * @author <a href="mailto:schneider@lat-lon.de">Markus Schneider</a> 056 * @author last edited by: $Author: mschneider $ 057 * 058 * @version $Revision: 18195 $, $Date: 2009-06-18 17:55:39 +0200 (Do, 18 Jun 2009) $ 059 */ 060 public class SchemaAnnotator { 061 062 private static final String XSL_FILE = "annotator.xsl"; 063 064 private static XSLTDocument xslSheet = new XSLTDocument(); 065 066 private static Properties featureTypeMappings = new Properties(); 067 068 private static Properties propertyMappings = new Properties(); 069 070 static { 071 URL sheetURL = SchemaAnnotator.class.getResource( XSL_FILE ); 072 try { 073 xslSheet.load( sheetURL ); 074 } catch ( Exception e ) { 075 e.printStackTrace(); 076 } 077 } 078 079 /** 080 * @param columnName 081 * @return the converted column name 082 */ 083 public static String getColumnName( String columnName ) { 084 columnName = columnName.toLowerCase(); 085 String colName = (String) propertyMappings.get( columnName ); 086 if ( colName == null ) { 087 System.out.println( "No field name -> column mapping for feature type '" + columnName 088 + "'. Using field name as column name." ); 089 colName = columnName; 090 } 091 return colName; 092 } 093 094 /** 095 * @param featureTypeName 096 * @return the table name 097 */ 098 public static String getTableName( String featureTypeName ) { 099 featureTypeName = featureTypeName.toLowerCase(); 100 String tableName = (String) featureTypeMappings.get( featureTypeName ); 101 if ( tableName == null ) { 102 System.out.println( "No feature type -> table mapping for feature type '" + featureTypeName 103 + "'. Using feature type name as table name." ); 104 tableName = featureTypeName; 105 } 106 return tableName; 107 } 108 109 /** 110 * @param args 111 * @throws SAXException 112 * @throws IOException 113 * @throws TransformerException 114 */ 115 public static void main( String[] args ) 116 throws IOException, SAXException, TransformerException { 117 118 if ( args.length < 2 ) { 119 System.out.println( "Usage: SchemaAnnotator <input.xsd> <output.xsd> [tableName.properties] [columnName.properties]" ); 120 System.exit( 0 ); 121 } 122 123 String tableNames = args.length < 4 ? null : args[3]; 124 String columnNames = args.length < 3 ? null : args[2]; 125 URL outputFile = new File( args[1] ).toURL(); 126 URL inputFile = new File( args[0] ).toURL(); 127 128 if ( tableNames != null ) { 129 System.out.println( "Loading feature type -> table name translations from file '" + tableNames + "'..." ); 130 featureTypeMappings.load( new FileInputStream( new URL( tableNames ).getFile() ) ); 131 } 132 133 if ( columnNames != null ) { 134 System.out.println( "Loading property -> column name translations from file '" + columnNames + "'..." ); 135 propertyMappings.load( new FileInputStream( new URL( columnNames ).getFile() ) ); 136 } 137 138 System.out.println( "Loading input schema file '" + inputFile + "'..." ); 139 XMLFragment inputSchema = new XMLFragment(); 140 inputSchema.load( inputFile ); 141 142 System.out.println( "Adding annotation information..." ); 143 XMLFragment outputSchema = xslSheet.transform( inputSchema ); 144 145 System.out.println( "Writing annotated schema file '" + outputFile + "'..." ); 146 FileWriter writer = new FileWriter( outputFile.getFile() ); 147 Properties properties = new Properties(); 148 149 properties.setProperty( OutputKeys.ENCODING, "UTF-8" ); 150 outputSchema.write( writer, properties ); 151 writer.close(); 152 } 153 }