PostGIS  2.5.7dev-r@@SVN_REVISION@@

◆ test_raster_colormap()

static void test_raster_colormap ( )
static

Definition at line 882 of file cu_mapalgebra.c.

882  {
883  rt_raster raster;
884  rt_raster rtn;
885  rt_band band;
886  int x;
887  int y;
888  rt_colormap colormap = NULL;
889  double value;
890  int nodata;
891 
892  raster = rt_raster_new(9, 9);
893  CU_ASSERT(raster != NULL); /* or we're out of virtual memory */
894  band = cu_add_band(raster, PT_8BUI, 0, 0);
895  CU_ASSERT(band != NULL);
896  rt_band_set_nodata(band, 0, NULL);
897 
898  for (y = 0; y < 9; y++) {
899  for (x = 0; x < 9; x++) {
900  rt_band_set_pixel(band, x, y, x, NULL);
901  }
902  }
903 
904  colormap = (rt_colormap) rtalloc(sizeof(struct rt_colormap_t));
905  CU_ASSERT(colormap != NULL);
906  colormap->nentry = 3;
907  colormap->entry = (rt_colormap_entry) rtalloc(sizeof(struct rt_colormap_entry_t) * colormap->nentry);
908  CU_ASSERT(colormap->entry != NULL);
909 
910  colormap->entry[0].isnodata = 0;
911  colormap->entry[0].value = 8;
912  colormap->entry[0].color[0] = 255;
913  colormap->entry[0].color[1] = 255;
914  colormap->entry[0].color[2] = 255;
915  colormap->entry[0].color[3] = 255;
916 
917  colormap->entry[1].isnodata = 0;
918  colormap->entry[1].value = 3;
919  colormap->entry[1].color[0] = 127;
920  colormap->entry[1].color[1] = 127;
921  colormap->entry[1].color[2] = 127;
922  colormap->entry[1].color[3] = 255;
923 
924  colormap->entry[2].isnodata = 0;
925  colormap->entry[2].value = 0;
926  colormap->entry[2].color[0] = 0;
927  colormap->entry[2].color[1] = 0;
928  colormap->entry[2].color[2] = 0;
929  colormap->entry[2].color[3] = 255;
930 
931  /* 2 colors, 3 entries, INTERPOLATE */
932  colormap->ncolor = 2;
933  colormap->method = CM_INTERPOLATE;
934 
935  rtn = rt_raster_colormap(
936  raster, 0,
937  colormap
938  );
939  CU_ASSERT(rtn != NULL);
940  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
941 
942  band = rt_raster_get_band(rtn, 0);
943  CU_ASSERT(band != NULL);
944  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 0, 0, &value, &nodata), ES_NONE);
945  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
946  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 3, 0, &value, &nodata), ES_NONE);
947  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
948  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 8, 0, &value, &nodata), ES_NONE);
949  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
950 
951  cu_free_raster(rtn);
952 
953  /* 4 colors, 3 entries, INTERPOLATE */
954  colormap->ncolor = 4;
955 
956  rtn = rt_raster_colormap(
957  raster, 0,
958  colormap
959  );
960  CU_ASSERT(rtn != NULL);
961  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
962  cu_free_raster(rtn);
963 
964  /* 4 colors, 3 entries, EXACT */
965  colormap->method = CM_EXACT;
966 
967  rtn = rt_raster_colormap(
968  raster, 0,
969  colormap
970  );
971  CU_ASSERT(rtn != NULL);
972  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
973 
974  band = rt_raster_get_band(rtn, 0);
975  CU_ASSERT(band != NULL);
976  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 0, 0, &value, &nodata), ES_NONE);
977  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
978  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 3, 0, &value, &nodata), ES_NONE);
979  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
980  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 8, 0, &value, &nodata), ES_NONE);
981  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
982  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 1, 0, &value, &nodata), ES_NONE);
983  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
984  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 7, 0, &value, &nodata), ES_NONE);
985  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
986 
987  cu_free_raster(rtn);
988 
989  /* 4 colors, 3 entries, NEAREST */
990  colormap->method = CM_NEAREST;
991 
992  rtn = rt_raster_colormap(
993  raster, 0,
994  colormap
995  );
996  CU_ASSERT(rtn != NULL);
997  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
998 
999  band = rt_raster_get_band(rtn, 0);
1000  CU_ASSERT(band != NULL);
1001  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 0, 0, &value, &nodata), ES_NONE);
1002  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
1003  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 3, 0, &value, &nodata), ES_NONE);
1004  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1005  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 8, 0, &value, &nodata), ES_NONE);
1006  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
1007  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 1, 0, &value, &nodata), ES_NONE);
1008  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
1009  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 2, 0, &value, &nodata), ES_NONE);
1010  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1011  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 4, 0, &value, &nodata), ES_NONE);
1012  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1013  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 7, 0, &value, &nodata), ES_NONE);
1014  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
1015 
1016  cu_free_raster(rtn);
1017 
1018  /* 4 colors, 2 entries, NEAREST */
1019  colormap->nentry = 2;
1020  colormap->method = CM_NEAREST;
1021 
1022  rtn = rt_raster_colormap(
1023  raster, 0,
1024  colormap
1025  );
1026  CU_ASSERT(rtn != NULL);
1027  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
1028 
1029  band = rt_raster_get_band(rtn, 0);
1030  CU_ASSERT(band != NULL);
1031  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 0, 0, &value, &nodata), ES_NONE);
1032  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1033  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 3, 0, &value, &nodata), ES_NONE);
1034  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1035  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 8, 0, &value, &nodata), ES_NONE);
1036  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
1037  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 1, 0, &value, &nodata), ES_NONE);
1038  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1039  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 2, 0, &value, &nodata), ES_NONE);
1040  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1041  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 4, 0, &value, &nodata), ES_NONE);
1042  CU_ASSERT_DOUBLE_EQUAL(value, 127, DBL_EPSILON);
1043  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 7, 0, &value, &nodata), ES_NONE);
1044  CU_ASSERT_DOUBLE_EQUAL(value, 255, DBL_EPSILON);
1045 
1046  cu_free_raster(rtn);
1047 
1048  rtdealloc(colormap->entry);
1049  rtdealloc(colormap);
1050 
1051  cu_free_raster(raster);
1052 
1053  /* new set of tests */
1054  raster = rt_raster_new(10, 10);
1055  CU_ASSERT(raster != NULL); /* or we're out of virtual memory */
1056  band = cu_add_band(raster, PT_8BUI, 0, 0);
1057  CU_ASSERT(band != NULL);
1058  rt_band_set_nodata(band, 0, NULL);
1059 
1060  for (y = 0; y < 10; y++) {
1061  for (x = 0; x < 10; x++) {
1062  rt_band_set_pixel(band, x, y, (x * y) + x, NULL);
1063  }
1064  }
1065 
1066  colormap = (rt_colormap) rtalloc(sizeof(struct rt_colormap_t));
1067  CU_ASSERT(colormap != NULL);
1068  colormap->nentry = 10;
1069  colormap->entry = (rt_colormap_entry) rtalloc(sizeof(struct rt_colormap_entry_t) * colormap->nentry);
1070  CU_ASSERT(colormap->entry != NULL);
1071 
1072  colormap->entry[0].isnodata = 0;
1073  colormap->entry[0].value = 90;
1074  colormap->entry[0].color[0] = 255;
1075  colormap->entry[0].color[1] = 255;
1076  colormap->entry[0].color[2] = 255;
1077  colormap->entry[0].color[3] = 255;
1078 
1079  colormap->entry[1].isnodata = 0;
1080  colormap->entry[1].value = 80;
1081  colormap->entry[1].color[0] = 255;
1082  colormap->entry[1].color[1] = 227;
1083  colormap->entry[1].color[2] = 227;
1084  colormap->entry[1].color[3] = 255;
1085 
1086  colormap->entry[2].isnodata = 0;
1087  colormap->entry[2].value = 70;
1088  colormap->entry[2].color[0] = 255;
1089  colormap->entry[2].color[1] = 198;
1090  colormap->entry[2].color[2] = 198;
1091  colormap->entry[2].color[3] = 255;
1092 
1093  colormap->entry[3].isnodata = 0;
1094  colormap->entry[3].value = 60;
1095  colormap->entry[3].color[0] = 255;
1096  colormap->entry[3].color[1] = 170;
1097  colormap->entry[3].color[2] = 170;
1098  colormap->entry[3].color[3] = 255;
1099 
1100  colormap->entry[4].isnodata = 0;
1101  colormap->entry[4].value = 50;
1102  colormap->entry[4].color[0] = 255;
1103  colormap->entry[4].color[1] = 142;
1104  colormap->entry[4].color[2] = 142;
1105  colormap->entry[4].color[3] = 255;
1106 
1107  colormap->entry[5].isnodata = 0;
1108  colormap->entry[5].value = 40;
1109  colormap->entry[5].color[0] = 255;
1110  colormap->entry[5].color[1] = 113;
1111  colormap->entry[5].color[2] = 113;
1112  colormap->entry[5].color[3] = 255;
1113 
1114  colormap->entry[6].isnodata = 0;
1115  colormap->entry[6].value = 30;
1116  colormap->entry[6].color[0] = 255;
1117  colormap->entry[6].color[1] = 85;
1118  colormap->entry[6].color[2] = 85;
1119  colormap->entry[6].color[3] = 255;
1120 
1121  colormap->entry[7].isnodata = 0;
1122  colormap->entry[7].value = 20;
1123  colormap->entry[7].color[0] = 255;
1124  colormap->entry[7].color[1] = 57;
1125  colormap->entry[7].color[2] = 57;
1126  colormap->entry[7].color[3] = 255;
1127 
1128  colormap->entry[8].isnodata = 0;
1129  colormap->entry[8].value = 10;
1130  colormap->entry[8].color[0] = 255;
1131  colormap->entry[8].color[1] = 28;
1132  colormap->entry[8].color[2] = 28;
1133  colormap->entry[8].color[3] = 255;
1134 
1135  colormap->entry[9].isnodata = 0;
1136  colormap->entry[9].value = 0;
1137  colormap->entry[9].color[0] = 255;
1138  colormap->entry[9].color[1] = 0;
1139  colormap->entry[9].color[2] = 0;
1140  colormap->entry[9].color[3] = 255;
1141 
1142  /* 2 colors, 3 entries, INTERPOLATE */
1143  colormap->ncolor = 4;
1144  colormap->method = CM_INTERPOLATE;
1145 
1146  rtn = rt_raster_colormap(
1147  raster, 0,
1148  colormap
1149  );
1150  CU_ASSERT(rtn != NULL);
1151  CU_ASSERT_EQUAL(rt_raster_get_num_bands(rtn), colormap->ncolor);
1152 
1153  band = rt_raster_get_band(rtn, 2);
1154  CU_ASSERT(band != NULL);
1155  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 0, 0, &value, &nodata), ES_NONE);
1156  CU_ASSERT_DOUBLE_EQUAL(value, 0, DBL_EPSILON);
1157  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 5, 0, &value, &nodata), ES_NONE);
1158  CU_ASSERT_DOUBLE_EQUAL(value, 14, DBL_EPSILON);
1159  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 6, 0, &value, &nodata), ES_NONE);
1160  CU_ASSERT_DOUBLE_EQUAL(value, 17, DBL_EPSILON);
1161  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 9, 0, &value, &nodata), ES_NONE);
1162  CU_ASSERT_DOUBLE_EQUAL(value, 25, DBL_EPSILON);
1163 
1164  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 2, 4, &value, &nodata), ES_NONE);
1165  CU_ASSERT_DOUBLE_EQUAL(value, 28, DBL_EPSILON);
1166  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 3, 4, &value, &nodata), ES_NONE);
1167  CU_ASSERT_DOUBLE_EQUAL(value, 43, DBL_EPSILON);
1168  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 4, 4, &value, &nodata), ES_NONE);
1169  CU_ASSERT_DOUBLE_EQUAL(value, 57, DBL_EPSILON);
1170 
1171  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 6, 9, &value, &nodata), ES_NONE);
1172  CU_ASSERT_DOUBLE_EQUAL(value, 170, DBL_EPSILON);
1173  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 7, 9, &value, &nodata), ES_NONE);
1174  CU_ASSERT_DOUBLE_EQUAL(value, 198, DBL_EPSILON);
1175  CU_ASSERT_EQUAL(rt_band_get_pixel(band, 8, 9, &value, &nodata), ES_NONE);
1176  CU_ASSERT_DOUBLE_EQUAL(value, 227, DBL_EPSILON);
1177 
1178  cu_free_raster(rtn);
1179 
1180  rtdealloc(colormap->entry);
1181  rtdealloc(colormap);
1182 
1183  cu_free_raster(raster);
1184 }
rtalloc
void * rtalloc(size_t size)
Wrappers used for managing memory.
Definition: rt_context.c:171
rt_raster_colormap
rt_raster rt_raster_colormap(rt_raster raster, int nband, rt_colormap colormap)
Returns a new raster with up to four 8BUI bands (RGBA) from applying a colormap to the user-specified...
Definition: rt_mapalgebra.c:1523
rt_band_get_pixel
rt_errorstate rt_band_get_pixel(rt_band band, int x, int y, double *value, int *nodata)
Get pixel value.
Definition: rt_band.c:1221
PT_8BUI
@ PT_8BUI
Definition: librtcore.h:190
rt_raster_new
rt_raster rt_raster_new(uint32_t width, uint32_t height)
Construct a raster with given dimensions.
Definition: rt_raster.c:48
rt_colormap_entry
struct rt_colormap_entry_t * rt_colormap_entry
Definition: librtcore.h:160
rt_band_set_nodata
rt_errorstate rt_band_set_nodata(rt_band band, double val, int *converted)
Set nodata value.
Definition: rt_band.c:733
rt_band_set_pixel
rt_errorstate rt_band_set_pixel(rt_band band, int x, int y, double val, int *converted)
Set single pixel's value.
Definition: rt_band.c:974
ES_NONE
@ ES_NONE
Definition: librtcore.h:180
rt_raster_get_num_bands
uint16_t rt_raster_get_num_bands(rt_raster raster)
Definition: rt_raster.c:372
rtdealloc
void rtdealloc(void *mem)
Definition: rt_context.c:186
rt_colormap
struct rt_colormap_t * rt_colormap
Definition: librtcore.h:161
rt_raster_get_band
rt_band rt_raster_get_band(rt_raster raster, int bandNum)
Return Nth band, or NULL if unavailable.
Definition: rt_raster.c:381
genraster.value
int value
Definition: genraster.py:61
ovdump.band
band
Definition: ovdump.py:57
rtrowdump.raster
raster
Be careful!! Zeros function's input parameter can be a (height x width) array, not (width x height): ...
Definition: rtrowdump.py:121
cu_add_band
rt_band cu_add_band(rt_raster raster, rt_pixtype pixtype, int hasnodata, double nodataval)
Definition: raster/test/cunit/cu_tester.c:224
cu_free_raster
void cu_free_raster(rt_raster raster)
Definition: raster/test/cunit/cu_tester.c:211
rt_band_t
Definition: librtcore.h:2313
rt_colormap_entry_t::color
uint8_t color[4]
Definition: librtcore.h:2484
rt_colormap_entry_t::value
double value
Definition: librtcore.h:2483
rt_colormap_entry_t::isnodata
int isnodata
Definition: librtcore.h:2482
rt_colormap_entry_t
Definition: librtcore.h:2481
rt_colormap_t::entry
rt_colormap_entry entry
Definition: librtcore.h:2496
rt_colormap_t::method
enum rt_colormap_t::@9 method
rt_colormap_t::ncolor
int ncolor
Definition: librtcore.h:2494
rt_colormap_t::nentry
uint16_t nentry
Definition: librtcore.h:2495
rt_colormap_t
Definition: librtcore.h:2487
rt_raster_t
Definition: librtcore.h:2284

References ovdump::band, rt_colormap_entry_t::color, cu_add_band(), cu_free_raster(), rt_colormap_t::entry, ES_NONE, rt_colormap_entry_t::isnodata, rt_colormap_t::method, rt_colormap_t::ncolor, rt_colormap_t::nentry, PT_8BUI, rtrowdump::raster, rt_band_get_pixel(), rt_band_set_nodata(), rt_band_set_pixel(), rt_raster_colormap(), rt_raster_get_band(), rt_raster_get_num_bands(), rt_raster_new(), rtalloc(), rtdealloc(), rt_colormap_entry_t::value, genraster::value, pixval::x, and pixval::y.

Referenced by mapalgebra_suite_setup().

Here is the call graph for this function:
Here is the caller graph for this function: