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]