Work on character-sets
[idzebra-moved-to-github.git] / index / kinput.c
index e152c75..92833bb 100644 (file)
@@ -1,10 +1,30 @@
 /*
- * Copyright (C) 1994-1995, Index Data I/S 
+ * Copyright (C) 1994-1996, Index Data I/S 
  * All rights reserved.
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: kinput.c,v $
- * Revision 1.11  1995-12-06 16:06:43  adam
+ * Revision 1.17  1996-05-14 15:47:07  adam
+ * Cleanup of various buffer size entities.
+ *
+ * Revision 1.16  1996/04/09  10:05:20  adam
+ * Bug fix: prev_name buffer possibly too small; allocated in key_file_init.
+ *
+ * Revision 1.15  1996/03/21  14:50:09  adam
+ * File update uses modify-time instead of change-time.
+ *
+ * Revision 1.14  1996/02/07  14:06:37  adam
+ * Better progress report during register merge.
+ * New command: clean - removes temporary shadow files.
+ *
+ * Revision 1.13  1996/02/05  12:30:00  adam
+ * Logging reduced a bit.
+ * The remaining running time is estimated during register merge.
+ *
+ * Revision 1.12  1995/12/06  17:49:19  adam
+ * Uses dict_delete now.
+ *
+ * Revision 1.11  1995/12/06  16:06:43  adam
  * Better diagnostics. Work on 'real' dictionary deletion.
  *
  * Revision 1.10  1995/12/06  12:41:22  adam
@@ -55,7 +75,7 @@
 #include "index.h"
 
 #define KEY_SIZE (1+sizeof(struct it_key))
-#define INP_NAME_MAX 8192
+#define INP_NAME_MAX 768
 #define INP_BUF_START 60000
 #define INP_BUF_ADD  400000
 
@@ -75,20 +95,36 @@ struct key_file {
     char *prev_name;     /* last word read */
     int   sysno;         /* last sysno */
     int   seqno;         /* last seqno */
+    off_t length;        /* length of file */
+                         /* handler invoked in each read */
+    void (*readHandler)(struct key_file *keyp, void *rinfo);
+    void *readInfo;
 };
 
+void getFnameTmp (char *fname, int no)
+{
+    sprintf (fname, TEMP_FNAME, no);
+}
+
 void key_file_chunk_read (struct key_file *f)
 {
     int nr = 0, r, fd;
-    char fname[256];
-    sprintf (fname, TEMP_FNAME, f->no);
+    char fname[1024];
+    getFnameTmp (fname, f->no);
     fd = open (fname, O_RDONLY);
     if (fd == -1)
     {
         logf (LOG_FATAL|LOG_ERRNO, "cannot open %s", fname);
         exit (1);
     }
-    logf (LOG_LOG, "reading chunk from %s", fname);
+    if (!f->length)
+    {
+        if ((f->length = lseek (fd, 0L, SEEK_END)) == (off_t) -1)
+        {
+            logf (LOG_FATAL|LOG_ERRNO, "cannot seek %s", fname);
+            exit (1);
+        }
+    }
     if (lseek (fd, f->offset, SEEK_SET) == -1)
     {
         logf (LOG_FATAL|LOG_ERRNO, "cannot seek %s", fname);
@@ -108,6 +144,8 @@ void key_file_chunk_read (struct key_file *f)
     }
     f->buf_size = nr;
     f->buf_ptr = 0;
+    if (f->readHandler)
+        (*f->readHandler)(f, f->readInfo);
     close (fd);
 }
 
@@ -121,8 +159,10 @@ struct key_file *key_file_init (int no, int chunk)
     f->no = no;
     f->chunk = chunk;
     f->offset = 0;
+    f->length = 0;
+    f->readHandler = NULL;
     f->buf = xmalloc (f->chunk);
-    f->prev_name = xmalloc (256);
+    f->prev_name = xmalloc (INP_NAME_MAX);
     *f->prev_name = '\0';
     key_file_chunk_read (f);
     return f;
@@ -231,7 +271,7 @@ struct heap_info *key_heap_init (int nkeys,
     for (i = 0; i<= nkeys; i++)
     {
         hi->ptr[i] = i;
-        hi->info.buf[i] = xmalloc (512);
+        hi->info.buf[i] = xmalloc (INP_NAME_MAX);
     }
     return hi;
 }
