/*
  Copyright (c) 2003 by Stefan Kurtz and The Institute for
  Genomic Research.  This is OSI Certified Open Source Software.
  Please see the file LICENSE for licensing information and
  the file ACKNOWLEDGEMENTS for names of contributors to the
  code base.
*/

//}

//\FILEINFO{construct.c}{Stefan Kurtz}{November 1999}

//\Ignore{

#include "st_intbits.h"
#include "st_spacedef.h"
#include "st_megabytes.h"
#include "st_debugdef.h"
#include "st_streemac.h"
#include "st_streetyp.h"
#include "st_streeacc.h"

#define FUNCLEVEL 4

#define DEBUGDEFAULT DEBUG1(FUNCLEVEL,">%s\n",__func__);\
                     DEBUGCODE(5,showstree(stree))

#define VALIDINIT     0

#define CHECKTEXTLEN\
        if(textlen > MAXTEXTLEN)\
        {\
          ERROR2("suffix tree construction failed: "\
                 "textlen=%lu larger than maximal textlen=%lu",\
                  (Showuint) textlen,(Showuint) MAXTEXTLEN);\
          return -1;\
        }

#ifdef ST_DEBUG


static void showvalues(void)
{
  SHOWVAL(SMALLINTS);
  SHOWVAL(LARGEINTS);
  SHOWVAL(MAXDISTANCE);
#if defined(STREELARGE) || defined(STREESMALL)
  SHOWVAL(SMALLDEPTH);
#endif
  SHOWVAL(MAXTEXTLEN);
}

#endif

//}

/*
 This file contains code for the improved linked list implementation,
 as described in \cite{KUR:1998,KUR:BAL:1999}. It can be 
 compiled with two options: 
 \begin{itemize}
 \item
 for short strings of length \(\leq 2^{21}-1=2\) megabytes, 
 we recommend the option 
 \texttt{STREESMALL}. This results in a representation which requires 
 \(2\) integers for each small node, and three integers for each large node.
 See also the header file \texttt{streesmall.h}.
 \item
 for long strings of length \(\leq 2^{27}-1=12\) megabytes, we 
 recommend the option 
 \texttt{STREELARGE}. This results in a representation which requires 
 \(2\) integers for each small node, and four integers for each large node.
 See also the header file \texttt{streelarge.h}.
 \end{itemize}
*/

//\subsection{Space Management}

/*
 For a string of length \(n\) we initially allocate space for
 \(\texttt{STARTFACTOR}\cdot\texttt{SMALLINTS}\cdot n\) integers to store
 the branching nodes. This usually suffices for most cases. In case we need 
 more integers, we allocate space for \(\texttt{ADDFACTOR}\cdot n\) 
 (at least 16) extra branching nodes. 
*/

#ifndef STARTFACTOR
#define STARTFACTOR 0.5
#endif

#define ADDFACTOR   0.05
#define MINEXTRA    16

/*
 Before a new node is stored, we check if there is enough space available.
 If not, the space is enlarged by a small amount. Since some global pointers
 directly refer into the table, these have to be adjusted after reallocation.
*/

static void spaceforbranchtab(Suffixtree *stree)
{
  DEBUG1(FUNCLEVEL,">%s\n",__func__);

  if(stree->nextfreebranch >= stree->firstnotallocated)
  {
    Uint tmpheadnode, 
         tmpchainstart = 0, 
         extra = (Uint) (ADDFACTOR * (MULTBYSMALLINTS(stree->textlen+1)));
    if(extra < MINEXTRA)
    {
      extra = MULTBYSMALLINTS(MINEXTRA);
    }
    DEBUG1(2,"#all suffixes up to suffix %lu have been scanned\n",
              (Showuint) stree->nextfreeleafnum);
    DEBUG1(2,"#current space peak %f\n",MEGABYTES(getspacepeak()));
    DEBUG1(2,"#to get %lu extra space do ",(Showuint) extra);
    stree->currentbranchtabsize += extra;
    tmpheadnode = BRADDR2NUM(stree,stree->headnode);
    if(stree->chainstart != NULL)
    {
      tmpchainstart = BRADDR2NUM(stree,stree->chainstart);
    }

    stree->branchtab 
     = ALLOCSPACE_TLB(stree->branchtab,Uint,stree->currentbranchtabsize);
    stree->nextfreebranch = stree->branchtab + stree->nextfreebranchnum;
    stree->headnode = stree->branchtab + tmpheadnode;
    if(stree->chainstart != NULL)
    {
      stree->chainstart = stree->branchtab + tmpchainstart;
    }
    stree->firstnotallocated 
       = stree->branchtab + stree->currentbranchtabsize - LARGEINTS;
  }
}

//\subsection{Initializing and Retrieving Headpositions, Depth, and Suffixlinks}

/*
 We have three functions to initialize and retrieve head positions, depth, and
 suffix links. The implementation depends on the bit layout.
 \begin{enumerate}
 \item
 The function \emph{setdepthnum} stores the \emph{depth} and the 
 \emph{head position} of a new large node.
 \item
 The function \emph{setsuffixlink} stores the \emph{suffixlink} 
 of a new large node.
 \item
 The function \emph{getlargelinkconstruction} retrieves the \emph{suffixlink} 
 of a large node, which is referenced by \emph{headposition}.
 \end{enumerate}
*/

