h264文件数据帧前添加帧长度
haungdezhi
parent
a79b70c200
commit
8c562c19a6
176
H264Parser.java
176
H264Parser.java
|
|
@ -1,58 +1,104 @@
|
|||
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.InputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Formatter;
|
||||
|
||||
public class H264Parser {
|
||||
|
||||
//0x00000001或0x000001是一个nalu的起始标志，遇到下一个此标志时为该nalu的结尾。
|
||||
//起始标志的后面第一个字节（type）里包含有nalu的类型，type & 0x1F即为该nalu的类型（nalu_type）。
|
||||
|
||||
private static final int NALU_TYPE_SLICE = 1; //非IDR图像中不采用数据划分的片段，为P或B帧
|
||||
private static final int NALU_TYPE_DPA = 2; //非IDR图像中A类数据划分片段
|
||||
private static final int NALU_TYPE_DPB = 3; //非IDR图像中B类数据划分片段
|
||||
private static final int NALU_TYPE_DPC = 4; //非IDR图像中C类数据划分片段
|
||||
private static final int NALU_TYPE_IDR = 5; //IDR图像的片段，关键帧，I帧
|
||||
private static final int NALU_TYPE_SEI = 6; //补充增强信息
|
||||
private static final int NALU_TYPE_SPS = 7; //序列参数集（Sequence Parameter Set）
|
||||
private static final int NALU_TYPE_PPS = 8; //图像参数集PPS（Picture Parameter Set）
|
||||
private static final int NALU_TYPE_AUD = 9; //分割符
|
||||
private static final int NALU_TYPE_EOSEQ = 10; //序列结束符
|
||||
private static final int NALU_TYPE_EOSTREAM = 11; //流结束符
|
||||
private static final int NALU_TYPE_FILL = 12; //填充数据
|
||||
|
||||
//0x00000001或0x000001是一个nalu的起始标志，遇到下一个此标志时为该nalu的结尾。
|
||||
//起始标志的后面第一个字节（type）里包含有nalu的类型，type & 0x1F即为该nalu的类型（nalu_type）。
|
||||
|
||||
private static final int NALU_TYPE_SLICE = 1; //非IDR图像中不采用数据划分的片段，为P或B帧
|
||||
private static final int NALU_TYPE_DPA = 2; //非IDR图像中A类数据划分片段
|
||||
private static final int NALU_TYPE_DPB = 3; //非IDR图像中B类数据划分片段
|
||||
private static final int NALU_TYPE_DPC = 4; //非IDR图像中C类数据划分片段
|
||||
private static final int NALU_TYPE_IDR = 5; //IDR图像的片段，关键帧，I帧
|
||||
private static final int NALU_TYPE_SEI = 6; //补充增强信息
|
||||
private static final int NALU_TYPE_SPS = 7; //序列参数集（Sequence Parameter Set）
|
||||
private static final int NALU_TYPE_PPS = 8; //图像参数集PPS（Picture Parameter Set）
|
||||
private static final int NALU_TYPE_AUD = 9; //分割符
|
||||
private static final int NALU_TYPE_EOSEQ = 10; //序列结束符
|
||||
private static final int NALU_TYPE_EOSTREAM = 11; //流结束符
|
||||
private static final int NALU_TYPE_FILL = 12; //填充数据
|
||||
|
||||
private static final int NALU_PRIORITY_DISPOSABLE = 0;
|
||||
private static final int NALU_PRIORITY_LOW = 1;
|
||||
private static final int NALU_PRIORITY_HIGH = 2;
|
||||
private static final int NALU_PRIORITY_HIGHEST = 3;
|
||||
|
||||
private boolean mFirstFind = true; //第一次查找起始码
|
||||
private int mStartCodeLen = 0; //起始码长度
|
||||
private boolean mFirstFind = true; //第一次查找起始码
|
||||
private int mStartCodeLen = 0; //起始码长度
|
||||
private int mCurStartCodeLen = 0;
|
||||
private int mFrameLen = 0; //帧长度
|
||||
private int mFrameFirstByte = 0; //帧数据中的第一个字节
|
||||
private int mFrameLen = 0; //帧长度
|
||||
private int mFrameFirstByte = 0; //帧数据中的第一个字节
|
||||
private int mCurFrameFirstByte = 0;
|
||||
private int mPos = 0;
|
||||
private int mLen = 0;
|
||||
private String mStartCode = "";
|
||||
|
||||
|
||||
private int mNaluType = 0;
|
||||
private int mForbiddenBit = 0;
|
||||
private int mNalReferenceIdc = 0;
|
||||
|
||||
|
||||
private byte[] mFrameBuffer = null;
|
||||
private FileOutputStream mFileOutputStream = null;
|
||||
private boolean mIsAddFrameLen = false;
|
||||
|
||||
public static void main(String[] args) {
|
||||
if (args.length > 0) {
|
||||
H264Parser h264Parser = new H264Parser();
|
||||
H264Parser h264Parser = null;
|
||||
if (args.length == 2) {
|
||||
h264Parser = new H264Parser();
|
||||
if ("1".equals(args[1])) {
|
||||
h264Parser.mIsAddFrameLen = true;
|
||||
}
|
||||
h264Parser.init();
|
||||
h264Parser.ParseH264(args[0]);
|
||||
} else if (args.length == 1) {
|
||||
h264Parser = new H264Parser();
|
||||
h264Parser.init();
|
||||
h264Parser.ParseH264(args[0]);
|
||||
} else {
|
||||
System.out.println("Missing h264 filename.");
|
||||
String path = "C:\\Users\\hw\\Desktop\\test.h264";
|
||||
h264Parser = new H264Parser();
|
||||
h264Parser.init();
|
||||
h264Parser.ParseH264(path);
|
||||
}
|
||||
if (h264Parser != null) {
|
||||
h264Parser.clear();
|
||||
}
|
||||
}
|
||||
|
||||
//如果NALU对应的Slice为一帧的开始就用0x00000001(即4个字节长度)，否则就用0x000001(即3个字节长度)。
|
||||
|
||||
public H264Parser() {
|
||||
|
||||
}
|
||||
public void init() {
|
||||
mFrameBuffer = new byte[2 * 1024 * 1024];
|
||||
if (mIsAddFrameLen) {
|
||||
File file = new File("test_2.h264");
|
||||
try {
|
||||
mFileOutputStream = new FileOutputStream(file);
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
public void clear() {
|
||||
if (mFileOutputStream != null) {
|
||||
try {
|
||||
mFileOutputStream.close();
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//如果NALU对应的Slice为一帧的开始就用0x00000001(即4个字节长度)，否则就用0x000001(即3个字节长度)。
|
||||
public int FindStartCode(InputStream inputStream) {
|
||||
|
||||
byte[] startCode = new byte[4];
|
||||
|
|
@ -102,7 +148,7 @@ public class H264Parser {
|
|||
public int GetNalu(InputStream inputStream) {
|
||||
mPos = mPos + mLen;
|
||||
|
||||
//第一次查找起始码，只是为了移动文件指针
|
||||
//第一次查找起始码，只是为了移动文件指针
|
||||
if (mFirstFind) {
|
||||
mFirstFind = false;
|
||||
if (-1 == (mStartCodeLen = FindStartCode(inputStream))) {
|
||||
|
|
@ -112,40 +158,55 @@ public class H264Parser {
|
|||
|
||||
if (mStartCodeLen == 3) {
|
||||
mStartCode = "001";
|
||||
// 添加起始码
|
||||
mFrameBuffer[0] = 0;
|
||||
mFrameBuffer[1] = 0;
|
||||
mFrameBuffer[2] = 1;
|
||||
} else {
|
||||
mStartCode = "0001";
|
||||
// 添加起始码
|
||||
mFrameBuffer[0] = 0;
|
||||
mFrameBuffer[1] = 0;
|
||||
mFrameBuffer[2] = 0;
|
||||
mFrameBuffer[3] = 1;
|
||||
}
|
||||
|
||||
|
||||
List<Integer> startCodeList = new ArrayList<>();
|
||||
int frameLen = 0;
|
||||
while(true) {
|
||||
int tmp = -1;
|
||||
|
||||
//一个字节一个字节从文件流中读取
|
||||
//一个字节一个字节从文件流中读取
|
||||
try {
|
||||
tmp = inputStream.read();
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
//读取到文件尾
|
||||
//读取到文件尾
|
||||
if (tmp == -1) {
|
||||
mFrameLen = frameLen;
|
||||
// 另存为
|
||||
saveAs(frameLen);
|
||||
return -1;
|
||||
}
|
||||
|
||||
//帧数据长度累加
|
||||
|
||||
//读取到的数据保存到buffer
|
||||
mFrameBuffer[mStartCodeLen+frameLen] = (byte)(0xff & tmp);
|
||||
//帧数据长度累加
|
||||
frameLen++;
|
||||
|
||||
//将从文件中读取的数据保存到list中
|
||||
//将从文件中读取的数据保存到list中
|
||||
startCodeList.add(tmp);
|
||||
|
||||
//从一帧的第一个字节中获取NALU信息
|
||||
//从一帧的第一个字节中获取NALU信息
|
||||
if (frameLen == 1) {
|
||||
//如果上一次查找到的起始码长度是3，则上次已经保存了第一个字节的数据
|
||||
//如果上一次查找到的起始码长度是3，则上次已经保存了第一个字节的数据
|
||||
if (mStartCodeLen == 3) {
|
||||
tmp = mFrameFirstByte;
|
||||
frameLen++; //帧长度加上第一个字节
|
||||
//读取到的数据保存到buffer
|
||||
mFrameBuffer[mStartCodeLen+frameLen] = (byte)(0xff & tmp);
|
||||
frameLen++; //帧长度加上第一个字节
|
||||
}
|
||||
mCurFrameFirstByte = tmp;
|
||||
mForbiddenBit = tmp & 0x80;
|
||||
|
|
@ -153,12 +214,12 @@ public class H264Parser {
|
|||
mNaluType = tmp & 0x1f;
|
||||
}
|
||||
|
||||
//保存四个字节到list中
|
||||
//保存四个字节到list中
|
||||
if (startCodeList.size() < 4) {
|
||||
continue;
|
||||
}
|
||||
|
||||
//查找起始码，如果返回值大于0就表示找到了
|
||||
|
||||
//查找起始码，如果返回值大于0就表示找到了
|
||||
int startCodeLen = FindStartCode(startCodeList);
|
||||
if (startCodeLen > 0) {
|
||||
if (startCodeLen == 3) {
|
||||
|
|
@ -172,9 +233,29 @@ public class H264Parser {
|
|||
startCodeList.remove(0);
|
||||
}
|
||||
mFrameLen = frameLen;
|
||||
// 另存为
|
||||
saveAs(frameLen);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 添加帧长度并另存为
|
||||
*
|
||||
* @param frameLen 除去起始码的帧长度
|
||||
*/
|
||||
public void saveAs(int frameLen) {
|
||||
if (mFileOutputStream != null) {
|
||||
try {
|
||||
//System.out.println("len = " + mStartCodeLen+frameLen);
|
||||
mFileOutputStream.write(intToBytes(mStartCodeLen+frameLen), 0, 4);
|
||||
mFileOutputStream.write(mFrameBuffer, 0, mStartCodeLen+frameLen);
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
mFrameBuffer[0] = '\0';
|
||||
}
|
||||
}
|
||||
|
||||
public boolean ParseH264(String fileName) {
|
||||
if (fileName.equals("")) {
|
||||
return false;
|
||||
|
|
@ -227,6 +308,9 @@ public class H264Parser {
|
|||
if (dataLen == -1) {
|
||||
break;
|
||||
}
|
||||
if (iNaluNum == 3) {
|
||||
//break;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
|
|
@ -236,4 +320,16 @@ public class H264Parser {
|
|||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* int型转四字节byte数组
|
||||
*/
|
||||
public static byte[] intToBytes(int i){
|
||||
byte[] result=new byte[4];
|
||||
result[3]=(byte)((i >> 24)& 0xFF);
|
||||
result[2]=(byte)((i >> 16)& 0xFF);
|
||||
result[1]=(byte)((i >> 8)& 0xFF);
|
||||
result[0]=(byte)(i & 0xFF);
|
||||
return result;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue