home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / sa104os2.zip / SATHR104.ZIP / SATHER / SYSTEM / GC / CORD / CORDBSCS.C < prev    next >
Text File  |  1994-10-03  |  27KB  |  916 lines
  1. /*
  2.  * Copyright (c) 1993-1994 by Xerox Corporation.  All rights reserved.
  3.  *
  4.  * THIS MATERIAL IS PROVIDED AS IS, WITH ABSOLUTELY NO WARRANTY EXPRESSED
  5.  * OR IMPLIED.  ANY USE IS AT YOUR OWN RISK.
  6.  *
  7.  * Permission is hereby granted to use or copy this program
  8.  * for any purpose,  provided the above notices are retained on all copies.
  9.  * Permission to modify the code and to distribute modified code is granted,
  10.  * provided the above notices are retained, and a notice that the code was
  11.  * modified is included with the above copyright notice.
  12.  *
  13.  * Author: Hans-J. Boehm (boehm@parc.xerox.com)
  14.  */
  15. /* Boehm, October 3, 1994 5:19 pm PDT */
  16. # include "gc.h"
  17. # include "cord.h"
  18. # include <stdlib.h>
  19. # include <stdio.h>
  20. # include <string.h>
  21.  
  22. /* An implementation of the cord primitives.  These are the only     */
  23. /* Functions that understand the representation.  We perform only    */
  24. /* minimal checks on arguments to these functions.  Out of bounds    */
  25. /* arguments to the iteration functions may result in client functions    */
  26. /* invoked on garbage data.  In most cases, client functions should be    */
  27. /* programmed defensively enough that this does not result in memory    */
  28. /* smashes.                                */ 
  29.  
  30. typedef void (* oom_fn)(void);
  31.  
  32. oom_fn CORD_oom_fn = (oom_fn) 0;
  33.  
  34. # define OUT_OF_MEMORY {  if (CORD_oom_fn != (oom_fn) 0) (*CORD_oom_fn)(); \
  35.               ABORT("Out of memory\n"); }
  36. # define ABORT(msg) { fprintf(stderr, "%s\n", msg); abort(); }
  37.  
  38. typedef unsigned long word;
  39.  
  40. typedef union {
  41.     struct Concatenation {
  42.         char null;
  43.     char header;
  44.     char depth;    /* concatenation nesting depth. */
  45.     unsigned char left_len;
  46.             /* Length of left child if it is sufficiently    */
  47.             /* short; 0 otherwise.                */
  48. #        define MAX_LEFT_LEN 255
  49.     word len;
  50.     CORD left;    /* length(left) > 0    */
  51.     CORD right;    /* length(right) > 0    */
  52.     } concatenation;
  53.     struct Function {
  54.     char null;
  55.     char header;
  56.     char depth;    /* always 0    */
  57.     char left_len;    /* always 0    */
  58.     word len;
  59.     CORD_fn fn;
  60.     void * client_data;
  61.     } function;
  62.     struct Generic {
  63.         char null;
  64.     char header;
  65.     char depth;
  66.     char left_len;
  67.     word len;
  68.     } generic;
  69.     char string[1];
  70. } CordRep;
  71.  
  72. # define CONCAT_HDR 1
  73.     
  74. # define FN_HDR 4
  75. # define SUBSTR_HDR 6
  76.     /* Substring nodes are a special case of function nodes.      */
  77.     /* The client_data field is known to point to a substr_args    */
  78.     /* structure, and the function is either CORD_apply_access_fn     */
  79.     /* or CORD_index_access_fn.                    */
  80.  
  81. /* The following may be applied only to function and concatenation nodes: */
  82. #define IS_CONCATENATION(s)  (((CordRep *)s)->generic.header == CONCAT_HDR)
  83.  
  84. #define IS_FUNCTION(s)  ((((CordRep *)s)->generic.header & FN_HDR) != 0)
  85.  
  86. #define IS_SUBSTR(s) (((CordRep *)s)->generic.header == SUBSTR_HDR)
  87.  
  88. #define LEN(s) (((CordRep *)s) -> generic.len)
  89. #define DEPTH(s) (((CordRep *)s) -> generic.depth)
  90. #define GEN_LEN(s) (CORD_IS_STRING(s) ? strlen(s) : LEN(s))
  91.  
  92. #define LEFT_LEN(c) ((c) -> left_len != 0? \
  93.                 (c) -> left_len \
  94.                 : (CORD_IS_STRING((c) -> left) ? \
  95.                     (c) -> len - GEN_LEN((c) -> right) \
  96.                     : LEN((c) -> left)))
  97.  
  98. #define SHORT_LIMIT (sizeof(CordRep) - 1)
  99.     /* Cords shorter than this are C strings */
  100.  
  101.  
  102. /* Dump the internal representation of x to stdout, with initial     */
  103. /* indentation level n.                            */
  104. void CORD_dump_inner(CORD x, unsigned n)
  105. {
  106.     register size_t i;
  107.     
  108.     for (i = 0; i < (size_t)n; i++) {
  109.         fputs("  ", stdout);
  110.     }
  111.     if (x == 0) {
  112.           fputs("NIL\n", stdout);
  113.     } else if (CORD_IS_STRING(x)) {
  114.         for (i = 0; i <= SHORT_LIMIT; i++) {
  115.             if (x[i] == '\0') break;
  116.             putchar(x[i]);
  117.         }
  118.         if (x[i] != '\0') fputs("...", stdout);
  119.         putchar('\n');
  120.     } else if (IS_CONCATENATION(x)) {
  121.         register struct Concatenation * conc =
  122.                     &(((CordRep *)x) -> concatenation);
  123.         printf("Concatenation: %p (len: %d, depth: %d)\n",
  124.                x, (int)(conc -> len), (int)(conc -> depth));
  125.         CORD_dump_inner(conc -> left, n+1);
  126.         CORD_dump_inner(conc -> right, n+1);
  127.     } else /* function */{
  128.         register struct Function * func =
  129.                     &(((CordRep *)x) -> function);
  130.         if (IS_SUBSTR(x)) printf("(Substring) ");
  131.         printf("Function: %p (len: %d): ", x, (int)(func -> len));
  132.         for (i = 0; i < 20 && i < func -> len; i++) {
  133.             putchar((*(func -> fn))(i, func -> client_data));
  134.         }
  135.         if (i < func -> len) fputs("...", stdout);
  136.         putchar('\n');
  137.     }
  138. }
  139.  
  140. /* Dump the internal representation of x to stdout    */
  141. void CORD_dump(CORD x)
  142. {
  143.     CORD_dump_inner(x, 0);
  144.     fflush(stdout);
  145. }
  146.  
  147. CORD CORD_cat_char_star(CORD x, const char * y, size_t leny)
  148. {
  149.     register size_t result_len;
  150.     register size_t lenx;
  151.     register int depth;
  152.     
  153.     if (x == CORD_EMPTY) return(y);
  154.     if (leny == 0) return(x);
  155.     if (CORD_IS_STRING(x)) {
  156.         lenx = strlen(x);
  157.         result_len = lenx + leny;
  158.         if (result_len <= SHORT_LIMIT) {
  159.             register char * result = GC_MALLOC_ATOMIC(result_len+1);
  160.         
  161.             if (result == 0) OUT_OF_MEMORY;
  162.             memcpy(result, x, lenx);
  163.             memcpy(result + lenx, y, leny);
  164.             result[result_len] = '\0';
  165.             return((CORD) result);
  166.         } else {
  167.             depth = 1;
  168.         }
  169.     } else {
  170.         register CORD right;
  171.         register CORD left;
  172.         register char * new_right;
  173.         register size_t right_len;
  174.         
  175.         lenx = LEN(x);
  176.         
  177.         if (leny <= SHORT_LIMIT/2
  178.             && IS_CONCATENATION(x)
  179.             && CORD_IS_STRING(right = ((CordRep *)x) -> concatenation.right)) {
  180.             /* Merge y into right part of x. */
  181.             if (!CORD_IS_STRING(left = ((CordRep *)x) -> concatenation.left)) {
  182.                 right_len = lenx - LEN(left);
  183.             } else if (((CordRep *)x) -> concatenation.left_len != 0) {
  184.                 right_len = lenx - ((CordRep *)x) -> concatenation.left_len;
  185.             } else {
  186.                 right_len = strlen(right);
  187.             }
  188.             result_len = right_len + leny;  /* length of new_right */
  189.             if (result_len <= SHORT_LIMIT) {
  190.                 new_right = GC_MALLOC_ATOMIC(result_len + 1);
  191.                 memcpy(new_right, right, right_len);
  192.                 memcpy(new_right + right_len, y, leny);
  193.                 new_right[result_len] = '\0';
  194.                 y = new_right;
  195.                 leny = result_len;
  196.                 x = left;
  197.                 lenx -= right_len;
  198.                 /* Now fall through to concatenate the two pieces: */
  199.             }
  200.             if (CORD_IS_STRING(x)) {
  201.                 depth = 1;
  202.             } else {
  203.                 depth = DEPTH(x) + 1;
  204.             }
  205.         } else {
  206.             depth = DEPTH(x) + 1;
  207.         }
  208.         result_len = lenx + leny;
  209.     }
  210.     {
  211.       /* The general case; lenx, result_len is known: */
  212.         register struct Concatenation * result;
  213.         
  214.         result = GC_NEW(struct Concatenation);
  215.         if (result == 0) OUT_OF_MEMORY;
  216.         result->header = CONCAT_HDR;
  217.         result->depth = depth;
  218.         if (lenx <= MAX_LEFT_LEN) result->left_len = lenx;
  219.         result->len = result_len;
  220.         result->left = x;
  221.         result->right = y;
  222.         if (depth > MAX_DEPTH) {
  223.             return(CORD_balance((CORD)result));
  224.         } else {
  225.             return((CORD) result);
  226.         }
  227.     }
  228. }
  229.  
  230.  
  231. CORD CORD_cat(CORD x, CORD y)
  232. {
  233.     register size_t result_len;
  234.     register int depth;
  235.     register size_t lenx;
  236.     
  237.     if (x == CORD_EMPTY) return(y);
  238.     if (y == CORD_EMPTY) return(x);
  239.     if (CORD_IS_STRING(y)) {
  240.         return(CORD_cat_char_star(x, y, strlen(y)));
  241.     } else if (CORD_IS_STRING(x)) {
  242.         lenx = strlen(x);
  243.         depth = DEPTH(y) + 1;
  244.     } else {
  245.         register int depthy = DEPTH(y);
  246.         
  247.         lenx = LEN(x);
  248.         depth = DEPTH(x) + 1;
  249.         if (depthy >= depth) depth = depthy + 1;
  250.     }
  251.     result_len = lenx + LEN(y);
  252.     {
  253.         register struct Concatenation * result;
  254.         
  255.         result = GC_NEW(struct Concatenation);
  256.         if (result == 0) OUT_OF_MEMORY;
  257.         result->header = CONCAT_HDR;
  258.         result->depth = depth;
  259.         if (lenx <= MAX_LEFT_LEN) result->left_len = lenx;
  260.         result->len = result_len;
  261.         result->left = x;
  262.         result->right = y;
  263.         return((CORD) result);
  264.     }
  265. }
  266.  
  267.  
  268.  
  269. CORD CORD_from_fn(CORD_fn fn, void * client_data, size_t len)
  270. {
  271.     if (len <= 0) return(0);
  272.     if (len <= SHORT_LIMIT) {
  273.         register char * result;
  274.         register size_t i;
  275.         char buf[SHORT_LIMIT+1];
  276.         register char c;
  277.         
  278.         for (i = 0; i < len; i++) {
  279.             c = (*fn)(i, client_data);
  280.             if (c == '\0') goto gen_case;
  281.             buf[i] = c;
  282.         }
  283.         buf[i] = '\0';
  284.         result = GC_MALLOC_ATOMIC(len+1);
  285.         if (result == 0) OUT_OF_MEMORY;
  286.         strcpy(result, buf);
  287.         result[len] = '\0';
  288.         return((CORD) result);
  289.     }
  290.   gen_case:
  291.     {
  292.         register struct Function * result;
  293.         
  294.         result = GC_NEW(struct Function);
  295.         if (result == 0) OUT_OF_MEMORY;
  296.         result->header = FN_HDR;
  297.         /* depth is already 0 */
  298.         result->len = len;
  299.         result->fn = fn;
  300.         result->client_data = client_data;
  301.         return((CORD) result);
  302.     }
  303. }
  304.  
  305. size_t CORD_len(CORD x)
  306. {
  307.     if (x == 0) {
  308.          return(0);
  309.     } else {
  310.     return(GEN_LEN(x));
  311.     }
  312. }
  313.  
  314. struct substr_args {
  315.     CordRep * sa_cord;
  316.     size_t sa_index;
  317. };
  318.  
  319. char CORD_index_access_fn(size_t i, void * client_data)
  320. {
  321.     register struct substr_args *descr = (struct substr_args *)client_data;
  322.     
  323.     return(((char *)(descr->sa_cord))[i + descr->sa_index]);
  324. }
  325.  
  326. char CORD_apply_access_fn(size_t i, void * client_data)
  327. {
  328.     register struct substr_args *descr = (struct substr_args *)client_data;
  329.     register struct Function * fn_cord = &(descr->sa_cord->function);
  330.     
  331.     return((*(fn_cord->fn))(i + descr->sa_index, fn_cord->client_data));
  332. }
  333.  
  334. /* A version of CORD_substr that simply returns a function node, thus    */
  335. /* postponing its work.    The fourth argument is a function that may    */
  336. /* be used for efficient access to the ith character.            */
  337. /* Assumes i >= 0 and i + n < length(x).                */
  338. CORD CORD_substr_closure(CORD x, size_t i, size_t n, CORD_fn f)
  339. {
  340.     register struct substr_args * sa = GC_NEW(struct substr_args);
  341.     CORD result;
  342.     
  343.     if (sa == 0) OUT_OF_MEMORY;
  344.     sa->sa_cord = (CordRep *)x;
  345.     sa->sa_index = i;
  346.     result = CORD_from_fn(f, (void *)sa, n);
  347.     ((CordRep *)result) -> function.header = SUBSTR_HDR;
  348.     return (result);
  349. }
  350.  
  351. # define SUBSTR_LIMIT (10 * SHORT_LIMIT)
  352.     /* Substrings of function nodes and flat strings shorter than     */
  353.     /* this are flat strings.  Othewise we use a functional     */
  354.     /* representation, which is significantly slower to access.    */
  355.  
  356. /* A version of CORD_substr that assumes i >= 0, n > 0, and i + n < length(x).*/
  357. CORD CORD_substr_checked(CORD x, size_t i, size_t n)
  358. {
  359.     if (CORD_IS_STRING(x)) {
  360.         if (n > SUBSTR_LIMIT) {
  361.             return(CORD_substr_closure(x, i, n, CORD_index_access_fn));
  362.         } else {
  363.             register char * result = GC_MALLOC_ATOMIC(n+1);
  364.             register char * p = result;
  365.             
  366.             if (result == 0) OUT_OF_MEMORY;
  367.             strncpy(result, x+i, n);
  368.             result[n] = '\0';
  369.             return(result);
  370.         }
  371.     } else if (IS_CONCATENATION(x)) {
  372.         register struct Concatenation * conc
  373.                 = &(((CordRep *)x) -> concatenation);
  374.         register size_t left_len;
  375.         register size_t right_len;
  376.         
  377.         left_len = LEFT_LEN(conc);
  378.         right_len = conc -> len - left_len;
  379.         if (i >= left_len) {
  380.             if (n == right_len) return(conc -> right);
  381.             return(CORD_substr_checked(conc -> right, i - left_len, n));
  382.         } else if (i+n <= left_len) {
  383.             if (n == left_len) return(conc -> left);
  384.             return(CORD_substr_checked(conc -> left, i, n));
  385.         } else {
  386.             /* Need at least one character from each side. */
  387.             register CORD left_part;
  388.             register CORD right_part;
  389.             register size_t left_part_len = left_len - i;
  390.          
  391.             if (i == 0) {
  392.                 left_part = conc -> left;
  393.             } else {
  394.                 left_part = CORD_substr_checked(conc -> left, i, left_part_len);
  395.             }
  396.             if (i + n == right_len + left_len) {
  397.                  right_part = conc -> right;
  398.             } else {
  399.                  right_part = CORD_substr_checked(conc -> right, 0,
  400.                                       n - left_part_len);
  401.             }
  402.             return(CORD_cat(left_part, right_part));
  403.         }
  404.     } else /* function */ {
  405.         if (n > SUBSTR_LIMIT) {
  406.             if (IS_SUBSTR(x)) {
  407.                 /* Avoid nesting substring nodes.    */
  408.                 register struct Function * f = &(((CordRep *)x) -> function);
  409.                 register struct substr_args *descr =
  410.                         (struct substr_args *)(f -> client_data);
  411.                 
  412.                 return(CORD_substr_closure((CORD)descr->sa_cord,
  413.                                i + descr->sa_index,
  414.                                n, f -> fn));
  415.             } else {
  416.                 return(CORD_substr_closure(x, i, n, CORD_apply_access_fn));
  417.             }
  418.         } else {
  419.             char * result;
  420.             register struct Function * f = &(((CordRep *)x) -> function);
  421.             char buf[SUBSTR_LIMIT+1];
  422.             register char * p = buf;
  423.             register char c;
  424.             register int j;
  425.             register int lim = i + n;
  426.             
  427.             for (j = i; j < lim; j++) {
  428.                 c = (*(f -> fn))(j, f -> client_data);
  429.                 if (c == '\0') {
  430.                     return(CORD_substr_closure(x, i, n, CORD_apply_access_fn));
  431.                 }
  432.                 *p++ = c;
  433.             }
  434.             *p = '\0';
  435.             result = GC_MALLOC_ATOMIC(n+1);
  436.             if (result == 0) OUT_OF_MEMORY;
  437.             strcpy(result, buf);
  438.             return(result);
  439.         }
  440.     }
  441. }
  442.  
  443. CORD CORD_substr(CORD x, size_t i, size_t n)
  444. {
  445.     register size_t len = CORD_len(x);
  446.     
  447.     if (i >= len || n <= 0) return(0);
  448.         /* n < 0 is impossible in a correct C implementation, but    */
  449.         /* quite possible  under SunOS 4.X.                */
  450.     if (i + n > len) n = len - i;
  451. #   ifndef __STDC__
  452.       if (i < 0) ABORT("CORD_substr: second arg. negative");
  453.         /* Possible only if both client and C implementation are buggy.    */
  454.         /* But empirically this happens frequently.            */
  455. #   endif
  456.     return(CORD_substr_checked(x, i, n));
  457. }
  458.  
  459. /* See cord.h for definition.  We assume i is in range.    */
  460. int CORD_iter5(CORD x, size_t i, CORD_iter_fn f1,
  461.              CORD_batched_iter_fn f2, void * client_data)
  462. {
  463.     if (x == 0) return(0);
  464.     if (CORD_IS_STRING(x)) {
  465.         register const char *p = x+i;
  466.         
  467.         if (*p == '\0') ABORT("2nd arg to CORD_iter5 too big");
  468.         if (f2 != CORD_NO_FN) {
  469.             return((*f2)(p, client_data));
  470.         } else {
  471.         while (*p) {
  472.                 if ((*f1)(*p, client_data)) return(1);
  473.                 p++;
  474.         }
  475.         return(0);
  476.         }
  477.     } else if (IS_CONCATENATION(x)) {
  478.         register struct Concatenation * conc
  479.                 = &(((CordRep *)x) -> concatenation);
  480.         
  481.         
  482.         if (i > 0) {
  483.             register size_t left_len = LEFT_LEN(conc);
  484.             
  485.             if (i >= left_len) {
  486.                 return(CORD_iter5(conc -> right, i - left_len, f1, f2,
  487.                           client_data));
  488.             }
  489.         }
  490.         if (CORD_iter5(conc -> left, i, f1, f2, client_data)) {
  491.             return(1);
  492.         }
  493.         return(CORD_iter5(conc -> right, 0, f1, f2, client_data));
  494.     } else /* function */ {
  495.         register struct Function * f = &(((CordRep *)x) -> function);
  496.         register size_t j;
  497.         register size_t lim = f -> len;
  498.         
  499.         for (j = i; j < lim; j++) {
  500.             if ((*f1)((*(f -> fn))(j, f -> client_data), client_data)) {
  501.                 return(1);
  502.             }
  503.         }
  504.         return(0);
  505.     }
  506. }
  507.             
  508. #undef CORD_iter
  509. int CORD_iter(CORD x, CORD_iter_fn f1, void * client_data)
  510. {
  511.     return(CORD_iter5(x, 0, f1, CORD_NO_FN, client_data));
  512. }
  513.  
  514. int CORD_riter4(CORD x, size_t i, CORD_iter_fn f1, void * client_data)
  515. {
  516.     if (x == 0) return(0);
  517.     if (CORD_IS_STRING(x)) {
  518.     register const char *p = x + i;
  519.     register char c;
  520.                
  521.     while (p >= x) {
  522.         c = *p;
  523.         if (c == '\0') ABORT("2nd arg to CORD_riter4 too big");
  524.             if ((*f1)(c, client_data)) return(1);
  525.             p--;
  526.     }
  527.     return(0);
  528.     } else if (IS_CONCATENATION(x)) {
  529.         register struct Concatenation * conc
  530.                 = &(((CordRep *)x) -> concatenation);
  531.         register CORD left_part = conc -> left;
  532.         register size_t left_len;
  533.         
  534.         left_len = LEFT_LEN(conc);
  535.         if (i >= left_len) {
  536.             if (CORD_riter4(conc -> right, i - left_len, f1, client_data)) {
  537.                 return(1);
  538.             }
  539.             return(CORD_riter4(left_part, left_len - 1, f1, client_data));
  540.         } else {
  541.             return(CORD_riter4(left_part, i, f1, client_data));
  542.         }
  543.     } else /* function */ {
  544.         register struct Function * f = &(((CordRep *)x) -> function);
  545.         register size_t j;
  546.         
  547.         for (j = i; ; j--) {
  548.             if ((*f1)((*(f -> fn))(j, f -> client_data), client_data)) {
  549.                 return(1);
  550.             }
  551.             if (j == 0) return(0);
  552.         }
  553.     }
  554. }
  555.  
  556. int CORD_riter(CORD x, CORD_iter_fn f1, void * client_data)
  557. {
  558.     return(CORD_riter4(x, CORD_len(x) - 1, f1, client_data));
  559. }
  560.  
  561. /*
  562.  * The following functions are concerned with balancing cords.
  563.  * Strategy:
  564.  * Scan the cord from left to right, keeping the cord scanned so far
  565.  * as a forest of balanced trees of exponentialy decreasing length.
  566.  * When a new subtree needs to be added to the forest, we concatenate all
  567.  * shorter ones to the new tree in the appropriate order, and then insert
  568.  * the result into the forest.
  569.  * Crucial invariants:
  570.  * 1. The concatenation of the forest (in decreasing order) with the
  571.  *     unscanned part of the rope is equal to the rope being balanced.
  572.  * 2. All trees in the forest are balanced.
  573.  * 3. forest[i] has depth at most i.
  574.  */
  575.  
  576. typedef struct {
  577.     CORD c;
  578.     size_t len;        /* Actual ength of c     */
  579. } ForestElement;
  580.  
  581. static size_t min_len [ MAX_DEPTH ];
  582.  
  583. static int min_len_init = 0;
  584.  
  585. int CORD_max_len;
  586.  
  587. typedef ForestElement Forest [ MAX_DEPTH ];
  588.             /* forest[i].min_length = fib(i+1)    */
  589.             /* The string is the concatenation    */
  590.             /* of the forest in order of DECREASING */
  591.             /* indices.                */
  592.  
  593. void CORD_init_min_len()
  594. {
  595.     register int i;
  596.     register size_t last, previous, current;
  597.         
  598.     min_len[0] = previous = 1;
  599.     min_len[1] = last = 2;
  600.     for (i = 2; i < MAX_DEPTH; i++) {
  601.         current = last + previous;
  602.         if (current < last) /* overflow */ current = last;
  603.         min_len[i] = current;
  604.         previous = last;
  605.         last = current;
  606.     }
  607.     CORD_max_len = last - 1;
  608.     min_len_init = 1;
  609. }
  610.  
  611.  
  612. void CORD_init_forest(ForestElement * forest, size_t max_len)
  613. {
  614.     register int i;
  615.     
  616.     for (i = 0; i < MAX_DEPTH; i++) {
  617.         forest[i].c = 0;
  618.         if (min_len[i] > max_len) return;
  619.     }
  620.     ABORT("Cord too long");
  621. }
  622.  
  623. /* Add a leaf to the appropriate level in the forest, cleaning        */
  624. /* out lower levels as necessary.                    */
  625. /* Also works if x is a balanced tree of concatenations; however    */
  626. /* in this case an extra concatenation node may be inserted above x;    */
  627. /* This node should not be counted in the statement of the invariants.    */
  628. void CORD_add_forest(ForestElement * forest, CORD x, size_t len)
  629. {
  630.     register int i = 0;
  631.     register CORD sum = CORD_EMPTY;
  632.     register size_t sum_len = 0;
  633.     
  634.     while (len > min_len[i + 1]) {
  635.         if (forest[i].c != 0) {
  636.             sum = CORD_cat(forest[i].c, sum);
  637.             sum_len += forest[i].len;
  638.             forest[i].c = 0;
  639.         }
  640.         i++;
  641.     }
  642.     /* Sum has depth at most 1 greter than what would be required     */
  643.     /* for balance.                            */
  644.     sum = CORD_cat(sum, x);
  645.     sum_len += len;
  646.     /* If x was a leaf, then sum is now balanced.  To see this        */
  647.     /* consider the two cases in whichforest[i-1] either is or is     */
  648.     /* not empty.                            */
  649.     while (sum_len >= min_len[i]) {
  650.         if (forest[i].c != 0) {
  651.             sum = CORD_cat(forest[i].c, sum);
  652.             sum_len += forest[i].len;
  653.             /* This is again balanced, since sum was balanced, and has    */
  654.             /* allowable depth that differs from i by at most 1.    */
  655.             forest[i].c = 0;
  656.         }
  657.         i++;
  658.     }
  659.     i--;
  660.     forest[i].c = sum;
  661.     forest[i].len = sum_len;
  662. }
  663.  
  664. CORD CORD_concat_forest(ForestElement * forest, size_t expected_len)
  665. {
  666.     register int i = 0;
  667.     CORD sum = 0;
  668.     size_t sum_len = 0;
  669.     
  670.     while (sum_len != expected_len) {
  671.         if (forest[i].c != 0) {
  672.             sum = CORD_cat(forest[i].c, sum);
  673.             sum_len += forest[i].len;
  674.         }
  675.         i++;
  676.     }
  677.     return(sum);
  678. }
  679.  
  680. /* Insert the frontier of x into forest.  Balanced subtrees are    */
  681. /* treated as leaves.  This potentially adds one to the depth    */
  682. /* of the final tree.                        */
  683. void CORD_balance_insert(CORD x, size_t len, ForestElement * forest)
  684. {
  685.     register int depth;
  686.     
  687.     if (CORD_IS_STRING(x)) {
  688.         CORD_add_forest(forest, x, len);
  689.     } else if (IS_CONCATENATION(x)
  690.                && ((depth = DEPTH(x)) >= MAX_DEPTH
  691.                    || len < min_len[depth])) {
  692.         register struct Concatenation * conc
  693.                 = &(((CordRep *)x) -> concatenation);
  694.         size_t left_len = LEFT_LEN(conc);
  695.         
  696.         CORD_balance_insert(conc -> left, left_len, forest);
  697.         CORD_balance_insert(conc -> right, len - left_len, forest);
  698.     } else /* function or balanced */ {
  699.         CORD_add_forest(forest, x, len);
  700.     }
  701. }
  702.  
  703.  
  704. CORD CORD_balance(CORD x)
  705. {
  706.     Forest forest;
  707.     register size_t len;
  708.     
  709.     if (x == 0) return(0);
  710.     if (CORD_IS_STRING(x)) return(x);
  711.     if (!min_len_init) CORD_init_min_len();
  712.     len = LEN(x);
  713.     CORD_init_forest(forest, len);
  714.     CORD_balance_insert(x, len, forest);
  715.     return(CORD_concat_forest(forest, len));
  716. }
  717.  
  718.  
  719. /* Position primitives    */
  720.  
  721. /* Private routines to deal with the hard cases only: */
  722.  
  723. /* P contains a prefix of the  path to cur_pos.    Extend it to a full    */
  724. /* path and set up leaf info.                        */
  725. /* Return 0 if past the end of cord, 1 o.w.                */
  726. void CORD__extend_path(register CORD_pos p)
  727. {
  728.      register struct CORD_pe * current_pe = &(p[0].path[p[0].path_len]);
  729.      register CORD top = current_pe -> pe_cord;
  730.      register size_t pos = p[0].cur_pos;
  731.      register size_t top_pos = current_pe -> pe_start_pos;
  732.      register size_t top_len = GEN_LEN(top);
  733.      
  734.      /* Fill in the rest of the path. */
  735.        while(!CORD_IS_STRING(top) && IS_CONCATENATION(top)) {
  736.           register struct Concatenation * conc =
  737.                   &(((CordRep *)top) -> concatenation);
  738.           register size_t left_len;
  739.           
  740.           left_len = LEFT_LEN(conc);
  741.           current_pe++;
  742.           if (pos >= top_pos + left_len) {
  743.               current_pe -> pe_cord = top = conc -> right;
  744.               current_pe -> pe_start_pos = top_pos = top_pos + left_len;
  745.               top_len -= left_len;
  746.           } else {
  747.               current_pe -> pe_cord = top = conc -> left;
  748.               current_pe -> pe_start_pos = top_pos;
  749.               top_len = left_len;
  750.           }
  751.           p[0].path_len++;
  752.        }
  753.      /* Fill in leaf description for fast access. */
  754.        if (CORD_IS_STRING(top)) {
  755.          p[0].cur_leaf = top;
  756.          p[0].cur_start = top_pos;
  757.          p[0].cur_end = top_pos + top_len;
  758.        } else {
  759.          p[0].cur_end = 0;
  760.        }
  761.        if (pos >= top_pos + top_len) p[0].path_len = CORD_POS_INVALID;
  762. }
  763.  
  764. char CORD__pos_fetch(register CORD_pos p)
  765. {
  766.     /* Leaf is a function node */
  767.     struct CORD_pe * pe = &((p)[0].path[(p)[0].path_len]);
  768.     CORD leaf = pe -> pe_cord;
  769.     register struct Function * f = &(((CordRep *)leaf) -> function);
  770.     
  771.     if (!IS_FUNCTION(leaf)) ABORT("CORD_pos_fetch: bad leaf");
  772.     return ((*(f -> fn))(p[0].cur_pos - pe -> pe_start_pos, f -> client_data));
  773. }
  774.  
  775. void CORD__next(register CORD_pos p)
  776. {
  777.     register size_t cur_pos = p[0].cur_pos + 1;
  778.     register struct CORD_pe * current_pe = &((p)[0].path[(p)[0].path_len]);
  779.     register CORD leaf = current_pe -> pe_cord;
  780.     
  781.     /* Leaf is not a string or we're at end of leaf */
  782.     p[0].cur_pos = cur_pos;
  783.     if (!CORD_IS_STRING(leaf)) {
  784.         /* Function leaf    */
  785.         register struct Function * f = &(((CordRep *)leaf) -> function);
  786.         register size_t start_pos = current_pe -> pe_start_pos;
  787.         register size_t end_pos = start_pos + f -> len;
  788.         
  789.         if (cur_pos < end_pos) {
  790.           /* Fill cache and return. */
  791.             register size_t i;
  792.             register size_t limit = cur_pos + FUNCTION_BUF_SZ;
  793.             register CORD_fn fn = f -> fn;
  794.             register void * client_data = f -> client_data;
  795.             
  796.             if (limit > end_pos) {
  797.                 limit = end_pos;
  798.             }
  799.             for (i = cur_pos; i < limit; i++) {
  800.                 p[0].function_buf[i - cur_pos] =
  801.                     (*fn)(i - start_pos, client_data);
  802.             }
  803.             p[0].cur_start = cur_pos;
  804.             p[0].cur_leaf = p[0].function_buf;
  805.             p[0].cur_end = limit;
  806.             return;
  807.         }
  808.     }
  809.     /* End of leaf    */
  810.     /* Pop the stack until we find two concatenation nodes with the     */
  811.     /* same start position: this implies we were in left part.        */
  812.     {
  813.         while (p[0].path_len > 0
  814.                && current_pe[0].pe_start_pos != current_pe[-1].pe_start_pos) {
  815.             p[0].path_len--;
  816.             current_pe--;
  817.         }
  818.         if (p[0].path_len == 0) {
  819.         p[0].path_len = CORD_POS_INVALID;
  820.             return;
  821.     }
  822.     }
  823.     p[0].path_len--;
  824.     CORD__extend_path(p);
  825. }
  826.  
  827. void CORD__prev(register CORD_pos p)
  828. {
  829.     register struct CORD_pe * pe = &(p[0].path[p[0].path_len]);
  830.     
  831.     if (p[0].cur_pos == 0) {
  832.         p[0].path_len = CORD_POS_INVALID;
  833.         return;
  834.     }
  835.     p[0].cur_pos--;
  836.     if (p[0].cur_pos >= pe -> pe_start_pos) return;
  837.     
  838.     /* Beginning of leaf    */
  839.     
  840.     /* Pop the stack until we find two concatenation nodes with the     */
  841.     /* different start position: this implies we were in right part.    */
  842.     {
  843.         register struct CORD_pe * current_pe = &((p)[0].path[(p)[0].path_len]);
  844.         
  845.         while (p[0].path_len > 0
  846.                && current_pe[0].pe_start_pos == current_pe[-1].pe_start_pos) {
  847.             p[0].path_len--;
  848.             current_pe--;
  849.         }
  850.     }
  851.     p[0].path_len--;
  852.     CORD__extend_path(p);
  853. }
  854.  
  855. #undef CORD_pos_fetch
  856. #undef CORD_next
  857. #undef CORD_prev
  858. #undef CORD_pos_to_index
  859. #undef CORD_pos_to_cord
  860. #undef CORD_pos_valid
  861.  
  862. char CORD_pos_fetch(register CORD_pos p)
  863. {
  864.     if (p[0].cur_start <= p[0].cur_pos && p[0].cur_pos < p[0].cur_end) {
  865.         return(p[0].cur_leaf[p[0].cur_pos - p[0].cur_start]);
  866.     } else {
  867.         return(CORD__pos_fetch(p));
  868.     }
  869. }
  870.  
  871. void CORD_next(CORD_pos p)
  872. {
  873.     if (p[0].cur_pos < p[0].cur_end - 1) {
  874.         p[0].cur_pos++;
  875.     } else {
  876.         CORD__next(p);
  877.     }
  878. }
  879.  
  880. void CORD_prev(CORD_pos p)
  881. {
  882.     if (p[0].cur_end != 0 && p[0].cur_pos > p[0].cur_start) {
  883.         p[0].cur_pos--;
  884.     } else {
  885.         CORD__prev(p);
  886.     }
  887. }
  888.  
  889. size_t CORD_pos_to_index(CORD_pos p)
  890. {
  891.     return(p[0].cur_pos);
  892. }
  893.  
  894. CORD CORD_pos_to_cord(CORD_pos p)
  895. {
  896.     return(p[0].path[0].pe_cord);
  897. }
  898.  
  899. int CORD_pos_valid(CORD_pos p)
  900. {
  901.     return(p[0].path_len != CORD_POS_INVALID);
  902. }
  903.  
  904. void CORD_set_pos(CORD_pos p, CORD x, size_t i)
  905. {
  906.     if (x == CORD_EMPTY) {
  907.         p[0].path_len = CORD_POS_INVALID;
  908.         return;
  909.     }
  910.     p[0].path[0].pe_cord = x;
  911.     p[0].path[0].pe_start_pos = 0;
  912.     p[0].path_len = 0;
  913.     p[0].cur_pos = i;
  914.     CORD__extend_path(p);
  915. }
  916.