Skip to content

Commit

Permalink
r.out.gdal: raster attribute table export enhancement (#4800)
Browse files Browse the repository at this point in the history 
* r.out.gdal: enhance raster attribute table export
  • Loading branch information
@metzm
metzm authored Dec 7, 2024
1 parent 2ae6362 commit 0b29c8e
Showing 1 changed file with 161 additions and 10 deletions.
171 changes: 161 additions & 10 deletions raster/r.out.gdal/attr.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,9 @@ int export_attr(GDALDatasetH hMEMDS, int band, const char *name,

rcount = 0;
Rast_init_colors(&sGrassColors);
if (Rast_read_colors(name, mapset, &sGrassColors) >= 0) {
/* only export color rules set by the user,
* skip automatically generated default color rules */
if (Rast_read_colors(name, mapset, &sGrassColors) > 0) {
rcount = Rast_colors_count(&sGrassColors);
}

Expand All @@ -47,28 +49,177 @@ int export_attr(GDALDatasetH hMEMDS, int band, const char *name,
* GFU_MaxCount
*/

/* TODO: cats.ncats > 0 && rcount > 0
* how to merge categories and color rules ?
* what to do for a cell value that has a category but no color rule ?
* what to do for a cell value that has a color rule but no category ?
*/
if (cats.ncats > 0) {
int use_minmax = 0;
if (cats.ncats > 0 && rcount > 0) {
bool use_minmax = false;
int r1, g1, b1, r2, g2, b2;

/* merge categories and color rules:
* go through categories and fetch corresponding color
* write out value, label, red, green, blue
* or the minmax variant
*/

if (maptype == CELL_TYPE) {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_c_cat(&cats, i, &CellMin, &CellMax);
if (!label || !*label) {
G_fatal_error(_("No label for category entry no %d "), i);
}
if (CellMin != CellMax) {
use_minmax = true;
break;
}
}
}
else {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_d_cat(&cats, i, &dfCellMin, &dfCellMax);
if (!label || !*label) {
G_fatal_error(_("No label for category entry no %d "), i);
}
if (dfCellMin != dfCellMax) {
use_minmax = true;
break;
}
}
}

/* create new raster attribute table */
hrat = GDALCreateRasterAttributeTable();

if (use_minmax) {
if (maptype == CELL_TYPE) {
GDALRATCreateColumn(hrat, "min", GFT_Integer, GFU_Min);
GDALRATCreateColumn(hrat, "max", GFT_Integer, GFU_Max);
}
else {
GDALRATCreateColumn(hrat, "min", GFT_Real, GFU_Min);
GDALRATCreateColumn(hrat, "max", GFT_Real, GFU_Max);
}
GDALRATCreateColumn(hrat, "label", GFT_String, GFU_Name);
GDALRATCreateColumn(hrat, "redmin", GFT_Integer, GFU_RedMin);
GDALRATCreateColumn(hrat, "redmax", GFT_Integer, GFU_RedMax);
GDALRATCreateColumn(hrat, "greenmin", GFT_Integer, GFU_GreenMin);
GDALRATCreateColumn(hrat, "greenmax", GFT_Integer, GFU_GreenMax);
GDALRATCreateColumn(hrat, "bluemin", GFT_Integer, GFU_BlueMin);
GDALRATCreateColumn(hrat, "bluemax", GFT_Integer, GFU_BlueMax);

GDALRATSetRowCount(hrat, cats.ncats);

if (maptype == CELL_TYPE) {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_c_cat(&cats, i, &CellMin, &CellMax);
GDALRATSetValueAsInt(hrat, i, 0, CellMin);
GDALRATSetValueAsInt(hrat, i, 1, CellMax);
GDALRATSetValueAsString(hrat, i, 2, label);

Rast_get_color(&CellMin, &r1, &g1, &b1, &sGrassColors,
maptype);
Rast_get_color(&CellMax, &r2, &g2, &b2, &sGrassColors,
maptype);
GDALRATSetValueAsInt(hrat, i, 3, r1);
GDALRATSetValueAsInt(hrat, i, 4, r2);
GDALRATSetValueAsInt(hrat, i, 5, g1);
GDALRATSetValueAsInt(hrat, i, 6, g2);
GDALRATSetValueAsInt(hrat, i, 7, b1);
GDALRATSetValueAsInt(hrat, i, 8, b2);
}
}
else {
for (i = 0; i < cats.ncats; i++) {
label =
Rast_get_ith_d_cat(&cats, i, &dfCellMin, &dfCellMax);
GDALRATSetValueAsDouble(hrat, i, 0, dfCellMin);
GDALRATSetValueAsDouble(hrat, i, 1, dfCellMax);
GDALRATSetValueAsString(hrat, i, 2, label);

Rast_get_color(&dfCellMin, &r1, &g1, &b1, &sGrassColors,
DCELL_TYPE);
Rast_get_color(&dfCellMax, &r2, &g2, &b2, &sGrassColors,
DCELL_TYPE);
GDALRATSetValueAsInt(hrat, i, 3, r1);
GDALRATSetValueAsInt(hrat, i, 4, r2);
GDALRATSetValueAsInt(hrat, i, 5, g1);
GDALRATSetValueAsInt(hrat, i, 6, g2);
GDALRATSetValueAsInt(hrat, i, 7, b1);
GDALRATSetValueAsInt(hrat, i, 8, b2);
}
}
}
else {
if (maptype == CELL_TYPE) {
GDALRATCreateColumn(hrat, "value", GFT_Integer, GFU_MinMax);
}
else {
GDALRATCreateColumn(hrat, "value", GFT_Real, GFU_MinMax);
}
GDALRATCreateColumn(hrat, "label", GFT_String, GFU_Name);
GDALRATCreateColumn(hrat, "red", GFT_Integer, GFU_Red);
GDALRATCreateColumn(hrat, "green", GFT_Integer, GFU_Green);
GDALRATCreateColumn(hrat, "blue", GFT_Integer, GFU_Blue);

GDALRATSetRowCount(hrat, cats.ncats);

if (maptype == CELL_TYPE) {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_c_cat(&cats, i, &CellMin, &CellMax);
GDALRATSetValueAsInt(hrat, i, 0, CellMin);
GDALRATSetValueAsString(hrat, i, 1, label);

Rast_get_color(&CellMin, &r1, &g1, &b1, &sGrassColors,
maptype);
GDALRATSetValueAsInt(hrat, i, 2, r1);
GDALRATSetValueAsInt(hrat, i, 3, g1);
GDALRATSetValueAsInt(hrat, i, 4, b1);
}
}
else {
for (i = 0; i < cats.ncats; i++) {
label =
Rast_get_ith_d_cat(&cats, i, &dfCellMin, &dfCellMax);
GDALRATSetValueAsDouble(hrat, i, 0, dfCellMin);
GDALRATSetValueAsString(hrat, i, 1, label);

Rast_get_color(&dfCellMin, &r1, &g1, &b1, &sGrassColors,
DCELL_TYPE);
GDALRATSetValueAsInt(hrat, i, 2, r1);
GDALRATSetValueAsInt(hrat, i, 3, g1);
GDALRATSetValueAsInt(hrat, i, 4, b1);
}
}
}

if (GDALSetDefaultRAT(hBand, hrat) != CE_None) {
G_warning(_("Failed to set raster attribute table"));
ret = -1;
}
/* GDALRATDumpReadable(hrat, stdout); */

GDALDestroyRasterAttributeTable(hrat);
}
else if (cats.ncats > 0 && rcount == 0) {
bool use_minmax = false;

if (maptype == CELL_TYPE) {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_c_cat(&cats, i, &CellMin, &CellMax);
if (!label || !*label) {
G_fatal_error(_("No label for category entry no %d "), i);
}
if (CellMin != CellMax) {
use_minmax = 1;
use_minmax = true;
break;
}
}
}
else {
for (i = 0; i < cats.ncats; i++) {
label = Rast_get_ith_d_cat(&cats, i, &dfCellMin, &dfCellMax);
if (!label || !*label) {
G_fatal_error(_("No label for category entry no %d "), i);
}
if (dfCellMin != dfCellMax) {
use_minmax = 1;
use_minmax = true;
break;
}
}
Expand Down

0 comments on commit 0b29c8e

Please sign in to comment.