C++ compilation.
[idzebra-moved-to-github.git] / index / retrieve.c
index 2bdadcf..00d1d9e 100644 (file)
@@ -1,10 +1,34 @@
 /*
- * Copyright (C) 1995-1998, Index Data
+ * Copyright (C) 1995-1999, Index Data
  * All rights reserved.
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: retrieve.c,v $
- * Revision 1.2  1998-10-16 08:14:33  adam
+ * Revision 1.10  1999-05-26 07:49:13  adam
+ * C++ compilation.
+ *
+ * Revision 1.9  1999/05/20 12:57:18  adam
+ * Implemented TCL filter. Updated recctrl system.
+ *
+ * Revision 1.8  1999/03/09 16:27:49  adam
+ * More work on SDRKit integration.
+ *
+ * Revision 1.7  1999/03/02 16:15:43  quinn
+ * Added "tagsysno" and "tagrank" directives to zebra.cfg.
+ *
+ * Revision 1.6  1999/02/18 15:01:25  adam
+ * Minor changes.
+ *
+ * Revision 1.5  1999/02/17 11:29:56  adam
+ * Fixed in record_fetch. Minor updates to API.
+ *
+ * Revision 1.4  1999/02/02 14:51:07  adam
+ * Updated WIN32 code specific sections. Changed header.
+ *
+ * Revision 1.3  1998/10/28 10:54:40  adam
+ * SDRKit integration.
+ *
+ * Revision 1.2  1998/10/16 08:14:33  adam
  * Updated record control system.
  *
  * Revision 1.1  1998/03/05 08:45:13  adam
@@ -17,7 +41,7 @@
 #include <assert.h>
 
 #include <fcntl.h>
-#ifdef WINDOWS
+#ifdef WIN32
 #include <io.h>
 #include <process.h>
 #else
 #include <recctrl.h>
 #include "zserver.h"
 
+#ifndef ZEBRASDR
+#define ZEBRASDR 0
+#endif
+
+#if ZEBRASDR
+#include "zebrasdr.h"
+#endif
+
 struct fetch_control {
     int record_offset;
     int record_int_pos;
@@ -37,41 +69,41 @@ struct fetch_control {
 
 static int record_ext_read (void *fh, char *buf, size_t count)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     return read (fc->fd, buf, count);
 }
 
 static off_t record_ext_seek (void *fh, off_t offset)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     return lseek (fc->fd, offset + fc->record_offset, SEEK_SET);
 }
 
 static off_t record_ext_tell (void *fh)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     return lseek (fc->fd, 0, SEEK_CUR) - fc->record_offset;
 }
 
 static off_t record_int_seek (void *fh, off_t offset)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     return (off_t) (fc->record_int_pos = offset);
 }
 
 static off_t record_int_tell (void *fh)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     return (off_t) fc->record_int_pos;
 }
 
 static int record_int_read (void *fh, char *buf, size_t count)
 {
-    struct fetch_control *fc = fh;
+    struct fetch_control *fc = (struct fetch_control *) fh;
     int l = fc->record_int_len - fc->record_int_pos;
     if (l <= 0)
         return 0;
-    l = (l < count) ? l : count;
+    l = (l < (int) count) ? l : count;
     memcpy (buf, fc->record_int_buf + fc->record_int_pos, l);
     fc->record_int_pos += l;
     return l;
@@ -89,6 +121,7 @@ int zebra_record_fetch (ZebraHandle zh, int sysno, int score, ODR stream,
     char subType[128];
     struct fetch_control fc;
     RecordAttr *recordAttr;
+    void *clientData;
 
     rec = rec_get (zh->records, sysno);
     if (!rec)
@@ -101,10 +134,10 @@ int zebra_record_fetch (ZebraHandle zh, int sysno, int score, ODR stream,
     file_type = rec->info[recInfo_fileType];
     fname = rec->info[recInfo_filename];
     basename = rec->info[recInfo_databaseName];
-    *basenamep = odr_malloc (stream, strlen(basename)+1);
+    *basenamep = (char *) odr_malloc (stream, strlen(basename)+1);
     strcpy (*basenamep, basename);
 
-    if (!(rt = recType_byName (zh->recTypes, file_type, subType)))
+    if (!(rt = recType_byName (zh->recTypes, file_type, subType, &clientData)))
     {
         logf (LOG_WARN, "Retrieve: Cannot handle type %s",  file_type);
        return 14;
@@ -122,7 +155,72 @@ int zebra_record_fetch (ZebraHandle zh, int sysno, int score, ODR stream,
         fc.record_int_pos = 0;
         logf (LOG_DEBUG, "Internal retrieve. %d bytes", fc.record_int_len);
     }
-    else 
+#if ZEBRASDR
+    else if (*fname == '%')
+    {  
+       ZebraSdrHandle h;
+       int segment = 0, r;
+       char *cp, xname[128];
+       unsigned char *buf;
+
+       logf (LOG_DEBUG, "SDR");
+       strcpy (xname, fname+1);
+       if ((cp = strrchr (xname, '.')))
+       {
+           *cp++ = 0;
+           segment = atoi(cp);
+       }
+       h = zebraSdr_open (xname);
+       if (!h)
+       {
+           logf (LOG_WARN, "sdr open %s", xname);
+           return 0;
+       }
+        if (zebraSdr_segment (h, &segment) < 0)
+       {
+           logf (LOG_WARN, "zebraSdr_segment fail segment=%d",
+               segment);
+            rec_rm (&rec);
+            return 14;
+       }    
+        r = zebraSdr_read (h, &buf);
+       if (r < 1)
+       {
+           logf (LOG_WARN, "zebraSdr_read fail segment=%d",
+               segment);
+            rec_rm (&rec);
+            return 14;
+       }
+       zebraSdr_close (h);
+
+        fc.record_int_len = recordAttr->recordSize;
+        fc.record_int_buf = buf + recordAttr->recordOffset;
+        fc.record_int_pos = 0;
+
+       logf (LOG_LOG, "segment = %d len=%d off=%d",
+           segment,
+           recordAttr->recordSize,
+           recordAttr->recordOffset);
+       if (fc.record_int_len > 180)
+       {
+           logf (LOG_LOG, "%.70s", fc.record_int_buf);
+           logf (LOG_LOG, "%.70s", fc.record_int_buf +
+               (fc.record_int_len - 70));
+       }
+       else
+           logf (LOG_LOG, "%.*s",
+               fc.record_int_len, fc.record_int_buf);
+
+       /* the following two lines makes rec_rm delete buf */
+        rec->size[recInfo_storeData] = r;
+        rec->info[recInfo_storeData] = buf;
+
+        retrieveCtrl.readf = record_int_read;
+        retrieveCtrl.seekf = record_int_seek;
+        retrieveCtrl.tellf = record_int_tell;
+    }
+#endif
+    else
     {
         if ((fc.fd = open (fname, O_BINARY|O_RDONLY)) == -1)
         {
@@ -148,9 +246,10 @@ int zebra_record_fetch (ZebraHandle zh, int sysno, int score, ODR stream,
     retrieveCtrl.comp = comp;
     retrieveCtrl.diagnostic = 0;
     retrieveCtrl.dh = zh->dh;
-    (*rt->retrieve)(&retrieveCtrl);
+    retrieveCtrl.res = zh->res;
+    (*rt->retrieve)(clientData, &retrieveCtrl);
     *output_format = retrieveCtrl.output_format;
-    *rec_bufp = retrieveCtrl.rec_buf;
+    *rec_bufp = (char *) retrieveCtrl.rec_buf;
     *rec_lenp = retrieveCtrl.rec_len;
     if (fc.fd != -1)
         close (fc.fd);