*исправлена запись новых тегов и 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" {
|
if frame.name.to_lower() == "apic" {
|
||||||
// Дополнительные байты с информацией (MIME)
|
// Дополнительные байты с информацией (MIME)
|
||||||
frame.addition = bytes[frame.start + 10..frame.start + 24]
|
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 {
|
} else {
|
||||||
// Выбор шага в записимости от версии ID3
|
// Выбор шага в записимости от версии ID3
|
||||||
stride := if idv3_version == 3 { 13 } else { 11 }
|
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]
|
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) {
|
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.encoded_data = encoded_data
|
||||||
|
|
||||||
|
// Размер тела фрейма: байт кодировки + данные + замыкающие нули
|
||||||
|
frame.cur_size = u64(1 + frame.encoded_data.len + frame.terminator_len())
|
||||||
|
|
||||||
// число -> [байт, байт, байт, байт]
|
// число -> [байт, байт, байт, байт]
|
||||||
frame.encode_size(idv3_version)
|
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 {
|
pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
|
||||||
frame.encode_frame(idv3_version)
|
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.encoding_name_idx
|
||||||
// Уже сами данные
|
// Уже сами данные
|
||||||
encoded_tag << frame.encoded_data
|
encoded_tag << frame.encoded_data
|
||||||
// Запас нулей чтобы не пересчитывать весь фрейм при маленьком редактировании
|
// Замыкающие нули
|
||||||
encoded_tag << []u8{len: 10}
|
encoded_tag << []u8{len: frame.terminator_len()}
|
||||||
// Замыкающий байт
|
|
||||||
encoded_tag << []u8{len: 1}
|
|
||||||
|
|
||||||
frame.cur_size = u64(encoded_tag.len)
|
|
||||||
|
|
||||||
return encoded_tag
|
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) {
|
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
|
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))
|
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)
|
current_frame.fill(all_tag_bytes, start_position, tag.version)
|
||||||
start_position = current_frame.end
|
start_position = current_frame.end
|
||||||
tag.size -= current_frame.cur_size
|
|
||||||
|
|
||||||
tag.add_frame(current_frame)
|
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)
|
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