Android移动直播，自定义采集视频数据推流

ImageReader mImageReader = ImageReader.newInstance(width, height, ImageFormat.YUV_420_888 ,1);

ImageReader.OnImageAvailableListener mOnImageAvailableListener = new ImageReader.OnImageAvailableListener() {


        @RequiresApi(api = Build.VERSION_CODES.KITKAT)
        @Override
        public void onImageAvailable(ImageReader reader) {

            // 获取捕获的照片数据
            Image image = reader.acquireNextImage();
            int width2 = image.getWidth();
            int height2 = image.getHeight();
            if(isPushFlag){

                byte[] yuv420pbuf = camera2ImageToI420(image);

                //TODO setConfig（）
                int result= mLivePusher.sendCustomVideoData(yuv420pbuf, TXLivePusher.YUV_420P, width2, height2);
            }
        }
}

@RequiresApi(api = Build.VERSION_CODES.KITKAT)
    public byte[] camera2ImageToI420(Image image){
        int width3 = image.getWidth();
        int height3 = image.getHeight();

        // 从image里获取三个plane
        Image.Plane[] planes = image.getPlanes();
//        for (int i = 0; i < planes.length; i++) {
//           ByteBuffer iBuffer = planes[i].getBuffer();
//            int iSize = iBuffer.remaining();
//            Log.i("TAG", "pixelStride  " + planes[i].getPixelStride());
//            Log.i("TAG", "rowStride   " + planes[i].getRowStride());
//            Log.i("TAG", "width  " + image.getWidth());
//            Log.i("TAG", "height  " + image.getHeight());
//            Log.i("TAG", "buffer size  " + iSize);
//            Log.i("TAG", "Finished reading data from plane  " + i);
//        }
        // Y-buffer
        ByteBuffer yBuffer = planes[0].getBuffer();
        int ySize = yBuffer.remaining();
        byte[] yBytes = new byte[ySize];
        yBuffer.get(yBytes);

        // U-buffer
        ByteBuffer uBuffer = planes[1].getBuffer();
        int uSize = uBuffer.remaining();
        byte[] uBytes = new byte[uSize];
        uBuffer.get(uBytes);

        // V-buffer
        ByteBuffer vBuffer = planes[2].getBuffer();
        int vSize = vBuffer.remaining();
        byte[] vBytes = new byte[vSize];
        vBuffer.get(vBytes);

        byte[] ret = new byte[width3 * height3 * 3/2];
        int yLength = yBytes.length;
        int uLength = uBytes.length+1;
        int vLength = vBytes.length+1;

        ByteBuffer bufferY = ByteBuffer.wrap(ret, 0, yLength);
        ByteBuffer bufferU = ByteBuffer.wrap(ret, yLength, uLength/ 2);
        ByteBuffer bufferV = ByteBuffer.wrap(ret, yLength+uLength/ 2, vLength/ 2);

        bufferY.put(yBytes,0,yLength);
        for(int i=0;i<uLength;i+=2){
            bufferU.put(uBytes[i]);
        }
        for(int i=0;i<vLength;i+=2){
            bufferV.put(vBytes[i]);
        }

        return ret;
    }

//以下是简单的实例，获取摄像机预览回调的视频数据并推流
@Override
public void onPreviewFrame(byte[] data, Camera camera) {
    // 假设摄像机获取的视频格式是 NV21, 预览画面大小为 1280X720
    // 即宽度 mPreviewWidth 值为1280，高度 mPreviewHeight 值为 720
    if (!isPush) {
    } else {
        // 开始自定义推流
        // 需要将视频格式转码为 I420
        byte[] buffer = new byte[data.length];
        buffer = nv21ToI420(data, mPreviewWidth, mPreviewHeight);

        int customModeType = 0;
        customModeType |= TXLiveConstants.CUSTOM_MODE_VIDEO_CAPTURE;
        // 只能分辨率的宽和高小于或者等于预览画面的宽和高的分辨率 
        // 还能选择 360x640 等，但不能选择 540x960。因指定分辨率的高(960) > 预览画面的高(720)，编码器无法裁剪画面。
        mLivePushConfig.setVideoResolution(TXLiveConstants.VIDEO_RESOLUTION_TYPE_1280_720);
        mLivePushConfig.setAutoAdjustBitrate(false);
        mLivePushConfig.setVideoBitrate(1500);
        mLivePushConfig.setVideoEncodeGop(3);
        mLivePushConfig.setVideoFPS(18);
        mLivePushConfig.setCustomModeType(customModeType);
        mLivePusher.setConfig(mLivePushConfig);

        int result= mLivePusher.sendCustomVideoData(buffer, TXLivePusher.YUV_420P, mPreviewWidth, mPreviewHeight);
    }
}

/**
 * nv21转I420
 * @param data
 * @param width
 * @param height
 * @return
 */
public static byte[] nv21ToI420(byte[] data, int width, int height) {
    byte[] ret = new byte[data.length];
    int total = width * height;

    ByteBuffer bufferY = ByteBuffer.wrap(ret, 0, total);
    ByteBuffer bufferU = ByteBuffer.wrap(ret, total, total / 4);
    ByteBuffer bufferV = ByteBuffer.wrap(ret, total + total / 4, total / 4);

    bufferY.put(data, 0, total);
    for (int i=total; i<data.length; i+=2) {
        bufferV.put(data[i]);
        bufferU.put(data[i+1]);
    }
    return ret;
}