feat: qvbr ffprobe metadata upload

Eyevinn · Aug 14, 2024 · 9ab10fe · 9ab10fe
1 parent 3b706d7
commit 9ab10fe
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/examples/aws/aws-job-qvbr.yml b/examples/aws/aws-job-qvbr.yml
@@ -1,6 +1,6 @@
 name: job-name
 pipeline: pipeline.yml
-encodingProfile: encoding-profile.json
+encodingProfile: encoding-profile-qvbr.json
 reference: reference.mp4
 models:
   - HD

diff --git a/src/pairVmaf.ts b/src/pairVmaf.ts
@@ -141,7 +141,7 @@ function fileExists(file) {
   return fs.existsSync(file);
 }
 
-async function runFfprobe(file) {
+export async function runFfprobe(file) {
   return new Promise((resolve, reject) => {
     ffmpeg.ffprobe(file, (err, metadata) => {
       if (err) {

diff --git a/src/pipelines/aws/aws-pipeline.ts b/src/pipelines/aws/aws-pipeline.ts
@@ -22,6 +22,7 @@ import {
   qualityAnalysisModelToString
 } from '../../models/quality-analysis-model';
 import logger from '../../logger';
+import { runFfprobe } from '../../pairVmaf';
 
 export function isS3URI(url: string): boolean {
   try {
@@ -235,6 +236,19 @@ export default class AWSPipeline implements Pipeline {
       `${outputFolder}/${outputObject}`
     );
     if (!s3Status) return '';
+    if (variables ? variables['QVBR'] : undefined) {
+      const url = this.generatePresignedUrl(outputBucket, outputObject, 5);
+      const metadata = await runFfprobe(url);
+      const upload = new Upload({
+        client: this.s3,
+        params: {
+          Bucket: outputBucket,
+          Key: `${outputObject}_metadata`,
+          Body: JSON.stringify(metadata)
+        }
+      });
+      await upload.done();
+    }
 
     logger.info('Finished transcoding ' + inputFilename + '.');
     return outputURI;