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
|
###############################################################################
# pytest_crop.py: crop
# Author(s): Pieter.Kempeneers@ec.europa.eu
# Copyright (c) 2016-2019 European Union (Joint Research Centre)
# License EUPLv1.2
#
# This file is part of jiplib
###############################################################################
# History
# 2018/03/01 - Created by Pieter Kempeneers (pieter.kempeneers@ec.europa.eu)
# Change log
import argparse
import os
import math
import jiplib as jl
parser=argparse.ArgumentParser()
parser.add_argument("-input","--input",help="Path of the raster dataset",dest="input",required=True,type=str)
parser.add_argument("-extent","--extent",help="Path of the vector dataset",dest="extent",required=False,type=str)
parser.add_argument("-output","--output",help="Path of the output vector dataset",dest="output",required=False,type=str)
parser.add_argument("-nodata","--nodata",help="nodata value to put in cut region",dest="nodata",required=False,default=0,type=int)
parser.add_argument("-cut_to_cutline","--cut_to_cutline",help="Crop the extent of the target dataset to the extent of the cutline, setting the outside area to nodata",dest="cut_to_cutline",required=False,type=bool,default=True)
parser.add_argument("-cut_in_cutline","--cut_in_cutline",help="Crop the extent of the target dataset to the extent of the cutline, setting the inner area to nodata",dest="cut_in_cutline",required=False,type=bool,default=False)
args = parser.parse_args()
try:
print("createJim")
jim0=jl.createJim(args.input)
rules=['centroid','min','max','mean','stdev']
print("createVector")
jlv=jl.createVector(args.extent)
bb=jlv.getBoundingBox()
if os.path.basename(args.extent)=='nuts_italy.sqlite':
if args.cut_to_cutline:
print("cut out")
print("milano")
jim_milano=jim0.cropOgr(jlv,{'ln':'milano','crop_to_cutline':True,'nodata':args.nodata,'align':True})
bb=jim_milano.getBoundingBox()
jim1=jim0.crop({'ulx':bb[0],'uly':bb[1],'lrx':bb[2],'lry':bb[3],'verbose':2})
print(jim1.getBoundingBox())
print(jim_milano.getBoundingBox())
if jim1.getBoundingBox() != jim_milano.getBoundingBox():
raise ValueError("Error: Bounding box milano not equal with crop and cropOgr")
print("lodi")
jim_lodi=jim0.cropOgr(jlv,{'ln':'lodi','crop_to_cutline':True,'nodata':args.nodata,'align':True})
elif args.cut_in_cutline:
print("cut within")
# to keep region
jim_milano=jim0.cropOgr(jlv,{'ln':'milano','crop_in_cutline':True,'nodata':args.nodata,'align':True})
jim_lodi=jim0.cropOgr(jlv,{'ln':'lodi','crop_in_cutline':True,'nodata':args.nodata,'align':True})
milanofn=os.path.join(os.path.dirname(args.output),'milano.tif')
print("write")
lodifn=os.path.join(os.path.dirname(args.output),'lodi.tif')
jim_milano.write({'filename':milanofn})
jim_milano.close()
jim_lodi.write({'filename':lodifn})
jim_lodi.close()
else:
jim0.cropOgr(jlv,{'align':True}).write({'filename':args.output}).close()
jim0.close()
jlv.close()
print("Success: crop")
except:
print("Failed: crop")
|
