コード例 #1
0
ファイル: SyntheticPopUs.java プロジェクト: msmobility/silo
  private void definePersonRolesInHousehold(Household hh, int[] relShp) {
    // define roles as single, married or child

    Person[] pp = hh.getPersons();
    HashMap<Integer, Integer> coupleCounter = new HashMap<>();
    coupleCounter.put(1, 0);
    coupleCounter.put(2, 0);
    for (int i = 0; i < pp.length; i++) {
      //      Householder      husband/wife  unmarried Partner
      if (relShp[i] == 1 || relShp[i] == 2 || relShp[i] == 19) {
        int cnt = coupleCounter.get(pp[i].getGender()) + 1;
        coupleCounter.put(pp[i].getGender(), cnt);
      }
    }
    int numberOfCouples = Math.min(coupleCounter.get(1), coupleCounter.get(2));
    int[] marriedPersons = new int[] {numberOfCouples, numberOfCouples};
    if (numberOfCouples > 0) {
      pp[0].setRole(PersonRole.married);
      marriedPersons[pp[0].getGender() - 1] -= 1;
    } else pp[0].setRole(PersonRole.single);

    for (int i = 1; i < pp.length; i++) {
      if ((relShp[i] == 2 || relShp[i] == 19) && marriedPersons[pp[i].getGender() - 1] > 0) {
        pp[i].setRole(PersonRole.married);
        marriedPersons[pp[i].getGender() - 1] -= 1;
        //   natural child     adopted child        step child        grandchild       foster child
      } else if (relShp[i] == 3
          || relShp[i] == 4
          || relShp[i] == 5
          || relShp[i] == 8
          || relShp[i] == 20) pp[i].setRole(PersonRole.child);
      else pp[i].setRole(PersonRole.single);
    }
  }
コード例 #2
0
ファイル: SyntheticPopUs.java プロジェクト: msmobility/silo
  private void savePumsRecord(
      int pumaZone,
      int weight,
      int hhSize,
      int pumsDdType,
      int bedRooms,
      int autos,
      int rent,
      int mortgage,
      int quality,
      int yearBuilt,
      int[] gender,
      int[] age,
      Race[] race,
      int[] relShp,
      int[] occupation,
      int[] workPumaZone,
      int[] workState,
      int[] income) {
    // add record to PUMS record storage

    if (pumsDdType == 10 || pumsDdType == -999)
      return; // skip this record if PUMS dwelling type is 10 (Boat, RV, Van) or -999 (unknown)
    DwellingType ddType = translateDwellingType(pumsDdType);
    for (int count = 0; count < weight; count++) {
      int newDdId = RealEstateDataManager.getNextDwellingId();
      int newHhId;
      if (gender[0] == 0) newHhId = -1;
      else newHhId = HouseholdDataManager.getNextHouseholdId();
      int taz = locateDwelling(pumaZone);

      int price = getDwellingPrice(rent, mortgage);
      int selectedYear = selectYear(yearBuilt);
      new Dwelling(newDdId, taz, newHhId, ddType, bedRooms, quality, price, 0, selectedYear);
      if (gender[0] == 0) return; // this dwelling is empty, do not create household
      Household hh = new Household(newHhId, newDdId, taz, hhSize, autos);
      for (int s = 0; s < hhSize; s++) {
        int newPpId = HouseholdDataManager.getNextPersonId();

        int occ = translateOccupation(occupation[s]);
        int workplace = -1;
        if (occ == 1) {

          if (workPumaZone[s] == 0 || workState[s] == 0) {
            // no workplace PUMA provided by PUMS (person did not go to work during week interviewed
            // because of vacation, leave, etc.)
            workplace =
                selectWorkplaceByTripLengthFrequencyDistribution(
                    workPumaZone[s], workState[s], taz);
          } else {
            // workplace PUMA provided by PUMS
            // workplace = selectWorkplace(workPumaZone[s], workState[s]);
            // update: As the distribution of requested workplaces according to PUMS is very
            // different from the available MSTM employment data all jobs are chosen based on trip
            // length frequency distributions
            workplace =
                selectWorkplaceByTripLengthFrequencyDistribution(
                    workPumaZone[s], workState[s], taz);
          }
          if (workplace != -2) {
            Job.getJobFromId(workplace)
                .setWorkerID(newPpId); // -2 for jobs outside of the study area
          }
        }
        Person pp =
            new Person(newPpId, newHhId, age[s], gender[s], race[s], occ, workplace, income[s]);
        hh.addPersonForInitialSetup(pp);
      }
      hh.setType();
      hh.setHouseholdRace();
      definePersonRolesInHousehold(hh, relShp);
      // trace persons, households and dwellings
      for (Person pp : hh.getPersons())
        if (pp.getId() == SiloUtil.trackPp) {
          SiloUtil.trackWriter.println("Generated person with following attributes:");
          Person.getPersonFromId(pp.getId()).logAttributes(SiloUtil.trackWriter);
        }
      if (newHhId == SiloUtil.trackHh) {
        SiloUtil.trackWriter.println("Generated household with following attributes:");
        Household.getHouseholdFromId(newHhId).logAttributes(SiloUtil.trackWriter);
      }
      if (newDdId == SiloUtil.trackDd) {
        SiloUtil.trackWriter.println("Generated dwelling with following attributes:");
        Dwelling.getDwellingFromId(newDdId).logAttributes(SiloUtil.trackWriter);
      }
    }
  }