*исправление ошибок: двой перезаписи, исправление записи с BOM, добавлена проверка в insert_tags_into_audio
This commit is contained in:
1 parent
ffa8d80875
commit
078e68efa8
5 files changed
+181
-55
No files matched your search
+59
-21
@@ -72,22 +72,16 @@ pub fn get_current_tag_size[T](mut tag T, mut audio os.File) {
|
||||
tag.size = (u64(first_ten_bytes[6]) << 21) | (u64(first_ten_bytes[7]) << 14) | (u64(first_ten_bytes[8]) << 7) | u64(first_ten_bytes[9])
|
||||
}
|
||||
|
||||
// Кодирование размера заголовка со всеми тэгами
|
||||
pub fn encode_header_size(size u64, idv3_version u8) []u8 {
|
||||
stride := if idv3_version == 3 {
|
||||
[u64(24), u64(16), u64(8), u64(0xFF)]
|
||||
} else {
|
||||
[u64(21), u64(14), u64(7), u64(0x7F)]
|
||||
}
|
||||
|
||||
// Кодирование размера заголовка со всеми тэгами.
|
||||
// В заголовке размер всегда syncsafe (по 7 бит в байте), в том числе в v2.3
|
||||
pub fn encode_header_size(size u64) []u8 {
|
||||
mut encoded_size := []u8{len: 4}
|
||||
encoded_size[0] = u8((size >> stride[0]) & stride[3])
|
||||
encoded_size[1] = u8((size >> stride[1]) & stride[3])
|
||||
encoded_size[2] = u8((size >> stride[2]) & stride[3])
|
||||
encoded_size[3] = u8(size & stride[3])
|
||||
encoded_size[0] = u8((size >> 21) & 0x7F)
|
||||
encoded_size[1] = u8((size >> 14) & 0x7F)
|
||||
encoded_size[2] = u8((size >> 7) & 0x7F)
|
||||
encoded_size[3] = u8(size & 0x7F)
|
||||
|
||||
return encoded_size
|
||||
|
||||
}
|
||||
|
||||
// Кодирование первых 10 байт ID3
|
||||
@@ -99,8 +93,7 @@ pub fn encode_header[T](mut id3v T) {
|
||||
// Минорная версия и спец. флаги
|
||||
id3v.encoded_header << []u8{len: 2}
|
||||
|
||||
// Размер всего тэга в заголовке всегда syncsafe (по 7 бит в байте), в том числе в v2.3
|
||||
id3v.encoded_header << metadata.encode_header_size(id3v.size, 4)
|
||||
id3v.encoded_header << metadata.encode_header_size(id3v.size)
|
||||
}
|
||||
|
||||
// Количество нулей в конце тэга
|
||||
@@ -116,16 +109,23 @@ pub fn encode_embed_tags[T](mut id3v2 T) {
|
||||
$if field.typ is Frame {
|
||||
// Читаются только заполненные тэги
|
||||
if id3v2.$(field.name).cur_size > 0 {
|
||||
if id3v2.$(field.name).name.to_lower() != "apic" {
|
||||
// Кодируем значения полей
|
||||
tag_bytes << id3v2.$(field.name).get_encoded_frame_tag(id3v2.version)
|
||||
} else {
|
||||
// Картинка не трогается
|
||||
tag_bytes << id3v2.$(field.name).apic or { []u8{} }
|
||||
// Фрейма не было в файле и init не вызывался: имя берётся из поля структуры
|
||||
if id3v2.$(field.name).name == "" && field.name.len == 4 {
|
||||
id3v2.$(field.name).init(field.name, id3v2.version)
|
||||
}
|
||||
|
||||
// Без 4-символьного ID фрейм испортит весь тэг
|
||||
if id3v2.$(field.name).name.len == 4 {
|
||||
tag_bytes << id3v2.$(field.name).to_bytes(id3v2.version)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Фреймы, для которых нет отдельного поля в структуре
|
||||
for mut frame in id3v2.others {
|
||||
tag_bytes << frame.to_bytes(id3v2.version)
|
||||
}
|
||||
// Запас нулей чтобы не пересчитывать весь тэг при маленьком редактировании
|
||||
tag_bytes << []u8{len: tag_padding}
|
||||
|
||||
@@ -136,3 +136,41 @@ pub fn encode_embed_tags[T](mut id3v2 T) {
|
||||
encode_header(mut id3v2)
|
||||
id3v2.encoded_header << tag_bytes
|
||||
}
|
||||
|
||||
// Добавляет аудио из файла после нового тэга в encoded_header, старый тэг пропускается
|
||||
pub fn insert_tags_into_audio[T](mut id3v2 T, mut audio os.File) ! {
|
||||
header := get_bytes(mut audio, 0, 10)
|
||||
if header.len < 10 || header[..3].bytestr() != "ID3" {
|
||||
return error("в файле нет тэга ID3v2")
|
||||
}
|
||||
|
||||
// Размер старого тэга берётся из файла, id3v2.size уже хранит размер нового.
|
||||
// В заголовке он syncsafe в любой версии, плюс 10 байт самого заголовка
|
||||
mut audio_start := get_size_v24(header[6..10]) + 10
|
||||
// В v2.4 после тэга может идти футер ещё на 10 байт
|
||||
if header[3] == 4 && header[5] & 0x10 != 0 {
|
||||
audio_start += 10
|
||||
}
|
||||
|
||||
audio.seek(0, .end)!
|
||||
file_size := u64(audio.tell()!)
|
||||
if audio_start > file_size {
|
||||
return error("размер тэга (${audio_start}) больше размера файла (${file_size})")
|
||||
}
|
||||
|
||||
// Аудио читается сразу в конец encoded_header, без промежуточной копии всего файла
|
||||
sound_len := int(file_size - audio_start)
|
||||
header_len := id3v2.encoded_header.len
|
||||
unsafe { id3v2.encoded_header.grow_len(sound_len) }
|
||||
|
||||
audio.seek(i64(audio_start), .start)!
|
||||
mut read := 0
|
||||
for read < sound_len {
|
||||
n := unsafe { audio.read_into_ptr(&u8(id3v2.encoded_header.data) + header_len + read, sound_len - read)! }
|
||||
if n == 0 {
|
||||
return error("файл закончился раньше ожидаемого")
|
||||
}
|
||||
read += n
|
||||
}
|
||||
audio.seek(0, .start)!
|
||||
}
|
||||
Reference in new issue
Block a user