*исправлена запись новых тегов и apic
This commit is contained in:
1 parent
abab08f133
commit
c06ef69c67
3 files changed
+63
-41
No files matched your search
+16
-9
@@ -51,11 +51,13 @@ fn (mut frame Frame) fill(bytes []u8, start_position u64, idv3_version u8) {
|
||||
if frame.name.to_lower() == "apic" {
|
||||
// Дополнительные байты с информацией (MIME)
|
||||
frame.addition = bytes[frame.start + 10..frame.start + 24]
|
||||
frame.apic = bytes[frame.start + 24..frame.end]
|
||||
// Фрейм целиком вместе с заголовком, чтобы записать его обратно без изменений
|
||||
frame.apic = bytes[frame.start..frame.end]
|
||||
} else {
|
||||
// Выбор шага в записимости от версии ID3
|
||||
stride := if idv3_version == 3 { 13 } else { 11 }
|
||||
frame.data = iconv.encoding_to_vstring(bytes[frame.start+u64(stride)..frame.end], frame.encoding_name) or { "" }
|
||||
// Замыкающие нули не относятся к данным
|
||||
frame.data = (iconv.encoding_to_vstring(bytes[frame.start+u64(stride)..frame.end], frame.encoding_name) or { "" }).trim_right('\x00')
|
||||
|
||||
frame.addition = bytes[frame.start..frame.start + 10]
|
||||
}
|
||||
@@ -88,14 +90,23 @@ pub fn (mut frame Frame) encode_size(idv3_version u8) {
|
||||
|
||||
// Кодируем кадр для дальнейшей записи
|
||||
pub fn (mut frame Frame) encode_frame(idv3_version u8) {
|
||||
encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or { [u8(0), u8(0), u8(0), u8(0)] }
|
||||
encoded_data := iconv.vstring_to_encoding(frame.data, frame.encoding_name) or { []u8{} }
|
||||
|
||||
frame.encoded_data = encoded_data
|
||||
|
||||
// Размер тела фрейма: байт кодировки + данные + замыкающие нули
|
||||
frame.cur_size = u64(1 + frame.encoded_data.len + frame.terminator_len())
|
||||
|
||||
// число -> [байт, байт, байт, байт]
|
||||
frame.encode_size(idv3_version)
|
||||
}
|
||||
|
||||
// Для UTF-16 строка завершается двумя нулями, для остальных кодировок одним
|
||||
@[inline]
|
||||
fn (frame Frame) terminator_len() int {
|
||||
return if frame.encoding_name_idx == 1 || frame.encoding_name_idx == 2 { 2 } else { 1 }
|
||||
}
|
||||
|
||||
// Получить закодированый тэг
|
||||
pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
|
||||
frame.encode_frame(idv3_version)
|
||||
@@ -112,12 +123,8 @@ pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
|
||||
encoded_tag << frame.encoding_name_idx
|
||||
// Уже сами данные
|
||||
encoded_tag << frame.encoded_data
|
||||
// Запас нулей чтобы не пересчитывать весь фрейм при маленьком редактировании
|
||||
encoded_tag << []u8{len: 10}
|
||||
// Замыкающий байт
|
||||
encoded_tag << []u8{len: 1}
|
||||
|
||||
frame.cur_size = u64(encoded_tag.len)
|
||||
// Замыкающие нули
|
||||
encoded_tag << []u8{len: frame.terminator_len()}
|
||||
|
||||
return encoded_tag
|
||||
}
|
||||
|
||||
+4
-30
@@ -70,39 +70,13 @@ pub fn (id3v24 ID3v24) to_v1() ID3v1 {
|
||||
}
|
||||
}
|
||||
|
||||
// Формируем полный полный тэг в байтах
|
||||
pub fn (mut id3v24 ID3v24) get_full_embed_tag() {
|
||||
// Проход по всем полям стуктуры
|
||||
mut tag_bytes := []u8{}
|
||||
|
||||
id3v24.size = 0
|
||||
|
||||
$for field in ID3v24.fields {
|
||||
$if field.typ is Frame {
|
||||
// Читаются только заполненные тэги
|
||||
if id3v24.$(field.name).cur_size > 0 {
|
||||
id3v24.size += id3v24.$(field.name).cur_size
|
||||
id3v24.size += 10
|
||||
|
||||
if id3v24.$(field.name).name.to_lower() != "apic" {
|
||||
// Кодируем значения полей
|
||||
tag_bytes << id3v24.$(field.name).get_encoded_frame_tag(id3v24.version)
|
||||
} else {
|
||||
// Картинка не трогается
|
||||
tag_bytes << id3v24.$(field.name).apic or { []u8{len: 4}}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
encode_header(mut id3v24)
|
||||
id3v24.encoded_header << tag_bytes
|
||||
}
|
||||
|
||||
|
||||
pub fn (mut id3v24 ID3v24) insert_tags_into_audio(mut audio File) {
|
||||
full_part := read_all(reader: audio) or { [u8(0), u8(0), u8(0), u8(0)] }
|
||||
full_part := read_all(reader: audio) or { panic(err) }
|
||||
|
||||
sound_part := full_part[int(id3v24.size) + 10 .. ]
|
||||
// Размер старого тэга берётся из файла, id3v24.size уже хранит размер нового
|
||||
old_size := get_size_v24(full_part[6..10])
|
||||
sound_part := full_part[int(old_size) + 10 .. ]
|
||||
|
||||
id3v24.encoded_header << sound_part
|
||||
}
|
||||
+43
-2
@@ -49,10 +49,17 @@ pub fn dump_data[T](mut tag T, mut audio os.File) {
|
||||
// Получаем вне возможные байты
|
||||
all_tag_bytes := get_bytes(mut audio, 10, int(tag.size + start_position))
|
||||
|
||||
for tag.size > start_position {
|
||||
// Конец тэга с учётом 10 байт заголовка
|
||||
tag_end := tag.size + 10
|
||||
|
||||
for start_position + 10 <= tag_end {
|
||||
// Нулевой байт вместо названия фрейма — начался запас нулей
|
||||
if all_tag_bytes[start_position] == 0 {
|
||||
break
|
||||
}
|
||||
|
||||
current_frame.fill(all_tag_bytes, start_position, tag.version)
|
||||
start_position = current_frame.end
|
||||
tag.size -= current_frame.cur_size
|
||||
|
||||
tag.add_frame(current_frame)
|
||||
}
|
||||
@@ -95,3 +102,37 @@ pub fn encode_header[T](mut id3v T) {
|
||||
id3v.encoded_header << metadata.encode_header_size(id3v.size, id3v.version)
|
||||
}
|
||||
|
||||
// Количество нулей в конце тэга
|
||||
const tag_padding = 256
|
||||
|
||||
type ID3vType = ID3v23 | ID3v24
|
||||
|
||||
// Формируем полный полный тэг в байтах
|
||||
pub fn get_full_embed_tag[ID3vType](mut id3v2 ID3vType) {
|
||||
// Проход по всем полям стуктуры
|
||||
mut tag_bytes := []u8{}
|
||||
|
||||
$for field in id3v2.fields {
|
||||
$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{} }
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
// Запас нулей чтобы не пересчитывать весь тэг при маленьком редактировании
|
||||
tag_bytes << []u8{len: tag_padding}
|
||||
|
||||
// Размер считается по уже закодированным фреймам
|
||||
id3v2.size = u64(tag_bytes.len)
|
||||
|
||||
id3v2.encoded_header = []u8{}
|
||||
encode_header(mut id3v2)
|
||||
id3v2.encoded_header << tag_bytes
|
||||
}
|
||||
Reference in new issue
Block a user