Leo Ma
9 years ago
parent
b2ec6c8e55
commit
c42c84a8a1
@ -0,0 +1,342 @@
|
||||
package net.ossrs.yasea;
|
||||
|
||||
import android.graphics.ImageFormat;
|
||||
import android.hardware.Camera;
|
||||
import android.media.AudioFormat;
|
||||
import android.media.AudioRecord;
|
||||
import android.media.MediaRecorder;
|
||||
import android.view.SurfaceHolder;
|
||||
import android.view.SurfaceView;
|
||||
|
||||
import net.ossrs.yasea.rtmp.RtmpPublisher;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* Created by Leo Ma on 2016/7/25.
|
||||
*/
|
||||
public class SrsPublisher implements SurfaceHolder.Callback, Camera.PreviewCallback {
|
||||
private static final String TAG = "SrsPublisher";
|
||||
|
||||
private AudioRecord mic;
|
||||
private boolean aloop = false;
|
||||
private Thread aworker;
|
||||
|
||||
private SurfaceView mCameraView;
|
||||
private Camera mCamera;
|
||||
|
||||
private int videoFrameCount;
|
||||
private long lastTimeMillis;
|
||||
private int mPreviewRotation = 90;
|
||||
private int mCamId = Camera.getNumberOfCameras() - 1;
|
||||
private double mSamplingFps;
|
||||
private byte[] mYuvPreviewFrame;
|
||||
|
||||
private SrsFlvMuxer mFlvMuxer;
|
||||
private SrsMp4Muxer mMp4Muxer;
|
||||
private SrsEncoder mEncoder = new SrsEncoder();
|
||||
|
||||
public SrsPublisher() {
|
||||
}
|
||||
|
||||
public void startEncode() {
|
||||
if (!mEncoder.start()) {
|
||||
return;
|
||||
}
|
||||
|
||||
startCamera();
|
||||
|
||||
aworker = new Thread(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
android.os.Process.setThreadPriority(android.os.Process.THREAD_PRIORITY_AUDIO);
|
||||
startAudio();
|
||||
}
|
||||
});
|
||||
aloop = true;
|
||||
aworker.start();
|
||||
}
|
||||
|
||||
public void stopEncode() {
|
||||
stopAudio();
|
||||
stopCamera();
|
||||
mEncoder.stop();
|
||||
}
|
||||
|
||||
public void startPublish(String rtmpUrl) {
|
||||
if (mFlvMuxer != null) {
|
||||
try {
|
||||
mFlvMuxer.start(rtmpUrl);
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
return;
|
||||
}
|
||||
mFlvMuxer.setVideoResolution(mEncoder.getOutputWidth(), mEncoder.getOutputHeight());
|
||||
|
||||
startEncode();
|
||||
}
|
||||
}
|
||||
|
||||
public void stopPublish() {
|
||||
if (mFlvMuxer != null) {
|
||||
stopEncode();
|
||||
mFlvMuxer.stop();
|
||||
}
|
||||
}
|
||||
|
||||
public void startRecord(String recPath) {
|
||||
if (mMp4Muxer != null) {
|
||||
try {
|
||||
mMp4Muxer.record(new File(recPath));
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void stopRecord() {
|
||||
if (mMp4Muxer != null) {
|
||||
mMp4Muxer.stop();
|
||||
}
|
||||
}
|
||||
|
||||
public void pauseRecord() {
|
||||
if (mMp4Muxer != null) {
|
||||
mMp4Muxer.pause();
|
||||
}
|
||||
}
|
||||
|
||||
public void resumeRecord() {
|
||||
if (mMp4Muxer != null) {
|
||||
mMp4Muxer.resume();
|
||||
}
|
||||
}
|
||||
|
||||
public void swithToSoftEncoder() {
|
||||
mEncoder.swithToSoftEncoder();
|
||||
}
|
||||
|
||||
public void swithToHardEncoder() {
|
||||
mEncoder.swithToHardEncoder();
|
||||
}
|
||||
|
||||
public boolean isSoftEncoder() {
|
||||
return mEncoder.isSoftEncoder();
|
||||
}
|
||||
|
||||
public double getmSamplingFps() {
|
||||
return mSamplingFps;
|
||||
}
|
||||
|
||||
public int getCamraId() {
|
||||
return mCamId;
|
||||
}
|
||||
|
||||
public int getNumberOfCameras() {
|
||||
return mCamera != null ? mCamera.getNumberOfCameras() : -1;
|
||||
}
|
||||
|
||||
public void setPreviewResolution(int width, int height) {
|
||||
mEncoder.setPreviewResolution(width, height);
|
||||
}
|
||||
|
||||
public void setOutputResolution(int width, int height) {
|
||||
mEncoder.setPortraitResolution(width, height);
|
||||
}
|
||||
|
||||
public void setScreenOrientation(int orientation) {
|
||||
mEncoder.setScreenOrientation(orientation);
|
||||
}
|
||||
|
||||
public void setPreviewRotation(int rotation) {
|
||||
mPreviewRotation = rotation;
|
||||
}
|
||||
|
||||
public void setVideoHDMode() {
|
||||
mEncoder.setVideoHDMode();
|
||||
}
|
||||
|
||||
public void setVideoSmoothMode() {
|
||||
mEncoder.setVideoSmoothMode();
|
||||
}
|
||||
|
||||
public void switchCameraFace(int id) {
|
||||
mCamId = id;
|
||||
stopCamera();
|
||||
mEncoder.swithCameraFace();
|
||||
startCamera();
|
||||
}
|
||||
|
||||
private void startCamera() {
|
||||
if (mCamera != null) {
|
||||
return;
|
||||
}
|
||||
if (mCamId > (Camera.getNumberOfCameras() - 1) || mCamId < 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
mCamera = Camera.open(mCamId);
|
||||
|
||||
Camera.Parameters params = mCamera.getParameters();
|
||||
Camera.Size size = mCamera.new Size(mEncoder.getPreviewWidth(), mEncoder.getPreviewHeight());
|
||||
if (!params.getSupportedPreviewSizes().contains(size) || !params.getSupportedPictureSizes().contains(size)) {
|
||||
Thread.getDefaultUncaughtExceptionHandler().uncaughtException(Thread.currentThread(),
|
||||
new IllegalArgumentException(String.format("Unsupported resolution %dx%d", size.width, size.height)));
|
||||
}
|
||||
|
||||
mYuvPreviewFrame = new byte[mEncoder.getPreviewWidth() * mEncoder.getPreviewHeight() * 3 / 2];
|
||||
|
||||
/***** set parameters *****/
|
||||
//params.set("orientation", "portrait");
|
||||
//params.set("orientation", "landscape");
|
||||
//params.setRotation(90);
|
||||
params.setPictureSize(mEncoder.getPreviewWidth(), mEncoder.getPreviewHeight());
|
||||
params.setPreviewSize(mEncoder.getPreviewWidth(), mEncoder.getPreviewHeight());
|
||||
int[] range = findClosestFpsRange(SrsEncoder.VFPS, params.getSupportedPreviewFpsRange());
|
||||
params.setPreviewFpsRange(range[0], range[1]);
|
||||
params.setPreviewFormat(ImageFormat.NV21);
|
||||
params.setFlashMode(Camera.Parameters.FLASH_MODE_OFF);
|
||||
params.setWhiteBalance(Camera.Parameters.WHITE_BALANCE_AUTO);
|
||||
params.setSceneMode(Camera.Parameters.SCENE_MODE_AUTO);
|
||||
if (!params.getSupportedFocusModes().isEmpty()) {
|
||||
params.setFocusMode(params.getSupportedFocusModes().get(0));
|
||||
}
|
||||
mCamera.setParameters(params);
|
||||
|
||||
mCamera.setDisplayOrientation(mPreviewRotation);
|
||||
|
||||
mCamera.addCallbackBuffer(mYuvPreviewFrame);
|
||||
mCamera.setPreviewCallbackWithBuffer(this);
|
||||
try {
|
||||
if (mCameraView != null){
|
||||
mCamera.setPreviewDisplay(mCameraView.getHolder());
|
||||
}
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
mCamera.startPreview();
|
||||
}
|
||||
|
||||
private static int[] findClosestFpsRange(int expectedFps, List<int[]> fpsRanges) {
|
||||
expectedFps *= 1000;
|
||||
int[] closestRange = fpsRanges.get(0);
|
||||
int measure = Math.abs(closestRange[0] - expectedFps) + Math.abs(closestRange[1] - expectedFps);
|
||||
for (int[] range : fpsRanges) {
|
||||
if (range[0] <= expectedFps && range[1] >= expectedFps) {
|
||||
int curMeasure = Math.abs(range[0] - expectedFps) + Math.abs(range[1] - expectedFps);
|
||||
if (curMeasure < measure) {
|
||||
closestRange = range;
|
||||
measure = curMeasure;
|
||||
}
|
||||
}
|
||||
}
|
||||
return closestRange;
|
||||
}
|
||||
|
||||
private void stopCamera() {
|
||||
if (mCamera != null) {
|
||||
// need to SET NULL CB before stop preview!!!
|
||||
mCamera.setPreviewCallback(null);
|
||||
mCamera.stopPreview();
|
||||
mCamera.release();
|
||||
mCamera = null;
|
||||
}
|
||||
}
|
||||
|
||||
private void onGetYuvFrame(byte[] data) {
|
||||
// Calculate YUV sampling FPS
|
||||
if (videoFrameCount == 0) {
|
||||
lastTimeMillis = System.nanoTime() / 1000000;
|
||||
videoFrameCount++;
|
||||
} else {
|
||||
if (++videoFrameCount >= 48) {
|
||||
long diffTimeMillis = System.nanoTime() / 1000000 - lastTimeMillis;
|
||||
mSamplingFps = (double) videoFrameCount * 1000 / diffTimeMillis;
|
||||
videoFrameCount = 0;
|
||||
}
|
||||
}
|
||||
mEncoder.onGetYuvFrame(data);
|
||||
}
|
||||
|
||||
private void startAudio() {
|
||||
if (mic != null) {
|
||||
return;
|
||||
}
|
||||
|
||||
int bufferSize = 2 * AudioRecord.getMinBufferSize(SrsEncoder.ASAMPLERATE, SrsEncoder.ACHANNEL, AudioFormat.ENCODING_PCM_16BIT);
|
||||
mic = new AudioRecord(MediaRecorder.AudioSource.MIC, SrsEncoder.ASAMPLERATE, SrsEncoder.ACHANNEL, AudioFormat.ENCODING_PCM_16BIT, bufferSize);
|
||||
mic.startRecording();
|
||||
|
||||
byte pcmBuffer[] = new byte[4096];
|
||||
while (aloop && !Thread.interrupted()) {
|
||||
int size = mic.read(pcmBuffer, 0, pcmBuffer.length);
|
||||
if (size <= 0) {
|
||||
break;
|
||||
}
|
||||
mEncoder.onGetPcmFrame(pcmBuffer, size);
|
||||
}
|
||||
}
|
||||
|
||||
private void stopAudio() {
|
||||
aloop = false;
|
||||
if (aworker != null) {
|
||||
aworker.interrupt();
|
||||
try {
|
||||
aworker.join();
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
aworker.interrupt();
|
||||
}
|
||||
aworker = null;
|
||||
}
|
||||
|
||||
if (mic != null) {
|
||||
mic.setRecordPositionUpdateListener(null);
|
||||
mic.stop();
|
||||
mic.release();
|
||||
mic = null;
|
||||
}
|
||||
}
|
||||
|
||||
public void setPublishEventHandler(RtmpPublisher.EventHandler handler) {
|
||||
mFlvMuxer = new SrsFlvMuxer(handler);
|
||||
mEncoder.setFlvMuxer(mFlvMuxer);
|
||||
}
|
||||
|
||||
public void setRecordEventHandler(SrsMp4Muxer.EventHandler handler) {
|
||||
mMp4Muxer = new SrsMp4Muxer(handler);
|
||||
mEncoder.setMp4Muxer(mMp4Muxer);
|
||||
}
|
||||
|
||||
public void setSurfaceView(SurfaceView surfaceView) {
|
||||
mCameraView = surfaceView;
|
||||
mCameraView.getHolder().addCallback(this);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onPreviewFrame(byte[] data, Camera camera) {
|
||||
onGetYuvFrame(data);
|
||||
camera.addCallbackBuffer(mYuvPreviewFrame);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void surfaceChanged(SurfaceHolder holder, int format, int width, int height) {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void surfaceCreated(SurfaceHolder arg0) {
|
||||
if (mCamera != null) {
|
||||
try {
|
||||
mCamera.setPreviewDisplay(mCameraView.getHolder());
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void surfaceDestroyed(SurfaceHolder arg0) {
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue