71 lines
1.8 KiB
Java
71 lines
1.8 KiB
Java
package de.hsel.spm.baudas.analysis;
|
|
|
|
import weka.core.Instance;
|
|
import weka.core.Instances;
|
|
|
|
import java.io.File;
|
|
import java.util.AbstractMap;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Week WeekOverviewDiagram Analysis.
|
|
*
|
|
* @author Julian Hinxlage
|
|
* @version 0.1
|
|
* @since 0.1
|
|
**/
|
|
public class WeekOverview implements Analysis<Map<String, Map.Entry<Double, Integer>>> {
|
|
|
|
private Instances instances;
|
|
private Map<String, Map.Entry<Double, Integer>> result;
|
|
|
|
public WeekOverview(File file) {
|
|
|
|
instances = load(file);
|
|
}
|
|
|
|
/**
|
|
* get results for analysis.
|
|
*
|
|
* @return day, amount, count
|
|
*/
|
|
@Override
|
|
public Map<String, Map.Entry<Double, Integer>> getResult() {
|
|
if (result != null) {
|
|
return result;
|
|
}
|
|
result = new HashMap<>();
|
|
int dayIndex = Attribute.SHOPPING_DAY;
|
|
int amountIndex = Attribute.PURCHASE_AMOUNT;
|
|
|
|
int startArticles = Attribute.POWER_TOOLS;
|
|
int endArticles = Attribute.GARDENING_TOOLS + 1;
|
|
|
|
for (int i = 0; i < instances.numInstances(); i++) {
|
|
Instance instance = instances.get(i);
|
|
double amount = instance.value(amountIndex);
|
|
String day = instance.stringValue(dayIndex);
|
|
|
|
|
|
int count = 0;
|
|
for (int j = startArticles; j < endArticles; j++) {
|
|
count += (int) instance.value(j);
|
|
}
|
|
|
|
if (!result.containsKey(day)) {
|
|
result.put(day, new AbstractMap.SimpleEntry<>(0.0, 0));
|
|
}
|
|
result.put(day,
|
|
new AbstractMap.SimpleEntry<>(
|
|
result.get(day).getKey() + amount,
|
|
result.get(day).getValue() + count
|
|
)
|
|
);
|
|
}
|
|
|
|
|
|
return result;
|
|
}
|
|
}
|