コード例 #1
0
ファイル: Loader.java プロジェクト: aaronsama/census_utils
  /**
   * @param source FeatureSource obtained from the shapefile of the area to redistrict
   * @param galFile GAL file containing contiguity information about the area to redistrict
   */
  public BlockGraph load(FeatureSource<SimpleFeatureType, SimpleFeature> source, String galFile) {
    BlockGraph bg = new BlockGraph(source);

    // Read GAL file
    parseGal(bg, galFile);

    // remove blocks with area 0
    ArrayList<Block> toRemove = new ArrayList<Block>();
    for (Block b : bg.getAllBlocks()) {
      if (b.getArea() == 0) {
        toRemove.add(b);
      }
    }
    bg.removeAllBlocks(toRemove);

    return bg;
  }
コード例 #2
0
ファイル: Loader.java プロジェクト: aaronsama/census_utils
  /**
   * Adds the edges to the input {@link BlockGraph}
   *
   * @param bg
   * @param filename
   */
  private void parseGal(BlockGraph bg, String filename) {
    File galFile = new File(filename);

    try {
      FileInputStream fstream = new FileInputStream(galFile);
      DataInputStream in = new DataInputStream(fstream);
      BufferedReader br = new BufferedReader(new InputStreamReader(in));

      int n_blocks = Integer.parseInt(br.readLine());

      String currentline;
      int line_num = 2;

      for (int i = 0; i < n_blocks; i++, line_num++) {
        // read current block id and number of neighbors
        currentline = br.readLine();

        if (currentline == null) {
          System.err.println("Error in GAL file at line " + line_num);
          System.exit(1);
        }

        StringTokenizer block_st = new StringTokenizer(currentline);
        int current_block_id = -1;
        int num_neighbors = -1;

        try {
          current_block_id = Integer.parseInt(block_st.nextToken());
          num_neighbors = Integer.parseInt(block_st.nextToken());
        } catch (NoSuchElementException e) {
          System.err.println("Error in GAL file at line " + line_num);
          System.exit(1);
        }

        Block currentBlock = bg.getBlock(current_block_id);

        // now read neighbors ids
        currentline = br.readLine();

        if (currentline == null) {
          System.err.println("Error in GAL file at line " + line_num);
          System.exit(1);
        }

        StringTokenizer neighbor_st = new StringTokenizer(currentline);

        for (int j = 0; j < num_neighbors; j++) {
          int neighbor_id = -1;
          try {
            neighbor_id = Integer.parseInt(neighbor_st.nextToken());
          } catch (NoSuchElementException e) {
            System.err.println("Error in GAL file at line " + line_num);
            System.exit(1);
          }

          // System.out.println("Adding " + neighbor_id +
          // " to the neighbors of " + current_block_id);
          currentBlock.neighbors.add(bg.getBlock(neighbor_id));
          bg.getBlock(neighbor_id).neighbors.add(currentBlock);
        }
      }

      in.close();
    } catch (Exception e) { // Catch exception if any
      System.err.println("Error: " + e.getMessage());
      e.printStackTrace();
    }
  }