contrib/pg_trgm/trgm.h

   1 #ifndef __TRGM_H__
   2 #define __TRGM_H__
   3
   4 #include "postgres.h"
   5
   6 #include "access/gist.h"
   7 #include "access/itup.h"
   8 #include "utils/elog.h"
   9 #include "utils/palloc.h"
  10 #include "utils/builtins.h"
  11 #include "storage/bufpage.h"
  12
  13 /* options */
  14 #define LPADDING                2
  15 #define RPADDING                1
  16 #define KEEPONLYALNUM
  17 #define IGNORECASE
  18 #define DIVUNION
  19
  20
  21 typedef char trgm[3];
  22
  23 #define CMPCHAR(a,b) ( ((a)==(b)) ? 0 : ( ((a)<(b)) ? -1 : 1 ) )
  24 #define CMPPCHAR(a,b,i)  CMPCHAR( *(((char*)(a))+i), *(((char*)(b))+i) )
  25 #define CMPTRGM(a,b) ( CMPPCHAR(a,b,0) ? CMPPCHAR(a,b,0) : ( CMPPCHAR(a,b,1) ? CMPPCHAR(a,b,1) : CMPPCHAR(a,b,2) ) )
  26
  27 #define CPTRGM(a,b) do {                                \
  28         *(((char*)(a))+0) = *(((char*)(b))+0);  \
  29         *(((char*)(a))+1) = *(((char*)(b))+1);  \
  30         *(((char*)(a))+2) = *(((char*)(b))+2);  \
  31 } while(0);
  32
  33
  34 typedef struct
  35 {
  36         int4            len;
  37         uint8           flag;
  38         char            data[1];
  39 }       TRGM;
  40
  41 #define TRGMHRDSIZE               (sizeof(int4)+sizeof(uint8))
  42
  43 /* gist */
  44 #define BITBYTE 8
  45 #define SIGLENINT  3                    /* >122 => key will toast, so very slow!!! */
  46 #define SIGLEN  ( sizeof(int)*SIGLENINT )
  47
  48 #define SIGLENBIT (SIGLEN*BITBYTE - 1)  /* see makesign */
  49
  50 typedef char BITVEC[SIGLEN];
  51 typedef char *BITVECP;
  52
  53 #define LOOPBYTE(a) \
  54                                 for(i=0;i<SIGLEN;i++) {\
  55                                                                 a;\
  56                                 }
  57
  58 #define LOOPBIT(a) \
  59                                 for(i=0;i<SIGLENBIT;i++) {\
  60                                                                 a;\
  61                                 }
  62
  63 #define GETBYTE(x,i) ( *( (BITVECP)(x) + (int)( (i) / BITBYTE ) ) )
  64 #define GETBITBYTE(x,i) ( ((char)(x)) >> i & 0x01 )
  65 #define CLRBIT(x,i)   GETBYTE(x,i) &= ~( 0x01 << ( (i) % BITBYTE ) )
  66 #define SETBIT(x,i)   GETBYTE(x,i) |=  ( 0x01 << ( (i) % BITBYTE ) )
  67 #define GETBIT(x,i) ( (GETBYTE(x,i) >> ( (i) % BITBYTE )) & 0x01 )
  68
  69 #define HASHVAL(val) (((unsigned int)(val)) % SIGLENBIT)
  70 #define HASH(sign, val) SETBIT((sign), HASHVAL(val))
  71
  72 #define ARRKEY                  0x01
  73 #define SIGNKEY                 0x02
  74 #define ALLISTRUE               0x04
  75
  76 #define ISARRKEY(x) ( ((TRGM*)x)->flag & ARRKEY )
  77 #define ISSIGNKEY(x)    ( ((TRGM*)x)->flag & SIGNKEY )
  78 #define ISALLTRUE(x)    ( ((TRGM*)x)->flag & ALLISTRUE )
  79
  80 #define CALCGTSIZE(flag, len) ( TRGMHRDSIZE + ( ( (flag) & ARRKEY ) ? ((len)*sizeof(trgm)) : (((flag) & ALLISTRUE) ? 0 : SIGLEN) ) )
  81 #define GETSIGN(x)              ( (BITVECP)( (char*)x+TRGMHRDSIZE ) )
  82 #define GETARR(x)               ( (trgm*)( (char*)x+TRGMHRDSIZE ) )
  83 #define ARRNELEM(x) ( ( ((TRGM*)x)->len - TRGMHRDSIZE )/sizeof(trgm) )
  84
  85 extern float4 trgm_limit;
  86 TRGM       *generate_trgm(char *str, int slen);
  87 float4          cnt_sml(TRGM * trg1, TRGM * trg2);
  88
  89 #endif