#ifdef STREESMALL

static void setdepthheadposition(Suffixtree *stree,Uint depth,
                                 Uint headposition)
{
  DEBUG4(FUNCLEVEL,">%s(%lu)=(depth=%lu,headposition=%lu)\n",
                   __func__,
                   (Showuint) stree->nextfreebranchnum,
                   (Showuint) depth,
                   (Showuint) headposition);
  DEBUGCODE(1,stree->maxset = stree->nextfreebranch + 2);
  if(ISSMALLDEPTH(depth))
  {
    *(stree->nextfreebranch+1) |= SMALLDEPTHMARK;
  } else
  {
    *(stree->nextfreebranch)
     = (*(stree->nextfreebranch) & (MAXINDEX | LARGEBIT)) | 
       ((depth << 11) & (7 << 29));
    *(stree->nextfreebranch+1) 
     = (*(stree->nextfreebranch+1) & (MAXINDEX | NILBIT)) |
       ((depth << 14) & (127 << 25));
  }
  *(stree->nextfreebranch+2) = (depth << 21) | headposition;
}

static void setsuffixlink(Suffixtree *stree,Uint slink)
{
  DEBUG3(FUNCLEVEL,">%s(%lu)=%lu\n",
                    __func__,
                    (Showuint) stree->nextfreebranchnum,
                    (Showuint) slink);
  *(stree->setlink) = (*(stree->setlink) & (255 << 24)) | (slink | NILBIT);
  if(ISSMALLDEPTH(stree->currentdepth))
  { 
    *(stree->nextfreebranch+1) |= (slink << 25);
    if(stree->nextfreebranchnum & (~((1 << 7) - 1)))
    {
      *(stree->nextfreebranch) |= ((slink << 17) & (255 << 24));
      if(stree->nextfreebranchnum & (~((1 << 15) - 1)))
      {
        stree->leaftab[stree->nextfreeleafnum-1] |= 
          ((slink << 9) & (255 << 24));
      }
    }
  } 
}

static Uint getlargelinkconstruction(Suffixtree *stree)
{
  SYMBOL secondchar;
  Uint succ, slink, headnodenum;

  DEBUG2(FUNCLEVEL,">%s(%lu)\n",
                   __func__,
                   (Showuint) BRADDR2NUM(stree,stree->headnode));

  DEBUGCODE(1,stree->largelinks++);
  if(stree->headnodedepth == 1)
  {
    return 0;               // link refers to the root
  }
  if(stree->headnodedepth == 2)   // determine second character of edge
  {
    if(stree->headend == NULL)   
    {
      secondchar = *(stree->tailptr-1);
    } else
    {
      secondchar = *(stree->tailptr - (stree->headend - stree->headstart + 2));
    }
    // this leads to the suffix link node
    return GETBRANCHINDEX(stree->rootchildren[(Uint) secondchar]);
  }
  if(ISSMALLDEPTH(stree->headnodedepth))   // retrieve link in constant time
  {
    slink = *(stree->headnode+1) >> 25;
    headnodenum = BRADDR2NUM(stree,stree->headnode);
    if(headnodenum & (~((1 << 7) - 1)))
    {
      slink |= ((*(stree->headnode) & (255 << 24)) >> 17);
      if(headnodenum & (~((1 << 15) - 1)))
      {
        slink |= ((stree->leaftab[GETHEADPOS(stree->headnode)] 
                   & (255 << 24)) >> 9);
      }
    }
    return slink;
  }
  succ = stree->onsuccpath;   // start at node on successor path
  DEBUGCODE(1,stree->largelinklinkwork++);
  while(!NILPTR(succ))        // linear retrieval of suffix links
  {
    DEBUGCODE(1,stree->largelinkwork++);
    if(ISLEAF(succ))
    {
      succ = LEAFBROTHERVAL(stree->leaftab[GETLEAFINDEX(succ)]);
    } else
    {
      succ = GETBROTHER(stree->branchtab + GETBRANCHINDEX(succ));
    } 
    DEBUGCODE(1,stree->largelinkwork++);
  }
  return succ & MAXINDEX;   // get only the index
}
#endif

#ifdef STREELARGE

static void setdepthheadposition(Suffixtree *stree,Uint depth,
                                 Uint headposition)
{
  DEBUG4(FUNCLEVEL,">%s(%lu)=(depth=%lu,headposition=%lu)\n",
                   __func__,
                   (Showuint) stree->nextfreebranchnum,
                   (Showuint) depth,
                   (Showuint) headposition);

  if(ISSMALLDEPTH(depth))
  {
    *(stree->nextfreebranch+2) = depth | SMALLDEPTHMARK;
  } else
  {
    *(stree->nextfreebranch+2) = depth;
  }
  *(stree->nextfreebranch+3) = headposition;
}

