*многое сделано, не помню что
This commit is contained in:
K1ndWha1e committed
1 parent
fdc72ae653
commit
abab08f133
8 files changed
+249
-85
No files matched your search
+52
-23
@@ -4,12 +4,14 @@ import encoding.iconv
|
||||
|
||||
@[heap]
|
||||
pub struct Frame {
|
||||
pub mut:
|
||||
data string
|
||||
mut:
|
||||
size u64
|
||||
prev_size u64
|
||||
cur_size u64
|
||||
start u64
|
||||
end u64
|
||||
encoding_name string
|
||||
data string
|
||||
name string
|
||||
encoding_name_idx u8
|
||||
addition []u8
|
||||
@@ -19,37 +21,37 @@ pub struct Frame {
|
||||
modified bool
|
||||
}
|
||||
|
||||
// Во врейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE
|
||||
// Во фрейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE
|
||||
@[inline]
|
||||
fn (mut frame Frame) fill_frame(bytes []u8, start_position u64, idv3_version u8) {
|
||||
fn (mut frame Frame) fill(bytes []u8, start_position u64, idv3_version u8) {
|
||||
// Начало фрейма
|
||||
frame.start = start_position
|
||||
// Название фрейма
|
||||
frame.name = bytes[frame.start..frame.start + 4].bytestr()
|
||||
|
||||
// Размер фрейма
|
||||
frame.size = if idv3_version == 3 {
|
||||
get_size_v23(bytes[frame.start+4..frame.start+8])
|
||||
frame.cur_size = if idv3_version == 3 {
|
||||
get_size_v23(bytes[frame.start + 4..frame.start + 8])
|
||||
} else {
|
||||
get_size_v24(bytes[frame.start+4..frame.start+8])
|
||||
get_size_v24(bytes[frame.start + 4..frame.start + 8])
|
||||
}
|
||||
// Конец фрейма
|
||||
frame.end = frame.start + frame.size + 10
|
||||
frame.end = frame.start + frame.cur_size + 10
|
||||
|
||||
// Содержимое фрейма
|
||||
frame.encoding_name = match bytes[frame.start+10] {
|
||||
frame.encoding_name = match bytes[frame.start + 10] {
|
||||
0 { "ISO-8859-1" }
|
||||
1 { "UTF-16LE" }
|
||||
2 { "UTF-16BE" }
|
||||
else { "UTF-8" }
|
||||
}
|
||||
|
||||
frame.encoding_name_idx = u8(bytes[frame.start+10])
|
||||
frame.encoding_name_idx = u8(bytes[frame.start + 10])
|
||||
|
||||
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.addition = bytes[frame.start + 10..frame.start + 24]
|
||||
frame.apic = bytes[frame.start + 24..frame.end]
|
||||
} else {
|
||||
// Выбор шага в записимости от версии ID3
|
||||
stride := if idv3_version == 3 { 13 } else { 11 }
|
||||
@@ -59,6 +61,13 @@ fn (mut frame Frame) fill_frame(bytes []u8, start_position u64, idv3_version u8)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn (mut frame Frame) init(tag_name string) {
|
||||
frame.name = tag_name.to_upper()
|
||||
|
||||
frame.encoding_name = "UTF-8"
|
||||
frame.encoding_name_idx = 3
|
||||
}
|
||||
|
||||
// Кодируем длину фрейма с нужным сдвигом
|
||||
@[inline]
|
||||
pub fn (mut frame Frame) encode_size(idv3_version u8) {
|
||||
@@ -69,10 +78,10 @@ pub fn (mut frame Frame) encode_size(idv3_version u8) {
|
||||
}
|
||||
|
||||
mut encoded_size := []u8{len: 4}
|
||||
encoded_size[0] = u8((frame.size >> stride[0]) & stride[3])
|
||||
encoded_size[1] = u8((frame.size >> stride[1]) & stride[3])
|
||||
encoded_size[2] = u8((frame.size >> stride[2]) & stride[3])
|
||||
encoded_size[3] = u8(frame.size & stride[3])
|
||||
encoded_size[0] = u8((frame.cur_size >> stride[0]) & stride[3])
|
||||
encoded_size[1] = u8((frame.cur_size >> stride[1]) & stride[3])
|
||||
encoded_size[2] = u8((frame.cur_size >> stride[2]) & stride[3])
|
||||
encoded_size[3] = u8(frame.cur_size & stride[3])
|
||||
|
||||
frame.encoded_size = encoded_size
|
||||
}
|
||||
@@ -80,12 +89,10 @@ 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)] }
|
||||
|
||||
|
||||
frame.encoded_data = encoded_data
|
||||
|
||||
//
|
||||
frame.size = u64(encoded_data.len + 10)
|
||||
|
||||
// число -> [байт, байт, байт, байт]
|
||||
frame.encode_size(idv3_version)
|
||||
}
|
||||
|
||||
@@ -95,16 +102,38 @@ pub fn (mut frame Frame) get_encoded_frame_tag(idv3_version u8) []u8 {
|
||||
|
||||
// Название тэга
|
||||
mut encoded_tag := frame.name.bytes()
|
||||
|
||||
|
||||
// Четыре числа на закодированный размер
|
||||
encoded_tag << frame.encoded_size
|
||||
// Флаги, оставляем пустыми
|
||||
encoded_tag << []u8{len: 2}
|
||||
// Идекс кодировщика текста
|
||||
encoded_tag << frame.encoding_name_idx
|
||||
// Четыре числа на закодированный размер
|
||||
encoded_tag << frame.encoded_size
|
||||
// Уже сами данные
|
||||
encoded_tag << frame.encoded_data
|
||||
encoded_tag << frame.encoded_data
|
||||
// Запас нулей чтобы не пересчитывать весь фрейм при маленьком редактировании
|
||||
encoded_tag << []u8{len: 10}
|
||||
// Замыкающий байт
|
||||
encoded_tag << []u8{len: 1}
|
||||
|
||||
frame.cur_size = u64(encoded_tag.len)
|
||||
|
||||
return encoded_tag
|
||||
}
|
||||
|
||||
// Запись данных в тэг
|
||||
@[inline]
|
||||
pub fn (mut frame Frame) set_data(data string) {
|
||||
if frame.data.len != 0 {
|
||||
frame.prev_size = u64(frame.data.len)
|
||||
} else {
|
||||
mut b_data := data.bytes()
|
||||
b_data << []u8{len: 1}
|
||||
frame.prev_size = u64(b_data.len)
|
||||
frame.cur_size = u64(b_data.len)
|
||||
}
|
||||
|
||||
frame.data = data
|
||||
frame.modified = true
|
||||
}
|
||||
Reference in new issue
Block a user