index_fst.c 22.9 KB
Newer Older
dengyihao's avatar
dengyihao 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "index_fst.h"

dengyihao's avatar
dengyihao 已提交
18 19 20 21 22 23 24 25 26

FstUnFinishedNodes *fstUnFinishedNodesCreate() {
  FstUnFinishedNodes *nodes = malloc(sizeof(FstUnFinishedNodes));
  if (nodes == NULL) { return NULL; }

  nodes->stack = (SArray *)taosArrayInit(64, sizeof(FstBuilderNodeUnfinished));
  fstUnFinishedNodesPushEmpty(nodes, false);
  return nodes;
}
dengyihao's avatar
dengyihao 已提交
27 28 29 30 31
void unFinishedNodeDestroyElem(void* elem) {
  FstBuilderNodeUnfinished *b = (FstBuilderNodeUnfinished*)elem;
  fstBuilderNodeDestroy(b->node); 
  free(b->last); 
}  
dengyihao's avatar
dengyihao 已提交
32
void fstUnFinishedNodesDestroy(FstUnFinishedNodes *nodes) {
dengyihao's avatar
dengyihao 已提交
33 34 35 36 37 38
  if (nodes == NULL) { return; } 

  taosArrayDestroyEx(nodes->stack, unFinishedNodeDestroyElem); 
  free(nodes);
}

dengyihao's avatar
dengyihao 已提交
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
void fstUnFinishedNodesPushEmpty(FstUnFinishedNodes *nodes, bool isFinal) {
  FstBuilderNode *node = malloc(sizeof(FstBuilderNode));
  node->isFinal     = isFinal;
  node->finalOutput = 0;
  node->trans       = NULL;

  FstBuilderNodeUnfinished un = {.node = node, .last = NULL}; 
  taosArrayPush(nodes->stack, &un);
  
}
FstBuilderNode *fstUnFinishedNodesPopRoot(FstUnFinishedNodes *nodes) {
  assert(taosArrayGetSize(nodes->stack) == 1);

  FstBuilderNodeUnfinished *un = taosArrayPop(nodes->stack);
  assert(un->last == NULL); 
  return un->node;  
}

FstBuilderNode *fstUnFinishedNodesPopFreeze(FstUnFinishedNodes *nodes, CompiledAddr addr) {
  FstBuilderNodeUnfinished *un = taosArrayPop(nodes->stack);
  fstBuilderNodeUnfinishedLastCompiled(un, addr);
  free(un->last); // TODO add func FstLastTransitionFree()
  return un->node; 
}

FstBuilderNode *fstUnFinishedNodesPopEmpty(FstUnFinishedNodes *nodes) {
  FstBuilderNodeUnfinished *un = taosArrayPop(nodes->stack);
  assert(un->last == NULL); 
  return un->node;  
  
}
void fstUnFinishedNodesSetRootOutput(FstUnFinishedNodes *nodes, Output out) {
  FstBuilderNodeUnfinished *un = taosArrayGet(nodes->stack, 0);
  un->node->isFinal     = true;
  un->node->finalOutput = out;
  //un->node->trans       = NULL;  
} 
void fstUnFinishedNodesTopLastFreeze(FstUnFinishedNodes *nodes, CompiledAddr addr) {
  size_t sz = taosArrayGetSize(nodes->stack) - 1; 
  FstBuilderNodeUnfinished *un = taosArrayGet(nodes->stack, sz);
  fstBuilderNodeUnfinishedLastCompiled(un, addr);
}
void fstUnFinishedNodesAddSuffix(FstUnFinishedNodes *nodes, FstSlice bs, Output out) {
  FstSlice *s = &bs;
  if (s->data == NULL || s->dLen == 0 || s->start > s->end) {
    return;
  }
  size_t sz = taosArrayGetSize(nodes->stack) - 1; 
  FstBuilderNodeUnfinished *un = taosArrayGet(nodes->stack, sz);
  assert(un->last == NULL);

  
  
dengyihao's avatar
dengyihao 已提交
92 93 94 95
  //FstLastTransition *trn = malloc(sizeof(FstLastTransition)); 
  //trn->inp = s->data[s->start]; 
  //trn->out = out;
  un->last = fstLastTransitionCreate(s->data[s->start], out); 
dengyihao's avatar
dengyihao 已提交
96 97 98 99 100 101 102

  for (uint64_t i = s->start; i <= s->end; i++) {
    FstBuilderNode *n = malloc(sizeof(FstBuilderNode));
    n->isFinal     = false;
    n->finalOutput = 0;
    n->trans       = NULL;
    
dengyihao's avatar
dengyihao 已提交
103 104 105 106
    //FstLastTransition *trn = malloc(sizeof(FstLastTransition)); 
    //trn->inp = s->data[i];
    //trn->out = out; 
    FstLastTransition *trn = fstLastTransitionCreate(s->data[i], out);
dengyihao's avatar
dengyihao 已提交
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131

    FstBuilderNodeUnfinished un = {.node = n, .last = trn}; 
    taosArrayPush(nodes->stack, &un); 
  }
  fstUnFinishedNodesPushEmpty(nodes, true);  
}


uint64_t fstUnFinishedNodesFindCommPrefix(FstUnFinishedNodes *node, FstSlice bs) {
  FstSlice *s = &bs;

  size_t lsz = (size_t)(s->end - s->start + 1);          // data len 
  size_t ssz = taosArrayGetSize(node->stack);  // stack size
    
  uint64_t count = 0;
  for (size_t i = 0; i < ssz && i < lsz; i++) {
    FstBuilderNodeUnfinished *un = taosArrayGet(node->stack, i); 
    if (un->last->inp == s->data[s->start + i]) {
      count++;
    } else {
      break;
    } 
  }
  return count;
}
dengyihao's avatar
dengyihao 已提交
132
uint64_t fstUnFinishedNodesFindCommPrefixAndSetOutput(FstUnFinishedNodes *node, FstSlice bs, Output in, Output *out) {
dengyihao's avatar
dengyihao 已提交
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
  FstSlice *s = &bs;

  size_t lsz = (size_t)(s->end - s->start + 1);          // data len 
  size_t ssz = taosArrayGetSize(node->stack);  // stack size

  uint64_t res = 0;
  for (size_t i = 0; i < lsz && i < ssz; i++) {
    FstBuilderNodeUnfinished *un = taosArrayGet(node->stack, i);

    FstLastTransition *last = un->last; 
    if (last->inp == s->data[s->start + i]) {
      uint64_t commPrefix = last->out;      
      uint64_t addPrefix  = last->out - commPrefix; 
      out = out - commPrefix; 
      last->out = commPrefix;
      if (addPrefix != 0) {
        fstBuilderNodeUnfinishedAddOutputPrefix(un, addPrefix);  
      }
    } else {
      break;
    }
  }   
  return res;
} 


dengyihao's avatar
dengyihao 已提交
159
FstState fstStateCreateFrom(FstSlice* slice, CompiledAddr addr) {
dengyihao's avatar
dengyihao 已提交
160
  FstState fs = {.state = EmptyFinal, .val = 0};
dengyihao's avatar
dengyihao 已提交
161 162 163
  if (addr == EMPTY_ADDRESS) {
    return fs; 
  }
dengyihao's avatar
dengyihao 已提交
164 165

  uint8_t v = slice->data[addr]; 
dengyihao's avatar
dengyihao 已提交
166 167 168 169
  uint8_t t = (v & 0b11000000) >> 6;
  if (t == 0b11) {
    fs.state = OneTransNext;
  } else if (t == 0b10) {
dengyihao's avatar
dengyihao 已提交
170
    fs.state = OneTrans; 
dengyihao's avatar
dengyihao 已提交
171
  } else {
dengyihao's avatar
dengyihao 已提交
172
    fs.state = AnyTrans;  
dengyihao's avatar
dengyihao 已提交
173
  }
dengyihao's avatar
dengyihao 已提交
174
  fs.val = v;
dengyihao's avatar
dengyihao 已提交
175 176
  return fs;
}
dengyihao's avatar
dengyihao 已提交
177

dengyihao's avatar
dengyihao 已提交
178 179 180 181 182 183 184 185 186 187
static FstState stateDict[] = {
  {.state = OneTransNext, .val = 0b11000000},
  {.state = OneTrans,     .val = 0b10000000},
  {.state = AnyTrans,     .val = 0b00000000},
  {.state = EmptyFinal,   .val = 0b00000000}
};

FstState fstStateCreate(State state){
  uint8_t idx = (uint8_t)state;
  return stateDict[idx];
dengyihao's avatar
dengyihao 已提交
188 189
}
//compile
dengyihao's avatar
dengyihao 已提交
190 191 192 193 194 195 196 197 198 199
void fstStateCompileForOneTransNext(FstCountingWriter *w, CompiledAddr addr, uint8_t inp) {
  FstState s = fstStateCreate(OneTransNext);  
  fstStateSetCommInput(&s, inp);

  bool null = false;
  uint8_t v = fstStateCommInput(&s, &null);
  if (null) {
    // w->write_all(&[inp])
  }  
  // w->write_all(&[s.val])
dengyihao's avatar
dengyihao 已提交
200 201
  return ;
}
dengyihao's avatar
dengyihao 已提交
202 203
void fstStateCompileForOneTrans(FstCountingWriter *w, CompiledAddr addr, FstTransition trn) {
  
dengyihao's avatar
dengyihao 已提交
204 205 206
  return ;

}
dengyihao's avatar
dengyihao 已提交
207
void fstStateCompileForAnyTrans(FstCountingWriter *w, CompiledAddr addr, FstBuilderNode *node) {
dengyihao's avatar
dengyihao 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220
  return;
}

