ingo@2844: package de.intevation.flys.importer.parsers;
ingo@2844: 
ingo@2845: import java.io.File;
ingo@2845: import java.io.IOException;
felix@5049: import java.math.BigDecimal;
ingo@2845: import java.text.NumberFormat;
ingo@2845: import java.text.ParseException;
ingo@2845: import java.util.ArrayList;
ingo@2844: import java.util.List;
ingo@2845: import java.util.regex.Matcher;
ingo@2845: import java.util.regex.Pattern;
ingo@2844: 
ingo@2845: import org.apache.log4j.Logger;
ingo@2845: 
ingo@2845: import de.intevation.flys.importer.ImportUnit;
ingo@2844: import de.intevation.flys.importer.ImportWaterlevel;
ingo@2845: import de.intevation.flys.importer.ImportWaterlevelQRange;
ingo@2845: import de.intevation.flys.importer.ImportWaterlevelValue;
ingo@2844: 
felix@5049: import de.intevation.flys.importer.ImportWstColumn;
felix@5049: import de.intevation.flys.importer.ImportRange;
felix@5049: import de.intevation.flys.importer.ImportWst;
felix@5049: import de.intevation.flys.importer.ImportWstQRange;
ingo@2844: 
felix@5049: 
felix@5049: /**
felix@5049:  * Parse CSV Waterlevel files.
felix@5049:  * As these waterlevels are probably used in fixation analysis
felix@5049:  * only, functionality to export them to "fixation"-wsts
felix@5049:  * has been added (the ImportWaterlevel*- stuff is actually
felix@5049:  * not needed to do so.)
felix@5049:  */
ingo@2844: public class WaterlevelParser extends LineParser {
ingo@2844: 
ingo@2845:     private static final Logger log = Logger.getLogger(WaterlevelParser.class);
ingo@2845: 
ingo@2845:     private static final NumberFormat nf =
ingo@2845:         NumberFormat.getInstance(DEFAULT_LOCALE);
ingo@2845: 
ingo@2845:     private static final Pattern META_Q_RANGE =
ingo@2845:         Pattern.compile("Abfluss\\s\\[(.*)\\];(.*)");
ingo@2845: 
ingo@2845:     public static final Pattern META_UNIT =
ingo@2845:         Pattern.compile("^Einheit: \\[(.*)\\].*");
ingo@2845: 
ingo@2844:     private List<ImportWaterlevel> waterlevels;
ingo@2844: 
ingo@2845:     private ImportWaterlevel current;
ingo@2845: 
ingo@2845:     private ImportWaterlevelQRange currentQ;
ingo@2845: 
ingo@2845:     private String currentDescription;
ingo@2845: 
ingo@2845: 
ingo@2844:     public WaterlevelParser() {
ingo@2845:         waterlevels = new ArrayList<ImportWaterlevel>();
ingo@2844:     }
ingo@2844: 
ingo@2844: 
felix@5049:     /**
felix@5049:      * Create ImportWst objects from ImportWaterlevel
felix@5049:      * objects.
felix@5049:      */
felix@5049:     public List<ImportWst> exportWsts() {
felix@5049:         List<ImportWst> wsts = new ArrayList<ImportWst>();
felix@5049:         for(ImportWaterlevel waterlevel: getWaterlevels()) {
felix@5049:             String description = waterlevel.getDescription();
felix@5049:             ImportWst wst = new ImportWst();
felix@5049:             wsts.add(wst);
felix@5049:             wst.setDescription(description);
felix@5049:             // Fixation kind.
felix@5049:             wst.setKind(2);
felix@5049:             wst.setUnit(waterlevel.getUnit());
felix@5049: 
felix@5049:             // Fake WST has but 1 column.
felix@5049:             wst.setNumberColumns(1);
felix@5049:             ImportWstColumn column = wst.getColumn(0);
felix@5049:             column.setDescription(description);
felix@5049:             column.setName(description);
felix@5049:             column.setPosition(0);
felix@5049: 
felix@5049:             // Build Q Range.
felix@5049:             List<ImportWaterlevelQRange> qRanges = waterlevel.getQRanges();
felix@5049:             for(ImportWaterlevelQRange range: qRanges) {
felix@5049:                 List<ImportWaterlevelValue> values = range.getValues();
felix@5049:                 if (values.size() < 2) {
felix@5049:                     log.warn ("Not enough values to build valid QRange");
felix@5049:                     continue;
felix@5049:                 }
felix@5049:                 ImportRange iRange = new ImportRange(
felix@5049:                    BigDecimal.valueOf(values.get(0).getStation()),
felix@5049:                    BigDecimal.valueOf(values.get(values.size() -1).getStation()));
felix@5049:                 column.addColumnQRange(
felix@5049:                      new ImportWstQRange(iRange, BigDecimal.valueOf(range.getQ())));
felix@5049:             }
felix@5049: 
felix@5049:             // The other W/KM values.
felix@5049:             for(ImportWaterlevelQRange range: qRanges) {
felix@5049:                 for(ImportWaterlevelValue value: range.getValues()) {
felix@5049:                     column.addColumnValue(BigDecimal.valueOf(value.getStation()),
felix@5049:                                           BigDecimal.valueOf(value.getW()));
felix@5049:                 }
felix@5049:             }
felix@5049:             // TODO Maybe set a timeinterval.
felix@5049:         }
felix@5049:         return wsts;
felix@5049:     }
felix@5049: 
felix@5049: 
ingo@2844:     public List<ImportWaterlevel> getWaterlevels() {
ingo@2844:         return waterlevels;
ingo@2844:     }
ingo@2844: 
ingo@2844: 
ingo@2844:     @Override
ingo@2845:     public void parse(File file) throws IOException {
ingo@2845:         currentDescription = file.getName();
ingo@2845: 
ingo@2845:         super.parse(file);
ingo@2845:     }
ingo@2845: 
ingo@2845: 
ingo@2845:     @Override
ingo@2844:     protected void reset() {
ingo@2845:         currentQ = null;
ingo@2845:         current  = new ImportWaterlevel(currentDescription);
ingo@2844:     }
ingo@2844: 
ingo@2844: 
ingo@2844:     @Override
ingo@2844:     protected void finish() {
ingo@2845:         if (current != null) {
ingo@2845:             if (currentQ != null) {
ingo@2845:                 current.addValue(currentQ);
ingo@2845:             }
ingo@2844: 
ingo@2845:             waterlevels.add(current);
ingo@2845:         }
ingo@2844:     }
ingo@2844: 
ingo@2844:     @Override
ingo@4193:     protected void handleLine(int lineNum, String line) {
ingo@2845:         if (line.startsWith(START_META_CHAR)) {
ingo@2845:             handleMetaLine(stripMetaLine(line));
ingo@2845:             return;
ingo@2845:         }
ingo@2845:         else if (handleQRange(line)) {
ingo@2845:             return;
ingo@2845:         }
ingo@2845:         else {
ingo@2845:             handleDataLine(line);
ingo@2845:             return;
ingo@2845:         }
ingo@2845:     }
ingo@2844: 
ingo@2845: 
ingo@2845:     private void handleMetaLine(String meta) {
ingo@2845:         Matcher m = META_UNIT.matcher(meta);
ingo@2845: 
ingo@2845:         if (m.matches()) {
ingo@2845:             String unit = m.group(1);
ingo@2845:             log.debug("Found unit: '" + unit + "'");
ingo@2845: 
ingo@2845:             current.setUnit(new ImportUnit(unit));
ingo@2845:         }
ingo@2845:     }
ingo@2845: 
ingo@2845: 
ingo@2845:     private boolean handleQRange(String line) {
ingo@2845:         Matcher m = META_Q_RANGE.matcher(line);
ingo@2845: 
ingo@2845:         if (m.matches()) {
ingo@2845:             String unitStr  = m.group(1);
ingo@2845:             String valueStr = m.group(2);
ingo@2845: 
ingo@2845:             if (currentQ != null) {
ingo@2845:                 if (current != null) {
ingo@2845:                     current.addValue(currentQ);
ingo@2845:                 }
ingo@2845:                 else {
ingo@2845:                     // this should never happen
ingo@2845:                     log.warn("Try to add Q range without waterlevel!");
ingo@2845:                 }
ingo@2845:             }
ingo@2845: 
ingo@2845:             try {
ingo@2845:                 log.debug("Found new Q range: Q=" + valueStr);
ingo@2845: 
ingo@2845:                 currentQ = new ImportWaterlevelQRange(
ingo@2845:                     nf.parse(valueStr).doubleValue());
ingo@2845: 
ingo@2845:                 return true;
ingo@2845:             }
ingo@2845:             catch (ParseException pe) {
ingo@2845:                 log.warn("Error while parsing Q range: '" + line + "'");
ingo@2845:             }
ingo@2845:         }
ingo@2845: 
ingo@2845:         return false;
ingo@2845:     }
ingo@2845: 
ingo@2845: 
ingo@2845:     private void handleDataLine(String line) {
ingo@2845:         String[] cols = line.split(SEPERATOR_CHAR);
ingo@2845: 
ingo@2845:         if (cols == null || cols.length < 2) {
ingo@2845:             log.warn("skip invalid waterlevel line: '" + line + "'");
ingo@2845:             return;
ingo@2845:         }
ingo@2845: 
ingo@2845:         try {
ingo@2845:             Double station = nf.parse(cols[0]).doubleValue();
ingo@2845:             Double value   = nf.parse(cols[1]).doubleValue();
ingo@2845: 
ingo@2845:             currentQ.addValue(new ImportWaterlevelValue(station, value));
ingo@2845:         }
ingo@2845:         catch (ParseException pe) {
ingo@2845:             log.warn("Error while parsing number values: '" + line + "'");
ingo@2845:         }
ingo@2844:     }
ingo@2844: }
ingo@2844: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :