Пример #1
0
 def write(self, data, pixelbuffer=0):
     if self.output.data_type == "vector":
         write_vector(self, self.tile.config, data, pixelbuffer=pixelbuffer, overwrite=self.config.overwrite)
     elif self.output.data_type == "raster":
         write_raster(self, self.tile.profile(pixelbuffer=pixelbuffer), data, pixelbuffer=pixelbuffer)
Пример #2
0
    def execute(self, tile, overwrite=True):
        """
        Processes and saves tile.
        Returns a tuple of three values:
        - Tile ID: zoom, row, col
        - status: "exists", "empty", "failed" or "processed"
        - error message: optional or None
        """
        # Do nothing if tile exists or overwrite is turned off.
        if not overwrite and tile.exists():
            return tile.id, "exists", None
        try:
            new_process = imp.load_source(self.process_name + "Process", self.config.process_file)
            tile_process = new_process.Process(config=self.config, tile=tile, params=self.config.at_zoom(tile.zoom))
        except:
            return tile.id, "failed", traceback.print_exc()

        # Generate tile using the user defined process.
        if not self.config.baselevel or (tile.zoom == self.config.baselevel["zoom"]):
            try:
                result = tile_process.execute()
            except:
                return tile.id, "failed", traceback.print_exc()
            finally:
                tile_process = None

            message = None
            if result:
                if result == "empty":
                    status = "empty"
                else:
                    status = "custom"
                    message = result
            else:
                status = "processed"
            return tile.id, status, message

        # If baselevel interpolation is activated, generate from neighbor zooms.
        else:
            if tile.zoom < self.config.baselevel["zoom"]:
                # determine tiles from next zoom level
                # TODO create option for on demand processing if subtile is not
                # available.

                # create temporary VRT and create new tile from resampled
                # subtiles.
                children_paths = [child.path for child in tile.get_children() if child.exists()]
                if len(children_paths) == 0:
                    return tile.id, "empty", None
                temp_vrt = NamedTemporaryFile()
                build_vrt = "gdalbuildvrt %s %s > /dev/null" % (temp_vrt.name, " ".join(children_paths))
                try:
                    os.system(build_vrt)
                    assert os.path.isfile(temp_vrt.name)
                except:
                    build_vrt = "gdalbuildvrt %s %s" % (temp_vrt.name, " ".join(children_paths))
                    os.system(build_vrt)
                    return tile.id, "failed", "GDAL VRT building"
                try:
                    assert os.path.isfile(temp_vrt.name)
                    bands = tuple(
                        read_raster_window(temp_vrt.name, tile, resampling=self.config.baselevel["resampling"])
                    )
                except:
                    return tile.id, "failed", traceback.print_exc()
                try:
                    write_raster(tile_process, self.output.profile, bands)
                    return tile.id, "processed", None
                except:
                    return tile.id, "failed", traceback.print_exc()

            elif tile.zoom > self.config.baselevel["zoom"]:
                # determine tiles from previous zoom level
                # check if tiles exist and if not, execute subtiles
                parent = tile.get_parent()
                if not overwrite and parent.exists():
                    # LOGGER.info((tile.id, "exists", None))
                    pass
                else:
                    pass
                if not parent.exists():
                    # TODO create option for on demand processing
                    return tile.id, "empty", "source tile does not exist"
                try:
                    bands = tuple(read_raster_window(parent.path, tile, resampling=self.config.baselevel["resampling"]))
                    write_raster(tile_process, self.output.profile, bands)
                    return tile.id, "processed", None
                except:
                    return tile.id, "failed", traceback.print_exc()