org.hortonmachine.modules.SaintGeo Maven / Gradle / Ivy
/*
* This file is part of HortonMachine (http://www.hortonmachine.org)
* (C) HydroloGIS - www.hydrologis.com
*
* The HortonMachine is free software: you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see lateralData = readIdData(inLateralId2Discharge);
saintGeo.inLateralId2DischargeMap = lateralData;
}
if (inConfluenceId2Discharge!=null) {
HashMap confluenceData = readIdData(inConfluenceId2Discharge);
saintGeo.inConfluenceId2DischargeMap = confluenceData;
}
saintGeo.pDeltaTMillis = pDeltaTMillis;
saintGeo.outputLevelFile = outputLevelFile;
saintGeo.outputDischargeFile = outputDischargeFile;
saintGeo.process();
}
private HashMap readIdData( String path ) throws Exception {
CSTable table = DataIO.table(new File(path), null);
HashMap> dataMap = new HashMap<>();
int columnCount = table.getColumnCount();
int[] ids = new int[columnCount - 1]; // minus type and timestamp
for( int i = 2; i <= columnCount; i++ ) {
Map columnInfo = table.getColumnInfo(i);
String idStr = columnInfo.get("id");
int id = Integer.parseInt(idStr);
ids[i - 2] = id;
}
TableIterator rowsIterator = (TableIterator) table.rows().iterator();
while( rowsIterator.hasNext() ) {
String[] row = rowsIterator.next();
for( int i = 2; i < row.length; i++ ) {
List dataList = dataMap.get(ids[i - 2]);
if (dataList == null) {
dataList = new ArrayList<>();
dataMap.put(ids[i - 2], dataList);
}
double value = -1;
if (row[i] == null || row[i].length() == 0) {
value = HMConstants.doubleNovalue;
} else {
String valueStr = row[i];
value = Double.parseDouble(valueStr);
}
dataList.add(value);
}
}
HashMap outDataMap = new HashMap<>();
for( Entry> entry : dataMap.entrySet() ) {
Integer id = entry.getKey();
List valueList = entry.getValue();
double[] values = new double[valueList.size()];
for( int i = 0; i < values.length; i++ ) {
values[i] = valueList.get(i);
}
outDataMap.put(id, values);
}
return outDataMap;
}
private double[] readToArray( String file ) throws IOException {
OmsTimeSeriesReader reader = new OmsTimeSeriesReader();
reader.file = file;
reader.fileNovalue = "-9999";
reader.read();
HashMap outData = reader.outData;
double[] data = new double[outData.size()];
int counter = 0;
for( double[] values : outData.values() ) {
data[counter++] = values[0];
}
return data;
}
public static void main( String[] args ) throws Exception {
String base = "D:/Dropbox/hydrologis/lavori/2015_phd_bz/gSoC2015/data/data_adige_test/";
String inSec = base + "sections_adige_75_rev.shp";
String inSecP = base + "sectionpoints_adige_75_rev.shp";
String inRiv = base + "riverpoints_adige_75_rev.shp";
String inDischarge = base + "head_discharge.csv";
String inLevel = base + "downstream_waterlevel.csv";
String inLateral = base + "q_lateral_offtakes.csv";
String outLevelFile = base + "saintgeo_level_out_offtakes.csv";
String outDischargeFile = base + "saintgeo_discharge_out_offtakes.csv";
SaintGeo sg = new SaintGeo();
sg.inRiverPoints = inRiv;
sg.inSectionPoints = inSecP;
sg.inSections = inSec;
sg.inDischarge = inDischarge;
sg.inDownstreamLevel = inLevel;
sg.inLateralId2Discharge = inLateral;
sg.pDeltaTMillis = 5000;
sg.outputLevelFile = outLevelFile;
sg.outputDischargeFile = outDischargeFile;
sg.process();
}
}