// set_comm_input
void fstStateSetCommInput(FstState* s, uint8_t inp) {
  assert(s->state == OneTransNext || s->state == OneTrans);

  uint8_t val;
  COMMON_INDEX(inp, 0x111111, val); 
  s->val = (s->val & stateDict[s->state].val) | val;    
}

// comm_input
dengyihao's avatar
dengyihao 已提交
221
uint8_t fstStateCommInput(FstState* s, bool *null) {
dengyihao's avatar
dengyihao 已提交
222 223
  assert(s->state == OneTransNext || s->state == OneTrans);
  uint8_t v = s->val & 0b00111111;
dengyihao's avatar
dengyihao 已提交
224 225 226 227
  if (v == 0) { 
    *null = true; 
    return v;
  } 
dengyihao's avatar
dengyihao 已提交
228 229 230 231 232 233 234 235
  //v = 0 indicate that common_input is None
  return v == 0 ? 0 : COMMON_INPUT(v);  
}

// input_len

uint64_t fstStateInputLen(FstState* s) {
  assert(s->state == OneTransNext || s->state == OneTrans);
dengyihao's avatar
dengyihao 已提交
236 237 238
  bool null = false;
  fstStateCommInput(s, &null); 
  return null ? 1 : 0 ;
dengyihao's avatar
dengyihao 已提交
239
} 
dengyihao's avatar
dengyihao 已提交
240
  
dengyihao's avatar
dengyihao 已提交
241 242 243 244 245 246

// end_addr 
uint64_t fstStateEndAddrForOneTransNext(FstState* s, FstSlice *data) {
  return FST_SLICE_LEN(data) - 1 - fstStateInputLen(s);
}
uint64_t fstStateEndAddrForOneTrans(FstState *s, FstSlice *data, PackSizes sizes) {
dengyihao's avatar
dengyihao 已提交
247 248 249 250 251 252
  return FST_SLICE_LEN(data) 
                - 1 
                - fstStateInputLen(s) 
                - 1 // pack size 
                - FST_GET_TRANSITION_PACK_SIZE(sizes) 
                - FST_GET_OUTPUT_PACK_SIZE(sizes); 
dengyihao's avatar
dengyihao 已提交
253 254
}
uint64_t fstStateEndAddrForAnyTrans(FstState *state, uint64_t version, FstSlice *date, PackSizes sizes, uint64_t nTrans) {
dengyihao's avatar
dengyihao 已提交
255 256 257 258 259 260 261 262 263
  uint8_t oSizes = FST_GET_OUTPUT_PACK_SIZE(sizes); 
  uint8_t finalOsize = !fstStateIsFinalState(state) ? 0 : oSizes;    
  return FST_SLICE_LEN(date)  
               - 1 
               - fstStateNtransLen(state) 
               - 1 //pack size 
               - fstStateTotalTransSize(state, version, sizes, nTrans) 
               - nTrans * oSizes  // output values
               - finalOsize;     // final output 
dengyihao's avatar
dengyihao 已提交
264 265
}
// input  
dengyihao's avatar
dengyihao 已提交
266 267 268
uint8_t  fstStateInput(FstState *s, FstNode *node) {
  assert(s->state == OneTransNext || s->state == OneTrans);
  FstSlice *slice = &node->data;
dengyihao's avatar
dengyihao 已提交
269 270 271
  bool null = false;
  uint8_t inp = fstStateCommInput(s, &null);
  return null == false ? inp : slice->data[slice->start - 1];
dengyihao's avatar
dengyihao 已提交
272
}
dengyihao's avatar
dengyihao 已提交
273 274 275 276 277 278 279 280 281 282 283
uint8_t  fstStateInputForAnyTrans(FstState *s, FstNode *node, uint64_t i) {
  assert(s->state == AnyTrans);
  FstSlice *slice = &node->data; 

  uint64_t at = node->start 
            - fstStateNtransLen(s) 
            - 1  // pack size
            - fstStateTransIndexSize(s, node->version, node->nTrans) 
            - i 
            - 1; // the output size 
  return slice->data[at];
dengyihao's avatar
dengyihao 已提交
284 285 286
}

// trans_addr
dengyihao's avatar
dengyihao 已提交
287 288 289 290 291 292 293 294 295 296 297 298 299 300
CompiledAddr fstStateTransAddr(FstState *s, FstNode *node) {
  assert(s->state == OneTransNext || s->state == OneTrans);
  FstSlice *slice = &node->data;
  if (s->state == OneTransNext) {
    return (CompiledAddr)(node->end);      
  } else {
    PackSizes sizes = node->sizes; 
    uint8_t tSizes = FST_GET_TRANSITION_PACK_SIZE(sizes);  
    uint64_t i = node->start 
                - fstStateInputLen(s)
                - 1 // PackSizes
                - tSizes;

    // refactor error logic 
dengyihao's avatar
dengyihao 已提交
301
    return unpackDelta(slice->data + slice->start + i, tSizes, node->end);      
dengyihao's avatar
dengyihao 已提交
302 303 304 305 306 307 308 309 310 311 312 313 314 315
  }  
}
CompiledAddr fstStateTransAddrForAnyTrans(FstState *s, FstNode *node, uint64_t i) {
  assert(s->state == AnyTrans);

  FstSlice *slice = &node->data;
  uint8_t tSizes = FST_GET_TRANSITION_PACK_SIZE(node->sizes);
  uint64_t at = node->start  
               - fstStateNtransLen(s)
               - 1
               - fstStateTransIndexSize(s, node->version, node->nTrans)
               - node->nTrans
               - (i * tSizes)
               - tSizes;
dengyihao's avatar
dengyihao 已提交
316
  return unpackDelta(slice->data + slice->start + at, tSizes, node->end); 
dengyihao's avatar
dengyihao 已提交
317 318
}

