Fix threshold calculation
[openjpeg.git] / src / lib / openjp3d / tgt.c
index c86cfe6dc5f21ca73934506baa055fa33a380561..5993cbd53a7cd3e6d19b9019884414f2b6658e01 100644 (file)
-/*\r
- * Copyright (c) 2001-2003, David Janssens\r
- * Copyright (c) 2002-2003, Yannick Verschueren\r
- * Copyright (c) 2003-2005, Francois Devaux and Antonin Descampe\r
- * Copyright (c) 2005, Herve Drolon, FreeImage Team\r
- * Copyright (c) 2002-2005, Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium\r
- * All rights reserved.\r
- *\r
- * Redistribution and use in source and binary forms, with or without\r
- * modification, are permitted provided that the following conditions\r
- * are met:\r
- * 1. Redistributions of source code must retain the above copyright\r
- *    notice, this list of conditions and the following disclaimer.\r
- * 2. Redistributions in binary form must reproduce the above copyright\r
- *    notice, this list of conditions and the following disclaimer in the\r
- *    documentation and/or other materials provided with the distribution.\r
- *\r
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'\r
- * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE\r
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE\r
- * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE\r
- * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR\r
- * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF\r
- * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS\r
- * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN\r
- * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)\r
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE\r
- * POSSIBILITY OF SUCH DAMAGE.\r
- */\r
-\r
-#include "opj_includes.h"\r
-\r
-/* \r
-==========================================================\r
-   Tag-tree coder interface\r
-==========================================================\r
-*/\r
-void tgt_tree_dump (FILE *fd, opj_tgt_tree_t * tree){\r
-       int nodesno;\r
-\r
-       fprintf(fd, "TGT_TREE {\n");\r
-       fprintf(fd, "  numnodes: %d \n", tree->numnodes);       \r
-       fprintf(fd, "  numleafsh: %d, numleafsv: %d, numleafsz: %d,\n", tree->numleafsh, tree->numleafsv, tree->numleafsz);\r
-\r
-       for (nodesno = 0; nodesno < tree->numnodes; nodesno++) {\r
-               fprintf(fd, "tgt_node %d {\n", nodesno);\r
-               fprintf(fd, "  value: %d \n", tree->nodes[nodesno].value);\r
-               fprintf(fd, "  low: %d \n", tree->nodes[nodesno].low);\r
-               fprintf(fd, "  known: %d \n", tree->nodes[nodesno].known);\r
-               if (tree->nodes[nodesno].parent) {\r
-                       fprintf(fd, "  parent.value: %d \n", tree->nodes[nodesno].parent->value);\r
-                       fprintf(fd, "  parent.low: %d \n", tree->nodes[nodesno].parent->low);\r
-                       fprintf(fd, "  parent.known: %d \n", tree->nodes[nodesno].parent->known);\r
-               }\r
-               fprintf(fd, "}\n");\r
-\r
-       }\r
-       fprintf(fd, "}\n");\r
-\r
-}\r
-\r
-\r
-opj_tgt_tree_t *tgt_create(int numleafsh, int numleafsv, int numleafsz) {\r
-       \r
-       int nplh[32];\r
-       int nplv[32];\r
-       int nplz[32];\r
-       opj_tgt_node_t *node = NULL;\r
-       opj_tgt_node_t *parentnode = NULL;\r
-       opj_tgt_node_t *parentnode0 = NULL;\r
-       opj_tgt_node_t *parentnode1 = NULL;\r
-       opj_tgt_tree_t *tree = NULL;\r
-       int i, j, k, p, p0;\r
-       int numlvls;\r
-       int n, z = 0;\r
-\r
-       tree = (opj_tgt_tree_t *) opj_malloc(sizeof(opj_tgt_tree_t));\r
-       if(!tree) \r
-               return NULL;\r
-       tree->numleafsh = numleafsh;\r
-       tree->numleafsv = numleafsv;\r
-       tree->numleafsz = numleafsz;\r
-\r
-       numlvls = 0;\r
-       nplh[0] = numleafsh;\r
-       nplv[0] = numleafsv;\r
-       nplz[0] = numleafsz;\r
-       tree->numnodes = 0;\r
-       do {\r
-               n = nplh[numlvls] * nplv[numlvls] * nplz[numlvls]; \r
-               nplh[numlvls + 1] = (nplh[numlvls] + 1) / 2;\r
-               nplv[numlvls + 1] = (nplv[numlvls] + 1) / 2;\r
-               nplz[numlvls + 1] = (nplz[numlvls] + 1) / 2;\r
-               tree->numnodes += n;\r
-               ++numlvls;\r
-       } while (n > 1);\r
-\r
-       if (tree->numnodes == 0) {\r
-               opj_free(tree);\r
-               return NULL;\r
-       }\r
-\r
-       tree->nodes = (opj_tgt_node_t *) opj_malloc(tree->numnodes * sizeof(opj_tgt_node_t));\r
-       if(!tree->nodes) {\r
-               opj_free(tree);\r
-               return NULL;\r
-       }\r
-\r
-       node = tree->nodes;\r
-       parentnode = &tree->nodes[tree->numleafsh * tree->numleafsv * tree->numleafsz];\r
-       parentnode0 = parentnode;\r
-       parentnode1 = parentnode;\r
-       /*fprintf(stdout,"\nH %d V %d Z %d numlvls %d nodes %d\n",tree->numleafsh,tree->numleafsv,tree->numleafsz,numlvls,tree->numnodes);*/\r
-       for (i = 0; i < numlvls - 1; ++i) {\r
-               for (z = 0; z < nplz[i]; ++z) {\r
-                       for (j = 0; j < nplv[i]; ++j) {\r
-                               k = nplh[i];\r
-                               while(--k >= 0) {\r
-                                       node->parent = parentnode;              /*fprintf(stdout,"node[%d].parent = node[%d]\n",n,p);*/\r
-                                       ++node;\r
-                                       if(--k >= 0) {\r
-                                               node->parent = parentnode;      /*fprintf(stdout,"node[%d].parent = node[%d]\n",n,p);*/\r
-                                               ++node;\r
-                                       }\r
-                                       ++parentnode;\r
-                               }\r
-                               if((j & 1) || j == nplv[i] - 1) {\r
-                                       parentnode0 = parentnode;\r
-                               } else {\r
-                                       parentnode = parentnode0;\r
-                               }\r
-                       }\r
-                       if ((z & 1) || z == nplz[i] - 1) {\r
-                               parentnode1 = parentnode;\r
-                       } else {\r
-                               parentnode0 = parentnode1;\r
-                               parentnode = parentnode1;\r
-                       }\r
-               }\r
-       }\r
-       node->parent = 0;\r
-\r
-       \r
-       tgt_reset(tree);\r
-\r
-       return tree;\r
-}\r
-\r
-void tgt_destroy(opj_tgt_tree_t *tree) {\r
-       opj_free(tree->nodes);\r
-       opj_free(tree);\r
-}\r
-\r
-void tgt_reset(opj_tgt_tree_t *tree) {\r
-       int i;\r
-\r
-       if (NULL == tree)\r
-               return;\r
-       \r
-       for (i = 0; i < tree->numnodes; i++) {\r
-               tree->nodes[i].value = 999;\r
-               tree->nodes[i].low = 0;\r
-               tree->nodes[i].known = 0;\r
-       }\r
-}\r
-\r
-void tgt_setvalue(opj_tgt_tree_t *tree, int leafno, int value) {\r
-       opj_tgt_node_t *node;\r
-       node = &tree->nodes[leafno];\r
-       while (node && node->value > value) {\r
-               node->value = value;\r
-               node = node->parent;\r
-       }\r
-}\r
-\r
-void tgt_encode(opj_bio_t *bio, opj_tgt_tree_t *tree, int leafno, int threshold) {\r
-       opj_tgt_node_t *stk[31];\r
-       opj_tgt_node_t **stkptr;\r
-       opj_tgt_node_t *node;\r
-       int low;\r
-\r
-       stkptr = stk;\r
-       node = &tree->nodes[leafno];\r
-       while (node->parent) {\r
-               *stkptr++ = node;\r
-               node = node->parent;\r
-       }\r
-       \r
-       low = 0;\r
-       for (;;) {\r
-               if (low > node->low) {\r
-                       node->low = low;\r
-               } else {\r
-                       low = node->low;\r
-               }\r
-               \r
-               while (low < threshold) {\r
-                       if (low >= node->value) {\r
-                               if (!node->known) {\r
-                                       bio_write(bio, 1, 1);\r
-                                       node->known = 1;\r
-                               }\r
-                               break;\r
-                       }\r
-                       bio_write(bio, 0, 1);\r
-                       ++low;\r
-               }\r
-               \r
-               node->low = low;\r
-               if (stkptr == stk)\r
-                       break;\r
-               node = *--stkptr;\r
-       }\r
-}\r
-\r
-int tgt_decode(opj_bio_t *bio, opj_tgt_tree_t *tree, int leafno, int threshold) {\r
-       opj_tgt_node_t *stk[31];\r
-       opj_tgt_node_t **stkptr;\r
-       opj_tgt_node_t *node;\r
-       int low;\r
-\r
-       stkptr = stk;\r
-       node = &tree->nodes[leafno];\r
-       while (node->parent) {\r
-               *stkptr++ = node;\r
-               node = node->parent;\r
-       }\r
-       \r
-       low = 0;\r
-       for (;;) {\r
-               if (low > node->low) {\r
-                       node->low = low;\r
-               } else {\r
-                       low = node->low;\r
-               }\r
-               while (low < threshold && low < node->value) {\r
-                       if (bio_read(bio, 1)) {\r
-                               node->value = low;\r
-                       } else {\r
-                               ++low;\r
-                       }\r
-               }\r
-               node->low = low;\r
-               if (stkptr == stk) {\r
-                       break;\r
-               }\r
-               node = *--stkptr;\r
-       }\r
-       \r
-       return (node->value < threshold) ? 1 : 0;\r
-}\r
+/*
+ * The copyright in this software is being made available under the 2-clauses 
+ * BSD License, included below. This software may be subject to other third 
+ * party and contributor rights, including patent rights, and no such rights
+ * are granted under this license.
+ *
+ * Copyright (c) 2001-2003, David Janssens
+ * Copyright (c) 2002-2003, Yannick Verschueren
+ * Copyright (c) 2003-2005, Francois Devaux and Antonin Descampe
+ * Copyright (c) 2005, Herve Drolon, FreeImage Team
+ * Copyright (c) 2002-2005, Communications and remote sensing Laboratory, Universite catholique de Louvain, Belgium
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "opj_includes.h"
+
+/* 
+==========================================================
+   Tag-tree coder interface
+==========================================================
+*/
+void tgt_tree_dump (FILE *fd, opj_tgt_tree_t * tree){
+       int nodesno;
+
+       fprintf(fd, "TGT_TREE {\n");
+       fprintf(fd, "  numnodes: %d \n", tree->numnodes);       
+       fprintf(fd, "  numleafsh: %d, numleafsv: %d, numleafsz: %d,\n", tree->numleafsh, tree->numleafsv, tree->numleafsz);
+
+       for (nodesno = 0; nodesno < tree->numnodes; nodesno++) {
+               fprintf(fd, "tgt_node %d {\n", nodesno);
+               fprintf(fd, "  value: %d \n", tree->nodes[nodesno].value);
+               fprintf(fd, "  low: %d \n", tree->nodes[nodesno].low);
+               fprintf(fd, "  known: %d \n", tree->nodes[nodesno].known);
+               if (tree->nodes[nodesno].parent) {
+                       fprintf(fd, "  parent.value: %d \n", tree->nodes[nodesno].parent->value);
+                       fprintf(fd, "  parent.low: %d \n", tree->nodes[nodesno].parent->low);
+                       fprintf(fd, "  parent.known: %d \n", tree->nodes[nodesno].parent->known);
+               }
+               fprintf(fd, "}\n");
+
+       }
+       fprintf(fd, "}\n");
+
+}
+
+
+opj_tgt_tree_t *tgt_create(int numleafsh, int numleafsv, int numleafsz) {
+       
+       int nplh[32];
+       int nplv[32];
+       int nplz[32];
+       opj_tgt_node_t *node = NULL;
+       opj_tgt_node_t *parentnode = NULL;
+       opj_tgt_node_t *parentnode0 = NULL;
+       opj_tgt_node_t *parentnode1 = NULL;
+       opj_tgt_tree_t *tree = NULL;
+       int i, j, k, p, p0;
+       int numlvls;
+       int n, z = 0;
+
+       tree = (opj_tgt_tree_t *) opj_malloc(sizeof(opj_tgt_tree_t));
+       if(!tree) 
+               return NULL;
+       tree->numleafsh = numleafsh;
+       tree->numleafsv = numleafsv;
+       tree->numleafsz = numleafsz;
+
+       numlvls = 0;
+       nplh[0] = numleafsh;
+       nplv[0] = numleafsv;
+       nplz[0] = numleafsz;
+       tree->numnodes = 0;
+       do {
+               n = nplh[numlvls] * nplv[numlvls] * nplz[numlvls]; 
+               nplh[numlvls + 1] = (nplh[numlvls] + 1) / 2;
+               nplv[numlvls + 1] = (nplv[numlvls] + 1) / 2;
+               nplz[numlvls + 1] = (nplz[numlvls] + 1) / 2;
+               tree->numnodes += n;
+               ++numlvls;
+       } while (n > 1);
+
+       if (tree->numnodes == 0) {
+               opj_free(tree);
+               return NULL;
+       }
+
+       tree->nodes = (opj_tgt_node_t *) opj_malloc(tree->numnodes * sizeof(opj_tgt_node_t));
+       if(!tree->nodes) {
+               opj_free(tree);
+               return NULL;
+       }
+
+       node = tree->nodes;
+       parentnode = &tree->nodes[tree->numleafsh * tree->numleafsv * tree->numleafsz];
+       parentnode0 = parentnode;
+       parentnode1 = parentnode;
+       /*fprintf(stdout,"\nH %d V %d Z %d numlvls %d nodes %d\n",tree->numleafsh,tree->numleafsv,tree->numleafsz,numlvls,tree->numnodes);*/
+       for (i = 0; i < numlvls - 1; ++i) {
+               for (z = 0; z < nplz[i]; ++z) {
+                       for (j = 0; j < nplv[i]; ++j) {
+                               k = nplh[i];
+                               while(--k >= 0) {
+                                       node->parent = parentnode;              /*fprintf(stdout,"node[%d].parent = node[%d]\n",n,p);*/
+                                       ++node;
+                                       if(--k >= 0) {
+                                               node->parent = parentnode;      /*fprintf(stdout,"node[%d].parent = node[%d]\n",n,p);*/
+                                               ++node;
+                                       }
+                                       ++parentnode;
+                               }
+                               if((j & 1) || j == nplv[i] - 1) {
+                                       parentnode0 = parentnode;
+                               } else {
+                                       parentnode = parentnode0;
+                               }
+                       }
+                       if ((z & 1) || z == nplz[i] - 1) {
+                               parentnode1 = parentnode;
+                       } else {
+                               parentnode0 = parentnode1;
+                               parentnode = parentnode1;
+                       }
+               }
+       }
+       node->parent = 0;
+
+       
+       tgt_reset(tree);
+
+       return tree;
+}
+
+void tgt_destroy(opj_tgt_tree_t *tree) {
+       opj_free(tree->nodes);
+       opj_free(tree);
+}
+
+void tgt_reset(opj_tgt_tree_t *tree) {
+       int i;
+
+       if (NULL == tree)
+               return;
+       
+       for (i = 0; i < tree->numnodes; i++) {
+               tree->nodes[i].value = 999;
+               tree->nodes[i].low = 0;
+               tree->nodes[i].known = 0;
+       }
+}
+
+void tgt_setvalue(opj_tgt_tree_t *tree, int leafno, int value) {
+       opj_tgt_node_t *node;
+       node = &tree->nodes[leafno];
+       while (node && node->value > value) {
+               node->value = value;
+               node = node->parent;
+       }
+}
+
+void tgt_encode(opj_bio_t *bio, opj_tgt_tree_t *tree, int leafno, int threshold) {
+       opj_tgt_node_t *stk[31];
+       opj_tgt_node_t **stkptr;
+       opj_tgt_node_t *node;
+       int low;
+
+       stkptr = stk;
+       node = &tree->nodes[leafno];
+       while (node->parent) {
+               *stkptr++ = node;
+               node = node->parent;
+       }
+       
+       low = 0;
+       for (;;) {
+               if (low > node->low) {
+                       node->low = low;
+               } else {
+                       low = node->low;
+               }
+               
+               while (low < threshold) {
+                       if (low >= node->value) {
+                               if (!node->known) {
+                                       bio_write(bio, 1, 1);
+                                       node->known = 1;
+                               }
+                               break;
+                       }
+                       bio_write(bio, 0, 1);
+                       ++low;
+               }
+               
+               node->low = low;
+               if (stkptr == stk)
+                       break;
+               node = *--stkptr;
+       }
+}
+
+int tgt_decode(opj_bio_t *bio, opj_tgt_tree_t *tree, int leafno, int threshold) {
+       opj_tgt_node_t *stk[31];
+       opj_tgt_node_t **stkptr;
+       opj_tgt_node_t *node;
+       int low;
+
+       stkptr = stk;
+       node = &tree->nodes[leafno];
+       while (node->parent) {
+               *stkptr++ = node;
+               node = node->parent;
+       }
+       
+       low = 0;
+       for (;;) {
+               if (low > node->low) {
+                       node->low = low;
+               } else {
+                       low = node->low;
+               }
+               while (low < threshold && low < node->value) {
+                       if (bio_read(bio, 1)) {
+                               node->value = low;
+                       } else {
+                               ++low;
+                       }
+               }
+               node->low = low;
+               if (stkptr == stk) {
+                       break;
+               }
+               node = *--stkptr;
+       }
+       
+       return (node->value < threshold) ? 1 : 0;
+}