def init_roi(config_file):
"""
1) Loads configuration file
2) Checks parameters
3) Selects the ROI
4) Checks the zoom factor
"""
# read the json configuration file
f = open(config_file)
user_cfg = json.load(f)
f.close()
# Check that all the mandatory arguments are defined, and warn about
# 'unknown' params
check_parameters(user_cfg)
# fill the config module: updates the content of the config.cfg dictionary
# with the content of the user_cfg dictionary
cfg.update(user_cfg)
# sets keys 'clr', 'cld' and 'roi' of the reference image to None if they
# are not already defined. The default values of these optional arguments
# can not be defined directly in the config.py module. They would be
# overwritten by the previous update, because they are in a nested dict.
cfg['images'][0].setdefault('clr')
cfg['images'][0].setdefault('cld')
cfg['images'][0].setdefault('roi')
# update roi definition if the full_img flag is set to true
if ('full_img' in cfg) and cfg['full_img']:
sz = common.image_size_tiffinfo(cfg['images'][0]['img'])
cfg['roi'] = {}
cfg['roi']['x'] = 0
cfg['roi']['y'] = 0
cfg['roi']['w'] = sz[0]
cfg['roi']['h'] = sz[1]
# check that the roi is well defined
if 'roi' not in cfg or any(p not in cfg['roi'] for p in ['x', 'y', 'w',
'h']):
print "missing or incomplete ROI definition"
print "ROI will be redefined by interactive selection"
x, y, w, h = common.get_roi_coordinates(cfg['images'][0]['img'],
cfg['images'][0]['prv'])
cfg['roi'] = {}
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
else :
x = cfg['roi']['x']
y = cfg['roi']['y']
w = cfg['roi']['w']
h = cfg['roi']['h']
try:
print "ROI x, y, w, h = %d, %d, %d, %d" % (x, y, w, h)
except TypeError:
print 'Neither a ROI nor a preview file are defined. Aborting.'
return
# check the zoom factor
z = cfg['subsampling_factor']
assert(z > 0 and z == np.floor(z))
# ensure that the coordinates of the ROI are multiples of the zoom factor,
# to avoid bad registration of tiles due to rounding problems.
x, y, w, h = common.round_roi_to_nearest_multiple(z, x, y, w, h)
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
def init_roi(config_file):
"""
1) Loads configuration file
2) Checks parameters
3) Selects the ROI
4) Checks the zoom factor
"""
# read the json configuration file
f = open(config_file)
user_cfg = json.load(f)
f.close()
# Check that all the mandatory arguments are defined, and warn about
# 'unknown' params
check_parameters(user_cfg)
# fill the config module: updates the content of the config.cfg dictionary
# with the content of the user_cfg dictionary
cfg.update(user_cfg)
# sets keys 'clr', 'cld' and 'roi' of the reference image to None if they
# are not already defined. The default values of these optional arguments
# can not be defined directly in the config.py module. They would be
# overwritten by the previous update, because they are in a nested dict.
cfg['images'][0].setdefault('clr')
cfg['images'][0].setdefault('cld')
cfg['images'][0].setdefault('roi')
cfg['images'][0].setdefault('wat')
# update roi definition if the full_img flag is set to true
if ('full_img' in cfg) and cfg['full_img']:
sz = common.image_size_tiffinfo(cfg['images'][0]['img'])
cfg['roi'] = {}
cfg['roi']['x'] = 0
cfg['roi']['y'] = 0
cfg['roi']['w'] = sz[0]
cfg['roi']['h'] = sz[1]
# check that the roi is well defined
if 'roi' not in cfg or any(p not in cfg['roi']
for p in ['x', 'y', 'w', 'h']):
print "missing or incomplete ROI definition"
print "ROI will be redefined by interactive selection"
x, y, w, h = common.get_roi_coordinates(cfg['images'][0]['img'],
cfg['images'][0]['prv'])
cfg['roi'] = {}
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
else:
x = cfg['roi']['x']
y = cfg['roi']['y']
w = cfg['roi']['w']
h = cfg['roi']['h']
try:
print "ROI x, y, w, h = %d, %d, %d, %d" % (x, y, w, h)
except TypeError:
print 'Neither a ROI nor a preview file are defined. Aborting.'
return
# check the zoom factor
z = cfg['subsampling_factor']
assert (z > 0 and z == np.floor(z))
# ensure that the coordinates of the ROI are multiples of the zoom factor,
# to avoid bad registration of tiles due to rounding problems.
x, y, w, h = common.round_roi_to_nearest_multiple(z, x, y, w, h)
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
# get utm zone
utm_zone = rpc_utils.utm_zone(
cfg['images'][0]['rpc'],
*[cfg['roi'][v] for v in ['x', 'y', 'w', 'h']])
cfg['utm_zone'] = utm_zone
def init_dirs_srtm_roi(config_file):
"""
1) Loads configuration file
2) Checks parameters
3) Selects the ROI
4) Checks the zoom factor
5) Creates different directories : output, temp...
Args:
config_file : path to a json configuration file
"""
# read the json configuration file
f = open(config_file)
user_cfg = json.load(f)
f.close()
# Check that all the mandatory arguments are defined, and warn about
# 'unknown' params
check_parameters(user_cfg)
# fill the config module: updates the content of the config.cfg dictionary
# with the content of the user_cfg dictionary
cfg.update(user_cfg)
# sets keys 'clr', 'cld' and 'roi' of the reference image to None if they
# are not already defined. The default values of these optional arguments
# can not be defined directly in the config.py module. They would be
# overwritten by the previous update, because they are in a nested dict.
cfg['images'][0].setdefault('clr')
cfg['images'][0].setdefault('cld')
cfg['images'][0].setdefault('roi')
# update roi definition if the full_img flag is set to true
if ('full_img' in cfg) and cfg['full_img']:
sz = common.image_size_tiffinfo(cfg['images'][0]['img'])
cfg['roi'] = {}
cfg['roi']['x'] = 0
cfg['roi']['y'] = 0
cfg['roi']['w'] = sz[0]
cfg['roi']['h'] = sz[1]
# check that the roi is well defined
if 'roi' not in cfg or any(p not in cfg['roi'] for p in ['x', 'y', 'w',
'h']):
print "missing or incomplete ROI definition"
print "ROI will be redefined by interactive selection"
x, y, w, h = common.get_roi_coordinates(cfg['images'][0]['img'],
cfg['images'][0]['prv'])
cfg['roi'] = {}
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
else:
x = cfg['roi']['x']
y = cfg['roi']['y']
w = cfg['roi']['w']
h = cfg['roi']['h']
try:
print "ROI x, y, w, h = %d, %d, %d, %d" % (x, y, w, h)
except TypeError:
print 'Neither a ROI nor a preview file are defined. Aborting.'
return
# check the zoom factor
z = cfg['subsampling_factor']
assert(z > 0 and z == np.floor(z))
# create tmp dir and output directory for the experiment, and store a json
# dump of the config.cfg dictionary there, download srtm files...
if not os.path.exists(cfg['out_dir']):
os.makedirs(cfg['out_dir'])
if not os.path.exists(cfg['temporary_dir']):
os.makedirs(cfg['temporary_dir'])
if not os.path.exists(os.path.join(cfg['temporary_dir'], 'meta')):
os.makedirs(os.path.join(cfg['temporary_dir'], 'meta'))
f = open('%s/config.json' % cfg['out_dir'], 'w')
json.dump(cfg, f, indent=2)
f.close()
# duplicate stdout and stderr to log file
tee.Tee('%s/stdout.log' % cfg['out_dir'], 'w')
# needed srtm tiles
srtm_tiles = srtm.list_srtm_tiles(cfg['images'][0]['rpc'],
*cfg['roi'].values())
for s in srtm_tiles:
srtm.get_srtm_tile(s, cfg['srtm_dir'])
def main(config_file):
"""
Launches s2p with the parameters given by a json file.
Args:
config_file: path to the config json file
"""
# read the json configuration file
f = open(config_file)
user_cfg = json.load(f)
f.close()
# Check that all the mandatory arguments are defined, and warn about
# 'unknown' params
check_parameters(user_cfg)
# fill the config module: updates the content of the config.cfg dictionary
# with the content of the user_cfg dictionary
cfg.update(user_cfg)
# sets keys 'clr', 'cld' and 'roi' of the reference image to None if they
# are not already defined. The default values of these optional arguments
# can not be defined directly in the config.py module. They would be
# overwritten by the previous update, because they are in a nested dict.
cfg['images'][0].setdefault('clr')
cfg['images'][0].setdefault('cld')
cfg['images'][0].setdefault('roi')
# update roi definition if the full_img flag is set to true
if ('full_img' in cfg) and cfg['full_img']:
sz = common.image_size_tiffinfo(cfg['images'][0]['img'])
cfg['roi'] = {}
cfg['roi']['x'] = 0
cfg['roi']['y'] = 0
cfg['roi']['w'] = sz[0]
cfg['roi']['h'] = sz[1]
# check that the roi is well defined
if 'roi' not in cfg or any(p not in cfg['roi'] for p in ['x', 'y', 'w',
'h']):
print "missing or incomplete ROI definition"
print "ROI will be redefined by interactive selection"
x, y, w, h = common.get_roi_coordinates(cfg['images'][0]['img'],
cfg['images'][0]['prv'])
cfg['roi'] = {}
cfg['roi']['x'] = x
cfg['roi']['y'] = y
cfg['roi']['w'] = w
cfg['roi']['h'] = h
# check the zoom factor
z = cfg['subsampling_factor']
assert(z > 0 and z == np.floor(z))
# create tmp dir and output directory for the experiment, and store a json
# dump of the config.cfg dictionary there
if not os.path.exists(cfg['temporary_dir']):
os.makedirs(cfg['temporary_dir'])
if not os.path.exists(os.path.join(cfg['temporary_dir'], 'meta')):
os.makedirs(os.path.join(cfg['temporary_dir'], 'meta'))
if not os.path.exists(cfg['out_dir']):
os.makedirs(cfg['out_dir'])
f = open('%s/config.json' % cfg['out_dir'], 'w')
json.dump(cfg, f, indent=2)
f.close()
# measure total runtime
t0 = time.time()
# needed srtm tiles
srtm_tiles = srtm.list_srtm_tiles(cfg['images'][0]['rpc'],
*cfg['roi'].values())
for s in srtm_tiles:
srtm.get_srtm_tile(s, cfg['srtm_dir'])
# height map
if len(cfg['images']) == 2:
height_map = process_pair(cfg['out_dir'], cfg['images'][0]['img'],
cfg['images'][0]['rpc'], cfg['images'][1]['img'],
cfg['images'][1]['rpc'], cfg['roi']['x'],
cfg['roi']['y'], cfg['roi']['w'], cfg['roi']['h'],
None, None, None, cfg['images'][0]['cld'],
cfg['images'][0]['roi'])
else:
height_map = process_triplet(cfg['out_dir'], cfg['images'][0]['img'],
cfg['images'][0]['rpc'], cfg['images'][1]['img'],
cfg['images'][1]['rpc'], cfg['images'][2]['img'],
cfg['images'][2]['rpc'], cfg['roi']['x'],
cfg['roi']['y'], cfg['roi']['w'], cfg['roi']['h'],
cfg['fusion_thresh'], None, None, None, None,
cfg['images'][0]['cld'], cfg['images'][0]['roi'])
# point cloud
generate_cloud(cfg['out_dir'], height_map, cfg['images'][0]['rpc'],
cfg['roi']['x'], cfg['roi']['y'], cfg['roi']['w'],
cfg['roi']['h'], cfg['images'][0]['img'],
cfg['images'][0]['clr'], cfg['offset_ply'])
# digital surface model
out_dsm = '%s/dsm.tif' % cfg['out_dir']
point_clouds_list = glob.glob('%s/cloud.ply' % cfg['out_dir'])
generate_dsm(out_dsm, point_clouds_list, cfg['dsm_resolution'])
# crop corresponding areas in the secondary images
if not cfg['full_img']:
crop_corresponding_areas(cfg['out_dir'], cfg['images'], cfg['roi'])
# runtime
t = int(time.time() - t0)
h = t/3600
m = (t/60) % 60
s = t % 60
print "Total runtime: %dh:%dm:%ds" % (h, m, s)
common.garbage_cleanup()
# read the json configuration file
import json
f = open(config)
cfg = json.load(f)
f.close()
if 'preview_coordinate_system' in cfg:
# roi definition in the preview frame
xx = cfg['roi_preview']['x']
yy = cfg['roi_preview']['y']
w = cfg['roi']['w']
h = cfg['roi']['h']
img1 = cfg['images'][0]['img']
prv_w, prv_h = common.image_size(preview)
full_w, full_h = common.image_size_tiffinfo(img1)
# convert x, y to the full image frame
x = int((float(xx) / prv_w) * full_w)
y = int((float(yy) / prv_h) * full_h)
# add offset
cfg['roi']['x'] = x - w / 2
cfg['roi']['y'] = y - h / 2
# convert roi definition in the preview coordinate system
ww = int(w * (float(prv_w) / full_w))
hh = int(h * (float(prv_h) / full_h))
xx -= ww / 2
yy -= hh / 2
cfg['roi_preview']['w'] = ww