int main(int argc, char *argv[])
{
char *input, *output;
int convertNull;
char nullValue[256];
int useTypeDefault, type, useCompressionDefault, doCompression;
int usePrecisionDefault, precision, useDimensionDefault, tileX, tileY,
tileZ;
RASTER3D_Region region;
FILE *fp;
struct GModule *module;
struct History history;
map = NULL;
G_gisinit(argv[0]);
module = G_define_module();
G_add_keyword(_("raster3d"));
G_add_keyword(_("import"));
G_add_keyword(_("voxel"));
G_add_keyword(_("conversion"));
G_add_keyword("ASCII");
module->description =
_("Converts a 3D ASCII raster text file into a (binary) 3D raster map.");
setParams();
Rast3d_set_standard3d_input_params();
if (G_parser(argc, argv))
exit(EXIT_FAILURE);
getParams(&input, &output, &convertNull, nullValue);
if (!Rast3d_get_standard3d_params(&useTypeDefault, &type,
&useCompressionDefault, &doCompression,
&usePrecisionDefault, &precision,
&useDimensionDefault, &tileX, &tileY,
&tileZ))
fatalError("Error getting standard parameters");
Rast3d_init_defaults();
fp = openAscii(input, ®ion);
/*Open the new RASTER3D map */
map = Rast3d_open_new_param(output, RASTER3D_TILE_SAME_AS_FILE, RASTER3D_USE_CACHE_XY,
®ion,
type, doCompression, precision, tileX, tileY,
tileZ);
if (map == NULL)
fatalError(_("Unable to open 3D raster map"));
/*Create the new RASTER3D Map */
asciiToG3d(fp, ®ion, convertNull, nullValue);
if (!Rast3d_close(map))
fatalError(_("Unable to close 3D raster map"));
/* write input name to map history */
Rast3d_read_history(output, G_mapset(), &history);
Rast_command_history(&history);
Rast_set_history(&history, HIST_DATSRC_1, input);
Rast3d_write_history(output, &history);
map = NULL;
if (fclose(fp))
fatalError(_("Unable to close ASCII file"));
return EXIT_SUCCESS;
}
static void makeMask(char *name, d_Mask * maskRules)
{
void *map, *mask;
RASTER3D_Region region;
int tileX, tileY, tileZ, x, y, z, cacheSize;
double value;
float floatNull;
cacheSize = Rast3d_cache_size_encode(RASTER3D_USE_CACHE_XY, 1);
if (NULL == G_find_raster3d(name, ""))
Rast3d_fatal_error(_("3D raster map <%s> not found"), name);
map = Rast3d_open_cell_old(name, G_mapset(), RASTER3D_DEFAULT_WINDOW,
DCELL_TYPE, cacheSize);
if (map == NULL)
Rast3d_fatal_error(_("Unable to open 3D raster map <%s>"), name);
Rast3d_get_region_struct_map(map, ®ion);
Rast3d_get_tile_dimensions_map(map, &tileX, &tileY, &tileZ);
mask = Rast3d_open_new_param(Rast3d_mask_file(), FCELL_TYPE, cacheSize,
®ion, FCELL_TYPE, RASTER3D_COMPRESSION, 0,
tileX, tileY, tileZ);
if (mask == NULL)
Rast3d_fatal_error(_("Unable to open 3D raster mask file"));
Rast3d_min_unlocked(map, RASTER3D_USE_CACHE_X);
Rast3d_autolock_on(map);
Rast3d_unlock_all(map);
Rast3d_min_unlocked(mask, RASTER3D_USE_CACHE_X);
Rast3d_autolock_on(mask);
Rast3d_unlock_all(mask);
Rast3d_set_null_value(&floatNull, 1, FCELL_TYPE);
for (z = 0; z < region.depths; z++) {
if ((z % tileZ) == 0) {
Rast3d_unlock_all(map);
Rast3d_unlock_all(mask);
}
for (y = 0; y < region.rows; y++) /* We count from north to south in the cube coordinate system */
for (x = 0; x < region.cols; x++) {
value = Rast3d_get_double_region(map, x, y, z);
if (Rast3d_mask_d_select((DCELL *) & value, maskRules))
Rast3d_put_float(mask, x, y, z, (float)floatNull); /* mask-out value */
else
Rast3d_put_float(mask, x, y, z, (float)0.0); /* not mask-out value */
}
if ((z % tileZ) == 0) {
if (!Rast3d_flush_tiles_in_cube
(mask, 0, 0, MAX(0, z - tileZ), region.rows - 1,
region.cols - 1, z))
Rast3d_fatal_error(_("makeMask: error flushing tiles in cube"));
}
}
if (!Rast3d_flush_all_tiles(mask))
Rast3d_fatal_error(_("makeMask: error flushing all tiles"));
Rast3d_autolock_off(map);
Rast3d_unlock_all(map);
Rast3d_autolock_off(mask);
Rast3d_unlock_all(mask);
if (!Rast3d_close(mask))
Rast3d_fatal_error(_("Unable to close 3D raster mask file"));
if (!Rast3d_close(map))
Rast3d_fatal_error(_("Unable to close raster map <%s>"), name);
}
static void
modifyNull(char *name, d_Mask * maskRules, int changeNull, double newNullVal)
{
void *map, *mapOut;
RASTER3D_Region region;
int tileX, tileY, tileZ, x, y, z;
double value;
int doCompress, doLzw, doRle, precision;
int cacheSize;
cacheSize = Rast3d_cache_size_encode(RASTER3D_USE_CACHE_XY, 1);
if (NULL == G_find_raster3d(name, ""))
Rast3d_fatal_error(_("3D raster map <%s> not found"), name);
fprintf(stderr, "name %s Mapset %s \n", name, G_mapset());
map = Rast3d_open_cell_old(name, G_mapset(), RASTER3D_DEFAULT_WINDOW,
DCELL_TYPE, cacheSize);
if (map == NULL)
Rast3d_fatal_error(_("Unable to open 3D raster map <%s>"), name);
Rast3d_get_region_struct_map(map, ®ion);
Rast3d_get_tile_dimensions_map(map, &tileX, &tileY, &tileZ);
Rast3d_get_compression_mode(&doCompress, &doLzw, &doRle, &precision);
mapOut = Rast3d_open_new_param(name, DCELL_TYPE, RASTER3D_USE_CACHE_XY,
®ion, Rast3d_file_type_map(map),
doLzw, doRle, Rast3d_tile_precision_map(map), tileX,
tileY, tileZ);
if (mapOut == NULL)
Rast3d_fatal_error(_("modifyNull: error opening tmp file"));
Rast3d_min_unlocked(map, RASTER3D_USE_CACHE_X);
Rast3d_autolock_on(map);
Rast3d_unlock_all(map);
Rast3d_min_unlocked(mapOut, RASTER3D_USE_CACHE_X);
Rast3d_autolock_on(mapOut);
Rast3d_unlock_all(mapOut);
for (z = 0; z < region.depths; z++) {
if ((z % tileZ) == 0) {
Rast3d_unlock_all(map);
Rast3d_unlock_all(mapOut);
}
for (y = 0; y < region.rows; y++)
for (x = 0; x < region.cols; x++) {
value = Rast3d_get_double_region(map, x, y, z);
if (Rast3d_is_null_value_num(&value, DCELL_TYPE)) {
if (changeNull) {
value = newNullVal;
}
}
else if (Rast3d_mask_d_select((DCELL *) & value, maskRules)) {
Rast3d_set_null_value(&value, 1, DCELL_TYPE);
}
Rast3d_put_double(mapOut, x, y, z, value);
}
if ((z % tileZ) == 0) {
if (!Rast3d_flush_tiles_in_cube
(mapOut, 0, 0, MAX(0, z - tileZ), region.rows - 1,
region.cols - 1, z))
Rast3d_fatal_error(_("modifyNull: error flushing tiles in cube"));
}
}
if (!Rast3d_flush_all_tiles(mapOut))
Rast3d_fatal_error(_("modifyNull: error flushing all tiles"));
Rast3d_autolock_off(map);
Rast3d_unlock_all(map);
Rast3d_autolock_off(mapOut);
Rast3d_unlock_all(mapOut);
if (!Rast3d_close(map))
Rast3d_fatal_error(_("Unable to close raster map"));
if (!Rast3d_close(mapOut))
Rast3d_fatal_error(_("modifyNull: Unable to close tmp file"));
}
