private static NumericMatrix getMatrix(Collection<PlainPerson> persons, LegPredicate pred) {
NumericMatrix m = new NumericMatrix();
for (Person person : persons) {
for (Episode episode : person.getEpisodes()) {
for (int i = 0; i < episode.getLegs().size(); i++) {
Segment leg = episode.getLegs().get(i);
if (pred.test(leg)) {
Segment prev = episode.getActivities().get(i);
Segment next = episode.getActivities().get(i + 1);
String origin = prev.getAttribute(SetZones.ZONE_KEY);
String dest = next.getAttribute(SetZones.ZONE_KEY);
if (origin != null && dest != null) {
m.add(origin, dest, 1);
String touched = leg.getAttribute(DBG_TOUCHED);
int cnt = 0;
if (touched != null) cnt = Integer.parseInt(touched);
leg.setAttribute(DBG_TOUCHED, String.valueOf(cnt + 1));
}
}
}
}
}
return m;
}
/*
* (non-Javadoc)
*
* @see
* playground.johannes.synpop.processing.EpisodeTask#apply(playground.johannes
* .gsv.synPop.PlainEpisode)
*/
@Override
public void apply(Episode plan) {
boolean hasVacations = false;
for (Attributable act : plan.getActivities()) {
if ("vacations".equalsIgnoreCase(act.getAttribute(CommonKeys.ACTIVITY_TYPE))) {
hasVacations = true;
break;
}
}
if (hasVacations) {
boolean isLong = false;
Attributable first = plan.getLegs().get(0);
Attributable last = plan.getLegs().get(plan.getLegs().size() - 1);
String startStr = first.getAttribute(CommonKeys.LEG_START_TIME);
String endStr = last.getAttribute(CommonKeys.LEG_END_TIME);
if (startStr != null && endStr != null) {
DateTime start = SplitPlanTask.formatter.parseDateTime(startStr);
DateTime end = SplitPlanTask.formatter.parseDateTime(endStr);
if (end.getDayOfYear() - start.getDayOfYear() > 3) {
isLong = true;
}
}
for (Attributable act : plan.getActivities()) {
if ("vacations".equalsIgnoreCase(act.getAttribute(CommonKeys.ACTIVITY_TYPE))) {
if (isLong) {
act.setAttribute(CommonKeys.ACTIVITY_TYPE, "vacations_long");
} else {
act.setAttribute(CommonKeys.ACTIVITY_TYPE, "vacations_short");
}
}
}
}
}
public static void main(String[] args) throws IOException {
String in = args[0];
String rootDir = args[1];
logger.info("Loading persons...");
XMLHandler reader = new XMLHandler(new PlainFactory());
reader.setValidating(false);
reader.readFile(in);
Collection<PlainPerson> persons = (Set<PlainPerson>) reader.getPersons();
logger.info(String.format("Loaded %s persons.", persons.size()));
logger.info("Restoring original activity types...");
TaskRunner.run(new RestoreActTypes(), persons, true);
logger.info("Replacing misc types...");
new ReplaceMiscType().apply(persons);
logger.info("Imputing month attributes...");
new ImputeMonth().apply(persons);
logger.info("Assigning leg purposes...");
TaskRunner.run(new SetLegPurposes(), persons);
logger.info("Inferring wecommuter purpose...");
TaskRunner.run(new InfereWeCommuter(100000), persons);
Map<String, LegPredicate> modePreds = new LinkedHashMap<>();
modePreds.put("car", new LegKeyValuePredicate(CommonKeys.LEG_MODE, "car"));
Map<String, LegPredicate> purposePreds = new LinkedHashMap<>();
purposePreds.put(
ActivityTypes.BUSINESS,
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.BUSINESS));
purposePreds.put(
ActivityTypes.EDUCATION,
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.EDUCATION));
PredicateORComposite leisurePred = new PredicateORComposite();
leisurePred.addComponent(
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.LEISURE));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "visit"));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "gastro"));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "culture"));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "private"));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "pickdrop"));
leisurePred.addComponent(new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, "sport"));
purposePreds.put(ActivityTypes.LEISURE, leisurePred);
purposePreds.put(
ActivityTypes.SHOP, new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.SHOP));
purposePreds.put(
ActivityTypes.WORK, new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.WORK));
purposePreds.put(
ActivityTypes.VACATIONS_SHORT,
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.VACATIONS_SHORT));
purposePreds.put(
ActivityTypes.VACATIONS_LONG,
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, ActivityTypes.VACATIONS_LONG));
purposePreds.put(
InfereWeCommuter.WECOMMUTER,
new LegKeyValuePredicate(CommonKeys.LEG_PURPOSE, InfereWeCommuter.WECOMMUTER));
Map<String, LegPredicate> dayPreds = new LinkedHashMap<>();
dayPreds.put(
CommonValues.MONDAY, new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.MONDAY));
dayPreds.put(
CommonValues.FRIDAY, new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.FRIDAY));
dayPreds.put(
CommonValues.SATURDAY, new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.SATURDAY));
dayPreds.put(
CommonValues.SUNDAY, new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.SUNDAY));
PredicateORComposite dimidoPred = new PredicateORComposite();
dimidoPred.addComponent(new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.TUESDAY));
dimidoPred.addComponent(new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.WEDNESDAY));
dimidoPred.addComponent(new PersonKeyValuePredicate(CommonKeys.DAY, CommonValues.THURSDAY));
dayPreds.put(DIMIDO, dimidoPred);
Map<String, LegPredicate> seasonPreds = new LinkedHashMap<>();
PredicateORComposite summerPred = new PredicateORComposite();
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.APRIL));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.MAY));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.JUNE));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.JULY));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.AUGUST));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.SEPTEMBER));
summerPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.OCTOBER));
seasonPreds.put(SUMMER, summerPred);
PredicateORComposite winterPred = new PredicateORComposite();
winterPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.NOVEMBER));
winterPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.DECEMBER));
winterPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.JANUARY));
winterPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.FEBRUARY));
winterPred.addComponent(new PersonKeyValuePredicate(MiDKeys.PERSON_MONTH, MiDValues.MARCH));
seasonPreds.put(WINTER, winterPred);
Map<String, LegPredicate> directionPreds = new LinkedHashMap<>();
directionPreds.put(
DirectionPredicate.OUTWARD, new DirectionPredicate(DirectionPredicate.OUTWARD));
directionPreds.put(
DirectionPredicate.RETURN, new DirectionPredicate(DirectionPredicate.RETURN));
directionPreds.put(
DirectionPredicate.INTERMEDIATE, new DirectionPredicate(DirectionPredicate.INTERMEDIATE));
logger.info("Extracting full matrix...");
NumericMatrixTxtIO.write(
getMatrix(persons, modePreds.get("car")), String.format("%s/car.txt.gz", rootDir));
for (Map.Entry<String, LegPredicate> mode : modePreds.entrySet()) {
for (Map.Entry<String, LegPredicate> purpose : purposePreds.entrySet()) {
for (Map.Entry<String, LegPredicate> day : dayPreds.entrySet()) {
for (Map.Entry<String, LegPredicate> season : seasonPreds.entrySet()) {
for (Map.Entry<String, LegPredicate> direction : directionPreds.entrySet()) {
PredicateANDComposite comp = new PredicateANDComposite();
comp.addComponent(mode.getValue());
comp.addComponent(purpose.getValue());
comp.addComponent(day.getValue());
comp.addComponent(season.getValue());
comp.addComponent(direction.getValue());
StringBuilder builder = new StringBuilder();
builder.append(mode.getKey());
builder.append(DIM_SEPARATOR);
builder.append(purpose.getKey());
builder.append(DIM_SEPARATOR);
builder.append(day.getKey());
builder.append(DIM_SEPARATOR);
builder.append(season.getKey());
builder.append(DIM_SEPARATOR);
builder.append(direction.getKey());
logger.info(String.format("Extracting matrix %s...", builder.toString()));
NumericMatrix m = getMatrix(persons, comp);
String out = String.format("%s/%s.txt.gz", rootDir, builder.toString());
NumericMatrixTxtIO.write(m, out);
}
}
}
}
}
int cnt = 0;
for (Person person : persons) {
for (Episode episode : person.getEpisodes()) {
for (Segment leg : episode.getLegs()) {
String touched = leg.getAttribute(DBG_TOUCHED);
if (touched == null) {
leg.setAttribute(DBG_TOUCHED, "0");
cnt++;
}
}
}
}
logger.info(String.format("%s trips are untouched.", cnt));
XMLWriter writer = new XMLWriter();
writer.write(rootDir + "/plans.xml.gz", persons);
logger.info("Done.");
}