-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathmp3queue.ml
1721 lines (1543 loc) · 66.6 KB
/
mp3queue.ml
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
(*******************************************************************************
This file is a part of mp3packer.
mp3packer is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
mp3packer 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. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with mp3packer; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*******************************************************************************)
open Types;;
open Mp3read;;
open Pack;;
(*
let t1_ref = ref (counter ());;
let printf f =
Printf.ksprintf (fun s ->
let t = counter () in
let t_delta = t - !t1_ref in
t1_ref := t;
Printf.printf "%10d %s" t_delta s
) f
;;
*)
let make_xing xing header_and_side_info =
let first_out = Ptr.make 8 0 in
Ptr.blit_from_string xing.xingTagType 0 first_out 0 4;
(* frames, bytes, toc, quality *)
let (ptrref_after_xingNumFrames, num_frames_flag) = (match xing.xingNumFrames with
| None -> (Ptr.Ref.null, 0x00)
| Some x -> (
let p = Ptr.make 4 0 in
Ptr.put_32_of_int_bswap p 0 x;
(Ptr.Ref.append Ptr.Ref.null (Ptr.Ref.of_ptr p), 0x01)
)
) in
let (ptrref_after_xingNumBytes, num_bytes_flag) = (match xing.xingNumBytes with
| None -> (ptrref_after_xingNumFrames, 0x00)
| Some x -> (
let p = Ptr.make 4 0 in
Ptr.put_32_of_int_bswap p 0 x;
(Ptr.Ref.append ptrref_after_xingNumFrames (Ptr.Ref.of_ptr p), 0x02)
)
) in
let (ptrref_after_xingTOC, toc_flag) = (match xing.xingTOC with
| None -> (ptrref_after_xingNumBytes, 0x00)
| Some x -> (
let toc_ptr = Ptr.make (Array.length x) 0 in
Array.iteri (fun i c -> Ptr.put_8_of_int toc_ptr i c) x;
(Ptr.Ref.append ptrref_after_xingNumBytes (Ptr.Ref.of_ptr toc_ptr), 0x04)
)
) in
let (ptrref_after_xingQuality, quality_flag) = (match xing.xingQuality with
| None -> (ptrref_after_xingTOC, 0x00)
| Some x -> (
let p = Ptr.make 4 0 in
Ptr.put_32_of_int_bswap p 0 x;
(Ptr.Ref.append ptrref_after_xingTOC (Ptr.Ref.of_ptr p), 0x08)
)
) in
(* Ptr.put_8_of_int first_out 7 (num_frames_flag lor num_bytes_flag lor toc_flag lor quality_flag);*)
Ptr.put_32_of_int_bswap first_out 4 (num_frames_flag lor num_bytes_flag lor toc_flag lor quality_flag);
(* first_out_str.[7] <- Char.chr (num_frames_flag lor num_bytes_flag lor toc_flag lor quality_flag);*)
let ptrref_after_xing = Ptr.Ref.append (Ptr.Ref.of_ptr first_out) ptrref_after_xingQuality in
let after_lame = (
match xing.xingLame with
| None -> (
let encoder = Ptr.clearret (Ptr.make 20 0) in
Ptr.blit_from_string xing.xingEncoder 0 encoder 0 (min (Ptr.length encoder) (String.length xing.xingEncoder));
(* let encoder_string = String.sub (xing.xingEncoder ^ (String.make 20 '\x00')) 0 20 in*)
Ptr.Ref.append ptrref_after_xing (Ptr.Ref.of_ptr encoder);
)
| Some l -> (
let encoder = Ptr.clearret (Ptr.make 9 0) in
Ptr.blit_from_string xing.xingEncoder 0 encoder 0 (min (Ptr.length encoder) (String.length xing.xingEncoder));
let ptrref_after_encoder = Ptr.Ref.append ptrref_after_xing (Ptr.Ref.of_ptr encoder) in
let lame = Ptr.make 25 0 in
let p = Ptr.put_bits lame in
p 0 4 l.lameRevision;
p 4 4 l.lameVBRMethod;
p 8 8 l.lameLowpass;
Ptr.put_32_of_float_bswap lame 2 l.lamePeakAmplitude;
p 48 16 l.lameRGTrack;
p 64 16 l.lameRGAlbum;
p 80 1 (if l.lameNoGapPrev then 1 else 0);
p 81 1 (if l.lameNoGapNext then 1 else 0);
p 82 1 (if l.lameNSSafeJoint then 1 else 0);
p 83 1 (if l.lameNSPsyTune then 1 else 0);
p 84 4 l.lameATHType;
p 88 8 l.lameABRBitrate;
p 96 12 l.lameDelayStart;
p 108 12 l.lameDelayEnd;
p 120 2 l.lameSourceFrequency;
p 122 1 (if l.lameUnwise then 1 else 0);
p 123 3 l.lameStereoMode;
p 126 2 l.lameNoiseShaping;
p 128 8 l.lameMP3Gain;
p 136 2 0; (* Unused *)
p 138 3 l.lameSurround;
p 141 11 l.lamePreset;
p 152 2 0; (* Only 30 bits used *)
p 154 30 l.lameMusicLength;
p 184 16 l.lameMusicCRC;
let ptrref_after_lame = Ptr.Ref.append ptrref_after_encoder (Ptr.Ref.of_ptr lame) in
let ptrref_before_crc = Ptr.Ref.append header_and_side_info ptrref_after_lame in
let crc = Crc.lame_create_ptrref ptrref_before_crc 0 in
let crc_p = Ptr.make 2 0 in
Ptr.put_16_of_int_bswap crc_p 0 crc;
Ptr.Ref.append ptrref_after_lame (Ptr.Ref.of_ptr crc_p);
)
) in
after_lame
;;
let bit_blit =
let rec b s1 o1 s2 o2 l = (
if l > 30 then (
packBits s2 o2 30 (unpackBits s1 o1 30);
b s1 (o1 + 30) s2 (o2 + 30) (l - 30)
) else (
packBits s2 o2 l (unpackBits s1 o1 l);
)
) in
fun s1 o1 s2 o2 l -> (
if l < 0 || o1 < 0 || o2 < 0 || o1 + l > String.length s1 lsl 3 || o2 + l > String.length s2 lsl 3 then (
invalid_arg "bit_blit"
) else (
b s1 o1 s2 o2 l
)
)
;;
let side_info_find_ok :
type id chan. bool -> (id,chan) side_t -> Ptr.Ref.ref_t -> int -> (id,chan) side_t * Ptr.Ref.ref_t * bool
= fun zero_whole_bad_frame -> function
| {side_bits = Bits_1_mono (a,b)} as input_side -> (fun reservoir input_offset ->
let first_bit = input_offset lsl 3 in (* The offset in bits *)
let second_bit = first_bit + a in (* The first bit of the second granule *)
let last_bit = second_bit + b in (* The first bit after the second granule *)
let reservoir_length_in_bits = Ptr.Ref.length reservoir lsl 3 in
let (first_granule_ok, second_granule_ok) = (
(* The first granule's OK if there's no data (always known) or if the first and second bits are in the string *)
let fgok = (first_bit = second_bit || (first_bit >= 0 && second_bit <= reservoir_length_in_bits)) in
(* Ditto for second granule *)
let sgok = (second_bit = last_bit || (second_bit >= 0 && last_bit <= reservoir_length_in_bits)) in
(* If zero_whole_bad_frame is set, then if one granule is bad count them both as bad *)
if zero_whole_bad_frame then (fgok && sgok, fgok && sgok) else (fgok, sgok)
) in
let first_granule_bits = if first_granule_ok then max 0 a else 0 in
let second_granule_bits = if second_granule_ok then max 0 b else 0 in
(* Printf.printf "BOOGA OK? %B %B\n" first_granule_ok second_granule_ok;*)
(* The beginning of the valid data *)
(* (the end of the valid data is output_offset + first_granule_bits + second_granule_bits) *)
(* If neither granule is good, set to 0 rather than last_bit since last_bit may result in a substring off the end of the string, but with length 0 *)
let output_offset_bits = if first_granule_ok then first_bit else if second_granule_ok then second_bit else 0 in
let output_raw = if first_granule_ok && second_granule_ok then (
input_side.side_raw
) else (
let output_raw_ptr = Ptr.Ref.to_ptr input_side.side_raw in
if not first_granule_ok then (
(* Zero out the first granule's data *)
Ptr.put_bits output_raw_ptr 18 30 0;
Ptr.put_bits output_raw_ptr 48 29 0;
);
if not second_granule_ok then (
Ptr.put_bits output_raw_ptr 77 30 0;
Ptr.put_bits output_raw_ptr 107 29 0;
);
Ptr.Ref.of_ptr output_raw_ptr
) in
let output_side = {
side_raw = output_raw;
side_offset = 0;
side_bits = Bits_1_mono (first_granule_bits,second_granule_bits);
side_bytes = (first_granule_bits + second_granule_bits + 7) asr 3;
} in
let output_data = (
let output_length_bytes = output_side.side_bytes in
if output_length_bytes = 0 then (
(* Don't try to get any data if there is no data to get! *)
Ptr.Ref.null
) else if output_offset_bits land 7 = 0 then (
(* Byte-aligned; just sub the ptrref *)
Ptr.Ref.sub reservoir (output_offset_bits asr 3) output_length_bytes
) else (
(* UH-OH! Need to do a bit-blit *)
let out = String.create output_length_bytes in
out.[output_length_bytes - 1] <- '\x00';
let reservoir_string = Ptr.Ref.to_string reservoir in
bit_blit reservoir_string output_offset_bits out 0 (first_granule_bits + second_granule_bits);
Ptr.Ref.of_string out
)
) in
(output_side, output_data, first_granule_ok && second_granule_ok)
)
| {side_bits = Bits_1_stereo (a,b,c,d)} as input_side -> (fun reservoir input_offset ->
let first_bit = input_offset lsl 3 in
let second_bit = first_bit + a + b in
let last_bit = second_bit + c + d in
let reservoir_length_in_bits = Ptr.Ref.length reservoir lsl 3 in
let (first_granule_ok, second_granule_ok) = (
let fgok = (first_bit = second_bit || (first_bit >= 0 && second_bit <= reservoir_length_in_bits)) in
let sgok = (second_bit = last_bit || (second_bit >= 0 && last_bit <= reservoir_length_in_bits)) in
if zero_whole_bad_frame then (fgok && sgok, fgok && sgok) else (fgok, sgok)
) in
let new_a = if first_granule_ok then a else 0 in
let new_b = if first_granule_ok then b else 0 in
let new_c = if second_granule_ok then c else 0 in
let new_d = if second_granule_ok then d else 0 in
let output_offset_bits = if first_granule_ok then max 0 first_bit else if second_granule_ok then max 0 second_bit else 0 in
let output_raw = if first_granule_ok && second_granule_ok then (
input_side.side_raw
) else (
let output_raw_ptr = Ptr.Ref.to_ptr input_side.side_raw in
if not first_granule_ok then (
Ptr.put_bits output_raw_ptr 20 30 0;
Ptr.put_bits output_raw_ptr 50 29 0;
Ptr.put_bits output_raw_ptr 79 30 0;
Ptr.put_bits output_raw_ptr 109 29 0;
);
if not second_granule_ok then (
Ptr.put_bits output_raw_ptr 138 30 0;
Ptr.put_bits output_raw_ptr 168 29 0;
Ptr.put_bits output_raw_ptr 197 30 0;
Ptr.put_bits output_raw_ptr 227 29 0;
);
Ptr.Ref.of_ptr output_raw_ptr
) in
let output_side = {
side_raw = output_raw;
side_offset = 0;
side_bits = Bits_1_stereo (new_a,new_b,new_c,new_d);
side_bytes = (new_a + new_b + new_c + new_d + 7) asr 3
} in
let output_data = (
let output_length_bytes = output_side.side_bytes in
if output_length_bytes = 0 then (
(* No data here *)
Ptr.Ref.null
) else if output_offset_bits land 7 = 0 then (
(* Sub the ptrref *)
Ptr.Ref.sub reservoir (output_offset_bits asr 3) output_length_bytes
) else (
(* bit-blit! *)
let out = String.create output_length_bytes in
(* Zero the last byte so that no random memory junk gets in after the data bits *)
out.[output_length_bytes - 1] <- '\x00';
let reservoir_string = Ptr.Ref.to_string reservoir in
bit_blit reservoir_string output_offset_bits out 0 (new_a + new_b + new_c + new_d);
Ptr.Ref.of_string out
)
) in
(output_side, output_data, first_granule_ok && second_granule_ok)
)
| {side_bits = Bits_2_mono a} as input_side -> (fun reservoir input_offset ->
let first_bit = input_offset lsl 3 in
let last_bit = first_bit + a in
let reservoir_length_in_bits = Ptr.Ref.length reservoir lsl 3 in
let granule_ok = (first_bit = last_bit || (first_bit >= 0 && last_bit <= reservoir_length_in_bits)) in
let granule_bits = if granule_ok then max 0 a else 0 in
let output_offset_bits = if granule_ok then first_bit else 0 in
let output_raw = if granule_ok then (
input_side.side_raw
) else (
let output_raw_ptr = Ptr.Ref.to_ptr input_side.side_raw in
Ptr.put_bits output_raw_ptr 9 30 0;
Ptr.put_bits output_raw_ptr 39 30 0;
Ptr.put_bits output_raw_ptr 69 3 0;
Ptr.Ref.of_ptr output_raw_ptr
) in
let output_side = {
side_raw = output_raw;
side_offset = 0;
side_bits = Bits_2_mono granule_bits;
side_bytes = (granule_bits + 7) asr 3;
} in
let output_data = (
let output_length_bytes = output_side.side_bytes in
if output_length_bytes = 0 then (
Ptr.Ref.null
) else if output_offset_bits land 7 = 0 then (
Ptr.Ref.sub reservoir (output_offset_bits asr 3) output_length_bytes
) else (
let out = String.create output_length_bytes in
out.[output_length_bytes - 1] <- '\x00';
let reservoir_string = Ptr.Ref.to_string reservoir in
bit_blit reservoir_string output_offset_bits out 0 granule_bits;
Ptr.Ref.of_string out
)
) in
(output_side, output_data, granule_ok)
)
| {side_bits = Bits_2_stereo (a,b)} as input_side -> (fun reservoir input_offset ->
let first_bit = input_offset lsl 3 in
let last_bit = first_bit + a + b in
let reservoir_length_in_bits = Ptr.Ref.length reservoir lsl 3 in
let granule_ok = (first_bit = last_bit || (first_bit >= 0 && last_bit <= reservoir_length_in_bits)) in
let new_a = if granule_ok then max 0 a else 0 in
let new_b = if granule_ok then max 0 b else 0 in
let output_offset_bits = if granule_ok then first_bit else 0 in
let output_raw = if granule_ok then (
input_side.side_raw
) else (
let output_raw_ptr = Ptr.Ref.to_ptr input_side.side_raw in
Ptr.put_bits output_raw_ptr 10 30 0;
Ptr.put_bits output_raw_ptr 40 30 0;
Ptr.put_bits output_raw_ptr 70 3 0;
Ptr.put_bits output_raw_ptr 73 30 0;
Ptr.put_bits output_raw_ptr 103 30 0;
Ptr.put_bits output_raw_ptr 133 3 0;
Ptr.Ref.of_ptr output_raw_ptr
) in
let output_side = {
side_raw = output_raw;
side_offset = 0;
side_bits = Bits_2_stereo (new_a,new_b);
side_bytes = (new_a + new_b + 7) asr 3;
} in
let output_data = (
let output_length_bytes = output_side.side_bytes in
if output_length_bytes = 0 then (
Ptr.Ref.null
) else if output_offset_bits land 7 = 0 then (
Ptr.Ref.sub reservoir (output_offset_bits asr 3) output_length_bytes
) else (
let out = String.create output_length_bytes in
out.[output_length_bytes - 1] <- '\x00';
let reservoir_string = Ptr.Ref.to_string reservoir in
bit_blit reservoir_string output_offset_bits out 0 (new_a + new_b);
Ptr.Ref.of_string out
)
) in
(output_side, output_data, granule_ok)
)
;;
let do_queue recompress_obj state file_state (in_obj : Mp3read.mp3read_ptr_2) out_obj =
let debug_queue = state.q_debug_queue in
let min_bitrate = state.q_min_bitrate in
let delete_beginning_junk = state.q_delete_beginning_junk in
let delete_end_junk = state.q_delete_end_junk in
let padding = state.q_padding in
let recompress = state.q_recompress in
let zero_whole_bad_frame = state.q_zero_whole_bad_frame in
let minimize_bit_reservoir = state.q_minimize_bit_reservoir in
let p = state.q_print_queue in
(*
(* Number of buffer errors so far *)
let buffer_errors_ref = ref 0 in
(* The number of sync errors which occured *)
let sync_errors_ref = ref 0 in
(* The number of frame CRC errors *)
let crc_errors_ref = ref 0 in
(* The number of errors encountered while recompressing the frames (will be 0 without -z) *)
let recompress_errors_ref = ref 0 in
*)
(* Set whether the recompress should warn about frequency overflows *)
(* let recompress_freq_overflow_warn_ref = ref (not state.q_silent) in*)
if state.q_silent then ignore file_state#freq_warn;
let t1 = Unix.gettimeofday () in
(*
let in_obj = new mp3read_unix ~debug:debug_in in_name in
let out_obj = new Mp3write.mp3write_unix out_name in
*)
(*
let out_file = open_out_bin out_name in
let output_this = output_string out_file in
*)
(* Sync to first frame *)
let (new_req, IF_ext first_frame, (first_wanted_at, first_got_at), in_xing_option) = (
let before_lame_reqs = {
req_id = Req_equal;
req_crc = Req_any;
req_bitrate = Req_any;
req_samplerate = Req_equal;
req_padding = Req_any;
req_private = Req_any;
req_channel_mode = Req_any;
req_channel_count = Req_any;
req_copyright = Req_any;
req_original = Req_any;
req_emphasis = Req_any;
} in
let after_lame_reqs = {
req_id = Req_equal;
req_crc = Req_any;
req_bitrate = Req_any;
req_samplerate = Req_equal;
req_padding = Req_any;
req_private = Req_any;
req_channel_mode = Req_any;
req_channel_count = Req_equal;
req_copyright = Req_any;
req_original = Req_any; (* Req_equal has had some problems in the past *)
req_emphasis = Req_any;
} in
let (first_req, IF_ext first_frame, (_ (* 0 *), first_got)) = in_obj#find_next_frame ~force_resync:true ~lame_search:true before_lame_reqs in
match first_frame.if_xing with
| None -> (
(* The first frame was NOT an XING frame; restart and use more strict after_lame_reqs *)
in_obj#seek first_got;
let (real_first_req, real_first_frame_ext, (_ (* first_got *), real_first_got)) = in_obj#find_next_frame ~force_resync:true after_lame_reqs in
(real_first_req, real_first_frame_ext, (first_got, real_first_got), None)
)
| Some x -> (
(* Found an XING frame; do another *)
let (second_req, second_frame_ext, (second_wanted, second_got)) = in_obj#find_next_frame ~force_resync:true after_lame_reqs in
(second_req, second_frame_ext, (second_wanted, second_got), Some x)
)
) in
(* Pretend that didn't happen, since it's easier to read the frame at the beginning of the frame loop *)
in_obj#seek first_got_at;
let k = first_frame.if_header in (* This replaces the old "k" global setting variable *)
let side_info_size = Ptr.Ref.length first_frame.if_side_raw in
let unpadded_frame_length = unpadded_frame_length k.header_samplerate in
(***********************)
(* DATA INITIALIZATION *)
(***********************)
(* An exact bitrate is taken to be dithered according to the table in Types *)
(* One more than an exact bitrate is assumed to be always padded *)
(* Anything larger than the max bitrate will be truncated to a padded max frame *)
(* Everything else rounds up to the next highest valid bitrate *)
let (number_to_bitrate, bytes_to_bitrate) = (
let (max_bitrate, lists) = match k.header_id with
| MPEG1 -> (320, [(1, 32);(2, 40);(3, 48);(4, 45);(5, 64);(6, 80);(7, 96);(8,112);(9,128);(10,160);(11,192);(12,224);(13,256);(14,320)])
| MPEG2 _ -> (160, [(1, 8);(2, 16);(3, 24);(4, 32);(5, 40);(6, 48);(7, 56);(8, 64);(9, 80);(10, 96);(11,112);(12,128);(13,144);(14,160)])
in
(fun num ->
let exact = List.exists (fun (_,a) -> a = num) lists in (* Did the caller specify an exact bitrate? *)
let exactP1 = List.exists (fun (_,a) -> a + 1 = num) lists in (* Did the caller specify 1 more than an exact bitrate? *)
let over = (num > max_bitrate) in (* Is the caller Way Out There? *)
let (index,real_bitrate) = try (List.find (fun (_,a) -> num <= a + 1) lists) with Not_found -> (14,max_bitrate) in
let unpad_length = unpadded_frame_length real_bitrate in
(fun i ->
let padded = if exact then (
padded_frame k.header_samplerate num i
) else (
exactP1 || over
) in
let pad_add = if padded then 1 else 0 in
{
bitrate_num = real_bitrate;
bitrate_padding = padded;
bitrate_size = unpad_length + pad_add;
bitrate_data = unpad_length + pad_add - 4 - side_info_size;
bitrate_index = index
}
)
), (fun bytes ->
let bph = bytes + 4 + side_info_size in (* bytes plus header *)
let out = ref None in
List.iter (fun (index, real_bitrate) ->
match !out with
| Some _ -> ()
| None -> (
let bytes_unpadded = unpadded_frame_length real_bitrate in
if bytes_unpadded >= bph then (
(* OK without padding *)
out := Some {
bitrate_num = real_bitrate;
bitrate_padding = false;
bitrate_size = bytes_unpadded;
bitrate_data = bytes_unpadded - 4 - side_info_size;
bitrate_index = index
}
) else if bytes_unpadded + 1 >= bph then (
(* Needs padding *)
out := Some {
bitrate_num = real_bitrate;
bitrate_padding = true;
bitrate_size = bytes_unpadded + 1;
bitrate_data = bytes_unpadded + 1 - 4 - side_info_size;
bitrate_index = index
}
) (* else keep going *)
)
) lists;
match !out with
| None -> raise Too_many_bytes (* No valid frame found! *)
| Some x -> x
)
) in
(* Returns a valid header given the bitrate info, common settings, and stereo mode *)
let (bitrate_to_header_string, bitrate_to_header) = (
let template = "\xFF\xFF\xFF\xFF" in
let pack_head = packBits template in
pack_head 0 11 0b11111111111;
pack_head 11 2 (match k.header_id with
| MPEG1 -> 0b11
| MPEG2 MPEG20 -> 0b10
| MPEG2 MPEG25 -> 0b00
);
pack_head 13 2 0b01; (* Layer 3 *)
pack_head 15 1 0b1; (* CRC *)
pack_head 20 2 (id_index_of_samplerate k.header_samplerate);
pack_head 23 1 0b0; (* Privates *)
pack_head 24 2 (match k.header_channel_mode with
| Stereo Stereo_simple -> 0b00
| Stereo Stereo_joint _ -> 0b01
| Stereo Stereo_dual -> 0b10
| Mono -> 0b11
);
pack_head 28 1 (if k.header_copyright then 1 else 0);
pack_head 29 1 (if k.header_original then 1 else 0);
pack_head 30 2 (match k.header_emphasis with
| Emphasis_none -> 0b00
| Emphasis_5015 -> 0b01
| Emphasis_CCITT -> 0b11
);
(
(fun br ms is -> (
let out_head = String.copy template in
let pack_head = packBits out_head in
pack_head 16 4 br.bitrate_index; (* Bitrate *)
pack_head 22 1 (if br.bitrate_padding then 1 else 0);
pack_head 26 1 (if ms then 1 else 0);
pack_head 27 1 (if is then 1 else 0);
out_head
))
,
(fun br ms is -> (
let out_head = Ptr.of_string template in
let p = Ptr.put_bits out_head in
p 16 4 br.bitrate_index;
p 22 1 (if br.bitrate_padding then 1 else 0);
p 26 1 (if ms then 1 else 0);
p 27 1 (if is then 1 else 0);
out_head
))
)
) in
let min_bitrate_now = number_to_bitrate min_bitrate in
(* Makes a string out of a given header and bitrate info *)
(* Always indicates no CRC *)
let ptrref_of_header_and_bitrate header bitrate =
let p = Ptr.Ref.to_ptr header.header_raw in
Ptr.put_bits p 15 1 1;
Ptr.put_bits p 16 4 bitrate.bitrate_index;
Ptr.put_bits p 22 1 (if bitrate.bitrate_padding then 1 else 0);
Ptr.Ref.of_ptr p
in
(* Changes a side info string to be the specified padding *)
(*
let ptrref_of_side_and_offset =
let bits = match k.header_id with
| MPEG1 -> 9
| _ -> 8
in
fun side offset -> (
let new_ptr = Ptr.Ref.to_ptr side.side_raw in
Ptr.put_bits new_ptr 0 bits offset;
Ptr.Ref.of_ptr new_ptr
)
in
*)
(*
let ptrref_of_f1_and_offset : type id chan. (id,chan) f1_t -> _ = fun f1 offset ->
let bits = match f1.f1_header.header_id with
| MPEG1 -> 9
| MPEG2 _ -> 8
in
let new_ptr = Ptr.Ref.to_ptr f1.f1_side.side_raw in
Ptr.put_bits new_ptr 0 bits offset;
Ptr.Ref.of_ptr new_ptr
in
*)
(* For some reason I can't take the bits calculation out of the function... *)
let ptrref_of_side_and_offset : type id chan. (id,chan) side_t -> _ = fun side offset ->
let bits = match k.header_id with
| MPEG1 -> 9
| MPEG2 _ -> 8
in
let new_ptr = Ptr.Ref.to_ptr side.side_raw in
Ptr.put_bits new_ptr 0 bits offset;
Ptr.Ref.of_ptr new_ptr
in
(* Pretty-printer for time from a frame number *)
let seconds_per_frame = bspfk_of_samplerate k.header_samplerate *. 0.008(*kbit/byte*) in
let p_time_of_frame frame_num p =
let s_float = float_of_int frame_num *. seconds_per_frame in
let (s_floatfrac, s_floatint) = modf s_float in
let s_unnormal = int_of_float s_floatint in
if s_unnormal < 60 then ( (* 0:ss.cc *)
p (Char '0');
p (Char ':');
p (Int0N (2, s_unnormal));
) else (
let m_unnormal = s_unnormal / 60 in
if m_unnormal < 60 then ( (* m:ss.cc *)
p (Int m_unnormal)
) else (
let h_unnormal = m_unnormal / 60 in
if h_unnormal < 24 then ( (* h:mm:ss.cc *)
p (Int h_unnormal)
) else ( (* d hh:mm:ss.cc *)
p (Int (h_unnormal / 24));
p (Char 'd');
p (Spaces 1);
p (Int0N (2, h_unnormal mod 24));
);
p (Char ':');
p (Int0N (2, m_unnormal mod 60));
);
p (Char ':');
p (Int0N (2, s_unnormal mod 60));
);
p (Char '.');
p (Int0N (2, int_of_float (s_floatfrac *. 100.0)));
in
(* Make room for the beginning data, if it is to be saved *)
if not delete_beginning_junk then (
p [Str "Writing the first "; Int in_obj#first_mp3_byte; Str " bytes to the output file"];
let in_pos = in_obj#pos in
in_obj#seek 0;
let length = in_obj#first_mp3_byte in
let ref = in_obj#read_ptrref length in
out_obj#output_this ref;
in_obj#seek in_pos;
);
(* Make room for the LAME/XING header *)
let (xing_bitrate, xing_pos, output_is_lame, xing_header_and_side_info) = (
let is_lame = (match in_xing_option with
| None -> false
| Some xing -> (match xing.xingLame with
| None -> false
| Some lame -> true
)
) in
let min_lame_bitrate = bytes_to_bitrate (if is_lame then 156 else 140) in
let bitrate = max (min_bitrate_now 0) min_lame_bitrate in
let output_header = Ptr.Ref.of_ptr (bitrate_to_header bitrate false false) in
p [Str "XING header "; Ptrref output_header];
out_obj#output_this output_header;
out_obj#seek (out_obj#pos + side_info_size);
let xing_header_and_side_info = Ptr.Ref.append_ptr (Ptr.Ref.of_ptr (bitrate_to_header bitrate false false)) (Ptr.clearret (Ptr.make side_info_size 0)) in
let wheresit = out_obj#pos in
out_obj#seek (out_obj#pos + bitrate.bitrate_data);
p [Str "XING frame located at "; Int wheresit];
(bitrate, wheresit, is_lame, xing_header_and_side_info)
) in
(* Reservoir initialization *)
let max_reservoir_size = match k.header_id with
| MPEG1 -> 511
| _ -> 255
in
let max_data_per_frame = match k.header_id with
| MPEG1 -> unpadded_frame_length 320 + 1 - 4 - side_info_size
| _ -> unpadded_frame_length 160 + 1 - 4 - side_info_size
in
p [Str "Max "; Int max_data_per_frame; Str " bytes of data per frame"];
(*****************)
(* INFORMATIONAL *)
(*****************)
(* Largest and smallest bitrates used *)
let max_output_bitrate_ref = ref {
bitrate_data = 0;
bitrate_size = 0;
bitrate_num = 0;
bitrate_padding = false;
bitrate_index = 0
} in
let min_output_bitrate_ref = ref {
bitrate_data = 500000;
bitrate_size = 500000;
bitrate_num = 500000;
bitrate_padding = true;
bitrate_index = 15
} in
(* let total_frames_ref = ref 0 in*)
let total_frame_bytes_ref = ref xing_bitrate.bitrate_size in
let frame_locations = Expandarray.create (match first_frame.if_xing with
| None -> (in_obj#length / 418) (* Assume a 128kbps file *)
| Some xing -> (match xing.xingNumFrames with
| None -> (in_obj#length / 418) (* Again with the 128 *)
| Some y -> (y + 2) (* Plus or minus a few *)
)
) in
(* Make the initial frame filled with whatever padding is *)
let ptr_template_padding =
let len = String.length padding in
let num = (max_data_per_frame - 1) / len + 1 in
let ptr = Ptr.make (num * len) 0 in
for i = 0 to num - 1 do
Ptr.blit_from_string padding 0 ptr (i * len) len;
done;
ptr
in
let pad_f3_frame f3 goto =
let len_now = Ptr.Ref.length f3.f3_output_data in
let len_add = goto - len_now in
if len_add < 0 then (
failwith "ERROR: immutable frame data was set to be overwritten";
) else if len_add = 0 then (
()
) else (
p [Str "padding frame..."];
let ptrref_padding = Ptr.Ref.of_subptr ptr_template_padding len_now len_add in
f3.f3_output_data <- Ptr.Ref.append f3.f3_output_data ptrref_padding
)
in
let pad_full_f3_frame f3 = pad_f3_frame f3 f3.f3_bitrate.bitrate_data in
let q0 = List2.create () in (* For the recompression *)
let q1 = List2.create () in
let q2 = List2.create () in
let q3 = List2.create () in
let q3_bytes_ref = ref 0 in (* The number of bytes currently in Q3 *)
let q3_current_reservoir_ref = ref 0 in (* The current byte reservoir in Q3. Although this is implied through the Q2 list, it must be explicitly set if Q2 is not used (minimize_bit_reservoir = false) *)
(************)
(************)
(** QUEUE! **)
(************)
(************)
let rec input_to_q0 frame_num update_percent bit_reservoir_so_far bit_reservoir_so_far_unused = (
(* XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX *)
(* Same as A but add to Q0 instead *)
p [];
let frame_stuff = (try
Some (in_obj#find_next_frame new_req)
with
End_of_file -> None
) in
match frame_stuff with
| Some (_, IF_ext if_now, (wanted_at, got_at)) -> (
p [Str "FRAME "; Int frame_num; Str " putting to Q0"];
let next_update_percent = if float_of_int in_obj#pos > float_of_int in_obj#length /. 100. *. float_of_int update_percent then (
(* if not debug_queue && not state.q_silent then printf "\r%2d%% done on frame %d%!" update_percent frame_num;*)
P.print_replace_always [IntN (2,update_percent); Str "% done on frame "; Int frame_num];
succ update_percent
) else (
update_percent
) in
if wanted_at <> got_at then (
(* if not state.q_silent then printf "\rWARNING: Sync error on frame %d at %s (wanted at %d, found at %d)\n" frame_num (string_time_of_frame frame_num) wanted_at got_at;*)
P.print_always [
Str "WARNING: Sync error on frame "; Int frame_num; Str " at "; Fun (p_time_of_frame frame_num);
Str " (wanted at "; Int wanted_at; Str ", found at "; Int got_at; Str ")"
];
file_state#add_sync_error;
);
if not if_now.if_crc_ok then (
(* if not state.q_silent then printf "\rWARNING: CRC error on frame %d at %s\n" frame_num (string_time_of_frame frame_num);*)
P.print_always [Str "WARNING: CRC error on frame "; Int frame_num; Str " at "; Fun (p_time_of_frame frame_num)];
file_state#add_crc_error;
);
let side = side_info_of_if_2 if_now in
if debug_queue then (
p [Str " Reservoir bytes available: "; Int bit_reservoir_so_far_unused];
p [Str " Reservoir bytes used: "; Int side.side_offset];
(* p [Str " Data bits: "; List (Array.fold_right (fun a b -> Int a :: Char ' ' :: b) side.side_bits [])];*)
p [Str " Data bits: "; Fun (fun p ->
match side.side_bits with
| Bits_1_mono (a,b) -> (p (Int a); p (Char ' '); p (Int b))
| Bits_1_stereo (a,b,c,d) -> (p (Int a); p (Char ' '); p (Int b); p (Char ' '); p (Int c); p (Char ' '); p (Int d))
| Bits_2_mono (a) -> (p (Int a))
| Bits_2_stereo (a,b) -> (p (Int a); p (Char ' '); p (Int b))
)];
p [Str " Total data bytes: "; Int side.side_bytes];
p [Str " Bit reservoir value length:"; Int (Ptr.Ref.length bit_reservoir_so_far)];
);
let combined_bit_reservoir = Ptr.Ref.append bit_reservoir_so_far if_now.if_data_raw in
if false then (
Printf.printf "Combined reservoir ptr:\n";
Ptr.Ref.print combined_bit_reservoir;
);
let start_offset = Ptr.Ref.length bit_reservoir_so_far - side.side_offset in
let (side_use, data_use, everythings_ok) = side_info_find_ok zero_whole_bad_frame side combined_bit_reservoir start_offset in
p [Str " Side: "; Ptrref side_use.side_raw];
p [Str " Data: "; Ptrref data_use];
let new_reservoir_unused = Ptr.Ref.length combined_bit_reservoir - start_offset - Ptr.Ref.length data_use in
if everythings_ok && if_now.if_crc_ok then (
(* printf "NOT CHECKING CRC\n";*)
p [Str " Putting to Q0"];
let frame_to_compress = {
f1_num = frame_num;
f1_header = if_now.if_header;
f1_side = side_use;
f1_data = data_use;
f1_pad_exact = None;
} in
let put_token = if true then (
recompress_obj#send (state, file_state, F1_ext frame_to_compress)
) else (
p [Str "ADDING TO Q0 SYNCHRONOUSLY"];
recompress_obj#bypass (state, file_state, F1_ext frame_to_compress) (recompress_obj#f (state, file_state, F1_ext frame_to_compress))
) in
List2.append q0 (F1_ext frame_to_compress, put_token);
) else (
(* Skip the repacking *)
p [Str " Skipping thread pool; putting to Q0"];
if if_now.if_crc_ok then (
(* if not state.q_silent then printf "\rWARNING: Buffer over/underflow on frame %d at %s\n" frame_num (string_time_of_frame frame_num);*)
P.print_always [Str "WARNING: Buffer over/underflow on frame "; Int frame_num; Str " at "; Fun (p_time_of_frame frame_num)];
file_state#add_buffer_error;
);
let frame_to_not_compress = {
f1_num = frame_num;
f1_header = if_now.if_header;
f1_side = side_use;
f1_data = data_use;
f1_pad_exact = None;
} in
let bypass_token = recompress_obj#bypass (state, file_state, F1_ext frame_to_not_compress) (Normal (F1_ext frame_to_not_compress, false)) in
List2.append q0 (F1_ext frame_to_not_compress, bypass_token)
);
let new_bit_reservoir = if Ptr.Ref.length combined_bit_reservoir < max_reservoir_size then (
combined_bit_reservoir
) else (
let over = Ptr.Ref.length combined_bit_reservoir - max_reservoir_size in
Ptr.Ref.sub combined_bit_reservoir over max_reservoir_size
) in
p [Str " X->Y"];
q0_to_q1 ();
p [Str " X->X"];
input_to_q0 (succ frame_num) next_update_percent new_bit_reservoir new_reservoir_unused
)
| None -> (
p [Str " X->Z (no more frames; target is 0)"];
flush_q0 ();
frame_num
)
) and q0_to_q1 () = (
(* YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY *)
if List2.length q0 > 4 * recompress_obj#max_threads then (
let (F1_ext orig, put_token) = List2.take_first q0 in
p [Str " requesting frame from Q0"];
(match recompress_obj#recv put_token with
| Normal (F1_ext new_f1, q_error) -> (
if Ptr.Ref.length new_f1.f1_data > Ptr.Ref.length orig.f1_data then (
p [
Str " Oops. The repacked frame is larger than the original (";
Int (Ptr.Ref.length new_f1.f1_data);
Str " > ";
Int (Ptr.Ref.length orig.f1_data);
Str "); reusing the input frame"
];
List2.append q1 (F1_ext orig)
) else (
List2.append q1 (F1_ext new_f1)
)
)
| Error e -> (
(* May print twice. Oh well *)
let print_this = [Str " Oops. Repacking the frame failed with error \""; Str (Printexc.to_string e); Str "\""] in
state.q_print_recompress print_this;
p print_this;
file_state#add_rehuff_error;
List2.append q1 (F1_ext orig);
)
);
p [Str " Y->B"];
mark_q1 ();
p [Str " Y->Y"];
q0_to_q1 ()
) else (
(* Not enough in the queue yet *)
p [Str " Y->_!"];
()
)
) and flush_q0 () = (
(* ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ *)
match List2.take_first_perhaps q0 with
| Some ((F1_ext orig), put_token) -> (
p [Str " requesting frame from Q0"];
match recompress_obj#recv put_token with
| Normal ((F1_ext new_f1), q_error) -> (
if Ptr.Ref.length new_f1.f1_data > Ptr.Ref.length orig.f1_data then (
p [
Str " Oops. The repacked frame is larger than the original (";
Int (Ptr.Ref.length new_f1.f1_data);
Str " > ";
Int (Ptr.Ref.length orig.f1_data);
Str "); reusing the input frame"
];
List2.append q1 (F1_ext orig)
) else (
List2.append q1 (F1_ext new_f1)
);
p [Str " Z->Z"];
flush_q0 ()
)
| Error e -> (
let print_this = [Str " Oops. Repacking the frame failed with error \""; Str (Printexc.to_string e); Str "\""] in
state.q_print_recompress print_this;
p print_this;
file_state#add_rehuff_error;
List2.append q1 (F1_ext orig);
p [Str " Z->Z"];
flush_q0 ()
)
)
| None -> (
p [Str " Z->H (Q0 flushed)"];
flush_q1 ()
)
) and input_to_q1 frame_num update_percent bit_reservoir_so_far bit_reservoir_so_far_unused = (
(* AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA *)
p [];
let frame_stuff = (try
Some (in_obj#find_next_frame new_req)
with
End_of_file -> None
) in
match frame_stuff with
| Some (_, IF_ext if_now, (wanted_at, got_at)) -> (
p [Str "FRAME "; Int frame_num];
(* This is not correct for some files when a frame is larger than 1% of the whole file length *)
(* However, these files must be fairly short (1000 frames in the worst case) when it goes very fast anyway *)
let next_update_percent = if float_of_int in_obj#pos > float_of_int in_obj#length /. 100. *. float_of_int update_percent then (
(* if not debug_queue && not state.q_silent then printf "\r%2d%% done on frame %d%!" update_percent frame_num;*)