public static void main(String[] args) {

    MutableScenario scenario =
        (MutableScenario) ScenarioUtils.createScenario(ConfigUtils.createConfig());
    MatsimReader populationReader = new PopulationReader(scenario);
    MatsimNetworkReader networkReader = new MatsimNetworkReader(scenario.getNetwork());
    networkReader.readFile(args[1]);
    populationReader.readFile(args[2]);

    ObjectAttributes bla = new ObjectAttributes();

    new ObjectAttributesXmlReader(bla).readFile(args[0]);

    for (Person p : scenario.getPopulation().getPersons().values()) {
      String act = "home";
      if (bla.getAttribute(p.getId().toString(), "earliestEndTime_leisure") != null)
        act = act + ",leisure";
      //	if (bla.getAttribute(p.getId().toString(), "earliestEndTime_work") != null)
      //		act = act + ",work";
      if (bla.getAttribute(p.getId().toString(), "earliestEndTime_shop") != null)
        act = act + ",shop";
      //	if (bla.getAttribute(p.getId().toString(), "earliestEndTime_education") != null)
      //		act = act + ",education";

      bla.putAttribute(p.getId().toString(), "activities", act);
    }

    ObjectAttributesXmlWriter betaWriter = new ObjectAttributesXmlWriter(bla);
    betaWriter.writeFile(args[3]);
  }
Esempio n. 2
0
 // does not matter which distribution is chosen here
 private void assignKValuesPersons() {
   for (Person p : this.scenario.getPopulation().getPersons().values()) {
     this.personsKValues.putAttribute(p.getId().toString(), "k", rnd.getUniform(1.0));
   }
   // write person k values
   ObjectAttributesXmlWriter attributesWriter = new ObjectAttributesXmlWriter(this.personsKValues);
   attributesWriter.writeFile(config.controler().getOutputDirectory() + pkValuesFile);
 }
Esempio n. 3
0
 private void assignKValuesAlternatives() {
   for (ActivityFacility facility :
       this.scenario.getActivityFacilities().getFacilities().values()) {
     this.facilitiesKValues.putAttribute(facility.getId().toString(), "k", rnd.getUniform(1.0));
   }
   ObjectAttributesXmlWriter attributesWriter =
       new ObjectAttributesXmlWriter(this.facilitiesKValues);
   attributesWriter.writeFile(config.controler().getOutputDirectory() + fkValuesFile);
 }
  /**
   * Writes the population and their attributes to file.
   *
   * @param outputfolder
   */
  public void writePopulation(String outputfolder) {
    if (this.sc.getPopulation().getPersons().size() == 0 || this.personAttributes == null) {
      throw new RuntimeException("Either no persons or person attributes to write.");
    } else {
      LOG.info("Writing population to file...");
      PopulationWriter pw = new PopulationWriter(this.sc.getPopulation(), this.sc.getNetwork());
      pw.writeV5(outputfolder + "Population.xml");

      LOG.info("Writing person attributes to file...");
      ObjectAttributesXmlWriter oaw = new ObjectAttributesXmlWriter(this.personAttributes);
      oaw.setPrettyPrint(true);
      oaw.writeFile(outputfolder + "PersonAttributes.xml");
    }
  }
  /**
   * Writes the households and their attributes to file.
   *
   * @param outputfolder
   */
  public void writeHouseholds(String outputfolder) {
    if (this.households == null || this.householdAttributes == null) {
      throw new RuntimeException("Either no households or household attributes to write.");
    } else {
      LOG.info("Writing households to file...");
      HouseholdsWriterV10 hw = new HouseholdsWriterV10(this.households);
      hw.setPrettyPrint(true);
      hw.writeFile(outputfolder + "Households.xml");

      LOG.info("Writing household attributes to file...");
      ObjectAttributesXmlWriter oaw = new ObjectAttributesXmlWriter(householdAttributes);
      oaw.setPrettyPrint(true);
      oaw.writeFile(outputfolder + "HouseholdAttributes.xml");
    }
  }