dengyihao's avatar
dengyihao 已提交
319
// sizes 
dengyihao's avatar
dengyihao 已提交
320 321 322 323 324 325 326 327 328 329
PackSizes fstStateSizes(FstState *s, FstSlice *slice) {
  assert(s->state == OneTrans || s->state == AnyTrans) ;
  uint64_t i; 
  if (s->state == OneTrans) {
    i = FST_SLICE_LEN(slice) - 1 - fstStateInputLen(s) - 1;  
  } else {
    i = FST_SLICE_LEN(slice) - 1 - fstStateNtransLen(s) - 1;
  }

  return (PackSizes)(slice->data[slice->start + i]);
dengyihao's avatar
dengyihao 已提交
330 331
}
// Output 
dengyihao's avatar
dengyihao 已提交
332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
Output fstStateOutput(FstState *s, FstNode *node) {
  assert(s->state == OneTrans);  
  
  uint8_t oSizes = FST_GET_OUTPUT_PACK_SIZE(node->sizes);
  if (oSizes == 0) {
    return 0;
  }
  FstSlice *slice = &node->data;
  uint8_t tSizes = FST_GET_TRANSITION_PACK_SIZE(node->sizes);

  uint64_t i = node->start 
                - fstStateInputLen(s);
                - 1
                - tSizes 
                - oSizes;
  return unpackUint64(slice->data + slice->start + i, oSizes);
dengyihao's avatar
dengyihao 已提交
348
  
dengyihao's avatar
dengyihao 已提交
349
}
dengyihao's avatar
dengyihao 已提交
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364
Output fstStateOutputForAnyTrans(FstState *s, FstNode *node, uint64_t i) {
  assert(s->state == AnyTrans);

  uint8_t oSizes = FST_GET_OUTPUT_PACK_SIZE(node->sizes); 
  if (oSizes == 0) {
    return 0;    
  }  
  FstSlice *slice = &node->data;
  uint64_t at = node->start
                - fstStateNtransLen(s)
                - 1 // pack size
                - fstStateTotalTransSize(s, node->version, node->sizes, node->nTrans)
                - (i * oSizes)
                - oSizes;
  return unpackUint64(slice->data + slice->start + at, oSizes);
dengyihao's avatar
dengyihao 已提交
365 366 367 368
}

// anyTrans specify function

dengyihao's avatar
dengyihao 已提交
369 370 371
void fstStateSetFinalState(FstState *s, bool yes) {
  assert(s->state == AnyTrans); 
  if (yes) { s->val |= 0b01000000; } 
dengyihao's avatar
dengyihao 已提交
372 373
  return;
}
dengyihao's avatar
dengyihao 已提交
374 375 376
bool fstStateIsFinalState(FstState *s) {
  assert(s->state == AnyTrans); 
  return (s->val & 0b01000000) == 0b01000000; 
dengyihao's avatar
dengyihao 已提交
377
} 
dengyihao's avatar
dengyihao 已提交
378 379 380 381 382 383

