from typing import List, Optional, Tuple
from rastervision.core.box import Box
from rastervision.pipeline.config import (Config, register_config, Field,
validator, ConfigError)
from rastervision.core.data.raster_transformer import RasterTransformerConfig
def rs_config_upgrader(cfg_dict: dict, version: int) -> dict:
if version == 6:
# removed in version 7
if cfg_dict.get('extent_crop') is not None:
raise ConfigError('RasterSourceConfig.extent_crop is deprecated.')
try:
del cfg_dict['extent_crop']
except KeyError:
pass
return cfg_dict
[docs]@register_config('raster_source', upgrader=rs_config_upgrader)
class RasterSourceConfig(Config):
"""Configure a :class:`.RasterSource`."""
channel_order: Optional[List[int]] = Field(
None,
description=
'The sequence of channel indices to use when reading imagery.')
transformers: List[RasterTransformerConfig] = []
extent: Optional[Tuple[int, int, int, int]] = Field(
None,
description='Use-specified extent in pixel coords in the form '
'(ymin, xmin, ymax, xmax). Useful for cropping the raster source so '
'that only part of the raster is read from.')
[docs] def build(self, tmp_dir, use_transformers=True):
raise NotImplementedError()
[docs] def update(self, pipeline=None, scene=None):
for t in self.transformers:
t.update(pipeline, scene)
[docs] @validator('extent')
def validate_extent(cls, v):
if v is None:
return None
return Box(*v)