12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040 |
- /*!
- * \file lib/raster/open.c
- *
- * \brief Raster Library - Open raster file
- *
- * (C) 1999-2009 by the GRASS Development Team
- *
- * This program is free software under the GNU General Public
- * License (>=v2). Read the file COPYING that comes with GRASS
- * for details.
- *
- * \author USACERL and many others
- */
- #include <rpc/types.h>
- #include <rpc/xdr.h>
- #include <unistd.h>
- #include <string.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <fcntl.h>
- #include <errno.h>
- #include <grass/config.h>
- #include <grass/gis.h>
- #include <grass/raster.h>
- #include <grass/glocale.h>
- #include "R.h"
- #define FORMAT_FILE "f_format"
- #define NULL_FILE "null"
- #define NULL2_FILE "null2"
- static int new_fileinfo(void)
- {
- int oldsize = R__.fileinfo_count;
- int newsize = oldsize;
- int i;
- for (i = 0; i < oldsize; i++)
- if (R__.fileinfo[i].open_mode <= 0) {
- memset(&R__.fileinfo[i], 0, sizeof(struct fileinfo));
- R__.fileinfo[i].open_mode = -1;
- return i;
- }
- if (newsize < 20)
- newsize += 20;
- else
- newsize *= 2;
- R__.fileinfo = G_realloc(R__.fileinfo, newsize * sizeof(struct fileinfo));
- /* Mark all cell files as closed */
- for (i = oldsize; i < newsize; i++) {
- memset(&R__.fileinfo[i], 0, sizeof(struct fileinfo));
- R__.fileinfo[i].open_mode = -1;
- }
- R__.fileinfo_count = newsize;
- return oldsize;
- }
- /*!
- * \brief Open raster file
- *
- * Arrange for the NULL-value bitmap to be read as well as the raster
- * map. If no NULL-value bitmap exists, arrange for the production of
- * NULL-values based on zeros in the raster map. If the map is
- * floating-point, arrange for quantization to integer for
- * Rast_get_c_row(), et. al., by reading the quantization rules
- * for the map using Rast_read_quant(). If the programmer wants to read
- * the floating point map using uing quant rules other than the ones
- * stored in map's quant file, he/she should call Rast_set_quant_rules()
- * after the call to Rast_open_old().
- *
- * \param name map name
- * \param open_mode mode
- * \param map_type map type (CELL, FCELL, DCELL)
- *
- * \return open file descriptor ( >= 0) if successful
- */
- static int open_raster_new(const char *name, int open_mode,
- RASTER_MAP_TYPE map_type);
- /*!
- \brief Open an existing integer raster map (cell)
- Opens the existing cell file <i>name</i> in the <i>mapset</i> for
- reading by Rast_get_row() with mapping into the current window.
- This routine opens the raster map <i>name</i> in <i>mapset</i> for
- reading. A nonnegative file descriptor is returned if the open is
- successful. Otherwise a diagnostic message is printed and a negative
- value is returned. This routine does quite a bit of work. Since
- GRASS users expect that all raster maps will be resampled into the
- current region, the resampling index for the raster map is prepared
- by this routine after the file is opened. The resampling is based on
- the active module region (see also \ref The_Region}. Preparation
- required for reading the various raster file formats (see \ref
- Raster_File_Format for an explanation of the various raster file
- formats) is also done.
- Diagnostics: warning message printed if open fails.
- \param name map name
- \param mapset mapset name where raster map <i>name</i> lives
- \return nonnegative file descriptor (int)
- */
- int Rast_open_old(const char *name, const char *mapset)
- {
- int fd = Rast__open_old(name, mapset);
- /* turn on auto masking, if not already on */
- Rast__check_for_auto_masking();
- /*
- if(R__.auto_mask <= 0)
- R__.mask_buf = Rast_allocate_c_buf();
- now we don't ever free it!, so no need to allocate it (Olga)
- */
- /* mask_buf is used for reading MASK file when mask is set and
- for reading map rows when the null file doesn't exist */
- return fd;
- }
- /*! \brief Lower level function, open cell files, supercell files,
- and the MASK file.
- Actions:
- - opens the named cell file, following reclass reference if
- named layer is a reclass layer.
- - creates the required mapping between the data and the window
- for use by the get_map_row family of routines.
- Diagnostics: Errors other than actual open failure will cause a
- diagnostic to be delivered thru G_warning() open failure messages
- are left to the calling routine since the masking logic will want to
- issue a different warning.
- Note: This routine does NOT open the MASK layer. If it did we would
- get infinite recursion. This routine is called to open the mask by
- Rast__check_for_auto_masking() which is called by Rast_open_old().
- \param name map name
- \param mapset mapset of cell file to be opened
- \return open file descriptor
- */
- int Rast__open_old(const char *name, const char *mapset)
- {
- struct fileinfo *fcb;
- int cell_fd, fd;
- char *cell_dir;
- const char *r_name;
- const char *r_mapset;
- struct Cell_head cellhd;
- int CELL_nbytes = 0; /* bytes per cell in CELL map */
- int INTERN_SIZE;
- int reclass_flag;
- int MAP_NBYTES;
- RASTER_MAP_TYPE MAP_TYPE;
- struct Reclass reclass;
- char xname[GNAME_MAX], xmapset[GMAPSET_MAX];
- struct GDAL_link *gdal;
- Rast__init();
- G_unqualified_name(name, mapset, xname, xmapset);
- name = xname;
- mapset = xmapset;
- if (!G_find_raster2(name, mapset))
- G_fatal_error(_("Raster map <%s> not found"),
- G_fully_qualified_name(name, mapset));
- /* Check for reclassification */
- reclass_flag = Rast_get_reclass(name, mapset, &reclass);
- switch (reclass_flag) {
- case 0:
- r_name = name;
- r_mapset = mapset;
- break;
- case 1:
- r_name = reclass.name;
- r_mapset = reclass.mapset;
- if (!G_find_raster2(r_name, r_mapset))
- G_fatal_error(_("Unable to open raster map <%s@%s> since it is a reclass "
- "of raster map <%s@%s> which does not exist"),
- name, mapset, r_name, r_mapset);
- break;
- default: /* Error reading cellhd/reclass file */
- G_fatal_error(_("Error reading reclass file for raster map <%s>"),
- G_fully_qualified_name(name, mapset));
- break;
- }
- /* read the cell header */
- Rast_get_cellhd(r_name, r_mapset, &cellhd);
- /* now check the type */
- MAP_TYPE = Rast_map_type(r_name, r_mapset);
- if (MAP_TYPE < 0)
- G_fatal_error(_("Error reading map type for raster map <%s>"),
- G_fully_qualified_name(name, mapset));
- if (MAP_TYPE == CELL_TYPE)
- /* set the number of bytes for CELL map */
- {
- CELL_nbytes = cellhd.format + 1;
- if (CELL_nbytes < 1)
- G_fatal_error(_("Raster map <%s@%s>: format field in header file invalid"),
- r_name, r_mapset);
- }
- /* compressor */
- if (MAP_TYPE != CELL_TYPE) {
- /* fp maps do not use RLE */
- /* previously, compressed simply meant yes (ZLIB) or no
- * now compressed encodes compressor type
- * 0: not compressed
- * 1, 2: ZLIB
- * 3: LZ4
- * 4: BZIP2
- * etc */
- if (cellhd.compressed == 1)
- cellhd.compressed = 2;
- }
- /* TODO: test if compressor type is supported */
- if (cellhd.proj != R__.rd_window.proj)
- G_fatal_error(_("Raster map <%s> is in different projection than current region. "
- "Found <%s>, should be <%s>."),
- G_fully_qualified_name(name, mapset),
- G_projection_name(cellhd.proj),
- G_projection_name(R__.rd_window.proj));
- if (cellhd.zone != R__.rd_window.zone)
- G_fatal_error(_("Raster map <%s> is in different zone (%d) than current region (%d)"),
- G_fully_qualified_name(name, mapset), cellhd.zone, R__.rd_window.zone);
- /* when map is int warn if too large cell size */
- if (MAP_TYPE == CELL_TYPE && (unsigned int)CELL_nbytes > sizeof(CELL))
- G_fatal_error(_("Raster map <%s>: bytes per cell (%d) too large"),
- G_fully_qualified_name(name, mapset), CELL_nbytes);
- /* record number of bytes per cell */
- if (MAP_TYPE == FCELL_TYPE) {
- cell_dir = "fcell";
- INTERN_SIZE = sizeof(FCELL);
- MAP_NBYTES = XDR_FLOAT_NBYTES;
- }
- else if (MAP_TYPE == DCELL_TYPE) {
- cell_dir = "fcell";
- INTERN_SIZE = sizeof(DCELL);
- MAP_NBYTES = XDR_DOUBLE_NBYTES;
- }
- else { /* integer */
- cell_dir = "cell";
- INTERN_SIZE = sizeof(CELL);
- MAP_NBYTES = CELL_nbytes;
- }
- gdal = Rast_get_gdal_link(r_name, r_mapset);
- if (gdal) {
- #ifdef HAVE_GDAL
- cell_fd = -1;
- #else
- G_fatal_error(_("Raster map <%s@%s> is a GDAL link but GRASS is compiled without GDAL support"),
- r_name, r_mapset);
- #endif
- }
- else {
- /* now actually open file for reading */
- cell_fd = G_open_old(cell_dir, r_name, r_mapset);
- if (cell_fd < 0)
- G_fatal_error(_("Unable to open %s file for raster map <%s@%s>"),
- cell_dir, r_name, r_mapset);
- }
- fd = new_fileinfo();
- fcb = &R__.fileinfo[fd];
- fcb->data_fd = cell_fd;
- fcb->map_type = MAP_TYPE;
- /* Save cell header */
- fcb->cellhd = cellhd;
- /* allocate null bitstream buffers for reading null rows */
- fcb->null_fd = -1;
- fcb->null_cur_row = -1;
- fcb->null_bits = Rast__allocate_null_bits(cellhd.cols);
- /* mark closed */
- fcb->open_mode = -1;
- /* save name and mapset */
- fcb->name = G_store(name);
- fcb->mapset = G_store(mapset);
- /* mark no data row in memory */
- fcb->cur_row = -1;
- /* if reclass, copy reclass structure */
- if ((fcb->reclass_flag = reclass_flag))
- fcb->reclass = reclass;
- fcb->gdal = gdal;
- if (!gdal)
- /* check for compressed data format, making initial reads if necessary */
- if (Rast__check_format(fd) < 0) {
- close(cell_fd); /* warning issued by check_format() */
- G_fatal_error(_("Error reading format for <%s@%s>"),
- r_name, r_mapset);
- }
- /* create the mapping from cell file to window */
- Rast__create_window_mapping(fd);
- /*
- * allocate the data buffer
- * number of bytes per cell is cellhd.format+1
- */
- /* for reading fcb->data is allocated to be fcb->cellhd.cols * fcb->nbytes
- (= XDR_FLOAT/DOUBLE_NBYTES) */
- fcb->data = (unsigned char *)G_calloc(fcb->cellhd.cols, MAP_NBYTES);
- /* initialize/read in quant rules for float point maps */
- if (fcb->map_type != CELL_TYPE) {
- if (fcb->reclass_flag)
- Rast_read_quant(fcb->reclass.name, fcb->reclass.mapset,
- &(fcb->quant));
- else
- Rast_read_quant(fcb->name, fcb->mapset, &(fcb->quant));
- }
- /* now mark open for read: this must follow create_window_mapping() */
- fcb->open_mode = OPEN_OLD;
- fcb->io_error = 0;
- fcb->map_type = MAP_TYPE;
- fcb->nbytes = MAP_NBYTES;
- fcb->null_row_ptr = NULL;
- if (!gdal) {
- /* First, check for compressed null file */
- fcb->null_fd = G_open_old_misc("cell_misc", NULL_FILE, r_name, r_mapset);
- if (fcb->null_fd < 0) {
- fcb->null_fd = G_open_old_misc("cell_misc", NULL2_FILE, r_name, r_mapset);
- if (fcb->null_fd >= 0) {
- fcb->null_row_ptr = G_calloc(fcb->cellhd.rows + 1, sizeof(off_t));
- if (Rast__read_null_row_ptrs(fd, fcb->null_fd) < 0) {
- close(fcb->null_fd);
- fcb->null_fd = -1;
- G_free(fcb->null_row_ptr);
- fcb->null_row_ptr = NULL;
- }
- }
- }
- fcb->null_file_exists = fcb->null_fd >= 0;
- }
- return fd;
- }
- /*!
- \brief Opens a new cell file in a database (compressed)
- Opens a new cell file <i>name</i> in the current mapset for writing
- by Rast_put_row().
- The file is created and filled with no data it is assumed that the
- new cell file is to conform to the current window.
- The file must be written sequentially. Use Rast_open_new_random()
- for non sequential writes.
- Note: the open actually creates a temporary file Rast_close() will
- move the temporary file to the cell file and write out the necessary
- support files (cellhd, cats, hist, etc.).
- Diagnostics: warning message printed if open fails
- Warning: calls to Rast_set_window() made after opening a new cell file
- may create confusion and should be avoided the new cell file will be
- created to conform to the window at the time of the open.
- \param name map name
- \return open file descriptor ( >= 0) if successful
- \return negative integer if error
- */
- int Rast_open_c_new(const char *name)
- {
- return open_raster_new(name, OPEN_NEW_COMPRESSED, CELL_TYPE);
- }
- /*!
- \brief Opens a new cell file in a database (uncompressed)
- See also Rast_open_new().
- \param name map name
- \return open file descriptor ( >= 0) if successful
- \return negative integer if error
- */
- int Rast_open_c_new_uncompressed(const char *name)
- {
- return open_raster_new(name, OPEN_NEW_UNCOMPRESSED, CELL_TYPE);
- }
- /*!
- \brief Save histogram for newly create raster map (cell)
- If newly created cell files should have histograms, set flag=1
- otherwise set flag=0. Applies to subsequent opens.
- \param flag flag indicator
- */
- void Rast_want_histogram(int flag)
- {
- R__.want_histogram = flag;
- }
- /*!
- \brief Sets the format for subsequent opens on new integer cell files
- (uncompressed and random only).
- Warning: subsequent put_row calls will only write n+1 bytes per
- cell. If the data requires more, the cell file will be written
- incorrectly (but with n+1 bytes per cell)
- When writing float map: format is -1
- \param n format
- */
- void Rast_set_cell_format(int n)
- /* sets the format for integer raster map */
- {
- R__.nbytes = n + 1;
- if (R__.nbytes <= 0)
- R__.nbytes = 1;
- if (R__.nbytes > sizeof(CELL))
- R__.nbytes = sizeof(CELL);
- }
- /*!
- \brief Get cell value format
- \param v cell
- \return cell format
- */
- int Rast_get_cell_format(CELL v)
- {
- unsigned int i;
- if (v >= 0)
- for (i = 0; i < sizeof(CELL); i++)
- if (!(v /= 256))
- return i;
- return sizeof(CELL) - 1;
- }
- /*!
- \brief Opens new fcell file in a database
- Opens a new floating-point map <i>name</i> in the current mapset for
- writing. The type of the file (i.e. either double or float) is
- determined and fixed at this point. The default is FCELL_TYPE. In
- order to change this default
- Use Rast_set_fp_type() where type is one of DCELL_TYPE or FCELL_TYPE.
- See warnings and notes for Rast_open_new().
- \param name map name
- \return nonnegative file descriptor (int)
- */
- int Rast_open_fp_new(const char *name)
- {
- return open_raster_new(name, OPEN_NEW_COMPRESSED, R__.fp_type);
- }
- /*!
- \brief Opens new fcell file in a database (uncompressed)
- See Rast_open_fp_new() for details.
- \param name map name
- \return nonnegative file descriptor (int)
- */
- int Rast_open_fp_new_uncompressed(const char *name)
- {
- return open_raster_new(name, OPEN_NEW_UNCOMPRESSED, R__.fp_type);
- }
- #ifdef HAVE_GDAL
- static int open_raster_new_gdal(char *map, char *mapset,
- RASTER_MAP_TYPE map_type)
- {
- int fd;
- struct fileinfo *fcb;
- fd = new_fileinfo();
- fcb = &R__.fileinfo[fd];
- fcb->data_fd = -1;
- /* mark closed */
- fcb->map_type = map_type;
- fcb->open_mode = -1;
- fcb->gdal = Rast_create_gdal_link(map, map_type);
- if (!fcb->gdal)
- G_fatal_error(_("Unable to create GDAL link"));
- fcb->cellhd = R__.wr_window;
- fcb->cellhd.compressed = 0;
- fcb->nbytes = Rast_cell_size(fcb->map_type);
- /* for writing fcb->data is allocated to be R__.wr_window.cols *
- sizeof(CELL or DCELL or FCELL) */
- fcb->data = G_calloc(R__.wr_window.cols, fcb->nbytes);
- fcb->name = map;
- fcb->mapset = mapset;
- fcb->cur_row = 0;
- fcb->row_ptr = NULL;
- fcb->temp_name = NULL;
- fcb->null_temp_name = NULL;
- fcb->null_cur_row = 0;
- fcb->null_bits = NULL;
- fcb->null_fd = -1;
- fcb->null_row_ptr = NULL;
- if (fcb->map_type != CELL_TYPE)
- Rast_quant_init(&(fcb->quant));
- /* init cell stats */
- /* now works only for int maps */
- if (fcb->map_type == CELL_TYPE)
- if ((fcb->want_histogram = R__.want_histogram))
- Rast_init_cell_stats(&fcb->statf);
- /* init range and if map is double/float init d/f_range */
- Rast_init_range(&fcb->range);
- if (fcb->map_type != CELL_TYPE)
- Rast_init_fp_range(&fcb->fp_range);
- /* mark file as open for write */
- fcb->open_mode = OPEN_NEW_UNCOMPRESSED;
- fcb->io_error = 0;
- return fd;
- }
- #endif /* HAVE_GDAL */
- static int open_raster_new(const char *name, int open_mode,
- RASTER_MAP_TYPE map_type)
- {
- char xname[GNAME_MAX], xmapset[GMAPSET_MAX];
- struct fileinfo *fcb;
- int fd, cell_fd;
- char *tempname;
- char *map;
- char *mapset;
- const char *cell_dir;
- int nbytes;
- Rast__init();
- switch (map_type) {
- case CELL_TYPE:
- cell_dir = "cell";
- nbytes = R__.nbytes;
- break;
- case FCELL_TYPE:
- nbytes = XDR_FLOAT_NBYTES;
- cell_dir = "fcell";
- break;
- case DCELL_TYPE:
- nbytes = XDR_DOUBLE_NBYTES;
- cell_dir = "fcell";
- break;
- default:
- G_fatal_error(_("Invalid map type <%d>"), map_type);
- break;
- }
- if (G_unqualified_name(name, G_mapset(), xname, xmapset) < 0)
- G_fatal_error(_("Raster map <%s> is not in the current mapset (%s)"),
- name, G_mapset());
- map = G_store(xname);
- mapset = G_store(xmapset);
- /* check for legal grass name */
- if (G_legal_filename(map) < 0)
- G_fatal_error(_("<%s> is an illegal file name"), map);
- #ifdef HAVE_GDAL
- if (G_find_file2("", "GDAL", G_mapset()))
- return open_raster_new_gdal(map, mapset, map_type);
- #endif
- /* open a tempfile name */
- tempname = G_tempfile();
- cell_fd = creat(tempname, 0666);
- if (cell_fd < 0) {
- G_free(mapset);
- G_free(tempname);
- G_free(map);
- G_fatal_error(_("No temp files available: %s"), strerror(errno));
- }
- fd = new_fileinfo();
- fcb = &R__.fileinfo[fd];
- fcb->data_fd = cell_fd;
- /*
- * since we are bypassing the normal open logic
- * must create the cell element
- */
- G_make_mapset_element(cell_dir);
- /* mark closed */
- fcb->map_type = map_type;
- fcb->open_mode = -1;
- fcb->gdal = NULL;
- /* for writing fcb->data is allocated to be R__.wr_window.cols *
- sizeof(CELL or DCELL or FCELL) */
- fcb->data = (unsigned char *)G_calloc(R__.wr_window.cols,
- Rast_cell_size(fcb->map_type));
- /*
- * copy current window into cell header
- * set format to cell/supercell
- * for compressed writing
- * allocate space to hold the row address array
- */
- fcb->cellhd = R__.wr_window;
- /* TODO: test if compressor type is supported */
- if (open_mode == OPEN_NEW_COMPRESSED && fcb->map_type == CELL_TYPE) {
- fcb->row_ptr = G_calloc(fcb->cellhd.rows + 1, sizeof(off_t));
- G_zero(fcb->row_ptr, (fcb->cellhd.rows + 1) * sizeof(off_t));
- Rast__write_row_ptrs(fd);
- fcb->cellhd.compressed = R__.compression_type;
- fcb->nbytes = 1; /* to the minimum */
- }
- else {
- fcb->nbytes = nbytes;
- if (open_mode == OPEN_NEW_COMPRESSED) {
- fcb->row_ptr = G_calloc(fcb->cellhd.rows + 1, sizeof(off_t));
- G_zero(fcb->row_ptr, (fcb->cellhd.rows + 1) * sizeof(off_t));
- Rast__write_row_ptrs(fd);
- fcb->cellhd.compressed = R__.compression_type;
- }
- else
- fcb->cellhd.compressed = 0;
- if (fcb->map_type != CELL_TYPE) {
- Rast_quant_init(&(fcb->quant));
- }
- }
- if (open_mode == OPEN_NEW_COMPRESSED && fcb->map_type != CELL_TYPE &&
- fcb->cellhd.compressed == 1) {
- /* fp maps do not use RLE */
- fcb->cellhd.compressed = 2;
- }
- /* save name and mapset, and tempfile name */
- fcb->name = map;
- fcb->mapset = mapset;
- fcb->temp_name = tempname;
- /* next row to be written (in order) is zero */
- fcb->cur_row = 0;
- /* open a null tempfile name */
- tempname = G_tempfile();
- fcb->null_fd = creat(tempname, 0666);
- if (fcb->null_fd < 0) {
- G_free(tempname);
- G_free(fcb->name);
- G_free(fcb->mapset);
- G_free(fcb->temp_name);
- close(cell_fd);
- G_fatal_error(_("No temp files available: %s"), strerror(errno));
- }
- fcb->null_temp_name = tempname;
- if (R__.compress_nulls) {
- fcb->null_row_ptr = G_calloc(fcb->cellhd.rows + 1, sizeof(off_t));
- G_zero(fcb->row_ptr, (fcb->cellhd.rows + 1) * sizeof(off_t));
- Rast__write_null_row_ptrs(fd, fcb->null_fd);
- }
- /* next row to be written (in order) is zero */
- fcb->null_cur_row = 0;
- /* allocate null bitstream buffer for writing */
- fcb->null_bits = Rast__allocate_null_bits(fcb->cellhd.cols);
- /* init cell stats */
- /* now works only for int maps */
- if (fcb->map_type == CELL_TYPE)
- if ((fcb->want_histogram = R__.want_histogram))
- Rast_init_cell_stats(&fcb->statf);
- /* init range and if map is double/float init d/f_range */
- Rast_init_range(&fcb->range);
- if (fcb->map_type != CELL_TYPE)
- Rast_init_fp_range(&fcb->fp_range);
- /* mark file as open for write */
- fcb->open_mode = open_mode;
- fcb->io_error = 0;
- return fd;
- }
- int Rast__open_null_write(const char *name)
- {
- char xname[GNAME_MAX], xmapset[GMAPSET_MAX];
- struct fileinfo *fcb;
- int fd;
- char *tempname;
- char *map;
- char *mapset;
- Rast__init();
- if (!G_find_raster2(name, G_mapset()))
- G_fatal_error(_("Raster map <%s> does not exist in the current mapset (%s)"),
- name, G_mapset());
- if (G_unqualified_name(name, G_mapset(), xname, xmapset) < 0)
- G_fatal_error(_("Raster map <%s> is not in the current mapset (%s)"),
- name, G_mapset());
- map = G_store(xname);
- mapset = G_store(xmapset);
- fd = new_fileinfo();
- fcb = &R__.fileinfo[fd];
- G_zero(fcb, sizeof(*fcb));
- fcb->name = map;
- fcb->mapset = mapset;
- Rast_get_cellhd(map, mapset, &fcb->cellhd);
- /* open a null tempfile name */
- tempname = G_tempfile();
- fcb->null_fd = creat(tempname, 0666);
- if (fcb->null_fd < 0) {
- G_free(tempname);
- G_free(fcb->name);
- G_free(fcb->mapset);
- G_fatal_error(_("No temp files available: %s"), strerror(errno));
- }
- fcb->null_temp_name = tempname;
- if (R__.compress_nulls) {
- fcb->null_row_ptr = G_calloc(fcb->cellhd.rows + 1, sizeof(off_t));
- G_zero(fcb->null_row_ptr, (fcb->cellhd.rows + 1) * sizeof(off_t));
- Rast__write_null_row_ptrs(fd, fcb->null_fd);
- }
- /* allocate null bitstream buffer for writing */
- fcb->null_bits = Rast__allocate_null_bits(fcb->cellhd.cols);
- return fd;
- }
- /*!
- \brief Set raster map floating-point data format.
- This controls the storage type for floating-point maps. It affects
- subsequent calls to G_open_fp_map_new(). The <i>type</i> must be
- one of FCELL_TYPE (float) or DCELL_TYPE (double). The use of this
- routine by applications is discouraged since its use would override
- user preferences.
- \param type raster data type
- \return void
- */
- void Rast_set_fp_type(RASTER_MAP_TYPE map_type)
- {
- Rast__init();
- switch (map_type) {
- case FCELL_TYPE:
- case DCELL_TYPE:
- R__.fp_type = map_type;
- break;
- default:
- G_fatal_error(_("Rast_set_fp_type(): can only be called with FCELL_TYPE or DCELL_TYPE"));
- break;
- }
- }
- /*!
- \brief Check if raster map is floating-point
- Returns true (1) if raster map <i>name</i> in <i>mapset</i>
- is a floating-point dataset; false(0) otherwise.
- \param name map name
- \param mapset mapset name
- \return 1 floating-point
- \return 0 int
- */
- int Rast_map_is_fp(const char *name, const char *mapset)
- {
- char path[GPATH_MAX];
- const char *xmapset;
- xmapset = G_find_raster2(name, mapset);
- if (!xmapset)
- G_fatal_error(_("Raster map <%s> not found"),
- G_fully_qualified_name(name, mapset));
- G_file_name(path, "fcell", name, xmapset);
- if (access(path, 0) == 0)
- return 1;
- G_file_name(path, "g3dcell", name, xmapset);
- if (access(path, 0) == 0)
- return 1;
- return 0;
- }
- /*!
- \brief Determine raster data type
- Determines if the raster map is floating point or integer. Returns
- DCELL_TYPE for double maps, FCELL_TYPE for float maps, CELL_TYPE for
- integer maps, -1 if error has occurred
- \param name map name
- \param mapset mapset where map <i>name</i> lives
- \return raster data type
- */
- RASTER_MAP_TYPE Rast_map_type(const char *name, const char *mapset)
- {
- char path[GPATH_MAX];
- const char *xmapset;
- xmapset = G_find_raster2(name, mapset);
- if (!xmapset) {
- if (mapset && *mapset)
- G_fatal_error(_("Raster map <%s> not found in mapset <%s>"),
- name, mapset);
- else
- G_fatal_error(_("Raster map <%s> not found"), name);
- }
- G_file_name(path, "fcell", name, xmapset);
- if (access(path, 0) == 0)
- return Rast__check_fp_type(name, xmapset);
- G_file_name(path, "g3dcell", name, xmapset);
- if (access(path, 0) == 0)
- return DCELL_TYPE;
- return CELL_TYPE;
- }
- /*!
- \brief Determine raster type from descriptor
- Determines if the raster map is floating point or integer. Returns
- DCELL_TYPE for double maps, FCELL_TYPE for float maps, CELL_TYPE for
- integer maps, -1 if error has occurred
- \param fd file descriptor
- \return raster data type
- */
- RASTER_MAP_TYPE Rast_get_map_type(int fd)
- {
- struct fileinfo *fcb = &R__.fileinfo[fd];
- return fcb->map_type;
- }
- /*!
- \brief Determines whether the floating points cell file has double or float type
- \param name map name
- \param mapset mapset where map <i>name</i> lives
- \return raster type (fcell, dcell)
- */
- RASTER_MAP_TYPE Rast__check_fp_type(const char *name, const char *mapset)
- {
- char path[GPATH_MAX];
- struct Key_Value *format_keys;
- const char *str, *str1;
- RASTER_MAP_TYPE map_type;
- const char *xmapset;
- xmapset = G_find_raster2(name, mapset);
- if (!xmapset)
- G_fatal_error(_("Raster map <%s> not found"),
- G_fully_qualified_name(name, mapset));
- G_file_name_misc(path, "cell_misc", FORMAT_FILE, name, xmapset);
- if (access(path, 0) != 0)
- G_fatal_error(_("Unable to find '%s'"), path);
- format_keys = G_read_key_value_file(path);
- if ((str = G_find_key_value("type", format_keys)) != NULL) {
- if (strcmp(str, "double") == 0)
- map_type = DCELL_TYPE;
- else if (strcmp(str, "float") == 0)
- map_type = FCELL_TYPE;
- else {
- G_free_key_value(format_keys);
- G_fatal_error(_("Invalid type: field '%s' in file '%s'"), str, path);
- }
- }
- else {
- G_free_key_value(format_keys);
- G_fatal_error(_("Missing type: field in file '%s'"), path);
- }
- if ((str1 = G_find_key_value("byte_order", format_keys)) != NULL) {
- if (strcmp(str1, "xdr") != 0)
- G_warning(_("Raster map <%s> is not xdr: byte_order: %s"),
- name, str);
- /* here read and translate byte order if not using xdr */
- }
- G_free_key_value(format_keys);
- return map_type;
- }
- /*!
- \brief Opens a new raster map
- Opens a new raster map of type <i>wr_type</i>
- See warnings and notes for Rast_open_new().
- Supported data types:
- - CELL_TYPE
- - FCELL_TYPE
- - DCELL_TYPE
- On CELL_TYPE calls Rast_open_new() otherwise Rast_open_fp_new().
- \param name map name
- \param wr_type raster data type
- \return nonnegative file descriptor (int)
- */
- int Rast_open_new(const char *name, RASTER_MAP_TYPE wr_type)
- {
- return open_raster_new(name, OPEN_NEW_COMPRESSED, wr_type);
- }
- /*!
- \brief Opens a new raster map (uncompressed)
- See Rast_open_new().
- \param name map name
- \param wr_type raster data type
- \return nonnegative file descriptor (int)
- */
- int Rast_open_new_uncompressed(const char *name, RASTER_MAP_TYPE wr_type)
- {
- return open_raster_new(name, OPEN_NEW_UNCOMPRESSED, wr_type);
- }
- /*!
- \brief Sets quant translation rules for raster map opened for
- reading.
- Returned by Rast_open_old(). After calling this function,
- Rast_get_c_row() and Rast_get_c_row() will use rules defined by q
- (instead of using rules defined in map's quant file) to convert floats to
- ints.
- \param fd file descriptor (cell file)
- \param q pointer to Quant structure
- \return void
- */
- void Rast_set_quant_rules(int fd, struct Quant *q)
- {
- struct fileinfo *fcb = &R__.fileinfo[fd];
- CELL cell;
- DCELL dcell;
- struct Quant_table *p;
- if (fcb->open_mode != OPEN_OLD)
- G_fatal_error(_("Rast_set_quant_rules() can be called only for "
- "raster maps opened for reading"));
- /* copy all info from q to fcb->quant) */
- Rast_quant_init(&fcb->quant);
- if (q->truncate_only) {
- Rast_quant_truncate(&fcb->quant);
- return;
- }
- for (p = &(q->table[q->nofRules - 1]); p >= q->table; p--)
- Rast_quant_add_rule(&fcb->quant, p->dLow, p->dHigh, p->cLow,
- p->cHigh);
- if (Rast_quant_get_neg_infinite_rule(q, &dcell, &cell) > 0)
- Rast_quant_set_neg_infinite_rule(&fcb->quant, dcell, cell);
- if (Rast_quant_get_pos_infinite_rule(q, &dcell, &cell) > 0)
- Rast_quant_set_pos_infinite_rule(&fcb->quant, dcell, cell);
- }
|