*многое сделано, не помню что

This commit is contained in:
K1ndWha1e committed 2026-09-25 21:57:30 +03:00
1 parent fdc72ae653
commit abab08f133
8 files changed
+240 -76

No files matched your search

+40 -6
View File
@@ -1,11 +1,13 @@
module main module main
import os import os
import io
import metadata import metadata
fn id3(){ fn id3(){
mut audio := os.open("./data/АЛИСА 11000.mp3") or { panic(err) } // mut audio := os.open("./data/АЛИСА 11000.mp3") or { panic(err) }
// mut audio := os.open("./data/Глава 2. 16 и 17 июля.mp3") or { panic(err) } // mut audio := os.open("./data/Глава 2. 16 и 17 июля.mp3") or { panic(err) }
mut audio := os.open("./out.mp3") or { panic(err) }
// mut audio := os.open("./data/01_01.mp3") or { panic(err) } // mut audio := os.open("./data/01_01.mp3") or { panic(err) }
// mut audio := os.open("./test.mp3") or { panic(err) } // mut audio := os.open("./test.mp3") or { panic(err) }
@@ -29,23 +31,55 @@ fn id3(){
if wav.id3v23.embedded { if wav.id3v23.embedded {
wav.id3v23.get_full_size(mut audio) wav.id3v23.get_full_size(mut audio)
wav.id3v23.fill_tag(mut audio) wav.id3v23.fill_tags(mut audio)
println(wav.id3v23.to_v1()) println(wav.id3v23.to_v1())
wav.id3v23.tit2.encode_size(3) wav.id3v23.tit2.encode_size(3)
tit2 := wav.id3v23.tit2.get_encoded_frame_tag(3)
println(tit2)
// println(wav.id3v23.tit2) // println(wav.id3v23.tit2)
} }
if wav.id3v24.embedded { if wav.id3v24.embedded {
wav.id3v24.get_full_size(mut audio) wav.id3v24.get_full_size(mut audio)
wav.id3v24.fill_tag(mut audio) wav.id3v24.fill_tags(mut audio)
println(wav.id3v24.to_v1()) println(wav.id3v24.to_v1())
// wav.id3v24.tit2.encode_size(4) // wav.id3v24.tit2.encode_size(4)
flex := wav.id3v24.tit2.get_encoded_frame_tag(4) // flex := wav.id3v24.talb.get_encoded_frame_tag(4)
print(flex) // println(flex)
// println(wav.id3v24.tit2) // println(wav.id3v24.tdrc.data)
// wav.id3v24.tdrc.set_data("2017")
// enc_size := metadata.encode_header_size(wav.id3v24.size, wav.id3v24.version)
// println(enc_size)
wav.id3v24.tit2.set_data("2017")
wav.id3v24.tpe1.init("tpe1")
wav.id3v24.tpe1.set_data("Hello")
// metadata.encode_header(mut wav.id3v24)
// println(wav.id3v24.encoded_header)
// println(wav.id3v24.tpe1.name)
wav.id3v24.get_full_embed_tag()
println(wav.id3v24.encoded_header.len)
wav.id3v24.insert_tags_into_audio(mut audio)
// println(wav.id3v24)
mut f := os.create("./temp.mp3") or { panic(err) }
f.write(wav.id3v24.encoded_header) or { panic(err) }
f.close()
// println(wav.id3v24)
// mut audio_2 := os.open("./out.mp3") or { panic(err) }
// data := io.read_all(reader: audio_2) or { [u8(0), u8(0), u8(0), u8(0)] }
// println(data)
println(wav.id3v24.size)
println(wav.id3v24.encoded_header)
} }
+29
View File
@@ -0,0 +1,29 @@
module metadata
pub enum EnumID3v24 as u8 {
tit2
tpe1
tpe2
talb
tdrc
trck
tcon
tcop
apic
comment
unknown
}
pub enum EnumID3v23 as u8 {
tit2
tpe1
tpe2
talb
tyer
trck
tcon
tcop
apic
comment
unknown
}
+44 -15
View File
@@ -4,12 +4,14 @@ import encoding.iconv
@[heap] @[heap]
pub struct Frame { pub struct Frame {
pub mut:
data string
mut: mut:
size u64 prev_size u64
cur_size u64
start u64 start u64
end u64 end u64
encoding_name string encoding_name string
data string
name string name string
encoding_name_idx u8 encoding_name_idx u8
addition []u8 addition []u8
@@ -19,22 +21,22 @@ pub struct Frame {
modified bool modified bool
} }
// Во врейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE // Во фрейме первый байт указывает кодировку 0 - ISO-8859-1 (Latin-1), 1 - UTF-16LE
@[inline] @[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.start = start_position
// Название фрейма // Название фрейма
frame.name = bytes[frame.start..frame.start + 4].bytestr() frame.name = bytes[frame.start..frame.start + 4].bytestr()
// Размер фрейма // Размер фрейма
frame.size = if idv3_version == 3 { frame.cur_size = if idv3_version == 3 {
get_size_v23(bytes[frame.start + 4..frame.start + 8]) get_size_v23(bytes[frame.start + 4..frame.start + 8])
} else { } 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] {
@@ -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] @[inline]
pub fn (mut frame Frame) encode_size(idv3_version u8) { 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} mut encoded_size := []u8{len: 4}
encoded_size[0] = u8((frame.size >> stride[0]) & stride[3]) encoded_size[0] = u8((frame.cur_size >> stride[0]) & stride[3])
encoded_size[1] = u8((frame.size >> stride[1]) & stride[3]) encoded_size[1] = u8((frame.cur_size >> stride[1]) & stride[3])
encoded_size[2] = u8((frame.size >> stride[2]) & stride[3]) encoded_size[2] = u8((frame.cur_size >> stride[2]) & stride[3])
encoded_size[3] = u8(frame.size & stride[3]) encoded_size[3] = u8(frame.cur_size & stride[3])
frame.encoded_size = encoded_size frame.encoded_size = encoded_size
} }
@@ -83,9 +92,7 @@ pub fn (mut frame Frame) encode_frame(idv3_version u8) {
frame.encoded_data = encoded_data frame.encoded_data = encoded_data
// // число -> [байт, байт, байт, байт]
frame.size = u64(encoded_data.len + 10)
frame.encode_size(idv3_version) 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() mut encoded_tag := frame.name.bytes()
// Четыре числа на закодированный размер
encoded_tag << frame.encoded_size
// Флаги, оставляем пустыми // Флаги, оставляем пустыми
encoded_tag << []u8{len: 2} encoded_tag << []u8{len: 2}
// Идекс кодировщика текста // Идекс кодировщика текста
encoded_tag << frame.encoding_name_idx 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: 10}
// Замыкающий байт
encoded_tag << []u8{len: 1}
frame.cur_size = u64(encoded_tag.len)
return encoded_tag 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
}
+16 -16
View File
@@ -11,19 +11,19 @@ import os { File }
// 13 - 1 (байт кодировки) - 2 (BOM) = 10 байт. // 13 - 1 (байт кодировки) - 2 (BOM) = 10 байт.
// В UTF-16 это будет всего 5 символов (так как 1 символ = 2 байта). // В UTF-16 это будет всего 5 символов (так как 1 символ = 2 байта).
enum EnumID3v23 as u8 { // enum EnumID3v23 as u8 {
tit2 // tit2
tpe1 // tpe1
tpe2 // tpe2
talb // talb
tyer // tyer
trck // trck
tcon // tcon
tcop // tcop
apic // apic
comment // comment
unknown // unknown
} // }
const start_idx = 10 const start_idx = 10
@@ -73,12 +73,12 @@ fn get_size_v23(size_bytes []u8) u64 {
// Получения размеров всех тэгов // Получения размеров всех тэгов
pub fn (mut id3v23 ID3v23) get_full_size(mut audio File) { pub fn (mut id3v23 ID3v23) get_full_size(mut audio File) {
get_full_size(mut id3v23, mut audio) get_current_tag_size(mut id3v23, mut audio)
} }
// Заполнение тэга // Заполнение тэга
pub fn (mut id3v23 ID3v23) fill_tag(mut audio File) { pub fn (mut id3v23 ID3v23) fill_tags(mut audio File) {
fill_tag[ID3v23](mut id3v23, mut audio) dump_data[ID3v23](mut id3v23, mut audio)
} }
// Конвертировать ID3 v2.3 в ID3 v1 // Конвертировать ID3 v2.3 в ID3 v1
+38 -17
View File
@@ -1,20 +1,8 @@
module metadata module metadata
import os { File } import os { File }
import io { read_all }
enum EnumID3v24 as u8 {
tit2
tpe1
tpe2
talb
tdrc
trck
tcon
tcop
apic
comment
unknown
}
pub fn (mut tag ID3v24) add_frame(frame Frame) { pub fn (mut tag ID3v24) add_frame(frame Frame) {
name := EnumID3v24.from(frame.name.to_lower()) or { EnumID3v24.unknown } name := EnumID3v24.from(frame.name.to_lower()) or { EnumID3v24.unknown }
@@ -53,6 +41,7 @@ pub fn (mut tag ID3v24) add_frame(frame Frame) {
} }
} }
@[inline] @[inline]
fn get_size_v24(size_bytes []u8) u64 { fn get_size_v24(size_bytes []u8) u64 {
return (u64(size_bytes[0]) << 21) | (u64(size_bytes[1]) << 14) | (u64(size_bytes[2]) << 7) | u64(size_bytes[3]) return (u64(size_bytes[0]) << 21) | (u64(size_bytes[1]) << 14) | (u64(size_bytes[2]) << 7) | u64(size_bytes[3])
@@ -60,12 +49,12 @@ fn get_size_v24(size_bytes []u8) u64 {
// Получения размеров всех тэгов // Получения размеров всех тэгов
pub fn (mut id3v24 ID3v24) get_full_size(mut audio File) { pub fn (mut id3v24 ID3v24) get_full_size(mut audio File) {
get_full_size(mut id3v24, mut audio) get_current_tag_size(mut id3v24, mut audio)
} }
// Заполнение тэга // Заполнение тэга
pub fn (mut id3v24 ID3v24) fill_tag(mut audio File) { pub fn (mut id3v24 ID3v24) fill_tags(mut audio File) {
fill_tag[ID3v24](mut id3v24, mut audio) dump_data[ID3v24](mut id3v24, mut audio)
} }
// Конвертировать ID3 v2.3 в ID3 v1 // Конвертировать ID3 v2.3 в ID3 v1
@@ -81,7 +70,39 @@ pub fn (id3v24 ID3v24) to_v1() ID3v1 {
} }
} }
// Формируем полный полный тэг в байтах
pub fn (mut id3v24 ID3v24) get_full_embed_tag() {
// Проход по всем полям стуктуры
mut tag_bytes := []u8{}
pub fn flexd() { 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)] }
sound_part := full_part[int(id3v24.size) + 10 .. ]
id3v24.encoded_header << sound_part
} }
+37 -17
View File
@@ -1,7 +1,7 @@
module metadata module metadata
@[heap] @[heap; minify]
pub struct ID3v1 { pub struct ID3v1 {
pub mut: pub mut:
genre string// Жанр genre string// Жанр
@@ -30,28 +30,47 @@ pub struct ID3v23 {
comment Frame // Комментарий comment Frame // Комментарий
unknown Frame // Затычка unknown Frame // Затычка
others []Frame // Остальные фреймы others []Frame // Остальные фреймы
version u8 = 3 // Версия тэга
embedded bool // Присутвует ли тэг embedded bool // Присутвует ли тэг
version u8 = 3 // Версия тэга
title string = "ID3"
} }
@[heap; minify] @[heap; minify]
pub struct ID3v24 { pub struct ID3v24 {
pub mut: pub mut:
size u64 // Размер тэга // Размер тэга
tit2 Frame // Название аудиодорожки size u64
tpe1 Frame // Автор // Название аудиодорожки
tpe2 Frame // Исполнитель tit2 Frame
talb Frame // Название альбома // Автор
tdrc Frame // Год выпуска tpe1 Frame
trck Frame // Номер трека // Исполнитель
tcon Frame // Жанр tpe2 Frame
tcop Frame // Композитор (автор) // Название альбома
apic Frame // Обложка talb Frame
comment Frame // Комментарий // Год выпуска
modified_frames []Frame // Затычка tdrc Frame
others []Frame // Остальные фреймы // Номер трека
version u8 = 4 // Версия тэга trck Frame
embedded bool // Присутвует ли тэг // Жанр
tcon Frame
// Композитор (автор)
tcop Frame
// Обложка
apic Frame
// Комментарий
comment Frame
// Затычка
modified_frames []Frame
// Остальные фреймы
others []Frame
// Версия тэга
version u8 = 4
// Присутвует ли тэг
embedded bool
// Закодированный хэдер
encoded_header []u8
title string = "ID3"
} }
@@ -63,3 +82,4 @@ pub struct Audio {
path string path string
extension string = ".mp3" extension string = ".mp3"
} }
+36 -5
View File
@@ -27,7 +27,7 @@ pub fn (mut audio Audio) detect_tags(mut data os.File) {
} }
pub fn get_bytes(mut audio os.File, position int, size int) []u8 { pub fn get_bytes(mut audio os.File, position int, size int) []u8 {
// ? ---------Работа с расположением указателя-------------- // > ---------Работа с расположением указателя--------------
// Ставим указатель на 128 байт с конца // Ставим указатель на 128 байт с конца
if position < 0 { if position < 0 {
audio.seek(position, os.SeekMode.end) or { panic(err) } audio.seek(position, os.SeekMode.end) or { panic(err) }
@@ -42,7 +42,7 @@ pub fn get_bytes(mut audio os.File, position int, size int) []u8 {
return last_bytes return last_bytes
} }
pub fn fill_tag[T](mut tag T, mut audio os.File) { pub fn dump_data[T](mut tag T, mut audio os.File) {
mut current_frame := &Frame{} mut current_frame := &Frame{}
// Самая первая позиция // Самая первая позиция
mut start_position := u64(10) mut start_position := u64(10)
@@ -50,17 +50,48 @@ pub fn fill_tag[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 { for tag.size > start_position {
current_frame.fill_frame(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.size tag.size -= current_frame.cur_size
tag.add_frame(current_frame) tag.add_frame(current_frame)
} }
} }
pub fn get_full_size[T](mut tag T, mut audio os.File) { pub fn get_current_tag_size[T](mut tag T, mut audio os.File) {
// Первые 10 байт тэга // Первые 10 байт тэга
first_ten_bytes := get_bytes(mut audio, 0, 10) first_ten_bytes := get_bytes(mut audio, 0, 10)
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]) 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)]
}
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])
return encoded_size
}
// Кодирование первых 10 байт ID3
pub fn encode_header[T](mut id3v T) {
// Байты для символов "ID3"
id3v.encoded_header << [u8(73), u8(68), u8(51)]
// Версия IDv3
id3v.encoded_header << [id3v.version]
// Минорная версия и спец. флаги
id3v.encoded_header << []u8{len: 2}
id3v.encoded_header << metadata.encode_header_size(id3v.size, id3v.version)
}
View File
Whitespace-only changes.