av1: add av1_obu_count() (#395)

baresip · Jun 9, 2022 · c2b9985 · c2b9985
1 parent acfb113
commit c2b9985
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/include/re_av1.h b/include/re_av1.h
@@ -28,6 +28,7 @@ int av1_obu_encode(struct mbuf *mb, uint8_t type, bool has_size,
 		   size_t len, const uint8_t *payload);
 int av1_obu_decode(struct av1_obu_hdr *hdr, struct mbuf *mb);
 int av1_obu_print(struct re_printf *pf, const struct av1_obu_hdr *hdr);
+unsigned av1_obu_count(const uint8_t *buf, size_t size);
 
 
 /*

diff --git a/src/av1/obu.c b/src/av1/obu.c
@@ -196,3 +196,33 @@ int av1_obu_print(struct re_printf *pf, const struct av1_obu_hdr *hdr)
 	return re_hprintf(pf, "type=%u x=%d s=%d size=%zu",
 			  hdr->type, hdr->x, hdr->s, hdr->size);
 }
+
+
+unsigned av1_obu_count(const uint8_t *buf, size_t size)
+{
+	struct mbuf wrap = {
+		.buf = (uint8_t *)buf,
+		.size = size,
+		.pos = 0,
+		.end = size
+	};
+	unsigned count = 0;
+
+	while (mbuf_get_left(&wrap) > 1) {
+
+		struct av1_obu_hdr hdr;
+
+		int err = av1_obu_decode(&hdr, &wrap);
+		if (err) {
+			DEBUG_WARNING("count: could not decode OBU"
+				      " [%zu bytes]: %m\n", size, err);
+			return 0;
+		}
+
+		mbuf_advance(&wrap, hdr.size);
+
+		++count;
+	}
+
+	return count;
+}