[GRASS-SVN] r73315 - grass-addons/grass7/raster/r.neighborhoodmatrix
svn_grass at osgeo.org
svn_grass at osgeo.org
Tue Sep 11 05:26:55 PDT 2018
Author: mlennert
Date: 2018-09-11 05:26:55 -0700 (Tue, 11 Sep 2018)
New Revision: 73315
Added:
grass-addons/grass7/raster/r.neighborhoodmatrix/main.c
grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.c
grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.h
Log:
r.neighborhoodmatrix: rewrite in C (MarkusM) and addition of a flag for column header output (add C files)
Added: grass-addons/grass7/raster/r.neighborhoodmatrix/main.c
===================================================================
--- grass-addons/grass7/raster/r.neighborhoodmatrix/main.c (rev 0)
+++ grass-addons/grass7/raster/r.neighborhoodmatrix/main.c 2018-09-11 12:26:55 UTC (rev 73315)
@@ -0,0 +1,317 @@
+
+/****************************************************************************
+ *
+ * MODULE: r.neigborhoodmatrix
+ *
+ * AUTHOR(S): Moritz Lennert (original Python version)
+ * Markus Metz (C version)
+ *
+ * PURPOSE: Calculates a neighborhood matrix for a raster map with regions
+ * (e.g. the output of r.clump or i.segment)
+ *
+ * COPYRIGHT: (C) 2018 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.
+ *
+ ***************************************************************************/
+
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+#include <math.h>
+#include <grass/gis.h>
+#include <grass/raster.h>
+#include <grass/glocale.h>
+#include "pavl.h"
+
+struct nbp
+{
+ int a, b, cnt;
+};
+
+static int cmp_nbp(const void *a, const void *b)
+{
+ struct nbp *nbpa = (struct nbp *)a;
+ struct nbp *nbpb = (struct nbp *)b;
+
+ if (nbpa->a < nbpb->a)
+ return -1;
+ if (nbpa->a > nbpb->a)
+ return 1;
+
+ return (nbpa->b < nbpb->b ? -1 : nbpa->b > nbpb->b);
+}
+
+static void free_pavl_item(void *p)
+{
+ G_free(p);
+}
+
+/* compare two cell values
+ * return 1 if different */
+static int cmp_cells(CELL a, CELL b, int a_null, int b_null)
+{
+ return (!a_null && !b_null && a != b);
+}
+
+int main(int argc, char *argv[])
+{
+ int row, col, nrows, ncols;
+
+ struct Range range;
+ CELL min, max;
+ int in_fd;
+ int i;
+ struct GModule *module;
+ struct Option *opt_in;
+ struct Option *opt_out;
+ struct Option *opt_sep;
+ struct Flag *flag_len;
+ struct Flag *flag_diag;
+ struct Flag *flag_nohead;
+ char *sep;
+ FILE *out_fp;
+ CELL *prev_in, *cur_in, *temp_in;
+ CELL cur, ngbr;
+ int cur_null, ngbr_null;
+ int len;
+ struct Cell_head cellhd;
+ struct pavl_table *nbptree;
+ struct pavl_traverser nbp_trav;
+ struct nbp *nbp_found, nbp_search, *nbp_new;
+
+ G_gisinit(argv[0]);
+
+ /* Define the different options */
+
+ module = G_define_module();
+ G_add_keyword(_("raster"));
+ G_add_keyword(_("statistics"));
+ G_add_keyword(_("reclass"));
+ G_add_keyword(_("clumps"));
+ module->description =
+ _("Calculates geometry parameters for raster objects.");
+
+ opt_in = G_define_standard_option(G_OPT_R_INPUT);
+ opt_in->description = _("Raster for which to calculate the neighboorhood matrix");
+
+ opt_out = G_define_standard_option(G_OPT_F_OUTPUT);
+ opt_out->required = NO;
+
+ opt_sep = G_define_standard_option(G_OPT_F_SEP);
+
+ flag_diag = G_define_flag();
+ flag_diag->key = 'd';
+ flag_diag->description = _("Also take into account diagonal neighbors");
+
+ flag_len = G_define_flag();
+ flag_len->key = 'l';
+ flag_len->description = _("Also output length of common border (in pixels)");
+
+ flag_nohead = G_define_flag();
+ flag_nohead->key = 'c';
+ flag_nohead->description = _("Include column names in output (meaning will be inversed soon)");
+
+ G_option_exclusive(flag_len, flag_diag, NULL);
+
+ /* parse options */
+ if (G_parser(argc, argv))
+ exit(EXIT_FAILURE);
+
+ sep = G_option_to_separator(opt_sep);
+ in_fd = Rast_open_old(opt_in->answer, "");
+
+ if (Rast_get_map_type(in_fd) != CELL_TYPE)
+ G_fatal_error(_("Input raster must be of type CELL"));
+
+ Rast_read_range(opt_in->answer, "", &range);
+ Rast_get_range_min_max(&range, &min, &max);
+ if (Rast_is_c_null_value(&min) || Rast_is_c_null_value(&max))
+ G_fatal_error(_("Empty input map <%s>"), opt_in->answer);
+ if (max == min)
+ G_fatal_error(_("Input map <%s> has only one value: %d"), opt_in->answer, max);
+
+ if (opt_out->answer != NULL && strcmp(opt_out->answer, "-") != 0) {
+ if(! (out_fp = fopen(opt_out->answer, "w")))
+ G_fatal_error(_("Unable to open file <%s> for writing"), opt_out->answer);
+ } else {
+ out_fp = stdout;
+ }
+
+ Rast_get_window(&cellhd);
+ nrows = cellhd.rows;
+ ncols = cellhd.cols;
+
+ /* allocate CELL buffers two columns larger than current window */
+ len = (ncols + 2) * sizeof(CELL);
+ prev_in = (CELL *) G_malloc(len);
+ cur_in = (CELL *) G_malloc(len);
+
+ /* fake a previous row which is all NULL */
+ Rast_set_c_null_value(prev_in, ncols + 2);
+
+ /* set left and right edge to NULL */
+ Rast_set_c_null_value(&cur_in[0], 1);
+ Rast_set_c_null_value(&cur_in[ncols + 1], 1);
+
+ nbptree = pavl_create(cmp_nbp, NULL);
+ nbp_new = NULL;
+
+ G_message(_("Calculating neighborhood matrix"));
+ for (row = 0; row < nrows; row++) {
+ G_percent(row, nrows, 2);
+
+ Rast_get_c_row(in_fd, cur_in + 1, row);
+ cur = cur_in[0];
+ cur_null = 1;
+ for (col = 1; col <= ncols; col++) {
+
+ cur = cur_in[col];
+ cur_null = Rast_is_c_null_value(&cur);
+ if (cur_null)
+ continue;
+
+ /* top */
+ ngbr = prev_in[col];
+ ngbr_null = Rast_is_c_null_value(&ngbr);
+ if (cmp_cells(cur, ngbr, cur_null, ngbr_null)) {
+ if (nbp_new == NULL)
+ nbp_new = G_malloc(sizeof(struct nbp));
+
+ if (cur < ngbr) {
+ nbp_new->a = cur;
+ nbp_new->b = ngbr;
+ }
+ else {
+ nbp_new->b = cur;
+ nbp_new->a = ngbr;
+ }
+ nbp_new->cnt = 1;
+ nbp_found = pavl_insert(nbptree, nbp_new);
+ if (nbp_found)
+ nbp_found->cnt++;
+ else
+ nbp_new = NULL;
+ }
+
+ /* left */
+ ngbr = cur_in[col - 1];
+ ngbr_null = Rast_is_c_null_value(&ngbr);
+ if (cmp_cells(cur, ngbr, cur_null, ngbr_null)) {
+ if (nbp_new == NULL)
+ nbp_new = G_malloc(sizeof(struct nbp));
+
+ if (cur < ngbr) {
+ nbp_new->a = cur;
+ nbp_new->b = ngbr;
+ }
+ else {
+ nbp_new->b = cur;
+ nbp_new->a = ngbr;
+ }
+ nbp_new->cnt = 1;
+ nbp_found = pavl_insert(nbptree, nbp_new);
+ if (nbp_found)
+ nbp_found->cnt++;
+ else
+ nbp_new = NULL;
+ }
+
+ if (flag_diag->answer) {
+ /* top left */
+ ngbr = prev_in[col - 1];
+ ngbr_null = Rast_is_c_null_value(&ngbr);
+ if (cmp_cells(cur, ngbr, cur_null, ngbr_null)) {
+ if (nbp_new == NULL)
+ nbp_new = G_malloc(sizeof(struct nbp));
+
+ if (cur < ngbr) {
+ nbp_new->a = cur;
+ nbp_new->b = ngbr;
+ }
+ else {
+ nbp_new->b = cur;
+ nbp_new->a = ngbr;
+ }
+ nbp_new->cnt = 1;
+ nbp_found = pavl_insert(nbptree, nbp_new);
+ if (nbp_found)
+ nbp_found->cnt++;
+ else
+ nbp_new = NULL;
+ }
+
+ /* top right */
+ ngbr = prev_in[col + 1];
+ ngbr_null = Rast_is_c_null_value(&ngbr);
+ if (cmp_cells(cur, ngbr, cur_null, ngbr_null)) {
+ if (nbp_new == NULL)
+ nbp_new = G_malloc(sizeof(struct nbp));
+
+ if (cur < ngbr) {
+ nbp_new->a = cur;
+ nbp_new->b = ngbr;
+ }
+ else {
+ nbp_new->b = cur;
+ nbp_new->a = ngbr;
+ }
+ nbp_new->cnt = 1;
+ nbp_found = pavl_insert(nbptree, nbp_new);
+ if (nbp_found)
+ nbp_found->cnt++;
+ else
+ nbp_new = NULL;
+ }
+ }
+ }
+
+ /* switch the buffers so that the current buffer becomes the previous */
+ temp_in = cur_in;
+ cur_in = prev_in;
+ prev_in = temp_in;
+ }
+
+ G_percent(1, 1, 1);
+
+ Rast_close(in_fd);
+ G_free(cur_in);
+ G_free(prev_in);
+
+ G_message(_("Writing output"));
+ /* print table */
+
+ if (flag_nohead->answer) {
+ /* print table header (column names) */
+ fprintf(out_fp, "acat%s", sep);
+ fprintf(out_fp, "bcat%s", sep);
+ if (flag_len->answer)
+ fprintf(out_fp, "border_length", sep);
+ fprintf(out_fp, "\n");
+ }
+
+ /* print table body */
+ pavl_t_init(&nbp_trav, nbptree);
+ while ((nbp_found = pavl_t_next(&nbp_trav)) != NULL) {
+ if (flag_len->answer && !flag_diag->answer) {
+ fprintf(out_fp, "%d%s%d%s%d\n", nbp_found->a, sep,
+ nbp_found->b, sep,
+ nbp_found->cnt);
+ fprintf(out_fp, "%d%s%d%s%d\n", nbp_found->b, sep,
+ nbp_found->a, sep,
+ nbp_found->cnt);
+ }
+ else {
+ fprintf(out_fp, "%d%s%d\n", nbp_found->a, sep,
+ nbp_found->b);
+ fprintf(out_fp, "%d%s%d\n", nbp_found->b, sep,
+ nbp_found->a);
+ }
+ }
+ if (out_fp != stdout)
+ fclose(out_fp);
+
+ exit(EXIT_SUCCESS);
+}
Added: grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.c
===================================================================
--- grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.c (rev 0)
+++ grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.c 2018-09-11 12:26:55 UTC (rev 73315)
@@ -0,0 +1,843 @@
+/* Produced by texiweb from libavl.w. */
+
+/* libavl - library for manipulation of binary trees.
+ Copyright (C) 1998, 1999, 2000, 2001, 2002, 2004 Free Software
+ Foundation, Inc.
+
+ This library is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Lesser General Public
+ License as published by the Free Software Foundation; either
+ version 3 of the License, or (at your option) any later version.
+
+ This library is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public
+ License along with this library; if not, write to the Free Software
+ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ 02110-1301 USA.
+ */
+
+/* Nov 2016, Markus Metz
+ * from libavl-2.0.3
+ * added safety checks and speed optimizations
+ */
+
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include "pavl.h"
+
+/* Creates and returns a new table
+ with comparison function |compare| using parameter |param|
+ and memory allocator |allocator|.
+ Returns |NULL| if memory allocation failed. */
+struct pavl_table *pavl_create(pavl_comparison_func * compare,
+ struct libavl_allocator *allocator)
+{
+ struct pavl_table *tree;
+
+ assert(compare != NULL);
+
+ if (allocator == NULL)
+ allocator = &pavl_allocator_default;
+
+ tree = allocator->libavl_malloc(sizeof *tree);
+ if (tree == NULL)
+ return NULL;
+
+ tree->pavl_root = NULL;
+ tree->pavl_compare = compare;
+ tree->pavl_alloc = allocator;
+ tree->pavl_count = 0;
+
+ return tree;
+}
+
+/* Search |tree| for an item matching |item|, and return it if found.
+ Otherwise return |NULL|. */
+void *pavl_find(const struct pavl_table *tree, const void *item)
+{
+ const struct pavl_node *p;
+
+ assert(tree != NULL && item != NULL);
+
+ p = tree->pavl_root;
+ while (p != NULL) {
+ int cmp = tree->pavl_compare(item, p->pavl_data);
+
+ if (cmp == 0)
+ return p->pavl_data;
+
+ p = p->pavl_link[cmp > 0];
+ }
+
+ return NULL;
+}
+
+/* Inserts |item| into |tree| and returns a pointer to |item|'s address.
+ If a duplicate item is found in the tree,
+ returns a pointer to the duplicate without inserting |item|.
+ Returns |NULL| in case of memory allocation failure. */
+void **pavl_probe(struct pavl_table *tree, void *item)
+{
+ struct pavl_node *y; /* Top node to update balance factor, and parent. */
+ struct pavl_node *p, *q; /* Iterator, and parent. */
+ struct pavl_node *n; /* Newly inserted node. */
+ struct pavl_node *w; /* New root of rebalanced subtree. */
+ int dir; /* Direction to descend. */
+
+ assert(tree != NULL && item != NULL);
+
+ y = p = tree->pavl_root;
+ q = NULL;
+ dir = 0;
+ while (p != NULL) {
+ int cmp = tree->pavl_compare(item, p->pavl_data);
+
+ if (cmp == 0)
+ return &p->pavl_data;
+
+ dir = cmp > 0;
+
+ if (p->pavl_balance != 0)
+ y = p;
+
+ q = p, p = p->pavl_link[dir];
+ }
+
+ n = tree->pavl_alloc->libavl_malloc(sizeof *p);
+ if (n == NULL)
+ return NULL;
+
+ tree->pavl_count++;
+ n->pavl_link[0] = n->pavl_link[1] = NULL;
+ n->pavl_parent = q;
+ n->pavl_data = item;
+ n->pavl_balance = 0;
+ if (q == NULL) {
+ tree->pavl_root = n;
+
+ return &n->pavl_data;
+ }
+ q->pavl_link[dir] = n;
+
+ p = n;
+ while (p != y) {
+ q = p->pavl_parent;
+ /*
+ dir = q->pavl_link[0] != p;
+ if (dir == 0)
+ q->pavl_balance--;
+ else
+ q->pavl_balance++;
+ */
+ if (q->pavl_link[0] != p)
+ q->pavl_balance++;
+ else
+ q->pavl_balance--;
+
+ p = q;
+ }
+
+ if (y->pavl_balance == -2) {
+ struct pavl_node *x = y->pavl_link[0];
+
+ if (x->pavl_balance == -1) {
+ w = x;
+ y->pavl_link[0] = x->pavl_link[1];
+ x->pavl_link[1] = y;
+ x->pavl_balance = y->pavl_balance = 0;
+ x->pavl_parent = y->pavl_parent;
+ y->pavl_parent = x;
+ if (y->pavl_link[0] != NULL)
+ y->pavl_link[0]->pavl_parent = y;
+ }
+ else {
+ assert(x->pavl_balance == +1);
+ w = x->pavl_link[1];
+ x->pavl_link[1] = w->pavl_link[0];
+ w->pavl_link[0] = x;
+ y->pavl_link[0] = w->pavl_link[1];
+ w->pavl_link[1] = y;
+ if (w->pavl_balance == -1)
+ x->pavl_balance = 0, y->pavl_balance = +1;
+ else if (w->pavl_balance == 0)
+ x->pavl_balance = y->pavl_balance = 0;
+ else /* |w->pavl_balance == +1| */
+ x->pavl_balance = -1, y->pavl_balance = 0;
+ w->pavl_balance = 0;
+ w->pavl_parent = y->pavl_parent;
+ x->pavl_parent = y->pavl_parent = w;
+ if (x->pavl_link[1] != NULL)
+ x->pavl_link[1]->pavl_parent = x;
+ if (y->pavl_link[0] != NULL)
+ y->pavl_link[0]->pavl_parent = y;
+ }
+ }
+ else if (y->pavl_balance == +2) {
+ struct pavl_node *x = y->pavl_link[1];
+
+ if (x->pavl_balance == +1) {
+ w = x;
+ y->pavl_link[1] = x->pavl_link[0];
+ x->pavl_link[0] = y;
+ x->pavl_balance = y->pavl_balance = 0;
+ x->pavl_parent = y->pavl_parent;
+ y->pavl_parent = x;
+ if (y->pavl_link[1] != NULL)
+ y->pavl_link[1]->pavl_parent = y;
+ }
+ else {
+ assert(x->pavl_balance == -1);
+ w = x->pavl_link[0];
+ x->pavl_link[0] = w->pavl_link[1];
+ w->pavl_link[1] = x;
+ y->pavl_link[1] = w->pavl_link[0];
+ w->pavl_link[0] = y;
+ if (w->pavl_balance == +1)
+ x->pavl_balance = 0, y->pavl_balance = -1;
+ else if (w->pavl_balance == 0)
+ x->pavl_balance = y->pavl_balance = 0;
+ else /* |w->pavl_balance == -1| */
+ x->pavl_balance = +1, y->pavl_balance = 0;
+ w->pavl_balance = 0;
+ w->pavl_parent = y->pavl_parent;
+ x->pavl_parent = y->pavl_parent = w;
+ if (x->pavl_link[0] != NULL)
+ x->pavl_link[0]->pavl_parent = x;
+ if (y->pavl_link[1] != NULL)
+ y->pavl_link[1]->pavl_parent = y;
+ }
+ }
+ else
+ return &n->pavl_data;
+
+ if (w->pavl_parent != NULL)
+ w->pavl_parent->pavl_link[y != w->pavl_parent->pavl_link[0]] = w;
+ else
+ tree->pavl_root = w;
+
+ return &n->pavl_data;
+}
+
+/* Inserts |item| into |table|.
+ Returns |NULL| if |item| was successfully inserted
+ or if a memory allocation error occurred.
+ Otherwise, returns the duplicate item. */
+void *pavl_insert(struct pavl_table *table, void *item)
+{
+ void **p = pavl_probe(table, item);
+
+ return p == NULL || *p == item ? NULL : *p;
+}
+
+/* Inserts |item| into |table|, replacing any duplicate item.
+ Returns |NULL| if |item| was inserted without replacing a duplicate,
+ or if a memory allocation error occurred.
+ Otherwise, returns the item that was replaced. */
+void *pavl_replace(struct pavl_table *table, void *item)
+{
+ void **p = pavl_probe(table, item);
+
+ if (p == NULL || *p == item)
+ return NULL;
+ else {
+ void *r = *p;
+
+ *p = item;
+
+ return r;
+ }
+}
+
+/* Deletes from |tree| and returns an item matching |item|.
+ Returns a null pointer if no matching item found. */
+void *pavl_delete(struct pavl_table *tree, const void *item)
+{
+ struct pavl_node *p; /* Traverses tree to find node to delete. */
+ struct pavl_node *q; /* Parent of |p|. */
+ int dir; /* Side of |q| on which |p| is linked. */
+ int cmp; /* Result of comparison between |item| and |p|. */
+
+ assert(tree != NULL && item != NULL);
+
+ p = tree->pavl_root;
+ dir = 0;
+ while (p != NULL) {
+ cmp = tree->pavl_compare(item, p->pavl_data);
+
+ if (cmp == 0)
+ break;
+
+ dir = cmp > 0;
+ p = p->pavl_link[dir];
+ }
+ if (p == NULL)
+ return NULL;
+
+ item = p->pavl_data;
+
+ q = p->pavl_parent;
+ if (q == NULL) {
+ q = (struct pavl_node *)&tree->pavl_root;
+ dir = 0;
+ }
+
+ if (p->pavl_link[1] == NULL) {
+ q->pavl_link[dir] = p->pavl_link[0];
+ if (q->pavl_link[dir] != NULL)
+ q->pavl_link[dir]->pavl_parent = p->pavl_parent;
+ }
+ else {
+ struct pavl_node *r = p->pavl_link[1];
+
+ if (r->pavl_link[0] == NULL) {
+ r->pavl_link[0] = p->pavl_link[0];
+ q->pavl_link[dir] = r;
+ r->pavl_parent = p->pavl_parent;
+ if (r->pavl_link[0] != NULL)
+ r->pavl_link[0]->pavl_parent = r;
+ r->pavl_balance = p->pavl_balance;
+ q = r;
+ dir = 1;
+ }
+ else {
+ struct pavl_node *s = r->pavl_link[0];
+
+ while (s->pavl_link[0] != NULL)
+ s = s->pavl_link[0];
+ r = s->pavl_parent;
+ r->pavl_link[0] = s->pavl_link[1];
+ s->pavl_link[0] = p->pavl_link[0];
+ s->pavl_link[1] = p->pavl_link[1];
+ q->pavl_link[dir] = s;
+ if (s->pavl_link[0] != NULL)
+ s->pavl_link[0]->pavl_parent = s;
+ s->pavl_link[1]->pavl_parent = s;
+ s->pavl_parent = p->pavl_parent;
+ if (r->pavl_link[0] != NULL)
+ r->pavl_link[0]->pavl_parent = r;
+ s->pavl_balance = p->pavl_balance;
+ q = r;
+ dir = 0;
+ }
+ }
+ tree->pavl_alloc->libavl_free(p);
+
+ while (q != (struct pavl_node *)&tree->pavl_root) {
+ struct pavl_node *y = q;
+
+ if (y->pavl_parent != NULL)
+ q = y->pavl_parent;
+ else
+ q = (struct pavl_node *)&tree->pavl_root;
+
+ if (dir == 0) {
+ dir = q->pavl_link[0] != y;
+ y->pavl_balance++;
+ if (y->pavl_balance == +1)
+ break;
+ else if (y->pavl_balance == +2) {
+ struct pavl_node *x = y->pavl_link[1];
+
+ if (x->pavl_balance == -1) {
+ struct pavl_node *w;
+
+ assert(x->pavl_balance == -1);
+ w = x->pavl_link[0];
+ x->pavl_link[0] = w->pavl_link[1];
+ w->pavl_link[1] = x;
+ y->pavl_link[1] = w->pavl_link[0];
+ w->pavl_link[0] = y;
+ if (w->pavl_balance == +1)
+ x->pavl_balance = 0, y->pavl_balance = -1;
+ else if (w->pavl_balance == 0)
+ x->pavl_balance = y->pavl_balance = 0;
+ else /* |w->pavl_balance == -1| */
+ x->pavl_balance = +1, y->pavl_balance = 0;
+ w->pavl_balance = 0;
+ w->pavl_parent = y->pavl_parent;
+ x->pavl_parent = y->pavl_parent = w;
+ if (x->pavl_link[0] != NULL)
+ x->pavl_link[0]->pavl_parent = x;
+ if (y->pavl_link[1] != NULL)
+ y->pavl_link[1]->pavl_parent = y;
+ q->pavl_link[dir] = w;
+ }
+ else {
+ y->pavl_link[1] = x->pavl_link[0];
+ x->pavl_link[0] = y;
+ x->pavl_parent = y->pavl_parent;
+ y->pavl_parent = x;
+ if (y->pavl_link[1] != NULL)
+ y->pavl_link[1]->pavl_parent = y;
+ q->pavl_link[dir] = x;
+ if (x->pavl_balance == 0) {
+ x->pavl_balance = -1;
+ y->pavl_balance = +1;
+ break;
+ }
+ else {
+ x->pavl_balance = y->pavl_balance = 0;
+ y = x;
+ }
+ }
+ }
+ }
+ else {
+ dir = q->pavl_link[0] != y;
+ y->pavl_balance--;
+ if (y->pavl_balance == -1)
+ break;
+ else if (y->pavl_balance == -2) {
+ struct pavl_node *x = y->pavl_link[0];
+
+ if (x->pavl_balance == +1) {
+ struct pavl_node *w;
+
+ assert(x->pavl_balance == +1);
+ w = x->pavl_link[1];
+ x->pavl_link[1] = w->pavl_link[0];
+ w->pavl_link[0] = x;
+ y->pavl_link[0] = w->pavl_link[1];
+ w->pavl_link[1] = y;
+ if (w->pavl_balance == -1)
+ x->pavl_balance = 0, y->pavl_balance = +1;
+ else if (w->pavl_balance == 0)
+ x->pavl_balance = y->pavl_balance = 0;
+ else /* |w->pavl_balance == +1| */
+ x->pavl_balance = -1, y->pavl_balance = 0;
+ w->pavl_balance = 0;
+ w->pavl_parent = y->pavl_parent;
+ x->pavl_parent = y->pavl_parent = w;
+ if (x->pavl_link[1] != NULL)
+ x->pavl_link[1]->pavl_parent = x;
+ if (y->pavl_link[0] != NULL)
+ y->pavl_link[0]->pavl_parent = y;
+ q->pavl_link[dir] = w;
+ }
+ else {
+ y->pavl_link[0] = x->pavl_link[1];
+ x->pavl_link[1] = y;
+ x->pavl_parent = y->pavl_parent;
+ y->pavl_parent = x;
+ if (y->pavl_link[0] != NULL)
+ y->pavl_link[0]->pavl_parent = y;
+ q->pavl_link[dir] = x;
+ if (x->pavl_balance == 0) {
+ x->pavl_balance = +1;
+ y->pavl_balance = -1;
+ break;
+ }
+ else {
+ x->pavl_balance = y->pavl_balance = 0;
+ y = x;
+ }
+ }
+ }
+ }
+ }
+
+ tree->pavl_count--;
+ return (void *)item;
+}
+
+/* Initializes |trav| for use with |tree|
+ and selects the null node. */
+void pavl_t_init(struct pavl_traverser *trav, struct pavl_table *tree)
+{
+ trav->pavl_table = tree;
+ trav->pavl_node = NULL;
+}
+
+/* Initializes |trav| for |tree|.
+ Returns data item in |tree| with the least value,
+ or |NULL| if |tree| is empty. */
+void *pavl_t_first(struct pavl_traverser *trav, struct pavl_table *tree)
+{
+ assert(tree != NULL && trav != NULL);
+
+ trav->pavl_table = tree;
+ trav->pavl_node = tree->pavl_root;
+ if (trav->pavl_node != NULL) {
+ while (trav->pavl_node->pavl_link[0] != NULL)
+ trav->pavl_node = trav->pavl_node->pavl_link[0];
+
+ return trav->pavl_node->pavl_data;
+ }
+ else
+ return NULL;
+}
+
+/* Initializes |trav| for |tree|.
+ Returns data item in |tree| with the greatest value,
+ or |NULL| if |tree| is empty. */
+void *pavl_t_last(struct pavl_traverser *trav, struct pavl_table *tree)
+{
+ assert(tree != NULL && trav != NULL);
+
+ trav->pavl_table = tree;
+ trav->pavl_node = tree->pavl_root;
+ if (trav->pavl_node != NULL) {
+ while (trav->pavl_node->pavl_link[1] != NULL)
+ trav->pavl_node = trav->pavl_node->pavl_link[1];
+
+ return trav->pavl_node->pavl_data;
+ }
+ else
+ return NULL;
+}
+
+/* Searches for |item| in |tree|.
+ If found, initializes |trav| to the item found and returns the item
+ as well.
+ If there is no matching item, initializes |trav| to the null item
+ and returns |NULL|. */
+void *pavl_t_find(struct pavl_traverser *trav, struct pavl_table *tree,
+ void *item)
+{
+ struct pavl_node *p;
+
+ assert(trav != NULL && tree != NULL && item != NULL);
+
+ trav->pavl_table = tree;
+
+ p = tree->pavl_root;
+ while (p != NULL) {
+ int cmp = tree->pavl_compare(item, p->pavl_data);
+
+ if (cmp == 0) {
+ trav->pavl_node = p;
+
+ return p->pavl_data;
+ }
+
+ p = p->pavl_link[cmp > 0];
+ }
+
+ trav->pavl_node = NULL;
+
+ return NULL;
+}
+
+/* Attempts to insert |item| into |tree|.
+ If |item| is inserted successfully, it is returned and |trav| is
+ initialized to its location.
+ If a duplicate is found, it is returned and |trav| is initialized to
+ its location. No replacement of the item occurs.
+ If a memory allocation failure occurs, |NULL| is returned and |trav|
+ is initialized to the null item. */
+void *pavl_t_insert(struct pavl_traverser *trav,
+ struct pavl_table *tree, void *item)
+{
+ void **p;
+
+ assert(trav != NULL && tree != NULL && item != NULL);
+
+ p = pavl_probe(tree, item);
+ if (p != NULL) {
+ trav->pavl_table = tree;
+ trav->pavl_node = ((struct pavl_node *)((char *)p -
+ offsetof(struct pavl_node,
+ pavl_data)));
+
+ return *p;
+ }
+ else {
+ pavl_t_init(trav, tree);
+
+ return NULL;
+ }
+}
+
+/* Initializes |trav| to have the same current node as |src|. */
+void *pavl_t_copy(struct pavl_traverser *trav,
+ const struct pavl_traverser *src)
+{
+ assert(trav != NULL && src != NULL);
+
+ trav->pavl_table = src->pavl_table;
+ trav->pavl_node = src->pavl_node;
+
+ return trav->pavl_node != NULL ? trav->pavl_node->pavl_data : NULL;
+}
+
+/* Returns the next data item in inorder
+ within the tree being traversed with |trav|,
+ or if there are no more data items returns |NULL|. */
+void *pavl_t_next(struct pavl_traverser *trav)
+{
+ assert(trav != NULL);
+
+ if (trav->pavl_node == NULL)
+ return pavl_t_first(trav, trav->pavl_table);
+ else if (trav->pavl_node->pavl_link[1] == NULL) {
+ struct pavl_node *q, *p; /* Current node and its child. */
+
+ for (p = trav->pavl_node, q = p->pavl_parent;;
+ p = q, q = q->pavl_parent)
+ if (q == NULL || p == q->pavl_link[0]) {
+ trav->pavl_node = q;
+
+ return trav->pavl_node != NULL ?
+ trav->pavl_node->pavl_data : NULL;
+ }
+ }
+ else {
+ trav->pavl_node = trav->pavl_node->pavl_link[1];
+ while (trav->pavl_node->pavl_link[0] != NULL)
+ trav->pavl_node = trav->pavl_node->pavl_link[0];
+
+ return trav->pavl_node->pavl_data;
+ }
+}
+
+/* Returns the previous data item in inorder
+ within the tree being traversed with |trav|,
+ or if there are no more data items returns |NULL|. */
+void *pavl_t_prev(struct pavl_traverser *trav)
+{
+ assert(trav != NULL);
+
+ if (trav->pavl_node == NULL)
+ return pavl_t_last(trav, trav->pavl_table);
+ else if (trav->pavl_node->pavl_link[0] == NULL) {
+ struct pavl_node *q, *p; /* Current node and its child. */
+
+ for (p = trav->pavl_node, q = p->pavl_parent;;
+ p = q, q = q->pavl_parent)
+ if (q == NULL || p == q->pavl_link[1]) {
+ trav->pavl_node = q;
+
+ return trav->pavl_node != NULL ?
+ trav->pavl_node->pavl_data : NULL;
+ }
+ }
+ else {
+ trav->pavl_node = trav->pavl_node->pavl_link[0];
+ while (trav->pavl_node->pavl_link[1] != NULL)
+ trav->pavl_node = trav->pavl_node->pavl_link[1];
+
+ return trav->pavl_node->pavl_data;
+ }
+}
+
+/* Returns |trav|'s current item. */
+void *pavl_t_cur(struct pavl_traverser *trav)
+{
+ assert(trav != NULL);
+
+ return trav->pavl_node != NULL ? trav->pavl_node->pavl_data : NULL;
+}
+
+/* Replaces the current item in |trav| by |new| and returns the item replaced.
+ |trav| must not have the null item selected.
+ The new item must not upset the ordering of the tree. */
+void *pavl_t_replace(struct pavl_traverser *trav, void *new)
+{
+ void *old;
+
+ assert(trav != NULL && trav->pavl_node != NULL && new != NULL);
+ old = trav->pavl_node->pavl_data;
+ trav->pavl_node->pavl_data = new;
+
+ return old;
+}
+
+/* Destroys |new| with |pavl_destroy (new, destroy)|,
+ first initializing right links in |new| that have
+ not yet been initialized at time of call. */
+static void
+copy_error_recovery(struct pavl_node *q,
+ struct pavl_table *new, pavl_item_func * destroy)
+{
+ assert(q != NULL && new != NULL);
+
+ for (;;) {
+ struct pavl_node *p = q;
+
+ q = q->pavl_parent;
+ if (q == NULL)
+ break;
+
+ if (p == q->pavl_link[0])
+ q->pavl_link[1] = NULL;
+ }
+
+ pavl_destroy(new, destroy);
+}
+
+/* Copies |org| to a newly created tree, which is returned.
+ If |copy != NULL|, each data item in |org| is first passed to |copy|,
+ and the return values are inserted into the tree;
+ |NULL| return values are taken as indications of failure.
+ On failure, destroys the partially created new tree,
+ applying |destroy|, if non-null, to each item in the new tree so far,
+ and returns |NULL|.
+ If |allocator != NULL|, it is used for allocation in the new tree.
+ Otherwise, the same allocator used for |org| is used. */
+struct pavl_table *pavl_copy(const struct pavl_table *org,
+ pavl_copy_func * copy, pavl_item_func * destroy,
+ struct libavl_allocator *allocator)
+{
+ struct pavl_table *new;
+ const struct pavl_node *x;
+ struct pavl_node *y;
+
+ assert(org != NULL);
+ new = pavl_create(org->pavl_compare,
+ allocator != NULL ? allocator : org->pavl_alloc);
+ if (new == NULL)
+ return NULL;
+
+ new->pavl_count = org->pavl_count;
+ if (new->pavl_count == 0)
+ return new;
+
+ x = (const struct pavl_node *)&org->pavl_root;
+ y = (struct pavl_node *)&new->pavl_root;
+ while (x != NULL) {
+ while (x->pavl_link[0] != NULL) {
+ y->pavl_link[0] =
+ new->pavl_alloc->libavl_malloc(sizeof *y->pavl_link[0]);
+ if (y->pavl_link[0] == NULL) {
+ if (y != (struct pavl_node *)&new->pavl_root) {
+ y->pavl_data = NULL;
+ y->pavl_link[1] = NULL;
+ }
+
+ copy_error_recovery(y, new, destroy);
+
+ return NULL;
+ }
+ y->pavl_link[0]->pavl_parent = y;
+
+ x = x->pavl_link[0];
+ y = y->pavl_link[0];
+ }
+ y->pavl_link[0] = NULL;
+
+ for (;;) {
+ y->pavl_balance = x->pavl_balance;
+ if (copy == NULL)
+ y->pavl_data = x->pavl_data;
+ else {
+ y->pavl_data = copy(x->pavl_data);
+ if (y->pavl_data == NULL) {
+ y->pavl_link[1] = NULL;
+ copy_error_recovery(y, new, destroy);
+
+ return NULL;
+ }
+ }
+
+ if (x->pavl_link[1] != NULL) {
+ y->pavl_link[1] =
+ new->pavl_alloc->libavl_malloc(sizeof *y->pavl_link[1]);
+ if (y->pavl_link[1] == NULL) {
+ copy_error_recovery(y, new, destroy);
+
+ return NULL;
+ }
+ y->pavl_link[1]->pavl_parent = y;
+
+ x = x->pavl_link[1];
+ y = y->pavl_link[1];
+ break;
+ }
+ else
+ y->pavl_link[1] = NULL;
+
+ for (;;) {
+ const struct pavl_node *w = x;
+
+ x = x->pavl_parent;
+ if (x == NULL) {
+ new->pavl_root->pavl_parent = NULL;
+ return new;
+ }
+ y = y->pavl_parent;
+
+ if (w == x->pavl_link[0])
+ break;
+ }
+ }
+ }
+
+ return new;
+}
+
+/* Frees storage allocated for |tree|.
+ If |destroy != NULL|, applies it to each data item in inorder. */
+void pavl_destroy(struct pavl_table *tree, pavl_item_func * destroy)
+{
+ struct pavl_node *p, *q;
+
+ assert(tree != NULL);
+
+ p = tree->pavl_root;
+ while (p != NULL) {
+ if (p->pavl_link[0] == NULL) {
+ q = p->pavl_link[1];
+ if (destroy != NULL && p->pavl_data != NULL)
+ destroy(p->pavl_data);
+ tree->pavl_alloc->libavl_free(p);
+ }
+ else {
+ q = p->pavl_link[0];
+ p->pavl_link[0] = q->pavl_link[1];
+ q->pavl_link[1] = p;
+ }
+ p = q;
+ }
+
+ tree->pavl_alloc->libavl_free(tree);
+}
+
+/* Allocates |size| bytes of space using |malloc()|.
+ Returns a null pointer if allocation fails. */
+void *pavl_malloc(size_t size)
+{
+ if (size > 0)
+ return malloc(size);
+
+ return NULL;
+}
+
+/* Frees |block|. */
+void pavl_free(void *block)
+{
+ if (block)
+ free(block);
+}
+
+/* Default memory allocator that uses |malloc()| and |free()|. */
+struct libavl_allocator pavl_allocator_default = {
+ pavl_malloc,
+ pavl_free
+};
+
+#undef NDEBUG
+#include <assert.h>
+
+/* Asserts that |pavl_insert()| succeeds at inserting |item| into |table|. */
+void (pavl_assert_insert) (struct pavl_table * table, void *item)
+{
+ void **p = pavl_probe(table, item);
+
+ assert(p != NULL && *p == item);
+}
+
+/* Asserts that |pavl_delete()| really removes |item| from |table|,
+ and returns the removed item. */
+void *(pavl_assert_delete) (struct pavl_table * table, void *item)
+{
+ void *p = pavl_delete(table, item);
+
+ assert(p != NULL);
+
+ return p;
+}
Added: grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.h
===================================================================
--- grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.h (rev 0)
+++ grass-addons/grass7/raster/r.neighborhoodmatrix/pavl.h 2018-09-11 12:26:55 UTC (rev 73315)
@@ -0,0 +1,106 @@
+/* Produced by texiweb from libavl.w. */
+
+/* libavl - library for manipulation of binary trees.
+ Copyright (C) 1998, 1999, 2000, 2001, 2002, 2004 Free Software
+ Foundation, Inc.
+
+ This library is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Lesser General Public
+ License as published by the Free Software Foundation; either
+ version 3 of the License, or (at your option) any later version.
+
+ This library is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public
+ License along with this library; if not, write to the Free Software
+ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ 02110-1301 USA.
+ */
+
+#ifndef PAVL_H
+#define PAVL_H 1
+
+#include <stddef.h>
+
+/* Function types. */
+typedef int pavl_comparison_func(const void *pavl_a, const void *pavl_b);
+typedef void pavl_item_func(void *pavl_item);
+typedef void *pavl_copy_func(void *pavl_item);
+
+#ifndef LIBAVL_ALLOCATOR
+#define LIBAVL_ALLOCATOR
+/* Memory allocator. */
+struct libavl_allocator
+{
+ void *(*libavl_malloc) (size_t libavl_size);
+ void (*libavl_free) (void *libavl_block);
+};
+#endif
+
+/* Default memory allocator. */
+extern struct libavl_allocator pavl_allocator_default;
+void *pavl_malloc(size_t);
+void pavl_free(void *);
+
+/* Maximum PAVL height, unused. */
+#ifndef PAVL_MAX_HEIGHT
+#define PAVL_MAX_HEIGHT 32
+#endif
+
+/* Tree data structure. */
+struct pavl_table
+{
+ struct pavl_node *pavl_root; /* Tree's root. */
+ pavl_comparison_func *pavl_compare; /* Comparison function. */
+ struct libavl_allocator *pavl_alloc; /* Memory allocator. */
+ size_t pavl_count; /* Number of items in tree. */
+};
+
+/* An PAVL tree node. */
+struct pavl_node
+{
+ struct pavl_node *pavl_link[2]; /* Subtrees. */
+ struct pavl_node *pavl_parent; /* Parent node. */
+ void *pavl_data; /* Pointer to data. */
+ signed char pavl_balance; /* Balance factor. */
+};
+
+/* PAVL traverser structure. */
+struct pavl_traverser
+{
+ struct pavl_table *pavl_table; /* Tree being traversed. */
+ struct pavl_node *pavl_node; /* Current node in tree. */
+};
+
+/* Table functions. */
+struct pavl_table *pavl_create(pavl_comparison_func *,
+ struct libavl_allocator *);
+struct pavl_table *pavl_copy(const struct pavl_table *, pavl_copy_func *,
+ pavl_item_func *, struct libavl_allocator *);
+void pavl_destroy(struct pavl_table *, pavl_item_func *);
+void **pavl_probe(struct pavl_table *, void *);
+void *pavl_insert(struct pavl_table *, void *);
+void *pavl_replace(struct pavl_table *, void *);
+void *pavl_delete(struct pavl_table *, const void *);
+void *pavl_find(const struct pavl_table *, const void *);
+void pavl_assert_insert(struct pavl_table *, void *);
+void *pavl_assert_delete(struct pavl_table *, void *);
+
+#define pavl_count(table) ((size_t) (table)->pavl_count)
+
+/* Table traverser functions. */
+void pavl_t_init(struct pavl_traverser *, struct pavl_table *);
+void *pavl_t_first(struct pavl_traverser *, struct pavl_table *);
+void *pavl_t_last(struct pavl_traverser *, struct pavl_table *);
+void *pavl_t_find(struct pavl_traverser *, struct pavl_table *, void *);
+void *pavl_t_insert(struct pavl_traverser *, struct pavl_table *, void *);
+void *pavl_t_copy(struct pavl_traverser *, const struct pavl_traverser *);
+void *pavl_t_next(struct pavl_traverser *);
+void *pavl_t_prev(struct pavl_traverser *);
+void *pavl_t_cur(struct pavl_traverser *);
+void *pavl_t_replace(struct pavl_traverser *, void *);
+
+#endif /* pavl.h */
More information about the grass-commit
mailing list