def _create_shapefile_test_data(): """ Create test data for tests """ import spatialvectors import osgeo.ogr fixture = {} # open the shapefiles test_poly_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/basins/test_poly_wgs84.shp"))) water_basin_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/basins/water_basin_wgs84.shp"))) water_basins_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/basins/water_basins_wgs84.shp"))) water_basin_pourpoint_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/basins/water_basin_pourpoint_wgs84.shp" ))) canes_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/gcm-tiles/CanES_wgs84.shp"))) wateruse_centroids_sample_wgs84_shapefile = osgeo.ogr.Open( os.path.abspath( os.path.join( os.getcwd(), "../../data/spatial-datafiles/wateruse-centroids/wateruse_centroids_sample_wgs84.shp" ))) fixture["test_poly_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=test_poly_wgs84_shapefile) fixture["water_basin_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=water_basin_wgs84_shapefile) fixture["water_basins_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=water_basins_wgs84_shapefile) fixture[ "water_basin_pourpoint_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=water_basin_pourpoint_wgs84_shapefile) fixture["canes_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=canes_wgs84_shapefile) fixture[ "wateruse_centroids_sample_wgs84"] = spatialvectors.fill_shapefile_dict( shapefile=wateruse_centroids_sample_wgs84_shapefile) return fixture
def get_shps_colors_fields(files_list, settings): """ Return shapefiles, projected to WGS84 (if needed) and lists of map display fields and colors corresponding to each shapefile. Parameters ---------- files_list : list List of shapefiles to create a map with settings : dictionary Dictionary of user settings """ # reproject all shapefiles shp_reproj_list = spatialvectors.reproject(shapefiles = files_list) shp_info_list = [] display_fields = [] colors = [] colors_index = 0 for shapefile in shp_reproj_list: shp = osgeo.ogr.Open(shapefile) shp_info = spatialvectors.fill_shapefile_dict(shapefile = shp) shp_info_list.append(shp_info) for key, values in settings["water_shapefiles"].iteritems(): if key in shp_info["name"].split("_")[0]: display_fields.append(values["field"]) colors.append(values["color"]) if not colors: display_fields.append("") colors.append(settings["map_colors_list"][colors_index]) colors_index += 1 return shp_info_list, display_fields, colors
def get_shps_colors_fields(files_list, settings): """ Return shapefiles, projected to WGS84 (if needed) and lists of map display fields and colors corresponding to each shapefile. Parameters ---------- files_list : list List of shapefiles to create a map with settings : dictionary Dictionary of user settings """ # reproject all shapefiles shp_reproj_list = spatialvectors.reproject(shapefiles=files_list) shp_info_list = [] display_fields = [] colors = [] colors_index = 0 for shapefile in shp_reproj_list: shp = osgeo.ogr.Open(shapefile) shp_info = spatialvectors.fill_shapefile_dict(shapefile=shp) shp_info_list.append(shp_info) for key, values in settings["water_shapefiles"].iteritems(): if key in shp_info["name"].split("_")[0]: display_fields.append(values["field"]) colors.append(values["color"]) if not colors: display_fields.append("") colors.append(settings["map_colors_list"][colors_index]) colors_index += 1 return shp_info_list, display_fields, colors
def _create_shapefile_test_data(): """ Create test data for tests """ import spatialvectors import osgeo.ogr fixture = {} # open the shapefiles test_poly_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/basins/test_poly_wgs84.shp"))) water_basin_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/basins/water_basin_wgs84.shp"))) water_basins_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/basins/water_basins_wgs84.shp"))) water_basin_pourpoint_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/basins/water_basin_pourpoint_wgs84.shp"))) canes_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/gcm-tiles/CanES_wgs84.shp"))) wateruse_centroids_sample_wgs84_shapefile = osgeo.ogr.Open(os.path.abspath(os.path.join(os.getcwd(), "../../data/spatial-datafiles/wateruse-centroids/wateruse_centroids_sample_wgs84.shp"))) fixture["test_poly_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = test_poly_wgs84_shapefile) fixture["water_basin_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = water_basin_wgs84_shapefile) fixture["water_basins_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = water_basins_wgs84_shapefile) fixture["water_basin_pourpoint_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = water_basin_pourpoint_wgs84_shapefile) fixture["canes_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = canes_wgs84_shapefile) fixture["wateruse_centroids_sample_wgs84"] = spatialvectors.fill_shapefile_dict(shapefile = wateruse_centroids_sample_wgs84_shapefile) return fixture