public static GroupAverageSource_NEW ofPlot(TsDB tsdb, String plotID, String... targetSchema) {
GeneralStation generalStation = null;
VirtualPlot virtualPlot = tsdb.getVirtualPlot(plotID);
if (virtualPlot != null) {
generalStation = virtualPlot.generalStation;
} else {
Station station = tsdb.getStation(plotID);
if (station != null) {
generalStation = station.generalStation;
}
}
throwNullText(generalStation, "plotID not found: " + plotID);
return new GroupAverageSource_NEW(tsdb, generalStation.group, targetSchema);
}
public static StationRawSource of(TsDB tsdb, String stationName, String[] querySchema) {
Station station = tsdb.getStation(stationName);
if (station == null) {
throw new RuntimeException("station not found: " + stationName);
}
return of(tsdb, station, querySchema);
}
public static void main(String[] args) {
// final int source_count = Interpolated.STATION_INTERPOLATION_COUNT;
final int source_count = 6;
TsDB tsdb = TsDBFactory.createDefault();
ContinuousGen continuousGen = QueryPlanGenerators.getContinuousGen(tsdb, DataQuality.STEP);
String[] schema = new String[] {"Ta_200"};
String targetPlot = "SEG29";
// String targetPlot = "HEG01";
long start = TimeUtil.ofDateStartHour(2014, 4);
long end = TimeUtil.ofDateEndHour(2014, 6);
long removeStart = TimeUtil.ofDateStartHour(2014, 6);
long removeEnd = TimeUtil.ofDateEndHour(2014, 6);
ArrayList<String> result = new ArrayList<String>();
Station targetStation = tsdb.getStation(targetPlot);
String[] sourcePlots =
targetStation
.nearestStations
.stream()
.limit(source_count)
.map(s -> s.stationID)
.toArray(String[]::new);
Continuous targetNode = continuousGen.get(targetStation.stationID, schema);
Continuous targetNodeIntervalRemoved = IntervalRemove.of(targetNode, removeStart, removeEnd);
Continuous[] sourceNodes =
Arrays.stream(sourcePlots)
.map(s -> continuousGen.get(s, schema))
.toArray(Continuous[]::new);
for (Continuous source : sourceNodes) {
TsIterator it = source.get(start, end);
int count = 0;
while (it.hasNext()) {
TsEntry e = it.next();
if (Float.isFinite(e.data[0])) {
count++;
}
}
result.add(source.getSourceStation().stationID + " " + count);
}
for (String e : result) {
System.out.println(e);
}
Continuous resultNode = Interpolated.of(tsdb, targetNodeIntervalRemoved, sourceNodes, schema);
// double[] maxMSEs = new double[]{};
// Continuous resultNode = new InterpolatedAverageLinear(tsdb, targetNodeIntervalRemoved,
// targetNodeIntervalRemoved, sourceNodes, schema, AggregationInterval.DAY, maxMSEs);
TsIterator it = resultNode.get(start, end);
int count = 0;
while (it.hasNext()) {
TsEntry e = it.next();
if (Float.isFinite(e.data[0])) {
count++;
}
// System.out.println(e);
}
System.out.println("" + count);
String path = "C:/timeseriesdatabase_R/";
targetNode.writeCSV(removeStart, removeEnd, path + targetPlot + "_real.csv");
resultNode.writeCSV(removeStart, removeEnd, path + targetPlot + "_interpolated.csv");
sourceNodes[0].writeCSV(
removeStart, removeEnd, path + sourceNodes[0].getSourceStation().stationID + "_real.csv");
}