void fstStateSetStateNtrans(FstState *s, uint8_t n) {
  assert(s->state == AnyTrans); 
  if (n <= 0b00111111) {
    s->val = (s->val & 0b11000000) | n;  
  } 
dengyihao's avatar
dengyihao 已提交
384 385 386
  return;
}
// state_ntrans
dengyihao's avatar
dengyihao 已提交
387 388 389 390 391 392 393 394 395
uint8_t fstStateStateNtrans(FstState *s, bool *null) {
  assert(s->state == AnyTrans); 
  *null = false;
  uint8_t n = s->val & 0b00111111; 

  if (n == 0) {
    *null = true; // None
  }  
  return n;
dengyihao's avatar
dengyihao 已提交
396
}
dengyihao's avatar
dengyihao 已提交
397 398 399 400
uint64_t fstStateTotalTransSize(FstState *s, uint64_t version, PackSizes sizes, uint64_t nTrans) {
  assert(s->state == AnyTrans); 
  uint64_t idxSize = fstStateTransIndexSize(s, version, nTrans); 
  return nTrans + (nTrans * FST_GET_TRANSITION_PACK_SIZE(sizes)) + idxSize;
dengyihao's avatar
dengyihao 已提交
401
}
dengyihao's avatar
dengyihao 已提交
402 403 404
uint64_t fstStateTransIndexSize(FstState *s, uint64_t version, uint64_t nTrans) {
  assert(s->state == AnyTrans); 
  return (version >= 2 &&nTrans > TRANS_INDEX_THRESHOLD) ?  256 : 0;
dengyihao's avatar
dengyihao 已提交
405
}
dengyihao's avatar
dengyihao 已提交
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
uint64_t fstStateNtransLen(FstState *s) {
  assert(s->state == AnyTrans);
  bool null = false;
  fstStateStateNtrans(s, &null);
  return null == true ?  1 : 0; 
}
uint64_t fstStateNtrans(FstState *s, FstSlice *slice) {
  bool null = false; 
  uint8_t n = fstStateStateNtrans(s, &null);
  if (null != true) {
    return n;   
  }  
  n = slice->data[slice->end - 1]; // data[data.len() - 2]
  return n == 1 ? 256: n; // // "1" is never a normal legal value here, because if there, // is only 1 transition, then it is encoded in the state byte  
}
Output  fstStateFinalOutput(FstState *s, uint64_t version, FstSlice *slice, PackSizes sizes, uint64_t nTrans) {
   uint8_t oSizes = FST_GET_OUTPUT_PACK_SIZE(sizes);
   if (oSizes == 0 || !fstStateIsFinalState(s)) {
      return 0;
   }
   
   uint64_t at = FST_SLICE_LEN(slice) 
                 - 1 
                 - fstStateNtransLen(s)
                 - fstStateTotalTransSize(s, version, sizes, nTrans)
                 - (nTrans * oSizes)
                 - oSizes;
  return unpackUint64(slice->data + slice->start + at, (uint8_t)oSizes);    

}
uint64_t fstStateFindInput(FstState *s, FstNode *node, uint8_t b, bool *null) {
  assert(s->state == AnyTrans);
  FstSlice *slice = &node->data;
  if (node->version >= 2 && node->nTrans > TRANS_INDEX_THRESHOLD) {
    uint64_t at = node->start
                  - fstStateNtransLen(s)
                  - 1 // pack size 
                  - fstStateTransIndexSize(s, node->version, node->nTrans);
    uint64_t i = slice->data[slice->start + at + b];  
    if (i >= node->nTrans) {
      *null = true;
    } 
    return i;
  } else {
    uint64_t start = node->start 
                    - fstStateNtransLen(s)
                    - 1 // pack size
                    - node->nTrans;
    uint64_t end =  start + node->nTrans;
    uint64_t len = end - start; 
    for(int i = 0; i < len; i++) {
      uint8_t v = slice->data[slice->start + i];
      if (v == b) {
        return node->nTrans - i - 1; // bug  
      }
    } 
  } 
dengyihao's avatar
dengyihao 已提交
463 464 465
}


dengyihao's avatar
dengyihao 已提交
466
// fst node function 
dengyihao's avatar
dengyihao 已提交
467 468

