def are_equal_color_table(color_table1: gdal.ColorTable, color_table2: gdal.ColorTable) -> bool:
if color_table1.GetCount() != color_table2.GetCount():
return False
for i in range(color_table1.GetCount()):
color_entry1: gdal.ColorEntry = color_table1.GetColorEntry(i)
color_entry2: gdal.ColorEntry = color_table2.GetColorEntry(i)
if color_entry1 != color_entry2:
return False
return True
def write_color_table_to_file(color_table: gdal.ColorTable, color_filename: Optional[PathLikeOrStr]):
if color_filename is None:
color_filename = tempfile.mktemp(suffix='.txt')
os.makedirs(os.path.dirname(color_filename), exist_ok=True)
with open(color_filename, mode='w') as fp:
for i in range(color_table.GetCount()):
color_entry = color_table.GetColorEntry(i)
color_entry = ' '.join(str(c) for c in color_entry)
fp.write('{} {}\n'.format(i, color_entry))
return color_filename
def color_table_from_color_palette(pal: ColorPalette, color_table: gdal.ColorTable,
fill_missing_colors=True, min_key=0, max_key=255) -> bool:
""" returns None if pal has no values, otherwise returns a gdal.ColorTable from the given ColorPalette"""
if not pal.pal or not pal.is_numeric():
# palette has no values or not numeric
return False
if fill_missing_colors:
keys = sorted(list(pal.pal.keys()))
if min_key is None:
min_key = keys[0]
if max_key is None:
max_key = keys[-1]
c = pal.color_to_color_entry(pal.pal[keys[0]])
for key in range(min_key, max_key + 1):
if key in keys:
c = pal.color_to_color_entry(pal.pal[key])
color_table.SetColorEntry(key, c)
else:
for key, col in pal.pal.items():
color_table.SetColorEntry(key, pal.color_to_color_entry(col)) # set color for each key
return True
def is_fixed_color_table(color_table: gdal.ColorTable, c=(0, 0, 0, 0)) -> bool:
for i in range(color_table.GetCount()):
color_entry: gdal.ColorEntry = color_table.GetColorEntry(i)
if color_entry != c:
return False
return True