io.appium.settings: recording: Make audio encoder thread's priority dynamic

Signed-off-by: sirmordred <[email protected]>

Author: sirmordred

app/src/main/java/io/appium/settings/Settings.java

@@ -48,18 +48,21 @@
 
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_BASE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_FILENAME;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_RESULT_CODE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_ROTATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_START;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_STOP;
 import static io.appium.settings.recorder.RecorderConstant.NO_ROTATION_SET;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MAX;
 import static io.appium.settings.recorder.RecorderConstant.REQUEST_CODE_SCREEN_CAPTURE;
 
 public class Settings extends Activity {
     private static final String TAG = "APPIUM SETTINGS";
 
     private String recordingOutputPath = "";
     private int recordingRotation = NO_ROTATION_SET;
+    private int recordingPriority = RECORDING_PRIORITY_MAX;
 
     @Override
     public void onCreate(Bundle savedInstanceState) {
@@ -154,6 +157,8 @@ so we need to call getExternalFilesDir() method twice
 
             recordingRotation = RecorderUtil.getDeviceRotation(getApplicationContext());
 
+            recordingPriority = RecorderUtil.getRecordingPriority(intent);
+
             // start record
             final MediaProjectionManager manager
                     = (MediaProjectionManager) getSystemService(
@@ -213,6 +218,7 @@ protected void onActivityResult(final int requestCode, final int resultCode, fin
         intent.putExtra(ACTION_RECORDING_RESULT_CODE, resultCode);
         intent.putExtra(ACTION_RECORDING_FILENAME, recordingOutputPath);
         intent.putExtra(ACTION_RECORDING_ROTATION, recordingRotation);
+        intent.putExtra(ACTION_RECORDING_PRIORITY, recordingPriority);
         intent.putExtras(data);
 
         startService(intent);

app/src/main/java/io/appium/settings/recorder/RecorderConstant.java

@@ -26,6 +26,7 @@ public class RecorderConstant {
     public static final String ACTION_RECORDING_RESULT_CODE = "result_code";
     public static final String ACTION_RECORDING_FILENAME = "recording_filename";
     public static final String ACTION_RECORDING_ROTATION = "recording_rotation";
+    public static final String ACTION_RECORDING_PRIORITY = "recording_priority";
     public static final float BITRATE_MULTIPLIER = 0.25f;
     public static final int AUDIO_CODEC_SAMPLE_RATE_HZ = 44100;
     public static final int AUDIO_CODEC_CHANNEL_COUNT = 1;
@@ -38,4 +39,8 @@ public class RecorderConstant {
     public static final long NO_TIMESTAMP_SET = -1;
     public static final int NO_ROTATION_SET = -1;
     public static final int NO_TRACK_INDEX_SET = -1;
+    public static final int RECORDING_PRIORITY_MAX = 0;
+    public static final int RECORDING_PRIORITY_MIN = 1;
+    public static final int RECORDING_PRIORITY_NORM = 2;
+    public static final int RECORDING_PRIORITY_DEFAULT = Thread.MAX_PRIORITY;
 }

app/src/main/java/io/appium/settings/recorder/RecorderService.java

@@ -31,10 +31,12 @@
 import io.appium.settings.helpers.NotificationHelpers;
 
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_FILENAME;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_RESULT_CODE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_ROTATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_START;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_STOP;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_DEFAULT;
 
 public class RecorderService extends Service {
     private static final String TAG = "RecorderService";
@@ -157,8 +159,11 @@ private void startScreenRecord(MediaProjectionManager mediaProjectionManager,
         Log.v(TAG, String.format("Starting recording with width/height(not clamped):(%d,%d)",
                 rawWidth, rawHeight));
 
+        int recordingPriority = intent.getIntExtra(ACTION_RECORDING_PRIORITY,
+                RECORDING_PRIORITY_DEFAULT);
+
         recorderThread = new RecorderThread(projection, outputFilePath,
-                rawWidth, rawHeight, recordingRotation);
+                rawWidth, rawHeight, recordingRotation, recordingPriority);
         recorderThread.startRecording();
     }
 

app/src/main/java/io/appium/settings/recorder/RecorderThread.java

@@ -42,6 +42,7 @@
 import static io.appium.settings.recorder.RecorderConstant.NANOSECOND_TO_MICROSECOND;
 import static io.appium.settings.recorder.RecorderConstant.NO_TIMESTAMP_SET;
 import static io.appium.settings.recorder.RecorderConstant.NO_TRACK_INDEX_SET;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_DEFAULT;
 
 public class RecorderThread implements Runnable {
 
@@ -52,6 +53,7 @@ public class RecorderThread implements Runnable {
     private final int videoWidth;
     private final int videoHeight;
     private final int recordingRotation;
+    private int recordingPriority = RECORDING_PRIORITY_DEFAULT;
 
     private boolean muxerStarted = false;
     private boolean isStartTimestampInitialized = false;
@@ -83,12 +85,14 @@ public void onStopped() {
     };
 
     public RecorderThread(MediaProjection mediaProjection, String outputFilePath,
-                          int videoWidth, int videoHeight, int recordingRotation) {
+                          int videoWidth, int videoHeight, int recordingRotation,
+                          int recordingPriority) {
         this.mediaProjection = mediaProjection;
         this.outputFilePath = outputFilePath;
         this.videoWidth = videoWidth;
         this.videoHeight = videoHeight;
         this.recordingRotation = recordingRotation;
+        this.recordingPriority = recordingPriority;
     }
 
     public void startRecording() {
@@ -170,11 +174,12 @@ private AudioRecord initAudioRecord(MediaProjection mediaProjection, int sampleR
     }
 
     @RequiresApi(api = Build.VERSION_CODES.LOLLIPOP)
-    private Thread initAudioRecordThread(MediaCodec audioEncoder, final AudioRecord audioRecord) {
+    private Thread initAudioRecordThread(MediaCodec audioEncoder, final AudioRecord audioRecord,
+                                         int priority) {
         return new Thread(new Runnable() {
             @Override
             public void run() {
-                Thread.currentThread().setPriority(Thread.MAX_PRIORITY);
+                Thread.currentThread().setPriority(priority);
                 try {
                     audioRecord.startRecording();
                 } catch (Exception e) {
@@ -387,7 +392,8 @@ public void run() {
             // note: this method must be run before muxer.start()
             muxer.setOrientationHint(recordingRotation);
 
-            audioRecordThread = initAudioRecordThread(audioEncoder, audioRecord);
+            audioRecordThread = initAudioRecordThread(audioEncoder, audioRecord,
+                    this.recordingPriority);
             audioRecordThread.start();
 
             MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();

app/src/main/java/io/appium/settings/recorder/RecorderUtil.java

@@ -18,18 +18,27 @@
 
 import android.Manifest;
 import android.content.Context;
+import android.content.Intent;
 import android.content.pm.PackageManager;
 import android.os.Build;
+import android.util.Log;
 import android.view.Display;
 import android.view.Surface;
 import android.view.WindowManager;
 
 import androidx.core.app.ActivityCompat;
 
 import static android.content.Context.WINDOW_SERVICE;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.NO_ROTATION_SET;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_DEFAULT;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MAX;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MIN;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_NORM;
 
 public class RecorderUtil {
+    private static final String TAG = "RecorderUtil";
+
     public static boolean isLowerThanQ() {
         return Build.VERSION.SDK_INT < Build.VERSION_CODES.Q;
     }
@@ -108,4 +117,28 @@ public static int getDeviceRotation(Context context) {
         }
         return NO_ROTATION_SET;
     }
+
+    public static int getRecordingPriority(Intent intent) {
+        if (intent.hasExtra(ACTION_RECORDING_PRIORITY)) {
+            try {
+                int userRequestedRecordingPriority =
+                        Integer.parseInt(intent.getStringExtra(ACTION_RECORDING_PRIORITY));
+                switch(userRequestedRecordingPriority) {
+                    case RECORDING_PRIORITY_MAX:
+                        return RECORDING_PRIORITY_DEFAULT;
+                    case RECORDING_PRIORITY_MIN:
+                        return Thread.MIN_PRIORITY;
+                    case RECORDING_PRIORITY_NORM:
+                        return Thread.NORM_PRIORITY;
+                    default:
+                        break;
+                }
+            } catch (NumberFormatException e) {
+                Log.e(TAG, "Exception while retrieving recording priority", e);
+            }
+        } else {
+            Log.e(TAG, "Unable to retrieve recording priority");
+        }
+        return RECORDING_PRIORITY_DEFAULT;
+    }
 }