static void setsuffixlink(Suffixtree *stree,Uint slink)
{
  Uint slinkhalf = slink >> 1;

  DEBUG3(FUNCLEVEL,">%s(%lu)=%lu\n",
                    __func__,
                    (Showuint) stree->nextfreebranchnum,
                    (Showuint) slink);
  *(stree->setlink) = (*(stree->setlink) & EXTRAPATT) | (slink | NILBIT);
  if(ISSMALLDEPTH(stree->currentdepth))
  { 
    *(stree->nextfreebranch+2) 
      |= ((slinkhalf << SMALLDEPTHBITS) & LOWERLINKPATT);
    if(stree->nextfreebranchnum & (~LOWERLINKSIZE))
    {
      *(stree->nextfreebranch+3) 
        |= ((slinkhalf << SHIFTMIDDLE) & MIDDLELINKPATT);
      if(stree->nextfreebranchnum & HIGHERSIZE)
      {
        stree->leaftab[stree->nextfreeleafnum-1] |= 
              ((slinkhalf << SHIFTHIGHER) & EXTRAPATT);
      }
    }
  } 
}

static Uint getlargelinkconstruction(Suffixtree *stree)
{
  SYMBOL secondchar;
  Uint succ, slinkhalf, headnodenum;

  DEBUG2(FUNCLEVEL,">%s(%lu)\n",
                   __func__,
                   (Showuint) BRADDR2NUM(stree,stree->headnode));
  DEBUGCODE(1,stree->largelinks++);
  if(stree->headnodedepth == 1)
  {
    return 0;        // link refers to root
  }
  if(stree->headnodedepth == 2)  // determine second char of egde
  {
    if(stree->headend == NULL)
    {
      secondchar = *(stree->tailptr-1);
    } else
    {
      secondchar = *(stree->tailptr - (stree->headend - stree->headstart + 2));
    }
    return stree->rootchildren[(Uint) secondchar]; 
  }
  if(ISSMALLDEPTH(stree->headnodedepth))  // retrieve link in constant time
  {
    slinkhalf = (*(stree->headnode+2) & LOWERLINKPATT) >> SMALLDEPTHBITS;
    headnodenum = BRADDR2NUM(stree,stree->headnode);
    if(headnodenum & (~LOWERLINKSIZE))
    {
      slinkhalf |= ((*(stree->headnode+3) & MIDDLELINKPATT) >> SHIFTMIDDLE);
      if(headnodenum & HIGHERSIZE)
      {
        slinkhalf 
          |= ((stree->leaftab[GETHEADPOS(stree->headnode)] & EXTRAPATT) 
             >> SHIFTHIGHER);
      }
    }
    return slinkhalf << 1;
  }
  succ = stree->onsuccpath;
  DEBUGCODE(1,stree->largelinklinkwork++);
  while(!NILPTR(succ))   // linear retrieval of suffix link
  {
    DEBUGCODE(1,stree->largelinkwork++);
    if(ISLEAF(succ))
    {
      succ = LEAFBROTHERVAL(stree->leaftab[GETLEAFINDEX(succ)]);
    } else
    {
      succ = GETBROTHER(stree->branchtab + GETBRANCHINDEX(succ));
    } 
    DEBUGCODE(1,stree->largelinkwork++);
  }
  return succ & MAXINDEX;   // get only the index
}
#endif

#ifdef STREEHUGE

static Uint getlargelinkconstruction(Suffixtree *stree)
{
  SYMBOL secondchar;

  DEBUG2(FUNCLEVEL,">%s(%lu)\n",
                   __func__,
                   (Showuint) BRADDR2NUM(stree,stree->headnode));
  if(stree->headnodedepth == 1)
  {
    return 0;        // link refers to root
  }
  if(stree->headnodedepth == 2)  // determine second char of egde
  {
    if(stree->headend == NULL)
    {
      secondchar = *(stree->tailptr-1);
    } else
    {
      secondchar = *(stree->tailptr - (stree->headend - stree->headstart + 2));
    }
    return stree->rootchildren[(Uint) secondchar]; 
  }
  return *(stree->headnode+4);
}
#endif

//\subsection{Insertion of Nodes}

/*
  The function \emph{insertleaf} inserts a leaf and a corresponding leaf
  edge outgoing from the current \emph{headnode}. 
  \emph{insertprev} refers to the node to the left of the leaf to be inserted.
  If the leaf is the first child, then \emph{insertprev} is 
  \texttt{UNDEFINEDREFERENCE}.
*/

static void insertleaf(Suffixtree *stree)
{
  Uint *ptr, newleaf;

  DEBUGDEFAULT;
  newleaf = MAKELEAF(stree->nextfreeleafnum);
  DEBUGCODE(1,stree->insertleafcalls++);
  if(stree->headnodedepth == 0)                // head is the root
  {
    if(stree->tailptr != stree->sentinel)      // no \$-edge initially
    {
      stree->rootchildren[(Uint) *(stree->tailptr)] = newleaf;
      *(stree->nextfreeleafptr) = VALIDINIT;
      DEBUG2(4,"%c-edge from root points to leaf %lu\n",
               *(stree->tailptr),(Showuint) stree->nextfreeleafnum);
    }
  } else
  {
    if (stree->insertprev == UNDEFINEDREFERENCE)  // newleaf = first child
    {
      *(stree->nextfreeleafptr) = GETCHILD(stree->headnode);
      SETCHILD(stree->headnode,newleaf);
    } else
    {
      if(ISLEAF(stree->insertprev))   // previous node is leaf
      {
        ptr = stree->leaftab + GETLEAFINDEX(stree->insertprev);
        *(stree->nextfreeleafptr) = LEAFBROTHERVAL(*ptr);
        SETLEAFBROTHER(ptr,newleaf);
      } else   // previous node is branching node
      {
        ptr = stree->branchtab + GETBRANCHINDEX(stree->insertprev);
        *(stree->nextfreeleafptr) = GETBROTHER(ptr);
        SETBROTHER(ptr,newleaf);
      }
    }
  }
  RECALLSUCC(newleaf);     // recall node on successor path of \emph{headnode}
  stree->nextfreeleafnum++;
  stree->nextfreeleafptr++;
}

