Compare geotiffs

compare_geotiffs.py

import sys
import numpy as np
import rasterio as rio


def compare(srcpath1, srcpath2):
    
    with rio.drivers():
        
        src1 = rio.open(srcpath1)
        src2 = rio.open(srcpath2)
        
        # Compare metadata
        # Compare pixels
        
        props = ['count', 'crs', 'dtypes', 'driver', 'bounds', 'height', 'width', 'shape', 'nodatavals']
        
        for prop in props:
            assert src1.__getattribute__(prop) == src2.__getattribute__(prop)
        
        for bidx in range(1, src1.count + 1):
            band1 = src1.read_band(1)
            band2 = src2.read_band(1)
            
            assert np.array_equal(band1, band2)
        
        src1.close()
        src2.close()



if __name__ == '__main__':
    sys.argv.pop(0)
    
    if len(sys.argv) != 2:
        print "Usage: python compare_geotiff.py [srcpath] [dstpath]"
        sys.exit()
    
    compare(*sys.argv)