/ Hex Artifact Content
Login

Artifact b0ff58fa643afa1d846786d51ea8d5c4b6b35aa0254ab5a82617db92f3adda64:


0000: 2f 2a 0a 2a 2a 20 32 30 31 38 2d 30 32 2d 30 39  /*.** 2018-02-09
0010: 0a 2a 2a 0a 2a 2a 20 54 68 65 20 61 75 74 68 6f  .**.** The autho
0020: 72 20 64 69 73 63 6c 61 69 6d 73 20 63 6f 70 79  r disclaims copy
0030: 72 69 67 68 74 20 74 6f 20 74 68 69 73 20 73 6f  right to this so
0040: 75 72 63 65 20 63 6f 64 65 2e 20 20 49 6e 20 70  urce code.  In p
0050: 6c 61 63 65 20 6f 66 0a 2a 2a 20 61 20 6c 65 67  lace of.** a leg
0060: 61 6c 20 6e 6f 74 69 63 65 2c 20 68 65 72 65 20  al notice, here 
0070: 69 73 20 61 20 62 6c 65 73 73 69 6e 67 3a 0a 2a  is a blessing:.*
0080: 2a 0a 2a 2a 20 20 20 20 4d 61 79 20 79 6f 75 20  *.**    May you 
0090: 64 6f 20 67 6f 6f 64 20 61 6e 64 20 6e 6f 74 20  do good and not 
00a0: 65 76 69 6c 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  evil..**    May 
00b0: 79 6f 75 20 66 69 6e 64 20 66 6f 72 67 69 76 65  you find forgive
00c0: 6e 65 73 73 20 66 6f 72 20 79 6f 75 72 73 65 6c  ness for yoursel
00d0: 66 20 61 6e 64 20 66 6f 72 67 69 76 65 20 6f 74  f and forgive ot
00e0: 68 65 72 73 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  hers..**    May 
00f0: 79 6f 75 20 73 68 61 72 65 20 66 72 65 65 6c 79  you share freely
0100: 2c 20 6e 65 76 65 72 20 74 61 6b 69 6e 67 20 6d  , never taking m
0110: 6f 72 65 20 74 68 61 6e 20 79 6f 75 20 67 69 76  ore than you giv
0120: 65 2e 0a 2a 2a 0a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  e..**.**********
0130: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0140: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0150: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0160: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0170: 2a 2a 2a 2a 0a 2a 2a 0a 2a 2a 20 53 51 4c 20 66  ****.**.** SQL f
0180: 75 6e 63 74 69 6f 6e 73 20 66 6f 72 20 7a 2d 6f  unctions for z-o
0190: 72 64 65 72 20 28 4d 6f 72 74 6f 6e 20 63 6f 64  rder (Morton cod
01a0: 65 29 20 74 72 61 6e 73 66 6f 72 6d 61 74 69 6f  e) transformatio
01b0: 6e 73 2e 0a 2a 2a 0a 2a 2a 20 20 20 20 20 20 7a  ns..**.**      z
01c0: 6f 72 64 65 72 28 58 30 2c 58 30 2c 2e 2e 2c 78  order(X0,X0,..,x
01d0: 4e 29 20 20 20 20 20 20 47 65 6e 65 72 61 74 65  N)      Generate
01e0: 20 61 6e 20 4e 2b 31 20 64 69 6d 65 6e 73 69 6f   an N+1 dimensio
01f0: 6e 20 4d 6f 72 74 6f 6e 20 63 6f 64 65 0a 2a 2a  n Morton code.**
0200: 0a 2a 2a 20 20 20 20 20 20 75 6e 7a 6f 72 64 65  .**      unzorde
0210: 72 28 5a 2c 4e 2c 49 29 20 20 20 20 20 20 20 20  r(Z,N,I)        
0220: 20 20 45 78 74 72 61 63 74 20 74 68 65 20 49 2d    Extract the I-
0230: 74 68 20 64 69 6d 65 6e 73 69 6f 6e 20 66 72 6f  th dimension fro
0240: 6d 20 4e 2d 64 69 6d 65 6e 73 69 6f 6e 61 6c 0a  m N-dimensional.
0250: 2a 2a 20 20 20 20 20 20 20 20 20 20 20 20 20 20  **              
0260: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0270: 20 4d 6f 72 74 6f 6e 20 63 6f 64 65 20 5a 2e 0a   Morton code Z..
0280: 2a 2f 0a 23 69 6e 63 6c 75 64 65 20 22 73 71 6c  */.#include "sql
0290: 69 74 65 33 65 78 74 2e 68 22 0a 53 51 4c 49 54  ite3ext.h".SQLIT
02a0: 45 5f 45 58 54 45 4e 53 49 4f 4e 5f 49 4e 49 54  E_EXTENSION_INIT
02b0: 31 0a 23 69 6e 63 6c 75 64 65 20 3c 61 73 73 65  1.#include <asse
02c0: 72 74 2e 68 3e 0a 23 69 6e 63 6c 75 64 65 20 3c  rt.h>.#include <
02d0: 73 74 72 69 6e 67 2e 68 3e 0a 0a 2f 2a 0a 2a 2a  string.h>../*.**
02e0: 20 46 75 6e 63 74 69 6f 6e 73 3a 20 20 20 20 20   Functions:     
02f0: 7a 6f 72 64 65 72 28 58 30 2c 58 31 2c 2e 2e 2e  zorder(X0,X1,...
0300: 2e 29 0a 2a 2a 0a 2a 2a 20 43 6f 6e 76 65 72 74  .).**.** Convert
0310: 20 69 6e 74 65 67 65 72 73 20 58 30 2c 20 58 31   integers X0, X1
0320: 2c 20 2e 2e 2e 20 69 6e 74 6f 20 6d 6f 72 74 6f  , ... into morto
0330: 6e 20 63 6f 64 65 2e 0a 2a 2a 0a 2a 2a 20 54 68  n code..**.** Th
0340: 65 20 6f 75 74 70 75 74 20 69 73 20 61 20 73 69  e output is a si
0350: 67 6e 65 64 20 36 34 2d 62 69 74 20 69 6e 74 65  gned 64-bit inte
0360: 67 65 72 2e 20 20 49 66 20 61 6e 79 20 61 72 67  ger.  If any arg
0370: 75 6d 65 6e 74 20 69 73 20 74 6f 6f 20 6c 61 72  ument is too lar
0380: 67 65 2c 0a 2a 2a 20 61 6e 20 65 72 72 6f 72 20  ge,.** an error 
0390: 69 73 20 74 68 72 6f 77 6e 2e 0a 2a 2f 0a 73 74  is thrown..*/.st
03a0: 61 74 69 63 20 76 6f 69 64 20 7a 6f 72 64 65 72  atic void zorder
03b0: 46 75 6e 63 28 0a 20 20 73 71 6c 69 74 65 33 5f  Func(.  sqlite3_
03c0: 63 6f 6e 74 65 78 74 20 2a 63 6f 6e 74 65 78 74  context *context
03d0: 2c 0a 20 20 69 6e 74 20 61 72 67 63 2c 0a 20 20  ,.  int argc,.  
03e0: 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 20 2a 2a  sqlite3_value **
03f0: 61 72 67 76 0a 29 7b 0a 20 20 73 71 6c 69 74 65  argv.){.  sqlite
0400: 33 5f 69 6e 74 36 34 20 7a 2c 20 78 5b 36 33 5d  3_int64 z, x[63]
0410: 3b 0a 20 20 69 6e 74 20 69 2c 20 6a 3b 0a 20 20  ;.  int i, j;.  
0420: 7a 20 3d 20 30 3b 0a 20 20 66 6f 72 28 69 3d 30  z = 0;.  for(i=0
0430: 3b 20 69 3c 61 72 67 63 3b 20 69 2b 2b 29 7b 0a  ; i<argc; i++){.
0440: 20 20 20 20 78 5b 69 5d 20 3d 20 73 71 6c 69 74      x[i] = sqlit
0450: 65 33 5f 76 61 6c 75 65 5f 69 6e 74 36 34 28 61  e3_value_int64(a
0460: 72 67 76 5b 69 5d 29 3b 0a 20 20 7d 0a 20 20 69  rgv[i]);.  }.  i
0470: 66 28 20 61 72 67 63 3e 30 20 29 7b 0a 20 20 20  f( argc>0 ){.   
0480: 20 66 6f 72 28 69 3d 30 3b 20 69 3c 36 33 3b 20   for(i=0; i<63; 
0490: 69 2b 2b 29 7b 0a 20 20 20 20 20 20 6a 20 3d 20  i++){.      j = 
04a0: 69 25 61 72 67 63 3b 0a 20 20 20 20 20 20 7a 20  i%argc;.      z 
04b0: 7c 3d 20 28 78 5b 6a 5d 26 31 29 3c 3c 69 3b 0a  |= (x[j]&1)<<i;.
04c0: 20 20 20 20 20 20 78 5b 6a 5d 20 3e 3e 3d 20 31        x[j] >>= 1
04d0: 3b 0a 20 20 20 20 7d 0a 20 20 7d 0a 20 20 73 71  ;.    }.  }.  sq
04e0: 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f 69 6e 74  lite3_result_int
04f0: 36 34 28 63 6f 6e 74 65 78 74 2c 20 7a 29 3b 0a  64(context, z);.
0500: 20 20 66 6f 72 28 69 3d 30 3b 20 69 3c 61 72 67    for(i=0; i<arg
0510: 63 3b 20 69 2b 2b 29 7b 0a 20 20 20 20 69 66 28  c; i++){.    if(
0520: 20 78 5b 69 5d 20 29 7b 0a 20 20 20 20 20 20 73   x[i] ){.      s
0530: 71 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f 65 72  qlite3_result_er
0540: 72 6f 72 28 63 6f 6e 74 65 78 74 2c 20 22 70 61  ror(context, "pa
0550: 72 61 6d 65 74 65 72 20 74 6f 6f 20 6c 61 72 67  rameter too larg
0560: 65 22 2c 20 2d 31 29 3b 0a 20 20 20 20 7d 0a 20  e", -1);.    }. 
0570: 20 7d 0a 7d 0a 0a 0a 2f 2a 0a 2a 2a 20 46 75 6e   }.}.../*.** Fun
0580: 63 74 69 6f 6e 73 3a 20 20 20 20 20 75 6e 7a 6f  ctions:     unzo
0590: 72 64 65 72 28 5a 2c 4e 2c 49 29 0a 2a 2a 0a 2a  rder(Z,N,I).**.*
05a0: 2a 20 41 73 73 75 6d 69 6e 67 20 74 68 61 74 20  * Assuming that 
05b0: 5a 20 69 73 20 61 6e 20 4e 2d 64 69 6d 65 6e 73  Z is an N-dimens
05c0: 69 6f 6e 61 6c 20 4d 6f 72 74 6f 6e 20 63 6f 64  ional Morton cod
05d0: 65 2c 20 65 78 74 72 61 63 74 20 74 68 65 20 49  e, extract the I
05e0: 2d 74 68 0a 2a 2a 20 64 69 6d 65 6e 73 69 6f 6e  -th.** dimension
05f0: 2e 0a 2a 2f 0a 73 74 61 74 69 63 20 76 6f 69 64  ..*/.static void
0600: 20 75 6e 7a 6f 72 64 65 72 46 75 6e 63 28 0a 20   unzorderFunc(. 
0610: 20 73 71 6c 69 74 65 33 5f 63 6f 6e 74 65 78 74   sqlite3_context
0620: 20 2a 63 6f 6e 74 65 78 74 2c 0a 20 20 69 6e 74   *context,.  int
0630: 20 61 72 67 63 2c 0a 20 20 73 71 6c 69 74 65 33   argc,.  sqlite3
0640: 5f 76 61 6c 75 65 20 2a 2a 61 72 67 76 0a 29 7b  _value **argv.){
0650: 0a 20 20 73 71 6c 69 74 65 33 5f 69 6e 74 36 34  .  sqlite3_int64
0660: 20 7a 2c 20 6e 2c 20 69 2c 20 78 3b 0a 20 20 69   z, n, i, x;.  i
0670: 6e 74 20 6a 2c 20 6b 3b 0a 20 20 7a 20 3d 20 73  nt j, k;.  z = s
0680: 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f 69 6e 74  qlite3_value_int
0690: 36 34 28 61 72 67 76 5b 30 5d 29 3b 0a 20 20 6e  64(argv[0]);.  n
06a0: 20 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65   = sqlite3_value
06b0: 5f 69 6e 74 36 34 28 61 72 67 76 5b 31 5d 29 3b  _int64(argv[1]);
06c0: 0a 20 20 69 20 3d 20 73 71 6c 69 74 65 33 5f 76  .  i = sqlite3_v
06d0: 61 6c 75 65 5f 69 6e 74 36 34 28 61 72 67 76 5b  alue_int64(argv[
06e0: 32 5d 29 3b 0a 20 20 78 20 3d 20 30 3b 0a 20 20  2]);.  x = 0;.  
06f0: 66 6f 72 28 6b 3d 30 2c 20 6a 3d 69 3b 20 6a 3c  for(k=0, j=i; j<
0700: 36 33 3b 20 6a 2b 3d 6e 2c 20 6b 2b 2b 29 7b 0a  63; j+=n, k++){.
0710: 20 20 20 20 78 20 7c 3d 20 28 28 7a 3e 3e 6a 29      x |= ((z>>j)
0720: 26 31 29 3c 3c 6b 3b 0a 20 20 7d 0a 20 20 73 71  &1)<<k;.  }.  sq
0730: 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f 69 6e 74  lite3_result_int
0740: 36 34 28 63 6f 6e 74 65 78 74 2c 20 78 29 3b 0a  64(context, x);.
0750: 7d 0a 0a 0a 23 69 66 64 65 66 20 5f 57 49 4e 33  }...#ifdef _WIN3
0760: 32 0a 5f 5f 64 65 63 6c 73 70 65 63 28 64 6c 6c  2.__declspec(dll
0770: 65 78 70 6f 72 74 29 0a 23 65 6e 64 69 66 0a 69  export).#endif.i
0780: 6e 74 20 73 71 6c 69 74 65 33 5f 7a 6f 72 64 65  nt sqlite3_zorde
0790: 72 5f 69 6e 69 74 28 0a 20 20 73 71 6c 69 74 65  r_init(.  sqlite
07a0: 33 20 2a 64 62 2c 20 0a 20 20 63 68 61 72 20 2a  3 *db, .  char *
07b0: 2a 70 7a 45 72 72 4d 73 67 2c 20 0a 20 20 63 6f  *pzErrMsg, .  co
07c0: 6e 73 74 20 73 71 6c 69 74 65 33 5f 61 70 69 5f  nst sqlite3_api_
07d0: 72 6f 75 74 69 6e 65 73 20 2a 70 41 70 69 0a 29  routines *pApi.)
07e0: 7b 0a 20 20 69 6e 74 20 72 63 20 3d 20 53 51 4c  {.  int rc = SQL
07f0: 49 54 45 5f 4f 4b 3b 0a 20 20 53 51 4c 49 54 45  ITE_OK;.  SQLITE
0800: 5f 45 58 54 45 4e 53 49 4f 4e 5f 49 4e 49 54 32  _EXTENSION_INIT2
0810: 28 70 41 70 69 29 3b 0a 20 20 28 76 6f 69 64 29  (pApi);.  (void)
0820: 70 7a 45 72 72 4d 73 67 3b 20 20 2f 2a 20 55 6e  pzErrMsg;  /* Un
0830: 75 73 65 64 20 70 61 72 61 6d 65 74 65 72 20 2a  used parameter *
0840: 2f 0a 20 20 72 63 20 3d 20 73 71 6c 69 74 65 33  /.  rc = sqlite3
0850: 5f 63 72 65 61 74 65 5f 66 75 6e 63 74 69 6f 6e  _create_function
0860: 28 64 62 2c 20 22 7a 6f 72 64 65 72 22 2c 20 2d  (db, "zorder", -
0870: 31 2c 20 53 51 4c 49 54 45 5f 55 54 46 38 2c 20  1, SQLITE_UTF8, 
0880: 30 2c 0a 20 20 20 20 20 20 20 20 20 20 20 20 20  0,.             
0890: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
08a0: 20 20 7a 6f 72 64 65 72 46 75 6e 63 2c 20 30 2c    zorderFunc, 0,
08b0: 20 30 29 3b 0a 20 20 69 66 28 20 72 63 3d 3d 53   0);.  if( rc==S
08c0: 51 4c 49 54 45 5f 4f 4b 20 29 7b 0a 20 20 20 20  QLITE_OK ){.    
08d0: 72 63 20 3d 20 73 71 6c 69 74 65 33 5f 63 72 65  rc = sqlite3_cre
08e0: 61 74 65 5f 66 75 6e 63 74 69 6f 6e 28 64 62 2c  ate_function(db,
08f0: 20 22 75 6e 7a 6f 72 64 65 72 22 2c 20 33 2c 20   "unzorder", 3, 
0900: 53 51 4c 49 54 45 5f 55 54 46 38 2c 20 30 2c 0a  SQLITE_UTF8, 0,.
0910: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0920: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 75                 u
0930: 6e 7a 6f 72 64 65 72 46 75 6e 63 2c 20 30 2c 20  nzorderFunc, 0, 
0940: 30 29 3b 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e  0);.  }.  return
0950: 20 72 63 3b 0a 7d 0a                              rc;.}.