/*
  The function \emph{insertbranch} inserts a branching node and splits
  the appropriate edges, according to the canonical location of the current
  head. \emph{insertprev} refers to the node to the left of the branching
  node to be inserted. If the branching node is the first child, then 
  \emph{insertprev} is \texttt{UNDEFINEDREFERENCE}. The edge to be split ends
  in the node referred to by \emph{insertnode}.
*/

static void insertbranchnode(Suffixtree *stree)
{
  Uint *ptr, *insertnodeptr, *insertleafptr, insertnodeptrbrother; 

  DEBUGDEFAULT;
  spaceforbranchtab(stree);
  if(stree->headnodedepth == 0)      // head is the root
  {
    stree->rootchildren[(Uint) *(stree->headstart)] 
      = MAKEBRANCHADDR(stree->nextfreebranchnum);
    *(stree->nextfreebranch+1) = VALIDINIT;
    DEBUG2(4,"%c-edge from root points to branch node with address %lu\n",
           *(stree->headstart),(Showuint) stree->nextfreebranchnum);
  } else
  {
    if(stree->insertprev == UNDEFINEDREFERENCE)  // new branch = first child
    {
      SETCHILD(stree->headnode,MAKEBRANCHADDR(stree->nextfreebranchnum));
    } else
    {
      if(ISLEAF(stree->insertprev))  // new branch = right brother of leaf
      {
        ptr = stree->leaftab + GETLEAFINDEX(stree->insertprev);
        SETLEAFBROTHER(ptr,MAKEBRANCHADDR(stree->nextfreebranchnum));
      } else                     // new branch = brother of branching node
      {
        SETBROTHER(stree->branchtab + GETBRANCHINDEX(stree->insertprev),
                   MAKEBRANCHADDR(stree->nextfreebranchnum));
      }
    }
  }
  if(ISLEAF(stree->insertnode))   // split edge is leaf edge
  {
    DEBUGCODE(1,stree->splitleafedge++);
    insertleafptr = stree->leaftab + GETLEAFINDEX(stree->insertnode);
    if (stree->tailptr == stree->sentinel || 
        *(stree->headend+1) < *(stree->tailptr)) 
    {
      SETNEWCHILDBROTHER(MAKELARGE(stree->insertnode),  // first child=oldleaf
                         LEAFBROTHERVAL(*insertleafptr));  // inherit brother
      RECALLNEWLEAFADDRESS(stree->nextfreeleafptr);
      SETLEAFBROTHER(insertleafptr,                     // new leaf =
                     MAKELEAF(stree->nextfreeleafnum)); // right brother of old leaf
    } else
    {
      SETNEWCHILDBROTHER(MAKELARGELEAF(stree->nextfreeleafnum),  // first child=new leaf
                         LEAFBROTHERVAL(*insertleafptr));  // inherit brother
      *(stree->nextfreeleafptr) = stree->insertnode;  // old leaf = right brother of of new leaf
      RECALLLEAFADDRESS(insertleafptr);
    }
  } else  // split edge leads to branching node
  {
    DEBUGCODE(1,stree->splitinternaledge++);
    insertnodeptr = stree->branchtab + GETBRANCHINDEX(stree->insertnode);
    insertnodeptrbrother = GETBROTHER(insertnodeptr);
    if (stree->tailptr == stree->sentinel || 
        *(stree->headend+1) < *(stree->tailptr)) 
    {
      SETNEWCHILDBROTHER(MAKELARGE(stree->insertnode), // first child new branch
                         insertnodeptrbrother);        // inherit right brother
      RECALLNEWLEAFADDRESS(stree->nextfreeleafptr);
      SETBROTHER(insertnodeptr,MAKELEAF(stree->nextfreeleafnum)); // new leaf = brother of old branch
    } else
    {
      SETNEWCHILDBROTHER(MAKELARGELEAF(stree->nextfreeleafnum), // first child is new leaf
                         insertnodeptrbrother);        // inherit brother
      *(stree->nextfreeleafptr) = stree->insertnode;   // new branch is brother of new leaf
      RECALLBRANCHADDRESS(insertnodeptr);
    }
  }
  SETNILBIT;
  RECALLSUCC(MAKEBRANCHADDR(stree->nextfreebranchnum)); // node on succ. path
  stree->currentdepth = stree->headnodedepth + (Uint) (stree->headend-stree->headstart+1);
  SETDEPTHHEADPOS(stree->currentdepth,stree->nextfreeleafnum);
  SETMAXBRANCHDEPTH(stree->currentdepth);
  stree->nextfreeleafnum++;
  stree->nextfreeleafptr++;
  DEBUGCODE(1,stree->nodecount++);
}

