import cv2 import pyexiv2 import re from fractions import Fraction from opensfm.exif import sensor_string from pyproj import Proj import log import io import system import context class ODM_Photo: """ ODMPhoto - a class for ODMPhotos """ def __init__(self, path_file, force_focal, force_ccd): # general purpose self.path_file = path_file self.filename = io.extract_file_from_path_file(path_file) # useful attibutes self.width = None self.height = None self.ccd_width = None self.focal_length = None self.focal_length_px = None # other attributes self.camera_make = '' self.camera_model = '' self.make_model = '' self.latitude = None self.longitude = None self.altitude = None # parse values from metadata self.parse_pyexiv2_values(self.path_file, force_focal, force_ccd) # compute focal length into pixels self.update_focal() # print log message log.ODM_DEBUG('Loaded {} | camera: {} | dimensions: {} x {} | focal: {} | ccd: {} | lat: {} | lon: {} | alt: {}' .format(self.filename, self.make_model, self.width, self.height, self.focal_length, self.ccd_width, self.latitude, self.longitude, self.altitude)) def update_focal(self): # compute focal length in pixels if self.focal_length and self.ccd_width: # take width or height as reference if self.width > self.height: # f(px) = w(px) * f(mm) / ccd(mm) self.focal_length_px = \ self.width * (self.focal_length / self.ccd_width) else: # f(px) = h(px) * f(mm) / ccd(mm) self.focal_length_px = \ self.height * (self.focal_length / self.ccd_width) def parse_pyexiv2_values(self, _path_file, _force_focal, _force_ccd): # read image metadata metadata = pyexiv2.ImageMetadata(_path_file) metadata.read() # loop over image tags for key in metadata: # try/catch tag value due to weird bug in pyexiv2 # ValueError: invalid literal for int() with base 10: '' GPS = 'Exif.GPSInfo.GPS' try: # parse tag names if key == 'Exif.Image.Make': self.camera_make = metadata[key].value elif key == 'Exif.Image.Model': self.camera_model = metadata[key].value elif key == 'Exif.Photo.FocalLength': self.focal_length = float(metadata[key].value) elif key == GPS + 'Latitude': self.latitude = self.dms_to_decimal(*metadata[key].value + [metadata[GPS + 'LatitudeRef'].value]) elif key == GPS + 'Longitude': self.longitude = self.dms_to_decimal(*metadata[key].value + [metadata[GPS + 'LongitudeRef'].value]) elif key == GPS + 'Altitude': self.altitude = float(metadata[key].value) if metadata[GPS + 'AltitudeRef'] and int(metadata[GPS + 'AltitudeRef'].value) > 0: self.altitude *= -1. except (pyexiv2.ExifValueError, ValueError) as e: pass except KeyError as e: log.ODM_DEBUG('Tag not set') except NotImplementedError as e: pass if self.camera_make and self.camera_model: self.make_model = sensor_string(self.camera_make, self.camera_model) # needed to do that since sometimes metadata contains wrong data img = cv2.imread(_path_file) self.width = img.shape[1] self.height = img.shape[0] # force focal and ccd_width with user parameter if _force_focal: self.focal_length = _force_focal if _force_ccd: self.ccd_width = _force_ccd # find ccd_width from file if needed if self.ccd_width is None and self.camera_model is not None: # load ccd_widths from file ccd_widths = system.get_ccd_widths() # search ccd by camera model key = [x for x in ccd_widths.keys() if self.make_model in x] # convert to float if found if key: self.ccd_width = float(ccd_widths[key[0]]) else: log.ODM_WARNING('Could not find ccd_width in file. Use --force-ccd or edit the sensor_data.json ' 'file to manually input ccd width') def dms_to_decimal(self, degrees, minutes, seconds, sign=' '): """Converts dms coords to decimal degrees""" return (-1 if sign[0] in 'SWsw' else 1) * ( float(degrees) + float(minutes) / 60 + float(seconds) / 3600 ) # TODO: finish this class class ODM_Reconstruction(object): """docstring for ODMReconstruction""" def __init__(self, photos, projstring = None, coords_file = None): self.photos = photos # list of ODM_Photos self.projection = None # Projection system the whole project will be in self.georef = None if projstring: self.projection = self.set_projection(projstring) self.georef = ODM_GeoRef(self.projection) else: self.projection = self.parse_coordinate_system(coords_file) if self.projection: self.georef = ODM_GeoRef(self.projection) def parse_coordinate_system(self, _file): """Write attributes to jobOptions from coord file""" # check for coordinate file existence if not io.file_exists(_file): log.ODM_WARNING('Could not find file %s' % _file) return with open(_file) as f: # extract reference system and utm zone from first line. # We will assume the following format: # 'WGS84 UTM 17N' or 'WGS84 UTM 17N \n' line = f.readline().rstrip() log.ODM_DEBUG('Line: %s' % line) ref = line.split(' ') # match_wgs_utm = re.search('WGS84 UTM (\d{1,2})(N|S)', line, re.I) if ref[0] == 'WGS84' and ref[1] == 'UTM': # match_wgs_utm: datum = ref[0] utm_pole = ref[2][len(ref[2]) - 1] utm_zone = int(ref[2][:len(ref[2]) - 1]) return Proj(proj="utm", zone=utm_zone, datum=datum, no_defs=True) elif '+proj' in line: return Proj(line.strip('\'')) elif 'epsg' in line.lower(): return Proj(init=line) else: raise log.ODM_ERROR('Could not parse coordinates. Bad CRS supplied: %s' % line) def set_projection(self, projstring): try: return Proj(projstring) except RuntimeError: log.ODM_EXCEPTION('Could not set projection. Please use a proj4 string') class ODM_GCPoint(object): """docstring for ODMPoint""" def __init__(self, x, y, z): self.x = x self.y = y self.z = z class ODM_GeoRef(object): """docstring for ODMUtmZone""" def __init__(self, projection): self.projection = projection self.datum = 'WGS84' self.epsg = None self.utm_zone = 0 self.utm_pole = 'N' self.utm_east_offset = 0 self.utm_north_offset = 0 self.transform = [] self.gcps = [] def calculate_EPSG(self, _utm_zone, _pole): """Calculate and return the EPSG""" if _pole == 'S': return 32700 + _utm_zone elif _pole == 'N': return 32600 + _utm_zone else: log.ODM_ERROR('Unknown pole format %s' % _pole) return def calculate_EPSG(self, proj): return proj def coord_to_fractions(self, coord, refs): deg_dec = abs(float(coord)) deg = int(deg_dec) minute_dec = (deg_dec - deg) * 60 minute = int(minute_dec) sec_dec = (minute_dec - minute) * 60 sec_dec = round(sec_dec, 3) sec_denominator = 1000 sec_numerator = int(sec_dec * sec_denominator) if float(coord) >= 0: latRef = refs[0] else: latRef = refs[1] output = str(deg) + '/1 ' + str(minute) + '/1 ' + str(sec_numerator) + '/' + str(sec_denominator) return output, latRef def convert_to_las(self, _file, _file_out, json_file): if not self.projection.srs: log.ODM_ERROR('Empty CRS: Could not convert to LAS') return kwargs = {'bin': context.pdal_path, 'f_in': _file, 'f_out': _file_out, 'east': self.utm_east_offset, 'north': self.utm_north_offset, 'srs': self.projection.srs, 'json': json_file} # create pipeline file transform.xml to enable transformation pipeline = '{{' \ ' "pipeline":[' \ ' "untransformed.ply",' \ ' {{' \ ' "a_srs":"{srs}",' \ ' "offset_x":"{east}",' \ ' "offset_y":"{north}",' \ ' "offset_z":"0",' \ ' "filename":"transformed.las"' \ ' }}' \ ' ]' \ '}}'.format(**kwargs) with open(json_file, 'w') as f: f.write(pipeline) # call pdal system.run('{bin}/pdal pipeline -i {json} --readers.ply.filename={f_in} ' '--writers.las.filename={f_out}'.format(**kwargs)) def utm_to_latlon(self, _file, _photo, idx): gcp = self.gcps[idx] kwargs = {'proj': self.projection, 'file': _file, 'x': gcp.x + self.utm_east_offset, 'y': gcp.y + self.utm_north_offset, 'z': gcp.z} latlon = system.run_and_return('echo {x} {y} {z} '.format(**kwargs), 'gdaltransform -s_srs \"{proj}\" ' '-t_srs \"EPSG:4326\"'.format(**kwargs)).split() # Example: 83d18'16.285"W # Example: 41d2'11.789"N # Example: 0.998 if len(latlon) == 3: lon_str, lat_str, alt_str = latlon elif len(latlon) == 2: lon_str, lat_str = latlon alt_str = '' else: log.ODM_ERROR('Something went wrong %s' % latlon) lat_frac = self.coord_to_fractions(latlon[1], ['N', 'S']) lon_frac = self.coord_to_fractions(latlon[0], ['E', 'W']) # read image metadata metadata = pyexiv2.ImageMetadata(_photo.path_file) metadata.read() # #set values # # # GPS latitude # key = 'Exif.GPSInfo.GPSLatitude' # value = lat_frac[0].split(' ') # log.ODM_DEBUG('lat_frac: %s %s %s' % (value[0], value[1], value[2])) # metadata[key] = pyexiv2.ExifTag(key, # [Fraction(value[0]), # Fraction(value[1]), # Fraction(value[2])]) # # key = 'Exif.GPSInfo.GPSLatitudeRef' # value = lat_frac[1] # metadata[key] = pyexiv2.ExifTag(key, value) # # # GPS longitude # key = 'Exif.GPSInfo.GPSLongitude' # value = lon_frac[0].split(' ') # metadata[key] = pyexiv2.ExifTag(key, # [Fraction(value[0]), # Fraction(value[1]), # Fraction(value[2])]) # # key = 'Exif.GPSInfo.GPSLongitudeRef' # value = lon_frac[1] # metadata[key] = pyexiv2.ExifTag(key, value) # # # GPS altitude # altitude = abs(int(float(latlon[2]) * 100)) # key = 'Exif.GPSInfo.GPSAltitude' # value = Fraction(altitude, 1) # metadata[key] = pyexiv2.ExifTag(key, value) # # if latlon[2] >= 0: # altref = '0' # else: # altref = '1' # key = 'Exif.GPSInfo.GPSAltitudeRef' # metadata[key] = pyexiv2.ExifTag(key, altref) # # # write values # metadata.write() def extract_offsets(self, _file): if not io.file_exists(_file): log.ODM_ERROR('Could not find file %s' % _file) return with open(_file) as f: offsets = f.readlines()[1].split(' ') self.utm_east_offset = float(offsets[0]) self.utm_north_offset = float(offsets[1]) def create_gcps(self, _file): if not io.file_exists(_file): log.ODM_ERROR('Could not find file %s' % _file) return with open(_file) as f: # parse coordinates lines = f.readlines()[2:] for l in lines: xyz = l.split(' ') if len(xyz) == 3: x, y, z = xyz[:3] elif len(xyz) == 2: x, y = xyz[:2] z = 0 self.gcps.append(ODM_GCPoint(float(x), float(y), float(z))) # Write to json file def parse_transformation_matrix(self, _file): if not io.file_exists(_file): log.ODM_ERROR('Could not find file %s' % _file) return # Create a nested list for the transformation matrix with open(_file) as f: for line in f: # Handle matrix formats that either # have leading or trailing brakets or just plain numbers. line = re.sub(r"[\[\],]", "", line).strip() self.transform += [[float(i) for i in line.split()]] self.utm_east_offset = self.transform[0][3] self.utm_north_offset = self.transform[1][3] class ODM_Tree(object): def __init__(self, root_path, images_path, gcp_file = None): # root path to the project self.root_path = io.absolute_path_file(root_path) if not images_path: self.input_images = io.join_paths(self.root_path, 'images') else: self.input_images = io.absolute_path_file(images_path) # modules paths # here are defined where all modules should be located in # order to keep track all files al directories during the # whole reconstruction process. self.dataset_raw = io.join_paths(self.root_path, 'images') self.opensfm = io.join_paths(self.root_path, 'opensfm') self.pmvs = io.join_paths(self.root_path, 'pmvs') self.odm_meshing = io.join_paths(self.root_path, 'odm_meshing') self.odm_texturing = io.join_paths(self.root_path, 'odm_texturing') self.odm_25dtexturing = io.join_paths(self.root_path, 'odm_texturing_25d') self.odm_georeferencing = io.join_paths(self.root_path, 'odm_georeferencing') self.odm_25dgeoreferencing = io.join_paths(self.root_path, 'odm_25dgeoreferencing') self.odm_orthophoto = io.join_paths(self.root_path, 'odm_orthophoto') self.odm_pdal = io.join_paths(self.root_path, 'pdal') # important files paths # benchmarking self.benchmarking = io.join_paths(self.root_path, 'benchmark.txt') self.dataset_list = io.join_paths(self.root_path, 'img_list.txt') # opensfm self.opensfm_tracks = io.join_paths(self.opensfm, 'tracks.csv') self.opensfm_bundle = io.join_paths(self.opensfm, 'bundle_r000.out') self.opensfm_bundle_list = io.join_paths(self.opensfm, 'list_r000.out') self.opensfm_image_list = io.join_paths(self.opensfm, 'image_list.txt') self.opensfm_reconstruction = io.join_paths(self.opensfm, 'reconstruction.json') self.opensfm_reconstruction_meshed = io.join_paths(self.opensfm, 'reconstruction.meshed.json') self.opensfm_reconstruction_nvm = io.join_paths(self.opensfm, 'reconstruction.nvm') self.opensfm_model = io.join_paths(self.opensfm, 'depthmaps/merged.ply') self.opensfm_transformation = io.join_paths(self.opensfm, 'geocoords_transformation.txt') # pmvs self.pmvs_rec_path = io.join_paths(self.pmvs, 'recon0') self.pmvs_bundle = io.join_paths(self.pmvs_rec_path, 'bundle.rd.out') self.pmvs_visdat = io.join_paths(self.pmvs_rec_path, 'vis.dat') self.pmvs_options = io.join_paths(self.pmvs_rec_path, 'pmvs_options.txt') self.pmvs_model = io.join_paths(self.pmvs_rec_path, 'models/option-0000.ply') # odm_meshing self.odm_mesh = io.join_paths(self.odm_meshing, 'odm_mesh.ply') self.odm_meshing_log = io.join_paths(self.odm_meshing, 'odm_meshing_log.txt') self.odm_25dmesh = io.join_paths(self.odm_meshing, 'odm_25dmesh.ply') self.odm_25dmeshing_log = io.join_paths(self.odm_meshing, 'odm_25dmeshing_log.txt') # texturing self.odm_texturing_undistorted_image_path = io.join_paths( self.odm_texturing, 'undistorted') self.odm_textured_model_obj = 'odm_textured_model.obj' self.odm_textured_model_mtl = 'odm_textured_model.mtl' # Log is only used by old odm_texturing self.odm_texuring_log = 'odm_texturing_log.txt' # odm_georeferencing self.odm_georeferencing_latlon = io.join_paths( self.odm_georeferencing, 'latlon.txt') self.odm_georeferencing_coords = io.join_paths( self.odm_georeferencing, 'coords.txt') self.odm_georeferencing_gcp = gcp_file or io.find('gcp_list.txt', self.root_path) self.odm_georeferencing_utm_log = io.join_paths( self.odm_georeferencing, 'odm_georeferencing_utm_log.txt') self.odm_georeferencing_log = 'odm_georeferencing_log.txt' self.odm_georeferencing_transform_file = 'odm_georeferencing_transform.txt' self.odm_georeferencing_proj = 'proj.txt' self.odm_georeferencing_model_txt_geo = 'odm_georeferencing_model_geo.txt' self.odm_georeferencing_model_ply_geo = 'odm_georeferenced_model.ply' self.odm_georeferencing_model_obj_geo = 'odm_textured_model_geo.obj' self.odm_georeferencing_xyz_file = io.join_paths( self.odm_georeferencing, 'odm_georeferenced_model.csv') self.odm_georeferencing_las_json = io.join_paths( self.odm_georeferencing, 'las.json') self.odm_georeferencing_model_las = io.join_paths( self.odm_georeferencing, 'odm_georeferenced_model.las') self.odm_georeferencing_dem = io.join_paths( self.odm_georeferencing, 'odm_georeferencing_model_dem.tif') # odm_orthophoto self.odm_orthophoto_file = io.join_paths(self.odm_orthophoto, 'odm_orthophoto.png') self.odm_orthophoto_tif = io.join_paths(self.odm_orthophoto, 'odm_orthophoto.tif') self.odm_orthophoto_corners = io.join_paths(self.odm_orthophoto, 'odm_orthophoto_corners.txt') self.odm_orthophoto_log = io.join_paths(self.odm_orthophoto, 'odm_orthophoto_log.txt') self.odm_orthophoto_tif_log = io.join_paths(self.odm_orthophoto, 'gdal_translate_log.txt') self.odm_orthophoto_gdaladdo_log = io.join_paths(self.odm_orthophoto, 'gdaladdo_log.txt') def path(self, *args): return io.join_paths(self.root_path, *args)