@@ -293,7 +333,7 @@ static void key_heap_insert (struct heap_info *hi, const char *buf, int nbytes,
 static int heap_read_one (struct heap_info *hi, char *name, char *key)
 {
     int n, r;
-    char rbuf[512];
+    char rbuf[INP_NAME_MAX];
     struct key_file *kf;
 
     if (!hi->heapnum)
@@ -313,8 +353,8 @@ static int heap_read_one (struct heap_info *hi, char *name, char *key)
 int heap_inp (Dict dict, ISAM isam, struct heap_info *hi)
 {
     char *info;
-    char next_name[INP_NAME_MAX+1];
-    char cur_name[INP_NAME_MAX+1];
+    char next_name[INP_NAME_MAX];
+    char cur_name[INP_NAME_MAX];
     int key_buf_size = INP_BUF_START;
     int key_buf_ptr;
     char *next_key;
@@ -322,7 +362,7 @@ int heap_inp (Dict dict, ISAM isam, struct heap_info *hi)
     int more;
     
     next_key = xmalloc (KEY_SIZE);
-    key_buf = xmalloc (key_buf_size * (KEY_SIZE));
+    key_buf = xmalloc (key_buf_size);
     more = heap_read_one (hi, cur_name, key_buf);
     while (more)                   /* EOF ? */
     {
@@ -355,14 +395,13 @@ int heap_inp (Dict dict, ISAM isam, struct heap_info *hi)
             logf (LOG_DEBUG, "updating %s", cur_name);
             memcpy (&isam_p, info+1, sizeof(ISAM_P));
             isam_p2 = is_merge (isam, isam_p, nmemb, key_buf);
-#if 0
             if (!isam_p2)
             {
                 no_deletions++;
-                dict_delete (dict, cur_name);
+                if (!dict_delete (dict, cur_name))
+                    abort ();
             }
             else 
-#endif
             {
                 no_updates++;
                 if (isam_p2 != isam_p)
@@ -383,6 +422,37 @@ int heap_inp (Dict dict, ISAM isam, struct heap_info *hi)
     return 0;
 }
 
+struct progressInfo {
+    time_t   startTime;
+    time_t   lastTime;
+    off_t    totalBytes;
+    off_t    totalOffset;
+};
+
+void progressFunc (struct key_file *keyp, void *info)
+{
+    struct progressInfo *p = info;
+    time_t now, remaining;
+
+    if (keyp->buf_size <= 0 || p->totalBytes <= 0)
+        return ;
+    time (&now);
+
+    if (now >= p->lastTime+10)
+    {
+        p->lastTime = now;
+        remaining = (now - p->startTime)*
+            ((double) p->totalBytes/p->totalOffset - 1.0);
+        if (remaining <= 130)
+            logf (LOG_LOG, "Merge %2.1f%% completed; %ld seconds remaining",
+                 (100.0*p->totalOffset) / p->totalBytes, (long) remaining);
+        else
+            logf (LOG_LOG, "Merge %2.1f%% completed; %ld minutes remaining",
+                (100.0*p->totalOffset) / p->totalBytes, (long) remaining/60);
+    }
+    p->totalOffset += keyp->buf_size;
+}
+
 void key_input (const char *dict_fname, const char *isam_fname,
                 int nkeys, int cache)
                 
@@ -393,7 +463,22 @@ void key_input (const char *dict_fname, const char *isam_fname,
     char rbuf[1024];
     int i, r;
     struct heap_info *hi;
+    struct progressInfo progressInfo;
 
+    if (nkeys < 0)
+    {
+        char fname[1024];
+        nkeys = 0;
+        while (1)
+        {
+            getFnameTmp (fname, nkeys+1);
+            if (access (fname, R_OK) == -1)
+                break;
+            nkeys++;
+        }
+        if (!nkeys)
+            return ;
+    }
     dict = dict_open (dict_fname, cache, 1);
     if (!dict)
     {
@@ -408,9 +493,18 @@ void key_input (const char *dict_fname, const char *isam_fname,
     }
 
     kf = xmalloc ((1+nkeys) * sizeof(*kf));
+    progressInfo.totalBytes = 0;
+    progressInfo.totalOffset = 0;
+    time (&progressInfo.startTime);
+    time (&progressInfo.lastTime);
     for (i = 1; i<=nkeys; i++)
+    {
         kf[i] = key_file_init (i, 32768);
-    
+        kf[i]->readHandler = progressFunc;
+        kf[i]->readInfo = &progressInfo;
+        progressInfo.totalBytes += kf[i]->length;
+        progressInfo.totalOffset += kf[i]->buf_size;
+    }
     hi = key_heap_init (nkeys, key_qsort_compare);
     for (i = 1; i<=nkeys; i++)
         if ((r = key_file_read (kf[i], rbuf)))
@@ -419,7 +513,12 @@ void key_input (const char *dict_fname, const char *isam_fname,
 
     dict_close (dict);
     is_close (isam);
-
+    
+    for (i = 1; i<=nkeys; i++)
+    {
+        getFnameTmp (rbuf, i);
+        unlink (rbuf);
+    }
     logf (LOG_LOG, "Iterations . . .%7d", no_iterations);
     logf (LOG_LOG, "Distinct words .%7d", no_diffs);
     logf (LOG_LOG, "Updates. . . . .%7d", no_updates);