//\subsection{Finding the Head-Locations}

/*
  The function \emph{rescan} finds the location of the current head.
  In order to scan down the tree, it suffices to look at the first 
  character of each edge.
*/

static void rescan (Suffixtree *stree)
{
  Uint *nodeptr, *largeptr = NULL, distance = 0, node, prevnode, 
       nodedepth, edgelen, wlen, leafindex, headposition;
  SYMBOL headchar, edgechar;

  DEBUGDEFAULT;
  if(stree->headnodedepth == 0)   // head is the root
  {
    headchar = *(stree->headstart);  // headstart is assumed to be not empty
    node = stree->rootchildren[(Uint) headchar];
    if(ISLEAF(node))   // stop if successor is leaf
    {
      stree->insertnode = node;
      return;
    } 
    nodeptr = stree->branchtab + GETBRANCHINDEX(node);
    GETONLYDEPTH(nodedepth,nodeptr);
    wlen = (Uint) (stree->headend - stree->headstart + 1);
    if(nodedepth > wlen)    // cannot reach the successor node
    {
      stree->insertnode = node;
      return;
    }
    stree->headnode = nodeptr;        // go to successor node
    stree->headnodedepth = nodedepth;
    if(nodedepth == wlen)             // location has been scanned
    {
      stree->headend = NULL;
      return;
    }
    (stree->headstart) += nodedepth;
  }
  while(True)   // \emph{headnode} is not the root
  {
    headchar = *(stree->headstart);  // \emph{headstart} is assumed to be nonempty
    prevnode = UNDEFINEDREFERENCE;
    node = GETCHILD(stree->headnode);
    while(True)             // traverse the list of successors
    {
      if(ISLEAF(node))   // successor is leaf
      {
        leafindex = GETLEAFINDEX(node);
        edgechar = stree->text[stree->headnodedepth + leafindex];
        if(edgechar == headchar)    // correct edge found
        {
          stree->insertnode = node;
          stree->insertprev = prevnode;
          return;
        }
        prevnode = node;
        node = LEAFBROTHERVAL(stree->leaftab[leafindex]);  
      } else   // successor is branch node
      {
        nodeptr = stree->branchtab + GETBRANCHINDEX(node);
        GETONLYHEADPOS(headposition,nodeptr);
        edgechar = stree->text[stree->headnodedepth + headposition];
        if(edgechar == headchar) // correct edge found
        {
          break;
        } 
        prevnode = node;
        node = GETBROTHER(nodeptr);
      }
    }

    GETDEPTHAFTERHEADPOS(nodedepth,nodeptr);     // get info about succ node
    edgelen = nodedepth - stree->headnodedepth;
    wlen = (Uint) (stree->headend - stree->headstart + 1);
    if(edgelen > wlen)     // cannot reach the succ node
    {
      stree->insertnode = node;
      stree->insertprev = prevnode;
      return;
    }
    stree->headnode = nodeptr;    // go to the successor node
    stree->headnodedepth = nodedepth;
    if(edgelen == wlen)                    // location is found
    {
      stree->headend = NULL;
      return;
    }
    (stree->headstart) += edgelen;
  }
}

/*
  The function \emph{taillcp} computes the length of the longest common prefix
  of two strings. The first string is between pointers \emph{start1} and 
  \emph{end1}. The second string is the current tail, which is between  the
  pointers \emph{tailptr} and \emph{sentinel}.
*/

static Uint taillcp(Suffixtree *stree,SYMBOL *start1, SYMBOL *end1)
{
  SYMBOL *ptr1 = start1, *ptr2 = stree->tailptr + 1;
  while(ptr1 <= end1 && ptr2 < stree->sentinel && *ptr1 == *ptr2)
  {
    ptr1++;
    ptr2++;
  }
  return (Uint) (ptr1-start1);
}

/*
 The function \emph{scanprefix} scans a prefix of the current tail 
 down from a given node.
*/