FstNode *fstNodeCreate(int64_t version, CompiledAddr addr, FstSlice *slice) {
dengyihao's avatar
dengyihao 已提交
469 470 471
  FstNode *n = (FstNode *)malloc(sizeof(FstNode)); 
  if (n == NULL) { return NULL; }

dengyihao's avatar
dengyihao 已提交
472
  FstState st = fstStateCreateFrom(slice, addr);  
dengyihao's avatar
dengyihao 已提交
473 474

  if (st.state == EmptyFinal) {
dengyihao's avatar
dengyihao 已提交
475
     n->data    = fstSliceCreate(NULL, 0);   
dengyihao's avatar
dengyihao 已提交
476
     n->version = version;
dengyihao's avatar
dengyihao 已提交
477
     n->state   = st;  
dengyihao's avatar
dengyihao 已提交
478 479 480 481 482
     n->start   = EMPTY_ADDRESS;
     n->end     = EMPTY_ADDRESS;  
     n->isFinal = true; 
     n->nTrans  = 0;
     n->sizes   = 0;  
dengyihao's avatar
dengyihao 已提交
483
     n->finalOutput = 0;  
dengyihao's avatar
dengyihao 已提交
484
  } else if (st.state == OneTransNext) {
dengyihao's avatar
dengyihao 已提交
485 486
     n->data    = fstSliceCopy(slice, 0, addr);     
     n->version = version;
dengyihao's avatar
dengyihao 已提交
487
     n->state   = st;  
dengyihao's avatar
dengyihao 已提交
488 489 490 491 492 493
     n->start   = addr;
     n->end     = addr; //? s.end_addr(data); 
     n->isFinal = false;
     n->sizes   = 0;
     n->nTrans  = 0;
     n->finalOutput = 0;
dengyihao's avatar
dengyihao 已提交
494
  } else if (st.state == OneTrans) {
dengyihao's avatar
dengyihao 已提交
495 496 497
     uint64_t sz; // fetch sz from addr
     n->data    = fstSliceCopy(slice, 0, addr); 
     n->version = version; 
dengyihao's avatar
dengyihao 已提交
498
     n->state   = st; 
dengyihao's avatar
dengyihao 已提交
499 500 501 502 503 504
     n->start   = addr;
     n->end     = addr; // s.end_addr(data, sz);
     n->isFinal = false; 
     n->nTrans  = 1; 
     n->sizes   = sz;   
     n->finalOutput = 0; 
dengyihao's avatar
dengyihao 已提交
505
  } else {
dengyihao's avatar
dengyihao 已提交
506 507 508 509
     uint64_t sz;    // s.sizes(data)
     uint32_t nTrans; // s.ntrans(data)  
     n->data    = *slice; 
     n->version = version;
dengyihao's avatar
dengyihao 已提交
510
     n->state   = st;
dengyihao's avatar
dengyihao 已提交
511 512 513 514 515 516
     n->start   = addr;
     n->end     = addr; // s.end_addr(version, data, sz, ntrans);
     n->isFinal = false; // s.is_final_state();
     n->nTrans  = nTrans;
     n->sizes   = sz;
     n->finalOutput = 0; // s.final_output(version, data, sz, ntrans);
dengyihao's avatar
dengyihao 已提交
517
  }
dengyihao's avatar
dengyihao 已提交
518 519
   return n; 
}
dengyihao's avatar
dengyihao 已提交
520 521 522
void fstNodeDestroy(FstNode *node) {
  free(node);
}
dengyihao's avatar
dengyihao 已提交
523 524 525 526 527 528 529
FstTransitions* fstNodeTransitions(FstNode *node) {
  FstTransitions *t = malloc(sizeof(FstTransitions));
  if (NULL == t) {
    return NULL; 
  }
  FstRange range = {.start = 0, .end = FST_NODE_LEN(node)};
  t->range = range;
dengyihao's avatar
dengyihao 已提交
530
  t->node  = node;
dengyihao's avatar
dengyihao 已提交
531 532
  return t; 
} 
dengyihao's avatar
dengyihao 已提交
533

dengyihao's avatar
dengyihao 已提交
534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549
// Returns the transition at index `i`. 
bool fstNodeGetTransitionAt(FstNode *node, uint64_t i, FstTransition *trn) {
  bool s = true;
  //FstState st = node->state;
  if (FST_STATE_ONE_TRNAS_NEXT(node)) {
    trn->inp  = 0; 
    trn->out  = 0;
    trn->addr = 0; 
  } else if (FST_STATE_ONE_TRNAS(node)) {
    trn->inp  = 0;    
    trn->out  = 0;
    trn->addr = 0; 
  } else if (FST_STATE_ANY_TRANS(node)) {
    trn->inp  = 0;
    trn->out  = 0;
    trn->addr = 0; 
dengyihao's avatar
dengyihao 已提交
550 551 552 553 554 555
  } else {
    s = false;
  }
  return s;
} 

dengyihao's avatar
dengyihao 已提交
556
// Returns the transition address of the `i`th transition
dengyihao's avatar
dengyihao 已提交
557 558
bool fstNodeGetTransitionAddrAt(FstNode *node, uint64_t i, CompiledAddr *res) {
  bool s = true;
dengyihao's avatar
dengyihao 已提交
559
  if (FST_STATE_ONE_TRNAS_NEXT(node)) {
dengyihao's avatar
dengyihao 已提交
560

dengyihao's avatar
dengyihao 已提交
561
  } else if (FST_STATE_ONE_TRNAS(node)) {
dengyihao's avatar
dengyihao 已提交
562

dengyihao's avatar
dengyihao 已提交
563
  } else if (FST_STATE_ANY_TRANS(node)) {
dengyihao's avatar
dengyihao 已提交
564

dengyihao's avatar
dengyihao 已提交
565
  } else if (FST_STATE_EMPTY_FINAL(node)){
dengyihao's avatar
dengyihao 已提交
566
    s = false;
dengyihao's avatar
dengyihao 已提交
567 568
  } else {
    assert(0);
dengyihao's avatar
dengyihao 已提交
569 570
  }
  return s;
dengyihao's avatar
dengyihao 已提交
571 572
}

dengyihao's avatar
dengyihao 已提交
573 574
//  Finds the `i`th transition corresponding to the given input byte.
//  If no transition for this byte exists, then `false` is returned. 
dengyihao's avatar
dengyihao 已提交
575 576
bool fstNodeFindInput(FstNode *node, uint8_t b, uint64_t *res) {
  bool s = true;
dengyihao's avatar
dengyihao 已提交
577 578 579
  uint8_t input; // get s.input
  FstState fs = node->state;
  if (FST_STATE_ONE_TRNAS_NEXT(node)) {
dengyihao's avatar
dengyihao 已提交
580 581
    if (b == input) { *res = 0; } 
    else { return s ; }
dengyihao's avatar
dengyihao 已提交
582
  } else if (FST_STATE_ONE_TRNAS(node)) {
dengyihao's avatar
dengyihao 已提交
583 584
    if (b == input) { *res = 0; }
    else {return s;}
dengyihao's avatar
dengyihao 已提交
585
  } else if (FST_STATE_ANY_TRANS(node)) {
dengyihao's avatar
dengyihao 已提交
586
     
dengyihao's avatar
dengyihao 已提交
587
  } else if (FST_STATE_EMPTY_FINAL(node)) {
dengyihao's avatar
dengyihao 已提交
588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617
    s = false;
  } 
  return s;
} 

