ingo@2850: package de.intevation.flys.importer.parsers;
ingo@2850: 
ingo@2850: import java.io.File;
ingo@2850: import java.io.IOException;
ingo@2850: import java.text.NumberFormat;
ingo@2850: import java.text.ParseException;
ingo@2850: import java.util.ArrayList;
ingo@2850: import java.util.List;
ingo@2850: import java.util.regex.Matcher;
ingo@2850: import java.util.regex.Pattern;
ingo@2850: 
ingo@2850: import org.apache.log4j.Logger;
ingo@2850: 
ingo@2850: import de.intevation.flys.importer.ImportUnit;
ingo@2850: import de.intevation.flys.importer.ImportWaterlevelDifference;
ingo@2850: import de.intevation.flys.importer.ImportWaterlevelDifferenceColumn;
ingo@2850: import de.intevation.flys.importer.ImportWaterlevelDifferenceValue;
ingo@2850: 
ingo@2850: 
ingo@2850: public class WaterlevelDifferencesParser extends LineParser {
ingo@2850: 
ingo@2850:     private static final Logger log =
ingo@2850:         Logger.getLogger(WaterlevelDifferencesParser.class);
ingo@2850: 
ingo@2850:     private static final NumberFormat nf =
ingo@2850:         NumberFormat.getInstance(DEFAULT_LOCALE);
ingo@2850: 
ingo@2850:     public static final Pattern META_UNIT =
ingo@2850:         Pattern.compile("^Einheit: \\[(.*)\\].*");
ingo@2850: 
ingo@2850: 
ingo@2850:     private List<ImportWaterlevelDifference> differences;
ingo@2850: 
ingo@2850:     private ImportWaterlevelDifferenceColumn[] columns;
ingo@2850: 
ingo@2850:     private ImportWaterlevelDifference current;
ingo@2850: 
ingo@2850: 
ingo@2850:     public WaterlevelDifferencesParser() {
ingo@2850:         differences = new ArrayList<ImportWaterlevelDifference>();
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     public List<ImportWaterlevelDifference> getDifferences() {
ingo@2850:         return differences;
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     @Override
ingo@2850:     public void parse(File file) throws IOException {
ingo@2850:         current = new ImportWaterlevelDifference(file.getName());
ingo@2850: 
ingo@2850:         super.parse(file);
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     @Override
ingo@2850:     protected void reset() {
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     @Override
ingo@2850:     protected void finish() {
ingo@2850:         if (columns != null && current != null) {
ingo@2850:             for (ImportWaterlevelDifferenceColumn col: columns) {
ingo@2850:                 current.addValue(col);
ingo@2850:             }
ingo@2850: 
ingo@2850:             differences.add(current);
ingo@2850:         }
ingo@2850: 
ingo@2850:         current = null;
ingo@2850:         columns = null;
ingo@2850:     }
ingo@2850: 
ingo@2850:     @Override
ingo@2850:     protected void handleLine(String line) {
ingo@2850:         if (line.startsWith(START_META_CHAR)) {
ingo@2850:             handleMetaLine(stripMetaLine(line));
ingo@2850:         }
ingo@2850:         else {
ingo@2850:             handleDataLine(line);
ingo@2850:         }
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     private void handleMetaLine(String meta) {
ingo@2850:         if (handleMetaUnit(meta)) {
ingo@2850:             return;
ingo@2850:         }
ingo@2850:         else {
ingo@2850:             handleMetaColumnNames(meta);
ingo@2850:         }
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     private boolean handleMetaUnit(String meta) {
ingo@2850:         Matcher m = META_UNIT.matcher(meta);
ingo@2850: 
ingo@2850:         if (m.matches()) {
ingo@2850:             String unit = m.group(1);
ingo@2850:             log.debug("Found unit: '" + unit + "'");
ingo@2850: 
ingo@2850:             current.setUnit(new ImportUnit(unit));
ingo@2850: 
ingo@2850:             return true;
ingo@2850:         }
ingo@2850: 
ingo@2850:         return false;
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     private boolean handleMetaColumnNames(String meta) {
ingo@2850:         Pattern META_COLUMN_NAMES = Pattern.compile("Fluss-km;(.*)");
ingo@2850:         Matcher m = META_COLUMN_NAMES.matcher(meta);
ingo@2850: 
ingo@2850:         if (m.matches()) {
ingo@2850:             String colStr = m.group(1);
ingo@2850:             String[] cols = colStr.split(SEPERATOR_CHAR);
ingo@2850: 
ingo@2850:             log.debug("Found " + cols.length + " columns.");
ingo@2850: 
ingo@2850:             initColumns(cols);
ingo@2850: 
ingo@2850:             return true;
ingo@2850:         }
ingo@2850: 
ingo@2850:         return false;
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     private void initColumns(String[] cols) {
ingo@2850:         columns = new ImportWaterlevelDifferenceColumn[cols.length];
ingo@2850: 
ingo@2850:         for (int i = 0; i < cols.length; i++) {
ingo@2850:             String name = cols[i].replace("\"", "");
ingo@2850: 
ingo@2850:             log.debug("Create new column '" + name + "'");
ingo@2850:             columns[i] = new ImportWaterlevelDifferenceColumn(name);
ingo@2850:         }
ingo@2850:     }
ingo@2850: 
ingo@2850: 
ingo@2850:     private void handleDataLine(String line) {
ingo@2850:         String[] cols = line.split(SEPERATOR_CHAR);
ingo@2850: 
ingo@2850:         if (cols == null || cols.length < 2) {
ingo@2850:             log.warn("skip invalid waterlevel line: '" + line + "'");
ingo@2850:             return;
ingo@2850:         }
ingo@2850: 
ingo@2850:         try {
ingo@2850:             Double station = nf.parse(cols[0]).doubleValue();
ingo@2850: 
ingo@2850:             for (int i = 0; i < columns.length; i++) {
ingo@3940:                 int idx = i+1;
ingo@3940: 
ingo@3940:                 if (idx >= cols.length) {
ingo@3940:                     log.warn("Insufficient column numbers: " + line);
ingo@3940:                     continue;
ingo@3940:                 }
ingo@3940: 
ingo@3940:                 String value = cols[idx];
ingo@2850: 
ingo@2850:                 try {
ingo@2850:                     columns[i].addValue(new ImportWaterlevelDifferenceValue(
ingo@2850:                         station,
ingo@2850:                         nf.parse(value).doubleValue()
ingo@2850:                     ));
ingo@2850:                 }
ingo@2850:                 catch (ParseException pe) {
ingo@2850:                     log.warn("Error while parsing value: '" + value + "'");
ingo@2850:                 }
ingo@2850:             }
ingo@2850:         }
ingo@2850:         catch (ParseException pe) {
ingo@2850:             log.warn("Error while parsing station: '" + line + "'");
ingo@2850:         }
ingo@2850:     }
ingo@2850: }
ingo@2850: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :