contrib/pg_trgm/trgm.h

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