基于ffmpeg 直播推流和播放rtmp (IOS源码)

　　ios直播推流每秒能达到30帧，比安卓要强，视频采用软编码的话手机会发烫，得采用码编码，播放视频采用opengl渲染。

 　

　　ffmpeg初始化代码如下：

int init_Code(int width, int height, const char *out_path) {
    av_log_set_callback(custom_log);
    //avcodec_register_all();
    av_register_all();
    avformat_network_init();
    avformat_alloc_output_context2(&ofmt_ctx, NULL, "flv", out_path);
    
    //Open output URL,set before avformat_write_header() for muxing
    AVOutputFormat * ofmt = ofmt_ctx->oformat;
    if (!(ofmt->flags & AVFMT_NOFILE)) { //Open output URL
        if (avio_open(&ofmt_ctx->pb, out_path, AVIO_FLAG_READ_WRITE) < 0) {
            return -1;
        }
    }
    if (isAudio == 1) {
        if (init_audio_Code() != 0)//初始化 音频参数
            return -1;
    }
    if (isVideo == 1) {
        if (init_video_code(width, height) != 0)//初始化 视频参数
            return -1;
    }
    av_dump_format(ofmt_ctx, 0, out_path, 1);
    if (avformat_write_header(ofmt_ctx, NULL) < 0) { //Write file header
        //LOGE("Error occurred when opening output file
");
        return -1;
    }
    
    start_thread_encode(); //开始编码
    return 0;
}

 　　

　　视频硬编码：

/* 视频 硬编码**/
int encodeVideo_h264(uint8_t* in, int64_t time, int size, int keyframe) {
    int ret;
    // 定义AVPacket对象后,请使用av_init_packet进行初始化
    av_init_packet(&video_pkt);
    //av_new_packet(&video_pkt,size);
    video_pkt.stream_index = video_st->index;
    video_pkt.data = in;
    
    video_pkt.size = size;
    video_pkt.pos = -1;
    ptsPacket(video_st, &video_pkt, time);
    if (video_pkt.buf != NULL) {
        video_pkt.buf->data = in;
        video_pkt.buf->size = size;
    }
    video_pkt.flags = keyframe;
    if (keyframe == 1) {
        //LOGE("硬编码-关键帧: %lld", time);
    }
    
    ret = av_interleaved_write_frame(ofmt_ctx, &video_pkt);
    if (ret != 0) {
        printf("----encodeVideo--encodeVideo  -ret: %d ", ret);
        //LOGE("----encodeVideo--encodeVideo  -ret: %d ", ret);
    }
    av_free_packet(&video_pkt);

    return 0;
}

　　音频硬编码：

/* 音频 硬编码**/
int encodeAudio_AAC(uint8_t* in, int64_t time, int size) {
    if (isAudio == 0)
        return 0;
    av_init_packet(&audio_pkt);
    int got_frame = 0;
    audio_pkt.stream_index = audio_st->index; //标识该AVPacket所属的视频/音频流。
    audio_pkt.data = in;
    audio_pkt.size = size;
    audio_pkt.pts = time;
    audio_pkt.dts = time;
    //audio_pkt.pos = -1;
    audio_pkt.flags = 1;
    //audio_pkt.duration = 10;
    int ret = av_interleaved_write_frame(ofmt_ctx, &audio_pkt);
    if (ret != 0) {
        //LOGE("----encodeAudio---ret: %d  size:%d  ,time:%lld ",
        //      ret, size, time);
    }
    return 0;
}

　　初始化相机：

- (void) initCamera:(BOOL)type
{
    NSError *deviceError;
    AVCaptureDeviceInput *inputCameraDevice;
    if (type==false)
    {
        inputCameraDevice = [AVCaptureDeviceInput deviceInputWithDevice:cameraDeviceB error:&deviceError];
    }
    else
    {
        inputCameraDevice = [AVCaptureDeviceInput deviceInputWithDevice:cameraDeviceF error:&deviceError];
    }
    AVCaptureVideoDataOutput *outputVideoDevice = [[AVCaptureVideoDataOutput alloc] init];
    
    NSString* key = (NSString*)kCVPixelBufferPixelFormatTypeKey;
    NSNumber* val = [NSNumber numberWithUnsignedInt:kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange];
    NSDictionary* videoSettings = [NSDictionary dictionaryWithObject:val forKey:key];
    outputVideoDevice.videoSettings = videoSettings;
    [outputVideoDevice setSampleBufferDelegate:self queue:dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_HIGH, 0)];
    captureSession = [[AVCaptureSession alloc] init];
    [captureSession addInput:inputCameraDevice];
    [captureSession addOutput:outputVideoDevice];
    [captureSession beginConfiguration];
    
    [captureSession setSessionPreset:[NSString stringWithString:AVCaptureSessionPreset352x288]];
    connectionVideo = [outputVideoDevice connectionWithMediaType:AVMediaTypeVideo];
#if TARGET_OS_IPHONE
    [self setRelativeVideoOrientation];
    
    NSNotificationCenter* notify = [NSNotificationCenter defaultCenter];
    [notify addObserver:self
               selector:@selector(statusBarOrientationDidChange:)
                   name:@"StatusBarOrientationDidChange"
                 object:nil];
