1 files changed, 96 insertions, 0 deletions
diff --git a/src/de/lmu/ifi/dbs/elki/datasource/FileBasedDatabaseConnection.java b/src/de/lmu/ifi/dbs/elki/datasource/FileBasedDatabaseConnection.java
new file mode 100644
index 00000000..0549fbab
--- /dev/null
+++ b/src/de/lmu/ifi/dbs/elki/datasource/FileBasedDatabaseConnection.java
@@ -0,0 +1,96 @@
+package de.lmu.ifi.dbs.elki.datasource;
+/*
+This file is part of ELKI:
+Environment for Developing KDD-Applications Supported by Index-Structures
+
+Copyright (C) 2011
+Ludwig-Maximilians-Universität München
+Lehr- und Forschungseinheit für Datenbanksysteme
+ELKI Development Team
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.List;
+
+import de.lmu.ifi.dbs.elki.datasource.filter.ObjectFilter;
+import de.lmu.ifi.dbs.elki.datasource.parser.Parser;
+import de.lmu.ifi.dbs.elki.utilities.FileUtil;
+import de.lmu.ifi.dbs.elki.utilities.optionhandling.OptionID;
+import de.lmu.ifi.dbs.elki.utilities.optionhandling.WrongParameterValueException;
+import de.lmu.ifi.dbs.elki.utilities.optionhandling.parameterization.Parameterization;
+import de.lmu.ifi.dbs.elki.utilities.optionhandling.parameters.FileParameter;
+
+/**
+ * Provides a file based database connection based on the parser to be set.
+ * 
+ * @author Arthur Zimek
+ */
+public class FileBasedDatabaseConnection extends InputStreamDatabaseConnection {
+  /**
+   * Parameter that specifies the name of the input file to be parsed.
+   * <p>
+   * Key: {@code -dbc.in}
+   * </p>
+   */
+  public static final OptionID INPUT_ID = OptionID.getOrCreateOptionID("dbc.in", "The name of the input file to be parsed.");
+
+  /**
+   * Constructor.
+   * 
+   * @param filters Filters, can be null
+   * @param parser the parser to provide a database
+   * @param in the input stream to parse from.
+   */
+  public FileBasedDatabaseConnection(List<ObjectFilter> filters, Parser parser, InputStream in) {
+    super(filters, parser);
+    this.in = in;
+  }
+
+  /**
+   * Parameterization class.
+   * 
+   * @author Erich Schubert
+   * 
+   * @apiviz.exclude
+   */
+  public static class Parameterizer extends InputStreamDatabaseConnection.Parameterizer {
+    protected InputStream inputStream;
+
+    @Override
+    protected void makeOptions(Parameterization config) {
+      // Add the input file first, for usability reasons.
+      final FileParameter inputParam = new FileParameter(INPUT_ID, FileParameter.FileType.INPUT_FILE);
+      if(config.grab(inputParam)) {
+        try {
+          inputStream = new FileInputStream(inputParam.getValue());
+          inputStream = FileUtil.tryGzipInput(inputStream);
+        }
+        catch(IOException e) {
+          config.reportError(new WrongParameterValueException(inputParam, inputParam.getValue().getPath(), e));
+          inputStream = null;
+        }
+      }
+      super.makeOptions(config);
+    }
+
+    @Override
+    protected FileBasedDatabaseConnection makeInstance() {
+      return new FileBasedDatabaseConnection(filters, parser, inputStream);
+    }
+  }
+}
+\ No newline at end of file