avcodec/ass: add ff_ass_add_rect_bprint() helper

2 files changed, 15 insertions, 0 deletions

diff --git a/libavcodec/ass.c b/libavcodec/ass.c
index a5b5ae5fb2..ba0593c936 100644
--- a/libavcodec/ass.c
+++ b/libavcodec/ass.c
@@ -151,6 +151,14 @@ err:
     return ret;
 }
 
+int ff_ass_add_rect_bprint(AVSubtitle *sub, const AVBPrint *buf,
+                           int ts_start, int duration, int raw)
+{
+    if (!av_bprint_is_complete(buf))
+        return AVERROR(ENOMEM);
+    return ff_ass_add_rect(sub, buf->str, ts_start, duration, raw);
+}
+
 void ff_ass_bprint_text_event(AVBPrint *buf, const char *p, int size,
                              const char *linebreaks, int keep_ass_markup)
 {
diff --git a/libavcodec/ass.h b/libavcodec/ass.h
index 2df38e6efa..5ed947ad9e 100644
--- a/libavcodec/ass.h
+++ b/libavcodec/ass.h
@@ -92,6 +92,13 @@ int ff_ass_add_rect(AVSubtitle *sub, const char *dialog,
                     int ts_start, int duration, int raw);
 
 /**
+ * Same as ff_ass_add_rect_bprint, but taking an AVBPrint buffer instead of a
+ * string.
+ */
+int ff_ass_add_rect_bprint(AVSubtitle *sub, const AVBPrint *buf,
+                           int ts_start, int duration, int raw);
+
+/**
  * Add an ASS dialog line to an AVBPrint buffer.
  *
  * @param buf pointer to an initialized AVBPrint buffer