import sys import cPickle sys.path.insert(0, '/home/shomea/a/anderovo/Dropbox/watershed/trapAnalysis/lib') sys.path.insert(0, '/home/shomea/a/anderovo/Dropbox/watershedLargeFiles') import load_geotiff import plot import util saved_files = '/home/shomea/a/anderovo/Dropbox/watershedLargeFiles/' file_name = saved_files + 'anders_hoh.tiff' """ Plot all combined minimums in the landscape after doing the standard method. """ landscape = load_geotiff.get_landscape(file_name) combined_minimums = cPickle.load(open(saved_files + 'minimumsInWatersheds.pkl', 'rb')) plot.plot_combined_minimums(combined_minimums, landscape)
import load_geotiff
import plot
import numpy as np
saved_files = '/home/shomea/a/anderovo/Dropbox/watershedLargeFiles/'
heights_file = saved_files + 'anders_hoh.tiff'
lakes_file = saved_files + 'anders_innsjo.tiff'
rivers_file = saved_files + 'anders_elvbekk.tiff'
small_rivers_file = saved_files + 'anders_elvmidtlinje.tiff'
marshes_file = saved_files + 'anders_myr.tiff'
"""
Plot all lakes, rivers and marshes from the data files.
"""
landscape = load_geotiff.get_landscape(heights_file)
lakes, rivers, small_rivers, marshes = load_geotiff.get_lake_river_marsh_information_tyrifjorden(
landscape, lakes_file, rivers_file, small_rivers_file, marshes_file)
lakes = lakes.flatten().astype(int)
rivers = rivers.flatten().astype(int)
small_rivers = small_rivers.flatten().astype(int)
marshes = marshes.flatten().astype(int)
lakes = np.where(lakes == 1)[0]
rivers = np.where(rivers == 1)[0]
small_rivers = np.where(small_rivers == 1)[0]
marshes = np.where(marshes == 1)[0]
