001: /*
002: * The Unified Mapping Platform (JUMP) is an extensible, interactive GUI
003: * for visualizing and manipulating spatial features with geometry and attributes.
004: *
005: * Copyright (C) 2003 Vivid Solutions
006: *
007: * This program is free software; you can redistribute it and/or
008: * modify it under the terms of the GNU General Public License
009: * as published by the Free Software Foundation; either version 2
010: * of the License, or (at your option) any later version.
011: *
012: * This program is distributed in the hope that it will be useful,
013: * but WITHOUT ANY WARRANTY; without even the implied warranty of
014: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
015: * GNU General Public License for more details.
016: *
017: * You should have received a copy of the GNU General Public License
018: * along with this program; if not, write to the Free Software
019: * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
020: *
021: * For more information, contact:
022: *
023: * Vivid Solutions
024: * Suite #1A
025: * 2328 Government Street
026: * Victoria BC V8T 5G5
027: * Canada
028: *
029: * (250)385-6040
030: * www.vividsolutions.com
031: */
032: package com.vividsolutions.jump.io.datasource;
033:
034: import java.io.File;
035: import java.util.ArrayList;
036: import java.util.Collection;
037: import java.util.Iterator;
038:
039: import com.vividsolutions.jump.feature.FeatureCollection;
040: import com.vividsolutions.jump.io.CompressedFile;
041: import com.vividsolutions.jump.io.DriverProperties;
042: import com.vividsolutions.jump.io.JUMPReader;
043: import com.vividsolutions.jump.util.FileUtil;
044: import com.vividsolutions.jump.util.StringUtil;
045:
046: /**
047: * If the file is a .zip or .gz file, mangles the DriverProperties into the format
048: * expected by many of the first JUMPReaders, which take responsibility
049: * for doing the decompression. Really, JUMPReaders should not
050: * have to be responsible for decompression -- they should be wrapped
051: * by a CompressedFileHandler (not yet written) which would decompress the
052: * data before handing it to the JUMPReader. Anyway, developers should now be
053: * writing DataSources instead of JUMPReaders.
054: */
055: public class DelegatingCompressedFileHandler implements JUMPReader {
056:
057: private Collection endings;
058:
059: private JUMPReader reader;
060:
061: /**
062: * Constructs a DelegatingCompressedFileHandler that wraps a JUMPReader.
063: * @param endings strings found at the ends of filenames, used to identify
064: * the file to extract from a .zip; does not apply to .gz files
065: */
066: public DelegatingCompressedFileHandler(JUMPReader reader,
067: Collection endings) {
068: this .reader = reader;
069: this .endings = new ArrayList(endings);
070: }
071:
072: /**
073: * @param dp
074: * only the "File" property is necessary; if it is a .zip or
075: * .gz, the required "CompressedFile" property will be
076: * determined automatically
077: */
078: public FeatureCollection read(DriverProperties dp) throws Exception {
079: mangle(dp, "File", "CompressedFile", endings);
080: return reader.read(dp);
081: }
082:
083: protected void mangle(DriverProperties dp, String fileProperty,
084: String compressedFileProperty, Collection myEndings)
085: throws Exception {
086: if (FileUtil.getExtension(
087: new File(dp.getProperty(fileProperty)))
088: .equalsIgnoreCase("zip")) {
089: String internalName = null;
090: for (Iterator i = myEndings.iterator(); internalName == null
091: && i.hasNext();) {
092: String ending = (String) i.next();
093: internalName = CompressedFile
094: .getInternalZipFnameByExtension(ending, dp
095: .getProperty(fileProperty));
096: }
097:
098: if (internalName == null) {
099: throw new Exception("Couldn't find a "
100: + StringUtil.toCommaDelimitedString(myEndings)
101: + " file inside the .zip file: "
102: + dp.getProperty(fileProperty));
103: }
104: dp
105: .set(compressedFileProperty, dp
106: .getProperty(fileProperty));
107: dp.set(fileProperty, internalName);
108: } else if (FileUtil.getExtension(
109: new File(dp.getProperty(fileProperty)))
110: .equalsIgnoreCase("gz")) {
111: dp
112: .set(compressedFileProperty, dp
113: .getProperty(fileProperty));
114: }
115: }
116:
117: }
|