X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;ds=sidebyside;f=app%2Fsrc%2Fmain%2Fjni%2Fai_suanzi_rtmpclient_Ffmpeg.cpp;h=decd9b4bca703210f32ca38c95c78cb62eae77c7;hb=8af8da070650175d3a35a840fbd339ee572d144f;hp=e6c8d158382d33318ed01ac71280df8aaff34eed;hpb=831cc09829bc6e18d8d0d8bb78063e89ea565ce9;p=rtmpclient.git

diff --git a/app/src/main/jni/ai_suanzi_rtmpclient_Ffmpeg.cpp b/app/src/main/jni/ai_suanzi_rtmpclient_Ffmpeg.cpp
index e6c8d15..decd9b4 100644
--- a/app/src/main/jni/ai_suanzi_rtmpclient_Ffmpeg.cpp
+++ b/app/src/main/jni/ai_suanzi_rtmpclient_Ffmpeg.cpp
@@ -109,6 +109,90 @@ JNIEXPORT jstring JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_getVersion (JNIEnv *e
     return env->NewStringUTF("====== Ffmpeg call =======");
 }
 
+//const char* out_path;
+
+JNIEXPORT void JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_setRtmpUrl (JNIEnv *env, jobject obj, jstring url){
+    LOGE("set rtmpurl");
+    //const char* out_path = "rtmp://gpussh.suanzi.ai:1935/myapp/suanzi_ac83f34ead90_cameraid";
+    //out_path  = env->GetStringUTFChars(url, 0);
+
+}
+JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_initnew (JNIEnv *env, jobject obj, jint width, jint height, jstring url)
+{
+    const char * out_path= env->GetStringUTFChars(url, 0);
+    LOGE("Ffmpeg init, width=%d, heigh=%d, url=%s", width, height, out_path);
+
+	yuv_width=width;
+	yuv_height=height;
+	y_length=width*height;
+	uv_length=width*height/4;
+
+
+	av_register_all();
+
+	//output initialize
+	avformat_alloc_output_context2(&ofmt_ctx, NULL, "flv", out_path);
+	//output encoder initialize
+	pCodec = avcodec_find_encoder(AV_CODEC_ID_H264);
+	if (!pCodec){
+		LOGE("Can not find encoder!\n");
+		return -1;
+	}
+	pCodecCtx = avcodec_alloc_context3(pCodec);
+	pCodecCtx->pix_fmt = AV_PIX_FMT_YUV420P;
+	pCodecCtx->width = width;
+	pCodecCtx->height = height;
+	pCodecCtx->time_base.num = 1;
+	pCodecCtx->time_base.den = 30;
+	pCodecCtx->bit_rate = 800000;
+	pCodecCtx->gop_size = 300;
+	/* Some formats want stream headers to be separate. */
+	if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
+		pCodecCtx->flags |= CODEC_FLAG_GLOBAL_HEADER;
+
+	//H264 codec param
+	//pCodecCtx->me_range = 16;
+	//pCodecCtx->max_qdiff = 4;
+	//pCodecCtx->qcompress = 0.6;
+	pCodecCtx->qmin = 10;
+	pCodecCtx->qmax = 51;
+	//Optional Param
+	pCodecCtx->max_b_frames = 3;
+	// Set H264 preset and tune
+	AVDictionary *param = 0;
+	av_dict_set(&param, "preset", "ultrafast", 0);
+	av_dict_set(&param, "tune", "zerolatency", 0);
+
+	if (avcodec_open2(pCodecCtx, pCodec, &param) < 0){
+		LOGE("Failed to open encoder!\n");
+		return -1;
+	}
+
+	//Add a new stream to output,should be called by the user before avformat_write_header() for muxing
+	video_st = avformat_new_stream(ofmt_ctx, pCodec);
+	if (video_st == NULL){
+		return -1;
+	}
+	video_st->time_base.num = 1;
+	video_st->time_base.den = 30;
+	video_st->codec = pCodecCtx;
+
+	//Open output URL,set before avformat_write_header() for muxing
+	jint ret = 0;
+	if (( ret = avio_open(&ofmt_ctx->pb, out_path, AVIO_FLAG_READ_WRITE)) < 0){
+		LOGE("Failed to open output file! return :%s(%d)\n", av_err2str(ret),ret);
+		return -1;
+	}
+
+	//Write File Header
+	avformat_write_header(ofmt_ctx, NULL);
+
+	start_time = av_gettime();
+    env->ReleaseStringUTFChars(url, out_path);
+    return 0;
+
+}
+
 JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_inithaha (JNIEnv *env, jobject obj, jint width, jint height) {
 
 	//const char* out_path = "/storage/emulated/0/Movies/output.flv";
@@ -245,12 +329,20 @@ JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_close (JNIEnv *env, jobj
     return 0;
 }
 
+
+
+JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_processnew (JNIEnv *env, jobject obj, jbyteArray yuv){
+
+}
+
+
+
 JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_process (JNIEnv *env, jobject obj, jbyteArray yuv){
 	int ret;
 	int enc_got_frame=0;
 	int i=0;
 
-    LOGE(" process data - ffmpeg");
+    //LOGE(" process data - ffmpeg");
 	pFrameYUV = av_frame_alloc();
 	uint8_t *out_buffer = (uint8_t *)av_malloc(avpicture_get_size(AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));
 	avpicture_fill((AVPicture *)pFrameYUV, out_buffer, AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);
@@ -275,7 +367,9 @@ JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_process (JNIEnv *env, jo
 	av_frame_free(&pFrameYUV);
 
 	if (enc_got_frame == 1){
-		//LOGE("Succeed to encode frame: %5d\tsize:%5d\n", framecnt, enc_pkt.size);
+        if (framecnt % (15 * 60) == 0){
+		    LOGE("Succeed to encode frame: %5d\tsize:%5d\n", framecnt, enc_pkt.size);
+        }
 		framecnt++;
 		enc_pkt.stream_index = video_st->index;
 
@@ -301,6 +395,7 @@ JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_process (JNIEnv *env, jo
 		ret = av_interleaved_write_frame(ofmt_ctx, &enc_pkt);
 		av_free_packet(&enc_pkt);
 	}
+    av_free(out_buffer);
     return 0;
 }
 
@@ -493,8 +588,8 @@ JNIEXPORT jint JNICALL Java_ai_suanzi_rtmpclient_Ffmpeg_push (JNIEnv *env, jobje
 //    av_log_set_callback(custom_log);
  // Open Output
     //const char* out_path = "rtmp://192.168.1.35:1935/myapp/peng2";
-    //const char* out_path =  "rtmp://gpussh.suanzi.ai:1935/myapp/suanzi_ac83f34ead90_cameraid";
-    const char* out_path =  env->GetStringUTFChars(url, 0);
+    const char* out_path =  "rtmp://gpussh.suanzi.ai:1935/myapp/suanzi_ac83f34ead90_cameraid";
+    //const char* out_path =  env->GetStringUTFChars(url, 0);
     //const char * file_name = env->GetStringUTFChars(fname, 0);