static void scanprefix(Suffixtree *stree)
{
  Uint *nodeptr = NULL, *largeptr = NULL, leafindex, nodedepth, edgelen, node, 
       distance = 0, prevnode, prefixlen, headposition;
  SYMBOL *leftborder = (SYMBOL *) NULL, tailchar, edgechar = 0;

  DEBUGDEFAULT;
  if(stree->headnodedepth == 0)   // headnode is root
  {
    if(stree->tailptr == stree->sentinel)   // there is no \$-edge
    {
      stree->headend = NULL;
      return;
    }
    tailchar = *(stree->tailptr);
    if((node = stree->rootchildren[(Uint) tailchar]) == UNDEFINEDREFERENCE)
    {
      stree->headend = NULL;   
      return;
    }
    if(ISLEAF(node)) // successor edge is leaf, compare tail and leaf edge label
    {
      leftborder = stree->text + GETLEAFINDEX(node);
      prefixlen = 1 + taillcp(stree,leftborder+1,stree->sentinel-1);
      (stree->tailptr) += prefixlen;
      stree->headstart = leftborder;
      stree->headend = leftborder + (prefixlen-1);
      stree->insertnode = node;
      return;
    }
    nodeptr = stree->branchtab + GETBRANCHINDEX(node);
    GETBOTH(nodedepth,headposition,nodeptr);  // get info for branch node
    leftborder = stree->text + headposition;
    prefixlen = 1 + taillcp(stree,leftborder+1,leftborder + nodedepth - 1);
    (stree->tailptr)+= prefixlen;
    if(nodedepth > prefixlen)   // cannot reach the successor, fall out of tree
    {
      stree->headstart = leftborder;
      stree->headend = leftborder + (prefixlen-1);
      stree->insertnode = node;
      return;
    }
    stree->headnode = nodeptr;
    stree->headnodedepth = nodedepth;
  }
  while(True)  // \emph{headnode} is not the root
  {
    prevnode = UNDEFINEDREFERENCE;
    node = GETCHILD(stree->headnode);
    if(stree->tailptr == stree->sentinel)  //  \$-edge
    {
      do // there is no \$-edge, so find last successor, of which it becomes right brother
      {
        prevnode = node; 
        if(ISLEAF(node))
        {
          node = LEAFBROTHERVAL(stree->leaftab[GETLEAFINDEX(node)]);
        } else
        {
          node = GETBROTHER(stree->branchtab + GETBRANCHINDEX(node));
        }
      } while(!NILPTR(node));
      stree->insertnode = NILBIT;
      stree->insertprev = prevnode;
      stree->headend = NULL;
      return;
    } 
    tailchar = *(stree->tailptr);

    do // find successor edge with firstchar = tailchar
    {
      if(ISLEAF(node))   // successor is leaf
      {
        leafindex = GETLEAFINDEX(node);
        leftborder = stree->text + (stree->headnodedepth + leafindex);
        if((edgechar = *leftborder) >= tailchar)   // edge will not come later
        {
          break;
        }
        prevnode = node;
        node = LEAFBROTHERVAL(stree->leaftab[leafindex]);
      } else  // successor is branch node
      {
        nodeptr = stree->branchtab + GETBRANCHINDEX(node);
        GETONLYHEADPOS(headposition,nodeptr);
        leftborder = stree->text + (stree->headnodedepth + headposition);
        if((edgechar = *leftborder) >= tailchar)  // edge will not come later
        {
          break;
        }
        prevnode = node;
        node = GETBROTHER(nodeptr);
      }
    } while(!NILPTR(node));
    if(NILPTR(node) || edgechar > tailchar)  // edge not found
    {
      stree->insertprev = prevnode;   // new edge will become brother of this
      stree->headend = NULL;
      return;
    } 
    if(ISLEAF(node))  // correct edge is leaf edge, compare its label with tail
    {
      prefixlen = 1 + taillcp(stree,leftborder+1,stree->sentinel-1);
      (stree->tailptr) += prefixlen;
      stree->headstart = leftborder;
      stree->headend = leftborder + (prefixlen-1);
      stree->insertnode = node;
      stree->insertprev = prevnode;
      return;
    }
    GETDEPTHAFTERHEADPOS(nodedepth,nodeptr); // we already know headposition
    edgelen = nodedepth - stree->headnodedepth;
    prefixlen = 1 + taillcp(stree,leftborder+1,leftborder + edgelen - 1);
    (stree->tailptr) += prefixlen;
    if(edgelen > prefixlen)  // cannot reach next node
    {
      stree->headstart = leftborder;
      stree->headend = leftborder + (prefixlen-1);
      stree->insertnode = node;
      stree->insertprev = prevnode;
      return;
    } 
    stree->headnode = nodeptr;
    stree->headnodedepth = nodedepth;
  }
}

//\subsection{Completion and Initialization}

/*
  The function \emph{completelarge} is called whenever a large node 
  is inserted. It basically sets the appropriate distance values of the small
  nodes of the current chain.
*/

static void completelarge(Suffixtree *stree)
{
  Uint distance, *backwards;

  DEBUGDEFAULT;
  if(stree->smallnotcompleted > 0)
  {
    backwards = stree->nextfreebranch;
    for(distance = 1; distance <= stree->smallnotcompleted; distance++)
    {
      backwards -= SMALLINTS;
      SETDISTANCE(backwards,distance);
    }
    stree->smallnotcompleted = 0;
    stree->chainstart = NULL;
  }
  stree->nextfreebranch += LARGEINTS;
  stree->nextfreebranchnum += LARGEINTS;
  stree->largenode++;
}

/*
  The function \emph{linkrootchildren} constructs the successor chain
  for the children of the root. This is done at the end of the algorithm
  in one sweep over table \emph{rootchildren}.
*/

static void linkrootchildren(Suffixtree *stree)
{
  Uint *rcptr, *prevnodeptr, prev = UNDEFINEDREFERENCE;

  DEBUGDEFAULT;
  stree->alphasize = 0;
  for(rcptr = stree->rootchildren; 
      rcptr <= stree->rootchildren + LARGESTCHARINDEX; rcptr++)
  {
    if(*rcptr != UNDEFINEDREFERENCE)
    {
      stree->alphasize++;
      if(prev == UNDEFINEDREFERENCE)
      {
        SETCHILD(stree->branchtab,MAKELARGE(*rcptr));
      } else
      {
        if(ISLEAF(prev))
        {
          stree->leaftab[GETLEAFINDEX(prev)] = *rcptr;
        } else
        {
          prevnodeptr = stree->branchtab + GETBRANCHINDEX(prev);
          SETBROTHER(prevnodeptr,*rcptr);
        }
      }
      prev = *rcptr;
    }
  }
  if(ISLEAF(prev))
  {
    stree->leaftab[GETLEAFINDEX(prev)] = MAKELEAF(stree->textlen);
  } else
  {
    prevnodeptr = stree->branchtab + GETBRANCHINDEX(prev);
    SETBROTHER(prevnodeptr,MAKELEAF(stree->textlen));
  }
  stree->leaftab[stree->textlen] = NILBIT;
}

/*
  \newpage
  \emph{initSuffixtree} allocates and initializes the data structures for 
  McCreight's Algorithm.
*/

static void initSuffixtree(Suffixtree *stree,SYMBOL *text,Uint textlen)
{
  Uint i, *ptr;

  DEBUG1(4,">%s\n",__func__);
  DEBUG1(2,"# MULTBYSMALLINTS(textlen+1)=%lu\n",
           (Showuint) MULTBYSMALLINTS(textlen+1));
  DEBUG1(2,"# STARTFACTOR=%.2f\n",STARTFACTOR);
  DEBUG1(2,"# MINEXTRA=%lu\n",(Showuint) MINEXTRA);
  stree->currentbranchtabsize 
    = (Uint) (STARTFACTOR * MULTBYSMALLINTS(textlen+1));
  if(stree->currentbranchtabsize < MINEXTRA)
  {
    stree->currentbranchtabsize = MULTBYSMALLINTS(MINEXTRA);
  }
  stree->leaftab = ALLOCSPACE_TLB(NULL,Uint,textlen+2);
  stree->rootchildren = ALLOCSPACE_TLB(NULL,Uint,LARGESTCHARINDEX + 1);
  stree->branchtab = ALLOCSPACE_TLB(NULL,Uint,stree->currentbranchtabsize);

  stree->text = stree->tailptr = text;
  stree->textlen = textlen;
  stree->sentinel = text + textlen;
  stree->firstnotallocated 
    = stree->branchtab + stree->currentbranchtabsize - LARGEINTS;
  stree->headnode = stree->nextfreebranch = stree->branchtab;
  stree->headend = NULL;
  stree->headnodedepth = stree->maxbranchdepth = 0;
  for(ptr=stree->rootchildren; ptr<=stree->rootchildren+LARGESTCHARINDEX; ptr++)
  {
    *ptr = UNDEFINEDREFERENCE;
  }
  for(i=0; i<LARGEINTS; i++)
  {
    stree->branchtab[i] = 0;
  }
  stree->nextfreebranch = stree->branchtab;
  stree->nextfreebranchnum = 0;
  SETDEPTHHEADPOS(0,0);
  SETNEWCHILDBROTHER(MAKELARGELEAF(0),0);
  SETBRANCHNODEOFFSET;
  stree->rootchildren[(Uint) *text] = MAKELEAF(0);
  stree->leaftab[0] = VALIDINIT;
  DEBUG2(4,"%c-edge from root points to leaf %lu\n",*text,(Showuint) 0);
  stree->leafcounts = NULL;
  stree->nextfreeleafnum = 1;
  stree->nextfreeleafptr = stree->leaftab + 1;
  stree->nextfreebranch = stree->branchtab + LARGEINTS;
  stree->nextfreebranchnum = LARGEINTS;
  stree->insertnode = stree->insertprev = UNDEFINEDREFERENCE;
  stree->smallnotcompleted = 0;
  stree->chainstart = NULL;
  stree->largenode = stree->smallnode = 0;

//\Ignore{

#ifdef ST_DEBUG
  stree->showsymbolstree = NULL;
  stree->alphabet = NULL;
  stree->nodecount = 1;
  stree->splitleafedge = 
  stree->splitinternaledge = 
  stree->artificial = 
  stree->multiplications = 0;
  stree->insertleafcalls = 1;
  stree->maxset = stree->branchtab + LARGEINTS - 1;
  stree->largelinks = stree->largelinkwork = stree->largelinklinkwork = 0;
#endif

//}

}

void freestree(Suffixtree *stree)
{
  FREESPACE_TLB(stree->leaftab);
  FREESPACE_TLB(stree->rootchildren);
  FREESPACE_TLB(stree->branchtab);
  if(stree->nonmaximal != NULL)
  {
    FREESPACE_TLB(stree->nonmaximal);
  }
  if(stree->leafcounts != NULL)
  {
    FREESPACE_TLB(stree->leafcounts);
  }
  // Added by TLB
  if(stree->subtreedataindex != NULL)
  {
    FREESPACE_TLB(stree->subtreedataindex);
    FREESPACE_TLB(stree->leafdataindex);
  }
}

//\subsection{Computing the Suffix Tree}

/*
  \emph{constructstree} implements McCreight Algorithm to compute the suffix 
  tree for a \texttt{text} of length \texttt{textlen}. For explanations, see
  \cite{KUR:1998}. The number \((i)\) refers to the cases of Section 6 in 
  \cite{KUR:1998}.
*/

