summaryrefslogtreecommitdiffstats
path: root/tdefile-plugins/mp4/tdefile_mp4.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tdefile-plugins/mp4/tdefile_mp4.cpp')
-rw-r--r--tdefile-plugins/mp4/tdefile_mp4.cpp276
1 files changed, 276 insertions, 0 deletions
diff --git a/tdefile-plugins/mp4/tdefile_mp4.cpp b/tdefile-plugins/mp4/tdefile_mp4.cpp
new file mode 100644
index 00000000..04291b2d
--- /dev/null
+++ b/tdefile-plugins/mp4/tdefile_mp4.cpp
@@ -0,0 +1,276 @@
+/* This file is part of the TDE project
+ * Copyright (C) 2025 mio <stigma@disroot.org>
+ *
+ * Portions of code based off tdefile_mp3.cpp
+ * Copyright (C) 2001, 2002 Rolf Magnus <ramagnus@kde.org>
+ * Copyright (C) 2002 Ryan Cumming <bodnar42@phalynx.dhs.org>
+ * Copyright (C) 2003 Scott Wheeler <wheeler@kde.org>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation version 2.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING. If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+#include "config.h"
+#include "tdefile_mp4.h"
+
+// TQt
+#include <tqfile.h>
+
+// TDE
+#include <kgenericfactory.h>
+#include <kstringvalidator.h>
+
+// TagLib
+#include <id3v1genres.h>
+#include <id3v2tag.h>
+#include <mp4file.h>
+#include <tfilestream.h>
+#include <tpropertymap.h>
+#include <tstring.h>
+
+#define TStringToTQString(s) TQString::fromUtf8(s.toCString(true))
+
+using Mp4Factory = KGenericFactory<TDEMp4Plugin>;
+
+K_EXPORT_COMPONENT_FACTORY(tdefile_mp4, Mp4Factory("tdefile_mp4"))
+
+/**
+ * Do translation between KFileMetaInfo items and TagLib::String in a tidy way.
+ */
+class Translator
+{
+public:
+ Translator(const KFileMetaInfo &info) : m_info(info) {}
+
+ TagLib::String operator[](const char *key) const
+ {
+ return QStringToTString(m_info["id3"][key].value().toString());
+ }
+
+ int toInt(const char *key) const
+ {
+ return m_info["id3"][key].value().toInt();
+ }
+
+private:
+ const KFileMetaInfo &m_info;
+};
+
+class Validator : public KStringListValidator
+{
+public:
+ Validator(const TQStringList& list, bool rejecting, bool fixupEnabled,
+ TQObject *parent, const char *name)
+ : KStringListValidator(list, rejecting, fixupEnabled, parent, name)
+ {
+
+ }
+
+ TQValidator::State validate(TQString&, int&) const override
+ {
+ return TQValidator::Acceptable;
+ }
+};
+
+TDEMp4Plugin::TDEMp4Plugin(TQObject *parent, const char *name, const TQStringList& args)
+ : KFilePlugin(parent, name, args)
+{
+ KFileMimeTypeInfo *info = addMimeTypeInfo(name);
+
+ // ID3 group
+ KFileMimeTypeInfo::GroupInfo *group = addGroupInfo(info, "id3", i18n("ID3 Tag"));
+ setAttributes(group, KFileMimeTypeInfo::Addable | KFileMimeTypeInfo::Removable);
+
+ KFileMimeTypeInfo::ItemInfo *item;
+
+ item = addItemInfo(group, "Title", i18n("Title"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+ setHint(item, KFileMimeTypeInfo::Name);
+
+ item = addItemInfo(group, "Artist", i18n("Artist"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+ setHint(item, KFileMimeTypeInfo::Author);
+
+ item = addItemInfo(group, "Album", i18n("Album"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+
+ item = addItemInfo(group, "Date", i18n("Year"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+
+ item = addItemInfo(group, "Comment", i18n("Comment"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+ setHint(item, KFileMimeTypeInfo::Description);
+
+ item = addItemInfo(group, "Tracknumber", i18n("Track"), TQVariant::Int);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+
+ item = addItemInfo(group, "Genre", i18n("Genre"), TQVariant::String);
+ setAttributes(item, KFileMimeTypeInfo::Modifiable);
+
+ // Technical group.
+ group = addGroupInfo(info, "Technical", i18n("Technical Details"));
+
+ item = addItemInfo(group, "Length", i18n("Length"), TQVariant::Int);
+ setAttributes(item, KFileMimeTypeInfo::Cummulative);
+ setUnit(item, KFileMimeTypeInfo::Seconds);
+
+ item = addItemInfo(group, "Bitrate", i18n("Average Bitrate"), TQVariant::Int);
+ setAttributes(item, KFileMimeTypeInfo::Averaged);
+ setHint(item, KFileMimeTypeInfo::Bitrate);
+ setSuffix(item, i18n(" kbps"));
+
+ item = addItemInfo(group, "Sample Rate", i18n("Sample Rate"), TQVariant::Int);
+ setSuffix(item, i18n("Hz"));
+
+ item = addItemInfo(group, "Channels", i18n("Channels"), TQVariant::Int);
+}
+
+TQValidator* TDEMp4Plugin::createValidator(const TQString& mimeType, const TQString& group,
+ const TQString& key, TQObject *parent, const char* name) const
+{
+ if (key == "Tracknumber" || key == "Date")
+ {
+ return new TQIntValidator(0, 9999, parent, name);
+ }
+
+ if (key == "Genre")
+ {
+ TQStringList genres;
+ for (const auto& genre : TagLib::ID3v1::genreList())
+ {
+ genres.append(TStringToTQString(genre));
+ }
+
+ return new Validator(genres, false, true, parent, name);
+ }
+
+ return nullptr;
+}
+
+bool TDEMp4Plugin::readInfo(KFileMetaInfo& info, unsigned what)
+{
+ using What = KFileMetaInfo::What;
+
+ bool readID3 = false;
+ bool readTech = false;
+
+ if (what & (What::Fastest | What::DontCare | What::ContentInfo))
+ {
+ readID3 = true;
+ }
+
+ if (what & (What::Fastest | What::DontCare | What::TechnicalInfo))
+ {
+ readTech = true;
+ }
+
+ if (!readID3 && !readTech)
+ {
+ return true;
+ }
+
+ if (info.path().isNull())
+ {
+ // Remote file.
+ return false;
+ }
+
+ TagLib::MP4::File file(TQFile::encodeName(info.path()), readTech);
+
+ if (!file.isOpen())
+ {
+ kdWarning(7034) << "Could not open " << file.name() << endl;
+ return false;
+ }
+
+ if (readID3)
+ {
+ KFileMetaInfoGroup id3group = appendGroup(info, "id3");
+ const TagLib::MP4::Tag* tag = file.tag();
+
+ if (!tag->title().isEmpty())
+ {
+ appendItem(id3group, "Title", TStringToTQString(tag->title()).stripWhiteSpace());
+ }
+ if (!tag->artist().isEmpty())
+ {
+ appendItem(id3group, "Artist", TStringToTQString(tag->artist()).stripWhiteSpace());
+ }
+ if (!tag->album().isEmpty())
+ {
+ appendItem(id3group, "Album", TStringToTQString(tag->album()).stripWhiteSpace());
+ }
+ if (!tag->comment().isEmpty())
+ {
+ appendItem(id3group, "Comment", TStringToTQString(tag->comment()).stripWhiteSpace());
+ }
+ if (!tag->genre().isEmpty())
+ {
+ appendItem(id3group, "Genre", TStringToTQString(tag->genre()).stripWhiteSpace());
+ }
+
+ TQString date = (tag->year() > 0) ? TQString::number(tag->year()) : TQString::null;
+ TQString track = (tag->track() > 0) ? TQString::number(tag->track()) : TQString::null;
+
+ appendItem(id3group, "Date", date);
+ appendItem(id3group, "Tracknumber", track);
+ }
+
+ if (readTech)
+ {
+ KFileMetaInfoGroup techGroup = appendGroup(info, "Technical");
+ const TagLib::AudioProperties* audioProperties = file.audioProperties();
+
+ appendItem(techGroup, "Length", audioProperties->lengthInSeconds());
+ appendItem(techGroup, "Bitrate", audioProperties->bitrate());
+ appendItem(techGroup, "Sample Rate", audioProperties->sampleRate());
+ appendItem(techGroup, "Channels", audioProperties->channels());
+ }
+
+ return true;
+}
+
+bool TDEMp4Plugin::writeInfo(const KFileMetaInfo& info) const
+{
+ TagLib::ID3v2::FrameFactory::instance()->setDefaultTextEncoding(TagLib::String::UTF8);
+ TagLib::FileStream stream(TQFile::encodeName(info.path()), false);
+
+ if (!stream.isOpen() || stream.readOnly())
+ {
+ kdDebug(7034) << "couldn't open " << info.path() << " for writing" << endl;
+ return false;
+ }
+
+ TagLib::MP4::File file(&stream, false);
+
+ if (!file.isValid())
+ {
+ return false;
+ }
+
+ Translator t(info);
+
+ file.tag()->setTitle(t["Title"]);
+ file.tag()->setArtist(t["Artist"]);
+ file.tag()->setAlbum(t["Album"]);
+ file.tag()->setYear(t.toInt("Date"));
+ file.tag()->setComment(t["Comment"]);
+ file.tag()->setTrack(t.toInt("Tracknumber"));
+ file.tag()->setGenre(t["Genre"]);
+
+ file.save();
+
+ return true;
+}
+
+#include "tdefile_mp4.moc"