Blame view

src/fstext/determinize-lattice-inl.h 52.3 KB
8dcb6dfcb   Yannick Estève   first commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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
132
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
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
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
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
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
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
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
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
  // fstext/determinize-lattice-inl.h
  
  // Copyright 2009-2012  Microsoft Corporation
  //           2012-2013  Johns Hopkins University (Author: Daniel Povey)
  
  // See ../../COPYING for clarification regarding multiple authors
  //
  // Licensed under the Apache License, Version 2.0 (the "License");
  // you may not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //  http://www.apache.org/licenses/LICENSE-2.0
  //
  // THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
  // WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
  // MERCHANTABLITY OR NON-INFRINGEMENT.
  // See the Apache 2 License for the specific language governing permissions and
  // limitations under the License.
  
  #ifndef KALDI_FSTEXT_DETERMINIZE_LATTICE_INL_H_
  #define KALDI_FSTEXT_DETERMINIZE_LATTICE_INL_H_
  // Do not include this file directly.  It is included by determinize-lattice.h
  
  #include <vector>
  #include <climits>
  
  namespace fst {
  
  // This class maps back and forth from/to integer id's to sequences of strings.
  // used in determinization algorithm.  It is constructed in such a way that
  // finding the string-id of the successor of (string, next-label) has constant time.
  
  // Note: class IntType, typically int32, is the type of the element in the
  // string (typically a template argument of the CompactLatticeWeightTpl).
  
  template<class IntType> class LatticeStringRepository {
   public:
    struct Entry {
      const Entry *parent; // NULL for empty string.
      IntType i;
      inline bool operator == (const Entry &other) const {
        return (parent == other.parent && i == other.i);
      }
      Entry() { }
      Entry(const Entry &e): parent(e.parent), i(e.i) {}
    };
    // Note: all Entry* pointers returned in function calls are
    // owned by the repository itself, not by the caller!
  
    // Interface guarantees empty string is NULL.
    inline const Entry *EmptyString() { return NULL; }
  
    // Returns string of "parent" with i appended.  Pointer
    // owned by repository
    const Entry *Successor(const Entry *parent, IntType i) {
      new_entry_->parent = parent;
      new_entry_->i = i;
  
      std::pair<typename SetType::iterator, bool> pr = set_.insert(new_entry_);
      if (pr.second) { // Was successfully inserted (was not there).  We need to
                       // replace the element we inserted, which resides on the
                       // stack, with one from the heap.
        const Entry *ans = new_entry_;
        new_entry_ = new Entry();
        return ans;
      } else { // Was not inserted because an equivalent Entry already
               // existed.
        return *pr.first;
      }
    }
  
    const Entry *Concatenate (const Entry *a, const Entry *b) {
      if (a == NULL) return b;
      else if (b == NULL) return a;
      vector<IntType> v;
      ConvertToVector(b, &v);
      const Entry *ans = a;
      for(size_t i = 0; i < v.size(); i++)
        ans = Successor(ans, v[i]);
      return ans;
    }
    const Entry *CommonPrefix (const Entry *a, const Entry *b) {
      vector<IntType> a_vec, b_vec;
      ConvertToVector(a, &a_vec);
      ConvertToVector(b, &b_vec);
      const Entry *ans = NULL;
      for(size_t i = 0; i < a_vec.size() && i < b_vec.size() &&
              a_vec[i] == b_vec[i]; i++)
        ans = Successor(ans, a_vec[i]);
      return ans;
    }
  
    // removes any elements from b that are not part of
    // a common prefix with a.
    void ReduceToCommonPrefix(const Entry *a,
                              vector<IntType> *b) {
      size_t a_size = Size(a), b_size = b->size();
      while (a_size> b_size) {
        a = a->parent;
        a_size--;
      }
      if (b_size > a_size)
        b_size = a_size;
      typename vector<IntType>::iterator b_begin = b->begin();
      while (a_size != 0) {
        if (a->i != *(b_begin + a_size - 1))
          b_size = a_size - 1;
        a = a->parent;
        a_size--;
      }
      if (b_size != b->size())
        b->resize(b_size);
    }
  
    // removes the first n elements of a.
    const Entry *RemovePrefix(const Entry *a, size_t n) {
      if (n==0) return a;
      vector<IntType> a_vec;
      ConvertToVector(a, &a_vec);
      assert(a_vec.size() >= n);
      const Entry *ans = NULL;
      for(size_t i = n; i < a_vec.size(); i++)
        ans = Successor(ans, a_vec[i]);
      return ans;
    }
  
  
  
    // Returns true if a is a prefix of b.  If a is prefix of b,
    // time taken is |b| - |a|.  Else, time taken is |b|.
    bool IsPrefixOf(const Entry *a, const Entry *b) const {
      if(a == NULL) return true; // empty string prefix of all.
      if (a == b) return true;
      if (b == NULL) return false;
      return IsPrefixOf(a, b->parent);
    }
  
  
    inline size_t Size(const Entry *entry) const {
      size_t ans = 0;
      while (entry != NULL) {
        ans++;
        entry = entry->parent;
      }
      return ans;
    }
  
    void ConvertToVector(const Entry *entry, vector<IntType> *out) const {
      size_t length = Size(entry);
      out->resize(length);
      if (entry != NULL) {
        typename vector<IntType>::reverse_iterator iter = out->rbegin();
      while (entry != NULL) {
        *iter = entry->i;
        entry = entry->parent;
          ++iter;
      }
    }
    }
  
    const Entry *ConvertFromVector(const vector<IntType> &vec) {
      const Entry *e = NULL;
      for(size_t i = 0; i < vec.size(); i++)
        e = Successor(e, vec[i]);
      return e;
    }
  
    LatticeStringRepository() { new_entry_ = new Entry; }
  
    void Destroy() {
      for (typename SetType::iterator iter = set_.begin();
           iter != set_.end();
           ++iter)
        delete *iter;
      SetType tmp;
      tmp.swap(set_);
      if (new_entry_) {
        delete new_entry_;
        new_entry_ = NULL;
      }
    }
  
    // Rebuild will rebuild this object, guaranteeing only
    // to preserve the Entry values that are in the vector pointed
    // to (this list does not have to be unique).  The point of
    // this is to save memory.
    void Rebuild(const std::vector<const Entry*> &to_keep) {
      SetType tmp_set;
      for (typename std::vector<const Entry*>::const_iterator
               iter = to_keep.begin();
           iter != to_keep.end(); ++iter)
        RebuildHelper(*iter, &tmp_set);
      // Now delete all elems not in tmp_set.
      for (typename SetType::iterator iter = set_.begin();
           iter != set_.end(); ++iter) {
        if (tmp_set.count(*iter) == 0)
          delete (*iter); // delete the Entry; not needed.
      }
      set_.swap(tmp_set);
    }
  
    ~LatticeStringRepository() { Destroy(); }
    int32 MemSize() const {
      return set_.size() * sizeof(Entry) * 2; // this is a lower bound
      // on the size this structure might take.
    }
   private:
    class EntryKey { // Hash function object.
     public:
      inline size_t operator()(const Entry *entry) const {
        size_t prime = 49109;
        return static_cast<size_t>(entry->i)
            + prime * reinterpret_cast<size_t>(entry->parent);
      }
    };
    class EntryEqual {
     public:
      inline bool operator()(const Entry *e1, const Entry *e2) const {
        return (*e1 == *e2);
      }
    };
    typedef unordered_set<const Entry*, EntryKey, EntryEqual> SetType;
  
    void RebuildHelper(const Entry *to_add, SetType *tmp_set) {
      while(true) {
        if (to_add == NULL) return;
        typename SetType::iterator iter = tmp_set->find(to_add);
        if (iter == tmp_set->end()) { // not in tmp_set.
          tmp_set->insert(to_add);
          to_add = to_add->parent; // and loop.
        } else {
          return;
        }
      }
    }
  
    KALDI_DISALLOW_COPY_AND_ASSIGN(LatticeStringRepository);
    Entry *new_entry_; // We always have a pre-allocated Entry ready to use,
                       // to avoid unnecessary news and deletes.
    SetType set_;
  
  };
  
  
  
  
  // class LatticeDeterminizer is templated on the same types that
  // CompactLatticeWeight is templated on: the base weight (Weight), typically
  // LatticeWeightTpl<float> etc. but could also be e.g. TropicalWeight, and the
  // IntType, typically int32, used for the output symbols in the compact
  // representation of strings [note: the output symbols would usually be
  // p.d.f. id's in the anticipated use of this code] It has a special requirement
  // on the Weight type: that there should be a Compare function on the weights
  // such that Compare(w1, w2) returns -1 if w1 < w2, 0 if w1 == w2, and +1 if w1 >
  // w2.  This requires that there be a total order on the weights.
  
  template<class Weight, class IntType> class LatticeDeterminizer {
   public:
    // Output to Gallic acceptor (so the strings go on weights, and there is a 1-1 correspondence
    // between our states and the states in ofst.  If destroy == true, release memory as we go
    // (but we cannot output again).
  
    typedef CompactLatticeWeightTpl<Weight, IntType> CompactWeight;
    typedef ArcTpl<CompactWeight> CompactArc; // arc in compact, acceptor form of lattice
    typedef ArcTpl<Weight> Arc; // arc in non-compact version of lattice
  
  
    // Output to standard FST with CompactWeightTpl<Weight> as its weight type (the
    // weight stores the original output-symbol strings).  If destroy == true,
    // release memory as we go (but we cannot output again).
    void Output(MutableFst<CompactArc>  *ofst, bool destroy = true) {
      assert(determinized_);
      typedef typename Arc::StateId StateId;
      StateId nStates = static_cast<StateId>(output_arcs_.size());
      if (destroy)
        FreeMostMemory();
      ofst->DeleteStates();
      ofst->SetStart(kNoStateId);
      if (nStates == 0) {
        return;
      }
      for (StateId s = 0;s < nStates;s++) {
        OutputStateId news = ofst->AddState();
        assert(news == s);
      }
      ofst->SetStart(0);
      // now process transitions.
      for (StateId this_state = 0; this_state < nStates; this_state++) {
        vector<TempArc> &this_vec(output_arcs_[this_state]);
        typename vector<TempArc>::const_iterator iter = this_vec.begin(), end = this_vec.end();
  
        for (;iter != end; ++iter) {
          const TempArc &temp_arc(*iter);
          CompactArc new_arc;
          vector<Label> seq;
          repository_.ConvertToVector(temp_arc.string, &seq);
          CompactWeight weight(temp_arc.weight, seq);
          if (temp_arc.nextstate == kNoStateId) {  // is really final weight.
            ofst->SetFinal(this_state, weight);
          } else {  // is really an arc.
            new_arc.nextstate = temp_arc.nextstate;
            new_arc.ilabel = temp_arc.ilabel;
            new_arc.olabel = temp_arc.ilabel;  // acceptor.  input == output.
            new_arc.weight = weight;  // includes string and weight.
            ofst->AddArc(this_state, new_arc);
          }
        }
        // Free up memory.  Do this inside the loop as ofst is also allocating memory
        if (destroy) { vector<TempArc> temp; std::swap(temp, this_vec); }
      }
      if (destroy) { vector<vector<TempArc> > temp; std::swap(temp, output_arcs_); }
    }
  
    // Output to standard FST with Weight as its weight type.  We will create extra
    // states to handle sequences of symbols on the output.  If destroy == true,
    // release memory as we go (but we cannot output again).
    void  Output(MutableFst<Arc> *ofst, bool destroy = true) {
      // Outputs to standard fst.
      OutputStateId nStates = static_cast<OutputStateId>(output_arcs_.size());
      ofst->DeleteStates();
      if (nStates == 0) {
        ofst->SetStart(kNoStateId);
        return;
      }
      if (destroy)
        FreeMostMemory();
      // Add basic states-- but we will add extra ones to account for strings on output.
      for (OutputStateId s = 0;s < nStates;s++) {
        OutputStateId news = ofst->AddState();
        assert(news == s);
      }
      ofst->SetStart(0);
      for (OutputStateId this_state = 0; this_state < nStates; this_state++) {
        vector<TempArc> &this_vec(output_arcs_[this_state]);
  
        typename vector<TempArc>::const_iterator iter = this_vec.begin(), end = this_vec.end();
        for (; iter != end; ++iter) {
          const TempArc &temp_arc(*iter);
          vector<Label> seq;
          repository_.ConvertToVector(temp_arc.string, &seq);
  
          if (temp_arc.nextstate == kNoStateId) {  // Really a final weight.
            // Make a sequence of states going to a final state, with the strings
            // as labels.  Put the weight on the first arc.
            OutputStateId cur_state = this_state;
            for (size_t i = 0; i < seq.size(); i++) {
              OutputStateId next_state = ofst->AddState();
              Arc arc;
              arc.nextstate = next_state;
              arc.weight = (i == 0 ? temp_arc.weight : Weight::One());
              arc.ilabel = 0;  // epsilon.
              arc.olabel = seq[i];
              ofst->AddArc(cur_state, arc);
              cur_state = next_state;
            }
            ofst->SetFinal(cur_state, (seq.size() == 0 ? temp_arc.weight : Weight::One()));
          } else {  // Really an arc.
            OutputStateId cur_state = this_state;
            // Have to be careful with this integer comparison (i+1 < seq.size()) because unsigned.
            // i < seq.size()-1 could fail for zero-length sequences.
            for (size_t i = 0; i+1 < seq.size();i++) {
              // for all but the last element of seq, create new state.
              OutputStateId next_state = ofst->AddState();
              Arc arc;
              arc.nextstate = next_state;
              arc.weight = (i == 0 ? temp_arc.weight : Weight::One());
              arc.ilabel = (i == 0 ? temp_arc.ilabel : 0);  // put ilabel on first element of seq.
              arc.olabel = seq[i];
              ofst->AddArc(cur_state, arc);
              cur_state = next_state;
            }
            // Add the final arc in the sequence.
            Arc arc;
            arc.nextstate = temp_arc.nextstate;
            arc.weight = (seq.size() <= 1 ? temp_arc.weight : Weight::One());
            arc.ilabel = (seq.size() <= 1 ? temp_arc.ilabel : 0);
            arc.olabel = (seq.size() > 0 ? seq.back() : 0);
            ofst->AddArc(cur_state, arc);
          }
        }
        // Free up memory.  Do this inside the loop as ofst is also allocating memory
        if (destroy) {
          vector<TempArc> temp; temp.swap(this_vec);
        }
      }
      if (destroy) {
        vector<vector<TempArc> > temp;
        temp.swap(output_arcs_);
        repository_.Destroy();
      }
    }
  
  
    // Initializer.  After initializing the object you will typically
    // call Determinize() and then call one of the Output functions.
    // Note: ifst.Copy() will generally do a
    // shallow copy.  We do it like this for memory safety, rather than
    // keeping a reference or pointer to ifst_.
    LatticeDeterminizer(const Fst<Arc> &ifst,
                        DeterminizeLatticeOptions opts):
        num_arcs_(0), num_elems_(0), ifst_(ifst.Copy()), opts_(opts),
        equal_(opts_.delta), determinized_(false),
        minimal_hash_(3, hasher_, equal_), initial_hash_(3, hasher_, equal_) {
      KALDI_ASSERT(Weight::Properties() & kIdempotent); // this algorithm won't
      // work correctly otherwise.
    }
  
    // frees all except output_arcs_, which contains the important info
    // we need to output the FST.
    void FreeMostMemory() {
      if (ifst_) {
        delete ifst_;
        ifst_ = NULL;
      }
      for (typename MinimalSubsetHash::iterator iter = minimal_hash_.begin();
          iter != minimal_hash_.end(); ++iter)
        delete iter->first;
      { MinimalSubsetHash tmp; tmp.swap(minimal_hash_); }
      for (typename InitialSubsetHash::iterator iter = initial_hash_.begin();
          iter != initial_hash_.end(); ++iter)
        delete iter->first;
      { InitialSubsetHash tmp; tmp.swap(initial_hash_); }
      { vector<vector<Element>* > output_states_tmp;
        output_states_tmp.swap(output_states_); }
      { vector<char> tmp;  tmp.swap(isymbol_or_final_); }
      { vector<OutputStateId> tmp; tmp.swap(queue_); }
      { vector<pair<Label, Element> > tmp; tmp.swap(all_elems_tmp_); }
    }
  
    ~LatticeDeterminizer() {
      FreeMostMemory(); // rest is deleted by destructors.
    }
    void RebuildRepository() { // rebuild the string repository,
      // freeing stuff we don't need.. we call this when memory usage
      // passes a supplied threshold.  We need to accumulate all the
      // strings we need the repository to "remember", then tell it
      // to clean the repository.
      std::vector<StringId> needed_strings;
      for (size_t i = 0; i < output_arcs_.size(); i++)
        for (size_t j = 0; j < output_arcs_[i].size(); j++)
          needed_strings.push_back(output_arcs_[i][j].string);
  
      // the following loop covers strings present in minimal_hash_
      // which are also accessible via output_states_.
      for (size_t i = 0; i < output_states_.size(); i++)
        for (size_t j = 0; j < output_states_[i]->size(); j++)
          needed_strings.push_back((*(output_states_[i]))[j].string);
  
      // the following loop covers strings present in initial_hash_.
      for (typename InitialSubsetHash::const_iterator
               iter = initial_hash_.begin();
           iter != initial_hash_.end(); ++iter) {
        const vector<Element> &vec = *(iter->first);
        Element elem = iter->second;
        for (size_t i = 0; i < vec.size(); i++)
          needed_strings.push_back(vec[i].string);
        needed_strings.push_back(elem.string);
      }
  
      std::sort(needed_strings.begin(), needed_strings.end());
      needed_strings.erase(std::unique(needed_strings.begin(),
                                       needed_strings.end()),
                           needed_strings.end()); // uniq the strings.
      repository_.Rebuild(needed_strings);
    }
  
    bool CheckMemoryUsage() {
      int32 repo_size = repository_.MemSize(),
          arcs_size = num_arcs_ * sizeof(TempArc),
          elems_size = num_elems_ * sizeof(Element),
          total_size = repo_size + arcs_size + elems_size;
      if (opts_.max_mem > 0 && total_size > opts_.max_mem) { // We passed the memory threshold.
        // This is usually due to the repository getting large, so we
        // clean this out.
        RebuildRepository();
        int32 new_repo_size = repository_.MemSize(),
            new_total_size = new_repo_size + arcs_size + elems_size;
  
        KALDI_VLOG(2) << "Rebuilt repository in determinize-lattice: repository shrank from "
                      << repo_size << " to " << new_repo_size << " bytes (approximately)";
  
        if (new_total_size > static_cast<int32>(opts_.max_mem * 0.8)) {
          // Rebuilding didn't help enough-- we need a margin to stop
          // having to rebuild too often.
          KALDI_WARN << "Failure in determinize-lattice: size exceeds maximum "
                     << opts_.max_mem << " bytes; (repo,arcs,elems) = ("
                     << repo_size << "," << arcs_size << "," << elems_size
                     << "), after rebuilding, repo size was " << new_repo_size;
          return false;
        }
      }
      return true;
    }
  
    // Returns true on success.  Can fail for out-of-memory
    // or max-states related reasons.
    bool Determinize(bool *debug_ptr) {
      assert(!determinized_);
      // This determinizes the input fst but leaves it in the "special format"
      // in "output_arcs_".  Must be called after Initialize().  To get the
      // output, call one of the Output routines.
      try {
        InitializeDeterminization(); // some start-up tasks.
        while (!queue_.empty()) {
          OutputStateId out_state = queue_.back();
          queue_.pop_back();
          ProcessState(out_state);
          if (debug_ptr && *debug_ptr) Debug();  // will exit.
          if (!CheckMemoryUsage()) return false;
        }
        return (determinized_ = true);
      } catch (const std::bad_alloc &) {
        int32 repo_size = repository_.MemSize(),
            arcs_size = num_arcs_ * sizeof(TempArc),
            elems_size = num_elems_ * sizeof(Element),
            total_size = repo_size + arcs_size + elems_size;
        KALDI_WARN << "Memory allocation error doing lattice determinization; using "
            << total_size << " bytes (max = " << opts_.max_mem
            << " (repo,arcs,elems) = ("
            << repo_size << "," << arcs_size << "," << elems_size << ")";
        return (determinized_ = false);
      } catch (const std::runtime_error &) {
        KALDI_WARN << "Caught exception doing lattice determinization";
        return (determinized_ = false);
      }
    }
   private:
  
    typedef typename Arc::Label Label;
    typedef typename Arc::StateId StateId;  // use this when we don't know if it's input or output.
    typedef typename Arc::StateId InputStateId;  // state in the input FST.
    typedef typename Arc::StateId OutputStateId;  // same as above but distinguish
                                                  // states in output Fst.
  
  
    typedef LatticeStringRepository<IntType> StringRepositoryType;
    typedef const typename StringRepositoryType::Entry* StringId;
  
    // Element of a subset [of original states]
    struct Element {
      StateId state; // use StateId as this is usually InputStateId but in one case
                     // OutputStateId.
      StringId string;
      Weight weight;
      bool operator != (const Element &other) const {
        return (state != other.state || string != other.string ||
                weight != other.weight);
      }
      // This operator is only intended to support sorting in EpsilonClosure()
      bool operator < (const Element &other) const {
        return state < other.state;
      }
    };
  
    // Arcs in the format we temporarily create in this class (a representation, essentially of
    // a Gallic Fst).
    struct TempArc {
      Label ilabel;
      StringId string;  // Look it up in the StringRepository, it's a sequence of Labels.
      OutputStateId nextstate;  // or kNoState for final weights.
      Weight weight;
    };
  
    // Hashing function used in hash of subsets.
    // A subset is a pointer to vector<Element>.
    // The Elements are in sorted order on state id, and without repeated states.
    // Because the order of Elements is fixed, we can use a hashing function that is
    // order-dependent.  However the weights are not included in the hashing function--
    // we hash subsets that differ only in weight to the same key.  This is not optimal
    // in terms of the O(N) performance but typically if we have a lot of determinized
    // states that differ only in weight then the input probably was pathological in some way,
    // or even non-determinizable.
    //   We don't quantize the weights, in order to avoid inexactness in simple cases.
    // Instead we apply the delta when comparing subsets for equality, and allow a small
    // difference.
  
    class SubsetKey {
     public:
      size_t operator ()(const vector<Element> * subset) const {  // hashes only the state and string.
        size_t hash = 0, factor = 1;
        for (typename vector<Element>::const_iterator iter= subset->begin(); iter != subset->end(); ++iter) {
          hash *= factor;
          hash += iter->state + reinterpret_cast<size_t>(iter->string);
          factor *= 23531;  // these numbers are primes.
        }
        return hash;
      }
    };
  
    // This is the equality operator on subsets.  It checks for exact match on state-id
    // and string, and approximate match on weights.
    class SubsetEqual {
     public:
      bool operator ()(const vector<Element> * s1, const vector<Element> * s2) const {
        size_t sz = s1->size();
        assert(sz>=0);
        if (sz != s2->size()) return false;
        typename vector<Element>::const_iterator iter1 = s1->begin(),
            iter1_end = s1->end(), iter2=s2->begin();
        for (; iter1 < iter1_end; ++iter1, ++iter2) {
          if (iter1->state != iter2->state ||
             iter1->string != iter2->string ||
              ! ApproxEqual(iter1->weight, iter2->weight, delta_)) return false;
        }
        return true;
      }
      float delta_;
      SubsetEqual(float delta): delta_(delta) {}
      SubsetEqual(): delta_(kDelta) {}
    };
  
    // Operator that says whether two Elements have the same states.
    // Used only for debug.
    class SubsetEqualStates {
     public:
      bool operator ()(const vector<Element> * s1, const vector<Element> * s2) const {
        size_t sz = s1->size();
        assert(sz>=0);
        if (sz != s2->size()) return false;
        typename vector<Element>::const_iterator iter1 = s1->begin(),
            iter1_end = s1->end(), iter2=s2->begin();
        for (; iter1 < iter1_end; ++iter1, ++iter2) {
          if (iter1->state != iter2->state) return false;
        }
        return true;
      }
    };
  
    // Define the hash type we use to map subsets (in minimal
    // representation) to OutputStateId.
    typedef unordered_map<const vector<Element>*, OutputStateId,
                          SubsetKey, SubsetEqual> MinimalSubsetHash;
  
    // Define the hash type we use to map subsets (in initial
    // representation) to OutputStateId, together with an
    // extra weight. [note: we interpret the Element.state in here
    // as an OutputStateId even though it's declared as InputStateId;
    // these types are the same anyway].
    typedef unordered_map<const vector<Element>*, Element,
                          SubsetKey, SubsetEqual> InitialSubsetHash;
  
  
    // converts the representation of the subset from canonical (all states) to
    // minimal (only states with output symbols on arcs leaving them, and final
    // states).  Output is not necessarily normalized, even if input_subset was.
    void ConvertToMinimal(vector<Element> *subset) {
      assert(!subset->empty());
      typename vector<Element>::iterator cur_in = subset->begin(),
          cur_out = subset->begin(), end = subset->end();
      while (cur_in != end) {
        if(IsIsymbolOrFinal(cur_in->state)) {  // keep it...
          *cur_out = *cur_in;
          cur_out++;
        }
        cur_in++;
      }
      subset->resize(cur_out - subset->begin());
    }
  
    // Takes a minimal, normalized subset, and converts it to an OutputStateId.
    // Involves a hash lookup, and possibly adding a new OutputStateId.
    // If it creates a new OutputStateId, it adds it to the queue.
    OutputStateId MinimalToStateId(const vector<Element> &subset) {
      typename MinimalSubsetHash::const_iterator iter
          = minimal_hash_.find(&subset);
      if (iter != minimal_hash_.end()) // Found a matching subset.
        return iter->second;
      OutputStateId ans = static_cast<OutputStateId>(output_arcs_.size());
      vector<Element> *subset_ptr = new vector<Element>(subset);
      output_states_.push_back(subset_ptr);
      num_elems_ += subset_ptr->size();
      output_arcs_.push_back(vector<TempArc>());
      minimal_hash_[subset_ptr] = ans;
      queue_.push_back(ans);
      return ans;
    }
  
  
    // Given a normalized initial subset of elements (i.e. before epsilon closure),
    // compute the corresponding output-state.
    OutputStateId InitialToStateId(const vector<Element> &subset_in,
                                   Weight *remaining_weight,
                                   StringId *common_prefix) {
      typename InitialSubsetHash::const_iterator iter
          = initial_hash_.find(&subset_in);
      if (iter != initial_hash_.end()) { // Found a matching subset.
        const Element &elem = iter->second;
        *remaining_weight = elem.weight;
        *common_prefix = elem.string;
        if (elem.weight == Weight::Zero())
          KALDI_WARN << "Zero weight!"; // TEMP
        return elem.state;
      }
      // else no matching subset-- have to work it out.
      vector<Element> subset(subset_in);
      // Follow through epsilons.  Will add no duplicate states.  note: after
      // EpsilonClosure, it is the same as "canonical" subset, except not
      // normalized (actually we never compute the normalized canonical subset,
      // only the normalized minimal one).
      EpsilonClosure(&subset); // follow epsilons.
      ConvertToMinimal(&subset); // remove all but emitting and final states.
  
      Element elem; // will be used to store remaining weight and string, and
                   // OutputStateId, in initial_hash_;
      NormalizeSubset(&subset, &elem.weight, &elem.string); // normalize subset; put
      // common string and weight in "elem".  The subset is now a minimal,
      // normalized subset.
  
      OutputStateId ans = MinimalToStateId(subset);
      *remaining_weight = elem.weight;
      *common_prefix = elem.string;
      if (elem.weight == Weight::Zero())
        KALDI_WARN << "Zero weight!"; // TEMP
  
      // Before returning "ans", add the initial subset to the hash,
      // so that we can bypass the epsilon-closure etc., next time
      // we process the same initial subset.
      vector<Element> *initial_subset_ptr = new vector<Element>(subset_in);
      elem.state = ans;
      initial_hash_[initial_subset_ptr] = elem;
      num_elems_ += initial_subset_ptr->size(); // keep track of memory usage.
      return ans;
    }
  
    // returns the Compare value (-1 if a < b, 0 if a == b, 1 if a > b) according
    // to the ordering we defined on strings for the CompactLatticeWeightTpl.
    // see function
    // inline int Compare (const CompactLatticeWeightTpl<WeightType,IntType> &w1,
    //                     const CompactLatticeWeightTpl<WeightType,IntType> &w2)
    // in lattice-weight.h.
    // this is the same as that, but optimized for our data structures.
    inline int Compare(const Weight &a_w, StringId a_str,
                       const Weight &b_w, StringId b_str) const {
      int weight_comp = fst::Compare(a_w, b_w);
      if (weight_comp != 0) return weight_comp;
      // now comparing strings.
      if (a_str == b_str) return 0;
      vector<IntType> a_vec, b_vec;
      repository_.ConvertToVector(a_str, &a_vec);
      repository_.ConvertToVector(b_str, &b_vec);
      // First compare their lengths.
      int a_len = a_vec.size(), b_len = b_vec.size();
      // use opposite order on the string lengths (c.f. Compare in
      // lattice-weight.h)
      if (a_len > b_len) return -1;
      else if (a_len < b_len) return 1;
      for(int i = 0; i < a_len; i++) {
        if (a_vec[i] < b_vec[i]) return -1;
        else if (a_vec[i] > b_vec[i]) return 1;
      }
      assert(0); // because we checked if a_str == b_str above, shouldn't reach here
      return 0;
    }
  
  
    // This function computes epsilon closure of subset of states by following epsilon links.
    // Called by InitialToStateId and Initialize.
    // Has no side effects except on the string repository.  The "output_subset" is not
    // necessarily normalized (in the sense of there being no common substring), unless
    // input_subset was.
    void EpsilonClosure(vector<Element> *subset) {
      // at input, subset must have only one example of each StateId.  [will still
      // be so at output].  This function follows input-epsilons, and augments the
      // subset accordingly.
  
      std::deque<Element> queue;
      unordered_map<InputStateId, Element> cur_subset;
      typedef typename unordered_map<InputStateId, Element>::iterator MapIter;
      typedef typename vector<Element>::const_iterator VecIter;
  
      for (VecIter iter = subset->begin(); iter != subset->end(); ++iter) {
        queue.push_back(*iter);
        cur_subset[iter->state] = *iter;
      }
  
      // find whether input fst is known to be sorted on input label.
      bool sorted = ((ifst_->Properties(kILabelSorted, false) & kILabelSorted) != 0);
      bool replaced_elems = false; // relates to an optimization, see below.
      int counter = 0; // stops infinite loops here for non-lattice-determinizable input;
      // useful in testing.
      while (queue.size() != 0) {
        Element elem = queue.front();
        queue.pop_front();
  
        // The next if-statement is a kind of optimization.  It's to prevent us
        // unnecessarily repeating the processing of a state.  "cur_subset" always
        // contains only one Element with a particular state.  The issue is that
        // whenever we modify the Element corresponding to that state in "cur_subset",
        // both the new (optimal) and old (less-optimal) Element will still be in
        // "queue".  The next if-statement stops us from wasting compute by
        // processing the old Element.
        if (replaced_elems && cur_subset[elem.state] != elem)
          continue;
        if (opts_.max_loop > 0 && counter++ > opts_.max_loop) {
          KALDI_ERR << "Lattice determinization aborted since looped more than "
                    << opts_.max_loop << " times during epsilon closure";
        }
        for (ArcIterator<Fst<Arc> > aiter(*ifst_, elem.state); !aiter.Done(); aiter.Next()) {
          const Arc &arc = aiter.Value();
          if (sorted && arc.ilabel != 0) break;  // Break from the loop: due to sorting there will be no
          // more transitions with epsilons as input labels.
          if (arc.ilabel == 0
              && arc.weight != Weight::Zero()) {  // Epsilon transition.
            Element next_elem;
            next_elem.state = arc.nextstate;
            next_elem.weight = Times(elem.weight, arc.weight);
            // now must append strings
            if (arc.olabel == 0)
              next_elem.string = elem.string;
            else
              next_elem.string = repository_.Successor(elem.string, arc.olabel);
  
            MapIter iter = cur_subset.find(next_elem.state);
            if (iter == cur_subset.end()) {
              // was no such StateId: insert and add to queue.
              cur_subset[next_elem.state] = next_elem;
              queue.push_back(next_elem);
            } else {
              // was not inserted because one already there.  In normal determinization we'd
              // add the weights.  Here, we find which one has the better weight, and
              // keep its corresponding string.
              int comp = Compare(next_elem.weight, next_elem.string,
                                 iter->second.weight, iter->second.string);
              if(comp == 1) { // next_elem is better, so use its (weight, string)
                iter->second.string = next_elem.string;
                iter->second.weight = next_elem.weight;
                queue.push_back(next_elem);
                replaced_elems = true;
              }
              // else it is the same or worse, so use original one.
            }
          }
        }
      }
  
      { // copy cur_subset to subset.
        subset->clear();
        subset->reserve(cur_subset.size());
        MapIter iter = cur_subset.begin(), end = cur_subset.end();
        for (; iter != end; ++iter) subset->push_back(iter->second);
        // sort by state ID, because the subset hash function is order-dependent(see SubsetKey)
        std::sort(subset->begin(), subset->end());
      }
    }
  
  
    // This function works out the final-weight of the determinized state.
    // called by ProcessSubset.
    // Has no side effects except on the variable repository_, and output_arcs_.
  
    void ProcessFinal(OutputStateId output_state) {
      const vector<Element> &minimal_subset = *(output_states_[output_state]);
      // processes final-weights for this subset.
  
      // minimal_subset may be empty if the graphs is not connected/trimmed, I think,
      // do don't check that it's nonempty.
      bool is_final = false;
      StringId final_string = NULL;  // = NULL to keep compiler happy.
      Weight final_weight = Weight::Zero();
      typename vector<Element>::const_iterator iter = minimal_subset.begin(), end = minimal_subset.end();
      for (; iter != end; ++iter) {
        const Element &elem = *iter;
        Weight this_final_weight = Times(elem.weight, ifst_->Final(elem.state));
        StringId this_final_string = elem.string;
        if (this_final_weight != Weight::Zero() &&
           (!is_final || Compare(this_final_weight, this_final_string,
                                 final_weight, final_string) == 1)) { // the new
          // (weight, string) pair is more in semiring than our current
          // one.
          is_final = true;
          final_weight = this_final_weight;
          final_string = this_final_string;
        }
      }
      if (is_final) {
        // store final weights in TempArc structure, just like a transition.
        TempArc temp_arc;
        temp_arc.ilabel = 0;
        temp_arc.nextstate = kNoStateId;  // special marker meaning "final weight".
        temp_arc.string = final_string;
        temp_arc.weight = final_weight;
        output_arcs_[output_state].push_back(temp_arc);
        num_arcs_++;
      }
    }
  
    // NormalizeSubset normalizes the subset "elems" by
    // removing any common string prefix (putting it in common_str),
    // and dividing by the total weight (putting it in tot_weight).
    void NormalizeSubset(vector<Element> *elems,
                         Weight *tot_weight,
                         StringId *common_str) {
      if(elems->empty()) { // just set common_str, tot_weight
        KALDI_WARN << "[empty subset]"; // TEMP
        // to defaults and return...
        *common_str = repository_.EmptyString();
        *tot_weight = Weight::Zero();
        return;
      }
      size_t size = elems->size();
      vector<IntType> common_prefix;
      repository_.ConvertToVector((*elems)[0].string, &common_prefix);
      Weight weight = (*elems)[0].weight;
      for (size_t i = 1; i < size; i++) {
        weight = Plus(weight, (*elems)[i].weight);
        repository_.ReduceToCommonPrefix((*elems)[i].string, &common_prefix);
      }
      assert(weight != Weight::Zero()); // we made sure to ignore arcs with zero
      // weights on them, so we shouldn't have zero here.
      size_t prefix_len = common_prefix.size();
      for (size_t i = 0; i < size; i++) {
        (*elems)[i].weight = Divide((*elems)[i].weight, weight, DIVIDE_LEFT);
        (*elems)[i].string =
            repository_.RemovePrefix((*elems)[i].string, prefix_len);
      }
      *common_str = repository_.ConvertFromVector(common_prefix);
      *tot_weight = weight;
    }
  
    // Take a subset of Elements that is sorted on state, and
    // merge any Elements that have the same state (taking the best
    // (weight, string) pair in the semiring).
    void MakeSubsetUnique(vector<Element> *subset) {
      typedef typename vector<Element>::iterator IterType;
  
      // This assert is designed to fail (usually) if the subset is not sorted on
      // state.
      assert(subset->size() < 2 || (*subset)[0].state <= (*subset)[1].state);
  
      IterType cur_in = subset->begin(), cur_out = cur_in, end = subset->end();
      size_t num_out = 0;
      // Merge elements with same state-id
      while (cur_in != end) {  // while we have more elements to process.
        // At this point, cur_out points to location of next place we want to put an element,
        // cur_in points to location of next element we want to process.
        if (cur_in != cur_out) *cur_out = *cur_in;
        cur_in++;
        while (cur_in != end && cur_in->state == cur_out->state) {
          if (Compare(cur_in->weight, cur_in->string,
                     cur_out->weight, cur_out->string) == 1) {
            // if *cur_in > *cur_out in semiring, then take *cur_in.
            cur_out->string = cur_in->string;
            cur_out->weight = cur_in->weight;
          }
          cur_in++;
        }
        cur_out++;
        num_out++;
      }
      subset->resize(num_out);
    }
  
    // ProcessTransition is called from "ProcessTransitions".  Broken out for
    // clarity.  Processes a transition from state "state".  The set of Elements
    // represents a set of next-states with associated weights and strings, each
    // one arising from an arc from some state in a determinized-state; the
    // next-states are not necessarily unique (i.e. there may be >1 entry
    // associated with each), and any such sets of Elements have to be merged
    // within this routine (we take the [weight, string] pair that's better in the
    // semiring).
    void ProcessTransition(OutputStateId state, Label ilabel, vector<Element> *subset) {
      MakeSubsetUnique(subset); // remove duplicates with the same state.
  
      StringId common_str;
      Weight tot_weight;
      NormalizeSubset(subset, &tot_weight, &common_str);
  
      OutputStateId nextstate;
      {
        Weight next_tot_weight;
        StringId next_common_str;
        nextstate = InitialToStateId(*subset,
                                     &next_tot_weight,
                                     &next_common_str);
        common_str = repository_.Concatenate(common_str, next_common_str);
        tot_weight = Times(tot_weight, next_tot_weight);
      }
  
      // Now add an arc to the next state (would have been created if necessary by
      // InitialToStateId).
      TempArc temp_arc;
      temp_arc.ilabel = ilabel;
      temp_arc.nextstate = nextstate;
      temp_arc.string = common_str;
      temp_arc.weight = tot_weight;
      output_arcs_[state].push_back(temp_arc);  // record the arc.
      num_arcs_++;
    }
  
  
    // "less than" operator for pair<Label, Element>.   Used in ProcessTransitions.
    // Lexicographical order, which only compares the state when ordering the
    // "Element" member of the pair.
  
    class PairComparator {
     public:
      inline bool operator () (const pair<Label, Element> &p1, const pair<Label, Element> &p2) {
        if (p1.first < p2.first) return true;
        else if (p1.first > p2.first) return false;
        else {
          return p1.second.state < p2.second.state;
        }
      }
    };
  
  
    // ProcessTransitions processes emitting transitions (transitions
    // with ilabels) out of this subset of states.
    // Does not consider final states.  Breaks the emitting transitions up by ilabel,
    // and creates a new transition in the determinized FST for each unique ilabel.
    // Does this by creating a big vector of pairs <Label, Element> and then sorting them
    // using a lexicographical ordering, and calling ProcessTransition for each range
    // with the same ilabel.
    // Side effects on repository, and (via ProcessTransition) on Q_, hash_,
    // and output_arcs_.
  
    void ProcessTransitions(OutputStateId output_state) {
      const vector<Element> &minimal_subset = *(output_states_[output_state]);
      // it's possible that minimal_subset could be empty if there are
      // unreachable parts of the graph, so don't check that it's nonempty.
      vector<pair<Label, Element> > &all_elems(all_elems_tmp_); // use class member
      // to avoid memory allocation/deallocation.
      {
        // Push back into "all_elems", elements corresponding to all
        // non-epsilon-input transitions out of all states in "minimal_subset".
        typename vector<Element>::const_iterator iter = minimal_subset.begin(), end = minimal_subset.end();
        for (;iter != end; ++iter) {
          const Element &elem = *iter;
          for (ArcIterator<Fst<Arc> > aiter(*ifst_, elem.state); ! aiter.Done(); aiter.Next()) {
            const Arc &arc = aiter.Value();
            if (arc.ilabel != 0
                && arc.weight != Weight::Zero()) {  // Non-epsilon transition -- ignore epsilons here.
              pair<Label, Element> this_pr;
              this_pr.first = arc.ilabel;
              Element &next_elem(this_pr.second);
              next_elem.state = arc.nextstate;
              next_elem.weight = Times(elem.weight, arc.weight);
              if (arc.olabel == 0) // output epsilon
                next_elem.string = elem.string;
              else
                next_elem.string = repository_.Successor(elem.string, arc.olabel);
              all_elems.push_back(this_pr);
            }
          }
        }
      }
      PairComparator pc;
      std::sort(all_elems.begin(), all_elems.end(), pc);
      // now sorted first on input label, then on state.
      typedef typename vector<pair<Label, Element> >::const_iterator PairIter;
      PairIter cur = all_elems.begin(), end = all_elems.end();
      vector<Element> this_subset;
      while (cur != end) {
        // Process ranges that share the same input symbol.
        Label ilabel = cur->first;
        this_subset.clear();
        while (cur != end && cur->first == ilabel) {
          this_subset.push_back(cur->second);
          cur++;
        }
        // We now have a subset for this ilabel.
        assert(!this_subset.empty()); // temp.
        ProcessTransition(output_state, ilabel, &this_subset);
      }
      all_elems.clear(); // as it's a class variable-- want it to stay
      // emtpy.
    }
  
  
  
    // ProcessState does the processing of a determinized state, i.e. it creates
    // transitions out of it and the final-probability if any.
    void ProcessState(OutputStateId output_state) {
      ProcessFinal(output_state);
      ProcessTransitions(output_state);
    }
  
  
    void Debug() {  // this function called if you send a signal
      // SIGUSR1 to the process (and it's caught by the handler in
      // fstdeterminizestar).  It prints out some traceback
      // info and exits.
  
      KALDI_WARN << "Debug function called (probably SIGUSR1 caught)";
      // free up memory from the hash as we need a little memory
      { MinimalSubsetHash hash_tmp; hash_tmp.swap(minimal_hash_); }
  
      if (output_arcs_.size() <= 2) {
        KALDI_ERR << "Nothing to trace back";
      }
      size_t max_state = output_arcs_.size() - 2;  // Don't take the last
      // one as we might be halfway into constructing it.
  
      vector<OutputStateId> predecessor(max_state+1, kNoStateId);
      for (size_t i = 0; i < max_state; i++) {
        for (size_t j = 0; j < output_arcs_[i].size(); j++) {
          OutputStateId nextstate = output_arcs_[i][j].nextstate;
          // Always find an earlier-numbered predecessor; this
          // is always possible because of the way the algorithm
          // works.
          if (nextstate <= max_state && nextstate > i)
            predecessor[nextstate] = i;
        }
      }
      vector<pair<Label, StringId> > traceback;
      // 'traceback' is a pair of (ilabel, olabel-seq).
      OutputStateId cur_state = max_state;  // A recently constructed state.
  
      while (cur_state != 0 && cur_state != kNoStateId) {
        OutputStateId last_state = predecessor[cur_state];
        pair<Label, StringId> p;
        size_t i;
        for (i = 0; i < output_arcs_[last_state].size(); i++) {
          if (output_arcs_[last_state][i].nextstate == cur_state) {
            p.first = output_arcs_[last_state][i].ilabel;
            p.second = output_arcs_[last_state][i].string;
            traceback.push_back(p);
            break;
          }
        }
        KALDI_ASSERT(i != output_arcs_[last_state].size());  // Or fell off loop.
        cur_state = last_state;
      }
      if (cur_state == kNoStateId)
        KALDI_WARN << "Traceback did not reach start state "
                   << "(possibly debug-code error)";
  
      std::stringstream ss;
      ss << "Traceback follows in format "
         << "ilabel (olabel olabel) ilabel (olabel) ... :";
      for (ssize_t i = traceback.size() - 1; i >= 0; i--) {
        ss << ' ' << traceback[i].first << " ( ";
        vector<Label> seq;
        repository_.ConvertToVector(traceback[i].second, &seq);
        for (size_t j = 0; j < seq.size(); j++)
          ss << seq[j] << ' ';
        ss << ')';
      }
      KALDI_ERR << ss.str();
    }
  
    bool IsIsymbolOrFinal(InputStateId state) { // returns true if this state
      // of the input FST either is final or has an osymbol on an arc out of it.
      // Uses the vector isymbol_or_final_ as a cache for this info.
      assert(state >= 0);
      if (isymbol_or_final_.size() <= state)
        isymbol_or_final_.resize(state+1, static_cast<char>(OSF_UNKNOWN));
      if (isymbol_or_final_[state] == static_cast<char>(OSF_NO))
        return false;
      else if (isymbol_or_final_[state] == static_cast<char>(OSF_YES))
        return true;
      // else work it out...
      isymbol_or_final_[state] = static_cast<char>(OSF_NO);
      if (ifst_->Final(state) != Weight::Zero())
        isymbol_or_final_[state] = static_cast<char>(OSF_YES);
      for (ArcIterator<Fst<Arc> > aiter(*ifst_, state);
           !aiter.Done();
           aiter.Next()) {
        const Arc &arc = aiter.Value();
        if (arc.ilabel != 0 && arc.weight != Weight::Zero()) {
          isymbol_or_final_[state] = static_cast<char>(OSF_YES);
          return true;
        }
      }
      return IsIsymbolOrFinal(state); // will only recurse once.
    }
  
    void InitializeDeterminization() {
      if(ifst_->Properties(kExpanded, false) != 0) { // if we know the number of
        // states in ifst_, it might be a bit more efficient
        // to pre-size the hashes so we're not constantly rebuilding them.
  #if !(__GNUC__ == 4 && __GNUC_MINOR__ == 0)
        StateId num_states =
            down_cast<const ExpandedFst<Arc>*, const Fst<Arc> >(ifst_)->NumStates();
        minimal_hash_.rehash(num_states/2 + 3);
        initial_hash_.rehash(num_states/2 + 3);
  #endif
      }
      InputStateId start_id = ifst_->Start();
      if (start_id != kNoStateId) {
        /* Insert determinized-state corresponding to the start state into hash and
           queue.  Unlike all the other states, we don't "normalize" the representation
           of this determinized-state before we put it into minimal_hash_.  This is actually
           what we want, as otherwise we'd have problems dealing with any extra weight
           and string and might have to create a "super-initial" state which would make
           the output nondeterministic.  Normalization is only needed to make the
           determinized output more minimal anyway, it's not needed for correctness.
           Note, we don't put anything in the initial_hash_.  The initial_hash_ is only
           a lookaside buffer anyway, so this isn't a problem-- it will get populated
           later if it needs to be.
        */
        Element elem;
        elem.state = start_id;
        elem.weight = Weight::One();
        elem.string = repository_.EmptyString();  // Id of empty sequence.
        vector<Element> subset;
        subset.push_back(elem);
        EpsilonClosure(&subset); // follow through epsilon-inputs links
        ConvertToMinimal(&subset); // remove all but final states and
        // states with input-labels on arcs out of them.
        vector<Element> *subset_ptr = new vector<Element>(subset);
        assert(output_arcs_.empty() && output_states_.empty());
        // add the new state...
        output_states_.push_back(subset_ptr);
        output_arcs_.push_back(vector<TempArc>());
        OutputStateId initial_state = 0;
        minimal_hash_[subset_ptr] = initial_state;
        queue_.push_back(initial_state);
      }
    }
  
    KALDI_DISALLOW_COPY_AND_ASSIGN(LatticeDeterminizer);
  
  
    vector<vector<Element>* > output_states_; // maps from output state to
                                              // minimal representation [normalized].
                                              // View pointers as owned in
                                              // minimal_hash_.
    vector<vector<TempArc> > output_arcs_;  // essentially an FST in our format.
  
    int num_arcs_; // keep track of memory usage: number of arcs in output_arcs_
    int num_elems_; // keep track of memory usage: number of elems in output_states_
  
    const Fst<Arc> *ifst_;
    DeterminizeLatticeOptions opts_;
    SubsetKey hasher_;  // object that computes keys-- has no data members.
    SubsetEqual equal_;  // object that compares subsets-- only data member is delta_.
    bool determinized_; // set to true when user called Determinize(); used to make
    // sure this object is used correctly.
    MinimalSubsetHash minimal_hash_;  // hash from Subset to OutputStateId.  Subset is "minimal
                                      // representation" (only include final and states and states with
                                      // nonzero ilabel on arc out of them.  Owns the pointers
                                      // in its keys.
    InitialSubsetHash initial_hash_;   // hash from Subset to Element, which
                                       // represents the OutputStateId together
                                       // with an extra weight and string.  Subset
                                       // is "initial representation".  The extra
                                       // weight and string is needed because after
                                       // we convert to minimal representation and
                                       // normalize, there may be an extra weight
                                       // and string.  Owns the pointers
                                      // in its keys.
    vector<OutputStateId> queue_; // Queue of output-states to process.  Starts with
    // state 0, and increases and then (hopefully) decreases in length during
    // determinization.  LIFO queue (queue discipline doesn't really matter).
  
    vector<pair<Label, Element> > all_elems_tmp_; // temporary vector used in ProcessTransitions.
  
    enum IsymbolOrFinal { OSF_UNKNOWN = 0, OSF_NO = 1, OSF_YES = 2 };
  
    vector<char> isymbol_or_final_; // A kind of cache; it says whether
    // each state is (emitting or final) where emitting means it has at least one
    // non-epsilon output arc.  Only accessed by IsIsymbolOrFinal()
  
    LatticeStringRepository<IntType> repository_;  // defines a compact and fast way of
    // storing sequences of labels.
  };
  
  
  // normally Weight would be LatticeWeight<float> (which has two floats),
  // or possibly TropicalWeightTpl<float>, and IntType would be int32.
  template<class Weight, class IntType>
  bool DeterminizeLattice(const Fst<ArcTpl<Weight> > &ifst,
                          MutableFst<ArcTpl<Weight> > *ofst,
                          DeterminizeLatticeOptions opts,
                          bool *debug_ptr) {
    ofst->SetInputSymbols(ifst.InputSymbols());
    ofst->SetOutputSymbols(ifst.OutputSymbols());
    LatticeDeterminizer<Weight, IntType> det(ifst, opts);
    if (!det.Determinize(debug_ptr))
      return false;
    det.Output(ofst);
    return true;
  }
  
  
  // normally Weight would be LatticeWeight<float> (which has two floats),
  // or possibly TropicalWeightTpl<float>, and IntType would be int32.
  template<class Weight, class IntType>
  bool DeterminizeLattice(const Fst<ArcTpl<Weight> >&ifst,
                          MutableFst<ArcTpl<CompactLatticeWeightTpl<Weight, IntType> > >*ofst,
                          DeterminizeLatticeOptions opts,
                          bool *debug_ptr) {
    ofst->SetInputSymbols(ifst.InputSymbols());
    ofst->SetOutputSymbols(ifst.OutputSymbols());
    LatticeDeterminizer<Weight, IntType> det(ifst, opts);
    if (!det.Determinize(debug_ptr))
      return false;
    det.Output(ofst);
    return true;
  }
  
  }  // namespace fst
  
  #endif  // KALDI_FSTEXT_DETERMINIZE_LATTICE_INL_H_