#define CONSTRUCT Sint constructstree(Suffixtree *stree,SYMBOL *text,Uint textlen)
#define DECLAREEXTRA        stree->nonmaximal = NULL
#define COMPLETELARGEFIRST  completelarge(stree)
#define COMPLETELARGESECOND completelarge(stree)
#define PROCESSHEAD         /* Nothing */
#define CHECKSTEP           /* Nothing */
#define FINALPROGRESS       /* Nothing */

#include "st_construct_gen.h"

#undef CONSTRUCT
#undef DECLAREEXTRA
#undef COMPLETELARGEFIRST
#undef COMPLETELARGESECOND
#undef PROCESSHEAD 
#undef CHECKSTEP
#undef FINALPROGRESS

/* --------------------------------------------------- */

#define CONSTRUCT Sint constructmarkmaxstree(Suffixtree *stree,SYMBOL *text,Uint textlen)

#define DECLAREEXTRA  Uint distance, headposition = 0, *largeptr,\
                           tabsize = 1 + DIVWORDSIZE(textlen+1), *tabptr;\
                      stree->nonmaximal = ALLOCSPACE_TLB(NULL,Uint,tabsize);\
                      for(tabptr = stree->nonmaximal;\
                          tabptr < stree->nonmaximal + tabsize;\
                          tabptr++)\
                          {\
                            *tabptr = 0;\
                          }

#define COMPLETELARGEFIRST\
        GETONLYHEADPOS(headposition,stree->headnode);\
        DEBUG2(3,"j=%lu: slink->%lu\n",(Showuint) stree->nextfreeleafnum,\
                                       (Showuint) headposition);\
        SETIBIT(stree->nonmaximal,headposition);\
        completelarge(stree);

#define COMPLETELARGESECOND\
        DEBUG2(3,"j=%lu: slink->%lu\n",(Showuint) stree->nextfreeleafnum,\
                                       (Showuint) stree->nextfreeleafnum);\
        SETIBIT(stree->nonmaximal,stree->nextfreeleafnum);\
        completelarge(stree)

#define PROCESSHEAD          /* Nothing */
#define CHECKSTEP            /* Nothing */
#define FINALPROGRESS        /* Nothing */

#include "st_construct_gen.h"

#undef CONSTRUCT
#undef DECLAREEXTRA
#undef COMPLETELARGEFIRST
#undef COMPLETELARGESECOND
#undef PROCESSHEAD 
#undef CHECKSTEP
#undef FINALPROGRESS

/* --------------------------------------------------- */

#define CONSTRUCT Sint constructheadstree(Suffixtree *stree,SYMBOL *text,Uint textlen,void(*processhead)(Suffixtree *,Uint,void *),void *processheadinfo)

#define DECLAREEXTRA         stree->nonmaximal = NULL
#define COMPLETELARGEFIRST   completelarge(stree)
#define COMPLETELARGESECOND  completelarge(stree)
#define PROCESSHEAD          processhead(stree,stree->nextfreeleafnum,\
                                               processheadinfo)

#define CHECKSTEP            /* Nothing */
#define FINALPROGRESS        /* Nothing */

#include "st_construct_gen.h"

#undef CONSTRUCT
#undef DECLAREEXTRA
#undef COMPLETELARGEFIRST
#undef COMPLETELARGESECOND
#undef PROCESSHEAD 
#undef CHECKSTEP
#undef FINALPROGRESS

/* --------------------------------------------------- */

#define LEASTSHOWPROGRESS 100000
#define NUMOFCALLS 100

#define CONSTRUCT Sint constructprogressstree(Suffixtree *stree,SYMBOL *text,Uint textlen,void (*progress)(Uint,void *),void (*finalprogress)(void *),void *info)
#define DECLAREEXTRA\
        Uint j = 0, step, nextstep;\
        stree->nonmaximal = NULL;\
        step = textlen/NUMOFCALLS;\
        nextstep = (textlen >= LEASTSHOWPROGRESS) ? step : (textlen+1);\
        if(progress == NULL && textlen >= LEASTSHOWPROGRESS)\
        {\
          fprintf(stderr,"# process %lu characters per dot\n",\
                 (Showuint) textlen/NUMOFCALLS);\
        }

#define COMPLETELARGEFIRST  completelarge(stree)
#define COMPLETELARGESECOND completelarge(stree)
#define PROCESSHEAD          /* Nothing */

#define CHECKSTEP            if(j == nextstep)\
                             {\
                               if(progress == NULL)\
                               {\
                                 if(nextstep == step)\
                                 {\
                                   fputc('#',stderr);\
                                 }\
                                 fputc('.',stderr);\
                                 fflush(stdout);\
                               } else\
                               {\
                                 progress(nextstep,info);\
                               }\
                               nextstep += step;\
                             }\
                             j++

#define FINALPROGRESS        if(textlen >= LEASTSHOWPROGRESS)\
                             {\
                               if(finalprogress == NULL)\
                               {\
                                 fputc('\n',stderr);\
                               } else\
                               {\
                                 finalprogress(info);\
                               }\
                             }

#include "st_construct_gen.h"

#undef CONSTRUCT
#undef DECLAREEXTRA
#undef COMPLETELARGEFIRST
#undef COMPLETELARGESECOND
#undef PROCESSHEAD 
#undef CHECKSTEP
#undef FINALPROGRESS