commit d777f7bd43cf02a1732977a798de5c0ce687068b
author Mufaddal Chakera <mufaddal.chakera@ittiam.com> Mon Jul 26 02:13:20 2021 +0530
committer Yunqing Wang <yunqingwang@google.com> Wed Jul 28 21:50:57 2021 +0000
tree 10ae22d79d6e0ae731f41599dcab8e1e62d3de3e
parent bf4f3581f415ec808afd4d9d9ab5a191f5fad215

Enable frame parallel encode of 2 frames in parallel

Integrated frame parallel encode modules. Frame parallel
encode is enabled for 2 frames in parallel for AOM_Q.

Change-Id: I7f7b6c87ac9ebb6b29b4a72f36e2cc50192f4020

av1/encoder/ethread.c

diff --git a/av1/encoder/ethread.c b/av1/encoder/ethread.c
index cad160c..7eff819 100644
--- a/av1/encoder/ethread.c
+++ b/av1/encoder/ethread.c
@@ -870,7 +870,30 @@
   if (had_error)
     aom_internal_error(&ppi->error, AOM_CODEC_ERROR, "Failed to encode frame");
 }
+
+static int get_compressed_data_hook(void *arg1, void *arg2) {
+  AV1_COMP *cpi = (AV1_COMP *)arg1;
+  AV1_COMP_DATA *cpi_data = (AV1_COMP_DATA *)arg2;
+  av1_get_compressed_data(cpi, cpi_data);
+
+  return 1;
+}
+
+// This function encodes the raw frame data for each frame in parallel encode
+// set, and outputs the frame bit stream to the designated buffers.
+int av1_compress_parallel_frames(AV1_PRIMARY *const ppi,
+                                 AV1_COMP_DATA *const first_cpi_data) {
+  int frames_in_parallel_set =
+      av1_init_parallel_frame_context(first_cpi_data, ppi);
+  prepare_fpmt_workers(ppi, first_cpi_data, get_compressed_data_hook,
+                       frames_in_parallel_set);
+  launch_fpmt_workers(ppi);
+  sync_fpmt_workers(ppi);
+
+  return AOM_CODEC_OK;
+}
 #endif  // CONFIG_FRAME_PARALLEL_ENCODE
+
 static AOM_INLINE void launch_workers(MultiThreadInfo *const mt_info,
                                       int num_workers) {
   const AVxWorkerInterface *const winterface = aom_get_worker_interface();