#endif
    
    [captureSession commitConfiguration];
    recordLayer = [AVCaptureVideoPreviewLayer    layerWithSession:captureSession];
    [recordLayer setVideoGravity:AVLayerVideoGravityResizeAspect];
}

　　

　　设置音频参数

　　

- (void)setupAudioFormat:(UInt32) inFormatID SampleRate:(int)sampeleRate
{
    //重置下
    memset(&_recordFormat, 0, sizeof(_recordFormat));
    //设置采样率，这里先获取系统默认的测试下 //TODO:
    //采样率的意思是每秒需要采集的帧数
    _recordFormat.mSampleRate = sampeleRate;//[[AVAudioSession sharedInstance] sampleRate];
    UInt32 size = sizeof(_recordFormat.mSampleRate);
    //AudioSessionGetProperty(    kAudioSessionProperty_CurrentHardwareSampleRate,
    //                                    &size,
    //                                  &_recordFormat.mSampleRate);
    size = sizeof(_recordFormat.mChannelsPerFrame);
    // AudioSessionGetProperty(    kAudioSessionProperty_CurrentHardwareInputNumberChannels,
    //                                      &size,
    //                                    &_recordFormat.mChannelsPerFrame);

    _recordFormat.mFormatID = inFormatID;
    if (inFormatID == kAudioFormatLinearPCM){
        //这个屌属性不知道干啥的。，//要看看是不是这里属性设置问题
        //结果分析: 8bit为1byte，即为1个通道里1帧需要采集2byte数据，再*通道数，即为所有通道采集的byte数目。
        //所以这里结果赋值给每帧需要采集的byte数目，然后这里的packet也等于一帧的数据。

        _recordFormat.mFramesPerPacket = 1;
        _recordFormat.mSampleRate =sampeleRate;// 16000.0;
        //每个通道里，一帧采集的bit数目 语音每采样点占用位数
        _recordFormat.mBitsPerChannel = 16;
        _recordFormat.mChannelsPerFrame = 2;// 1:单声道；2:立体声
        _recordFormat.mFramesPerPacket = 1;
        _recordFormat.mBytesPerFrame = (_recordFormat.mBitsPerChannel / 8) * _recordFormat.mChannelsPerFrame;
        _recordFormat.mBytesPerPacket = _recordFormat.mBytesPerFrame * _recordFormat.mFramesPerPacket;
        //_recordFormat.mBytesPerPacket = _recordFormat.mBytesPerFrame = (_recordFormat.mBitsPerChannel / 8) * _recordFormat.mChannelsPerFrame;
        _recordFormat.mFormatFlags = kLinearPCMFormatFlagIsSignedInteger | kLinearPCMFormatFlagIsPacked;
        //_recordFormat.mFormatFlags = kLinearPCMFormatFlagIsSignedInteger | kLinearPCMFormatFlagIsPacked;
    }
}

　　开始录音：

-(void)startRecording
{
    UInt32 size;
    NSError *error = nil;
    //设置audio session的category
    BOOL ret = [[AVAudioSession sharedInstance] setCategory:AVAudioSessionCategoryRecord error:&error];//注意，这里选的是AVAudioSessionCategoryPlayAndRecord参数，如果只需要录音，就选择Record就可以了，如果需要录音和播放，则选择PlayAndRecord，这个很重要
    if (!ret) {
        NSLog(@"设置声音环境失败");
        return;
    }
    //启用audio session
    ret = [[AVAudioSession sharedInstance] setActive:YES error:&error];
    if (!ret)
    {
        NSLog(@"启动失败");
        return;
    }
    //初始化音频输入队列
    AudioQueueNewInput(&_recordFormat, inputBufferHandler, (__bridge void *)(self), NULL, kCFRunLoopCommonModes, 0, &_audioQueue);//inputBufferHandler这个是回调函数名
    size = sizeof(_recordFormat);
    //AudioQueueGetProperty(_audioQueue, kAudioQueueProperty_StreamDescription,
    //                                   &_recordFormat, &size);
    //计算估算的缓存区大小
    //int frames = (int)ceil(kDefaultBufferDurationSeconds * _recordFormat.mSampleRate);//返回大于或者等于指定表达式的最小整数
    int bufferByteSize =4096;// frames * _recordFormat.mBytesPerFrame;//缓冲区大小在这里设置，这个很重要，在这里设置的缓冲区有多大，那么在回调函数的时候得到的inbuffer的大小就是多大。
    //bufferByteSize=[self ComputeRecordBufferSize:&_recordFormat sss:kDefaultBufferDurationSeconds];
    NSLog(@"缓冲区大小:%d",bufferByteSize);
    AudioQueueBufferRef         _audioBuffers[3];
    //创建缓冲器
    for (int i = 0; i < kNumberAudioQueueBuffers; i++){
        AudioQueueAllocateBuffer(_audioQueue, bufferByteSize, &_audioBuffers[i]);
        AudioQueueEnqueueBuffer(_audioQueue, _audioBuffers[i], 0, NULL);//将 _audioBuffers[i]添加到队列中
    }
    // 开始录音
    AudioQueueStart(_audioQueue, NULL);
   
}

　　源码地址：http://pan.baidu.com/s/1qXOSznA