Add functions to get track metadata

vapoursynth · Oct 27, 2024 · c25cf29 · c25cf29
1 parent 12a247d
commit c25cf29
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 3 deletions.
diff --git a/src/audiosource.h b/src/audiosource.h
@@ -55,7 +55,7 @@ struct BSAudioProperties {
     double StartTime; /* in seconds */
 };
 
-struct LWAudioDecoder {
+class LWAudioDecoder {
 private:
     AVFormatContext *FormatContext = nullptr;
     AVCodecContext *CodecContext = nullptr;

diff --git a/src/tracklist.cpp b/src/tracklist.cpp
@@ -92,4 +92,31 @@ int BestTrackList::GetNumTracks() const {
 
 const BestTrackList::TrackInfo &BestTrackList::GetTrackInfo(int Track) const {
     return TrackList[Track];
-}
+}
+
+std::map<std::string, std::string> BestTrackList::GetFileMetadata() const {
+    std::map<std::string, std::string> Result;
+
+    auto Entry = av_dict_iterate(FormatContext->metadata, nullptr);
+    while (Entry) {
+        Result[Entry->key] = Entry->value;
+        Entry = av_dict_iterate(FormatContext->metadata, Entry);
+    }
+
+    return Result;
+}
+
+std::map<std::string, std::string> BestTrackList::GetTrackMetadata(int Track) const {
+    if (Track < 0 || Track >= FormatContext->nb_streams)
+        throw BestSourceException("Invalid track number");
+
+    std::map<std::string, std::string> Result;
+
+    auto Entry = av_dict_iterate(FormatContext->streams[Track]->metadata, nullptr);
+    while (Entry) {
+        Result[Entry->key] = Entry->value;
+        Entry = av_dict_iterate(FormatContext->streams[Track]->metadata, Entry);
+    }
+
+    return Result;
+}
diff --git a/src/tracklist.h b/src/tracklist.h
@@ -50,6 +50,8 @@ struct BestTrackList {
     ~BestTrackList();
     [[nodiscard]] int GetNumTracks() const;
     [[nodiscard]] const TrackInfo &GetTrackInfo(int Track) const;
+    [[nodiscard]] std::map<std::string, std::string> GetFileMetadata() const;
+    [[nodiscard]] std::map<std::string, std::string> GetTrackMetadata(int Track) const;
 };
 
 #endif
diff --git a/src/videosource.h b/src/videosource.h
@@ -95,7 +95,7 @@ struct BSVideoProperties {
     int Rotation; /* A positive number in degrees */
 };
 
-struct LWVideoDecoder {
+class LWVideoDecoder {
 private:
     AVFormatContext *FormatContext = nullptr;
     AVCodecContext *CodecContext = nullptr;