예제 #1
0
  public static void printFileInformation(File file) throws IOException {
    RandomAccessFile r = new RandomAccessFile(file.getAbsolutePath(), "r");
    try {
      BinaryMapIndexReader index = new BinaryMapIndexReader(r);
      int i = 1;
      System.out.println("Binary index " + file.getName() + " version = " + index.getVersion());
      for (BinaryIndexPart p : index.getIndexes()) {
        String partname = "";
        if (p instanceof MapIndex) {
          partname = "Map";
        } else if (p instanceof TransportIndex) {
          partname = "Transport";
        } else if (p instanceof AddressRegion) {
          partname = "Address";
        }
        String name = p.getName() == null ? "" : p.getName();
        System.out.println(
            MessageFormat.format(
                "{0}. {1} data {3} - {2} bytes", i, partname, p.getLength(), name));
        if (p instanceof TransportIndex) {
          TransportIndex ti = ((TransportIndex) p);
          int sh = (31 - BinaryMapIndexReader.TRANSPORT_STOP_ZOOM);
          System.out.println(
              "\t Bounds "
                  + formatBounds(
                      ti.getLeft() << sh,
                      ti.getRight() << sh,
                      ti.getTop() << sh,
                      ti.getBottom() << sh));
        } else if (p instanceof MapIndex) {
          MapIndex m = ((MapIndex) p);
          int j = 1;
          for (MapRoot mi : m.getRoots()) {
            System.out.println(
                MessageFormat.format(
                    "\t{4}.{5} Map level minZoom = {0}, maxZoom = {1}, size = {2} bytes \n\t\tBounds {3}",
                    mi.getMinZoom(),
                    mi.getMaxZoom(),
                    mi.getLength(),
                    formatBounds(mi.getLeft(), mi.getRight(), mi.getTop(), mi.getBottom()),
                    i,
                    j++));
          }
        }
        i++;
      }

    } catch (IOException e) {
      System.err.println("File is not valid index : " + file.getAbsolutePath());
      throw e;
    }
  }
