GIS-Scripts/ClipDatasets.py at master · MACBIO/GIS-Scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import os
import subprocess
import sys

inFolder = sys.argv[1]
outFolder = os.path.join(os.path.dirname(inFolder), 'Clipped')

### Fiji
##ulx = 160
##uly = 0
##lrx = 195
##lry = -30

### Solomon Islands
##ulx = 145
##uly = 0
##lrx = 180
##lry = -20

### Tonga
##ulx = -180
##uly = -10
##lrx = -165
##lry = -30

# Vanuatu
ulx = 155
uly = -5
lrx = 180
lry = -30

ogr2ogrFile = r"C:\OSGeo4W64\bin\ogr2ogr.exe"
gdal_translateFile = r"C:\OSGeo4W64\bin\gdal_translate.exe"

# function that sends command to console
def check_output(command,console):
    if console == True:
        process = subprocess.Popen(command)
    else:
        process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
    output,error = process.communicate()
    returncode = process.poll()
    return returncode,output

# create list of shapefiles
fileList = []
for root, dirs, files in os.walk(inFolder):
    for f in files:
        if f.endswith('.shp'):
            fileList.append(os.path.join(root, f))

# process each shapefile
for f in fileList:
    inFile = f
    outTail = inFile.split(os.path.basename(inFolder))[1]
    outFile = outFolder + outTail
    if not os.path.exists(os.path.dirname(outFile)):
        os.makedirs(os.path.dirname(outFile))
    if not os.path.exists(outFile):
        args = []
        args.append('"'+ogr2ogrFile+'"')
        args.append('-skipfailures')
        args.append('-progress')
        args.append('"'+outFile+'"')
        args.append('"'+inFile+'"')
        args.append('-clipsrc')
        args.append(str(ulx))
        args.append(str(lry))
        args.append(str(lrx))
        args.append(str(uly))
        command = " ".join(args)
        print command
        returncode,output = check_output(command, True)
        print output

# create list of tiffs
fileList = []
for root, dirs, files in os.walk(inFolder):
    for f in files:
        if f.endswith('.tif'):
            fileList.append(os.path.join(root, f))

# process each tiff
for f in fileList:
    inFile = f
    outTail = inFile.split(os.path.basename(inFolder))[1]
    outFile = outFolder + outTail
    if not os.path.exists(os.path.dirname(outFile)):
        os.makedirs(os.path.dirname(outFile))
    if not os.path.exists(outFile):
        args = []
        args.append('"'+gdal_translateFile+'"')
        args.append('-projwin')
        args.append(str(ulx))
        args.append(str(uly))
        args.append(str(lrx))
        args.append(str(lry))
        args.append('"'+inFile+'"')
        args.append('"'+outFile+'"')
        command = " ".join(args)
        print command
        returncode,output = check_output(command, True)
        print output