bool fstNodeCompile(FstNode *node, void *w, CompiledAddr lastAddr, CompiledAddr addr, FstBuilderNode *builderNode) {
  size_t sz = taosArrayGetSize(builderNode->trans);  
  assert(sz < 256);
  if (sz == 0 && builderNode->isFinal && builderNode->finalOutput == 0) {
    return true; 
  } else if (sz != 1 || builderNode->isFinal) {
    // AnyTrans->Compile(w, addr, node);
  } else {
    FstTransition *tran = taosArrayGet(builderNode->trans, 0);   
    if (tran->addr == lastAddr && tran->out == 0) {
       //OneTransNext::compile(w, lastAddr, tran->inp);
       return true;
    } else {
      //OneTrans::Compile(w, lastAddr, *tran);
       return true;
    } 
  } 
  return true; 
} 


FstBuilder *fstBuilderCreate(void *w, FstType ty) {
  FstBuilder *b = malloc(sizeof(FstBuilder));  
  if (NULL == b) { return b; }

dengyihao's avatar
dengyihao 已提交
618 619 620 621
   
  b->wrt = fstCountingWriterCreate(w);
  b->unfinished = fstUnFinishedNodesCreate();   
  b->registry   = fstRegistryCreate(10000, 2) ;
dengyihao's avatar
dengyihao 已提交
622
  b->last       = fstSliceCreate(NULL, 0);
dengyihao's avatar
dengyihao 已提交
623 624
  b->lastAddr   = NONE_ADDRESS; 
  b->len        = 0;
dengyihao's avatar
dengyihao 已提交
625 626
  return b;
}
dengyihao's avatar
dengyihao 已提交
627 628
void fstBuilderDestroy(FstBuilder *b) {
  if (b == NULL) { return; }
dengyihao's avatar
dengyihao 已提交
629

dengyihao's avatar
dengyihao 已提交
630 631 632 633 634
  fstCountingWriterDestroy(b->wrt); 
  fstUnFinishedNodesDestroy(b->unfinished); 
  fstRegistryDestroy(b->registry);
  free(b);
}
dengyihao's avatar
dengyihao 已提交
635 636 637 638 639 640 641 642 643 644 645 646


bool fstBuilderInsert(FstBuilder *b, FstSlice bs, Output in) {
  OrderType t = fstBuilderCheckLastKey(b, bs, true);  
  if (t == Ordered) {
    // add log info
    fstBuilderInsertOutput(b, bs, in); 
    return true; 
  } 
  return false;
}

dengyihao's avatar
dengyihao 已提交
647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674
void fstBuilderInsertOutput(FstBuilder *b, FstSlice bs, Output in) {
   FstSlice *s = &bs;
   if (fstSliceEmpty(s)) {
     b->len = 1; 
     fstUnFinishedNodesSetRootOutput(b->unfinished, in);
     return;
   }
   Output out; 
   uint64_t prefixLen;
   if (in != 0) { //if let Some(in) = in 
      prefixLen = fstUnFinishedNodesFindCommPrefixAndSetOutput(b->unfinished, bs, in, &out);  
   } else {
      prefixLen = fstUnFinishedNodesFindCommPrefix(b->unfinished, bs);
      out = 0;
   }

   if (prefixLen == FST_SLICE_LEN(s)) {
      assert(out != 0);
      return;
   }

   b->len += 1;
   fstBuilderCompileFrom(b, prefixLen); 
   
   FstSlice sub = fstSliceCopy(s, prefixLen, s->end);
   fstUnFinishedNodesAddSuffix(b->unfinished, sub, out);
   return;
 }
dengyihao's avatar
dengyihao 已提交
675