예제 #2
0
  @SuppressWarnings("unchecked")
  public static List<Float> combineParts(File fileToExtract, Map<File, String> partsToExtractFrom)
      throws IOException {
    BinaryMapIndexReader[] indexes = new BinaryMapIndexReader[partsToExtractFrom.size()];
    RandomAccessFile[] rafs = new RandomAccessFile[partsToExtractFrom.size()];

    LinkedHashSet<Float>[] partsSet = new LinkedHashSet[partsToExtractFrom.size()];
    int c = 0;
    Set<String> addressNames = new LinkedHashSet<String>();

    int version = -1;
    // Go through all files and validate conistency
    for (File f : partsToExtractFrom.keySet()) {
      if (f.getAbsolutePath().equals(fileToExtract.getAbsolutePath())) {
        System.err.println("Error : Input file is equal to output file " + f.getAbsolutePath());
        return null;
      }
      rafs[c] = new RandomAccessFile(f, "r");
      indexes[c] = new BinaryMapIndexReader(rafs[c]);
      partsSet[c] = new LinkedHashSet<Float>();
      if (version == -1) {
        version = indexes[c].getVersion();
      } else {
        if (indexes[c].getVersion() != version) {
          System.err.println(
              "Error : Different input files has different input versions "
                  + indexes[c].getVersion()
                  + " != "
                  + version);
          return null;
        }
      }

      LinkedHashSet<Float> temp = new LinkedHashSet<Float>();
      String pattern = partsToExtractFrom.get(f);
      boolean minus = true;
      for (int i = 0; i < indexes[c].getIndexes().size(); i++) {
        partsSet[c].add(i + 1f);
        BinaryIndexPart part = indexes[c].getIndexes().get(i);
        if (part instanceof MapIndex) {
          List<MapRoot> roots = ((MapIndex) part).getRoots();
          for (int j = 0; j < roots.size(); j++) {
            partsSet[c].add((i + 1f) + (j + 1) / 10f);
          }
        }
      }
      if (pattern != null) {
        minus = pattern.startsWith("-");
        String[] split = pattern.substring(1).split(",");
        for (String s : split) {
          temp.add(Float.parseFloat(s));
        }
      }

      if (minus) {
        partsSet[c].removeAll(temp);
      } else {
        partsSet[c].retainAll(temp);
      }

      c++;
    }

    // write files
    FileOutputStream fout = new FileOutputStream(fileToExtract);
    CodedOutputStream ous = CodedOutputStream.newInstance(fout, BUFFER_SIZE);
    List<Float> list = new ArrayList<Float>();
    byte[] BUFFER_TO_READ = new byte[BUFFER_SIZE];

    ous.writeInt32(OsmandOdb.OsmAndStructure.VERSION_FIELD_NUMBER, version);

    for (int k = 0; k < indexes.length; k++) {
      LinkedHashSet<Float> partSet = partsSet[k];
      BinaryMapIndexReader index = indexes[k];
      RandomAccessFile raf = rafs[k];
      for (int i = 0; i < index.getIndexes().size(); i++) {
        if (!partSet.contains(i + 1f)) {
          continue;
        }
        list.add(i + 1f);

        BinaryIndexPart part = index.getIndexes().get(i);
        String map;
        if (part instanceof MapIndex) {
          ous.writeTag(
              OsmandOdb.OsmAndStructure.MAPINDEX_FIELD_NUMBER,
              WireFormat.WIRETYPE_FIXED32_LENGTH_DELIMITED);
          map = "Map";
          List<MapRoot> roots = ((MapIndex) part).getRoots();
          List<MapRoot> toSkip = new ArrayList<MapRoot>();
          int newL = 0;
          int tagAndFieldSize =
              CodedOutputStream.computeTagSize(OsmandOdb.OsmAndMapIndex.LEVELS_FIELD_NUMBER) + 4;
          for (int j = 0; j < roots.size(); j++) {
            if (!partSet.contains(i + 1f + (j + 1) * 0.1f)) {
              newL -= (roots.get(j).getLength() + tagAndFieldSize);
              toSkip.add(roots.get(j));
            }
          }

          writeInt(ous, part.getLength() + newL);
          long seek = part.getFilePointer();
          while (seek < (part.getFilePointer() + part.getLength())) {
            MapRoot next = null;
            for (MapRoot r : toSkip) {
              if (seek < r.getFilePointer()) {
                if (next == null || next.getFilePointer() > r.getFilePointer()) {
                  next = r;
                }
              }
            }
            if (next == null) {
              copyBinaryPart(
                  ous,
                  BUFFER_TO_READ,
                  raf,
                  seek,
                  (int) (part.getLength() - (seek - part.getFilePointer())));
              break;
            } else {
              int l = (int) (next.getFilePointer() - seek - tagAndFieldSize);
              if (l > 0) {
                copyBinaryPart(ous, BUFFER_TO_READ, raf, seek, l);
              }
              seek += next.getLength() + tagAndFieldSize + l;
            }
          }

          System.out.println(
              MessageFormat.format(
                  "{2} part {0} is extracted {1} bytes",
                  part.getName(), part.getLength() + newL, map));
        } else {
          if (part instanceof AddressRegion) {
            ous.writeTag(
                OsmandOdb.OsmAndStructure.ADDRESSINDEX_FIELD_NUMBER,
                WireFormat.WIRETYPE_FIXED32_LENGTH_DELIMITED);
            map = "Address";
            if (addressNames.contains(part.getName())) {
              System.err.println(
                  "Error : going to merge 2 addresses with same names. Skip " + part.getName());
              continue;
            }
            addressNames.add(part.getName());
          } else if (part instanceof TransportIndex) {
            ous.writeTag(
                OsmandOdb.OsmAndStructure.TRANSPORTINDEX_FIELD_NUMBER,
                WireFormat.WIRETYPE_FIXED32_LENGTH_DELIMITED);
            map = "Transport";
          } else {
            throw new UnsupportedOperationException();
          }
          writeInt(ous, part.getLength());
          copyBinaryPart(ous, BUFFER_TO_READ, raf, part.getFilePointer(), part.getLength());
          System.out.println(
              MessageFormat.format(
                  "{2} part {0} is extracted {1} bytes", part.getName(), part.getLength(), map));
        }
      }
    }

    ous.writeInt32(OsmandOdb.OsmAndStructure.VERSIONCONFIRM_FIELD_NUMBER, version);
    ous.flush();
    fout.close();

    return list;
  }