dengyihao's avatar
dengyihao 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
OrderType fstBuilderCheckLastKey(FstBuilder *b, FstSlice bs, bool ckDup) {
  FstSlice *input = &bs;
  if (fstSliceEmpty(&b->last)) {
    // deep copy or not
    b->last = fstSliceCopy(&bs, input->start, input->end);
  } else {
    int comp = fstSliceCompare(&b->last, &bs);
    if (comp == 0 && ckDup) {
      return DuplicateKey;  
    } else if (comp == 1) {
      return OutOfOrdered;
    }
    // deep copy or not
    b->last = fstSliceCopy(&bs, input->start, input->end); 
  }       
  return Ordered;
dengyihao's avatar
dengyihao 已提交
692
} 
dengyihao's avatar
dengyihao 已提交
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708
void fstBuilderCompileFrom(FstBuilder *b, uint64_t istate) {
  CompiledAddr addr = NONE_ADDRESS;
  while (istate + 1 < FST_UNFINISHED_NODES_LEN(b->unfinished)) {
    FstBuilderNode *n = NULL;
    if (addr == NONE_ADDRESS) {
      n = fstUnFinishedNodesPopEmpty(b->unfinished);
    } else {
      n = fstUnFinishedNodesPopFreeze(b->unfinished, addr);
    }
    addr =  fstBuilderCompile(b, n);
    assert(addr != NONE_ADDRESS);      
    fstBuilderNodeDestroy(n);
  }
  fstUnFinishedNodesTopLastFreeze(b->unfinished, addr);
  return; 
}
dengyihao's avatar
dengyihao 已提交
709 710 711 712 713 714 715 716 717
CompiledAddr fstBuilderCompile(FstBuilder *b, FstBuilderNode *bn) {
  if (FST_BUILDER_NODE_IS_FINAL(bn) 
      && FST_BUILDER_NODE_TRANS_ISEMPTY(bn) 
      && FST_BUILDER_NODE_FINALOUTPUT_ISZERO(bn)) {
    return EMPTY_ADDRESS; 
  }
  FstRegistryEntry *entry = fstRegistryGetEntry(b->registry, bn); 
  if (entry->state == FOUND) { 
    CompiledAddr ret = entry->addr;
dengyihao's avatar
dengyihao 已提交
718
    fstRegistryEntryDestroy(entry);
dengyihao's avatar
dengyihao 已提交
719 720 721 722 723
    return ret;
  } 
  CompiledAddr startAddr = (CompiledAddr)(FST_WRITER_COUNT(b->wrt));

  fstBuilderNodeCompileTo(bn, b->wrt, b->lastAddr, startAddr);  
dengyihao's avatar
dengyihao 已提交
724
  b->lastAddr =  (CompiledAddr)(FST_WRITER_COUNT(b->wrt) - 1);  
dengyihao's avatar
dengyihao 已提交
725 726 727
  if (entry->state == NOTFOUND) {
    FST_REGISTRY_CELL_INSERT(entry->cell, b->lastAddr);    
  }
dengyihao's avatar
dengyihao 已提交
728 729
  fstRegistryEntryDestroy(entry);
  
dengyihao's avatar
dengyihao 已提交
730 731 732
  return b->lastAddr;  
}

dengyihao's avatar
dengyihao 已提交
733

dengyihao's avatar
dengyihao 已提交
734 735


dengyihao's avatar
dengyihao 已提交
736 737 738 739 740
FstSlice fstNodeAsSlice(FstNode *node) {
  FstSlice *slice = &node->data; 
  FstSlice s = fstSliceCopy(slice, slice->end, slice->dLen - 1);   
  return s; 
}
dengyihao's avatar
dengyihao 已提交
741

dengyihao's avatar
dengyihao 已提交
742 743 744 745 746 747 748 749 750 751 752 753
FstLastTransition *fstLastTransitionCreate(uint8_t inp, Output out) {
  FstLastTransition *trn = malloc(sizeof(FstLastTransition));
  if (trn == NULL) { return NULL; }

  trn->inp = inp;
  trn->out = out;
  return trn;
}

void fstLastTransitionDestroy(FstLastTransition *trn) {
  free(trn);
}
dengyihao's avatar
dengyihao 已提交
754 755
void fstBuilderNodeUnfinishedLastCompiled(FstBuilderNodeUnfinished *unNode, CompiledAddr addr) {
  FstLastTransition *trn = unNode->last;       
dengyihao's avatar
dengyihao 已提交
756 757
  if (trn == NULL) { return; }  

dengyihao's avatar
dengyihao 已提交
758 759
  FstTransition t = {.inp = trn->inp, .out = trn->out, .addr = addr};      
  taosArrayPush(unNode->node->trans, &t); 
dengyihao's avatar
dengyihao 已提交
760 761 762
  return;
}

dengyihao's avatar
dengyihao 已提交
763 764 765 766 767 768 769 770 771 772 773 774
void fstBuilderNodeUnfinishedAddOutputPrefix(FstBuilderNodeUnfinished *unNode, Output out) {
  if (FST_BUILDER_NODE_IS_FINAL(unNode->node)) {
    unNode->node->finalOutput += out;  
  }
  size_t sz = taosArrayGetSize(unNode->node->trans);
  for (size_t i = 0; i < sz; i++) {
    FstTransition *trn = taosArrayGet(unNode->node->trans, i); 
    trn->out += out;
  }
  if (unNode->last) {
    unNode->last->out += out;  
  }
dengyihao's avatar
dengyihao 已提交
775 776 777
  return;
}

dengyihao's avatar
dengyihao 已提交
778