pt1: recpt1/tssplitter_lite.c comparison

comparison recpt1/tssplitter_lite.c @ 99:3a3f15b063e1

now recpt1 accepts single sid for terrestrial channels. technical summary: - mark pid to record instead of pid to drop - split after decoding for now

author	Yoshiki Yazawa <yaz@honeyplanet.jp>
date	Fri, 12 Feb 2010 21:40:21 +0900
parents	3fd15032fd3a
children	aeba1988234f

comparison

equal deleted inserted replaced

-:3fd15032fd3a
+:3a3f15b063e1
 #include <sys/stat.h>
 #include "decoder.h"
 #include "recpt1.h"
 #include "tssplitter_lite.h"
+/* prototypes */
+static char** AnalyzeSid(char* sid);
+static int ReadTs( unsigned char** pat, unsigned char* pids, char** sid_list, unsigned char* pmt_pids, ARIB_STD_B25_BUFFER *sbuf, int* pmt_remain, int* pmt_counter);
+static int AnalyzePat(unsigned char* buf, unsigned char** pat, unsigned char* pids, char** sid_list, unsigned char* pmt_pids, int* pmt_remain);
+static int RecreatePat(unsigned char* buf, unsigned char** pat, unsigned char* pids, int *pos);
+static int AnalyzePmt(unsigned char* buf, unsigned char* pids);
+static int GetCrc32(unsigned char* data, int len);
+static int GetPid(unsigned char* data);
 /**
 * サービスID解析
 */
-char** AnalyzeSid(
+static char** AnalyzeSid(
 	char* sid)						// [in]		サービスID(カンマ区切りテキスト)
 {
 	int i = 0;
 	char** sid_list = NULL;
 	char* p;
 	{
 		free(sp);
 		return NULL;
 	}
 	sp->pat_count	= 0xFF;
-	sp->pmt_drop = -1;
+	sp->pmt_remain = -1;
 	sp->pmt_counter = 0;
 	return sp;
 }
 	ARIB_STD_B25_BUFFER *sbuf			// [in]			入力TS
 )
 {
 	int result;
 	// TS解析
-	result = ReadTs(&(sp->pat), sp->pids, sp->sid_list, sp->pmt_pids, sbuf, &(sp->pmt_drop), &(sp->pmt_counter));
+	result = ReadTs(&(sp->pat), sp->pids, sp->sid_list, sp->pmt_pids, sbuf, &(sp->pmt_remain), &(sp->pmt_counter));
 	return result;
 }
 /**
 /**
 * TS 解析処理
 *
 * 対象のチャンネル番号のみの PAT の再構築と出力対象 PID の抽出を行う
 */
-int ReadTs(
+static int ReadTs(
 	unsigned char** pat,				// [out]	PAT 情報（再構築後）
 	unsigned char* pids,				// [out]	出力対象 PID 情報
 	char** sid_list,					// [in]		出力対象サービス ID のリスト
 	unsigned char* pmt_pids,			// [in]		出力対象PIDのPMT PID
 	ARIB_STD_B25_BUFFER *sbuf,			// [in]		pt1_drvの入力TS
-	int* pmt_drop,						// [in]		PMTの落とすべき数
+	int* pmt_remain,						// [in]		PMTの落とすべき数
 	int* pmt_counter					// [out]	PMTの落とした数
 )
 {
 	int length = sbuf->size;
 	int pid;
 	{
 		pid = GetPid(((unsigned char*)sbuf->data)+index+1);
 		// PAT
 		if (0x0000 == pid)
 		{
-			result = AnalyzePat(((unsigned char*)sbuf->data)+index, pat, pids, sid_list, pmt_pids, pmt_drop);
+			result = AnalyzePat(((unsigned char*)sbuf->data)+index, pat, pids, sid_list, pmt_pids, pmt_remain);
 			if (TSS_SUCCESS != result)
 			{
 				/* 下位の関数内部でmalloc error発生 */
 				return result;
 			}
 		}
 		// PMT
-		/* 落とすpmt_pidである場合には、pmtに書かれている
+		/* 残すpmt_pidである場合には、pmtに書かれている
-		 * 落とすべきPCR/AUDIO/VIDEO PIDを取得する */
+		 * 残すべきPCR/AUDIO/VIDEO PIDを取得する */
 		if (pmt_pids[pid] == 1)
 		{
+//			fprintf(stderr, "\npmt_pid=%d\n", pid);
 			/* この中にはPMT毎に一度しか入らないようにしておく */
 			AnalyzePmt(((unsigned char*)sbuf->data)+index, pids);
 			pmt_pids[pid]++;
-			*pmt_counter = *(pmt_counter)+1;
+			*pmt_counter += 1;
 		}
-		/* 全ての落とすPMTの中に書かれている、落とすPCR/AUDIO/VIDEOのPIDを得たら処理を抜ける */
+		/* 録画する全てのPMTについて、中にあるPCR/AUDIO/VIDEOのPIDを
-		/* pmt_counter と pmt_drop が一致する場合に条件は満たされる */
+		 * 得る */
-		if ((*pmt_counter == *pmt_drop)) {
+		/* pmt_counter と pmt_remain が一致する場合に条件は満たされる */
+		if ((*pmt_counter == *pmt_remain)) {
 			result = TSS_SUCCESS;
 			break;
 		}
-		else
+		else {
-		{
 			result = TSS_ERROR;
 		}
 		index += LENGTH_PACKET;
 	}
 * TS 分離処理
 */
 int split_ts(
 	splitter *splitter,					// [in]		splitterパラメータ
 	ARIB_STD_B25_BUFFER *sbuf,			// [in]		入力TS
-	BUFSZ *dbuf							// [out]	出力TS
+	splitbuf_t *dbuf							// [out]	出力TS
 )
 {
 	int pid;
+	unsigned char *sptr, *dptr;
 	int s_offset = 0;
 	int d_offset = 0;
 	/* 初期化 */
 	dbuf->size = 0;
-	if ( sbuf->size < 0 )
+	if (sbuf->size < 0) {
-	{
 		return TSS_ERROR;
 	}
-	while (sbuf->size > s_offset)
+	sptr = sbuf->data;
-	{
+	dptr = dbuf->buffer;
-		pid = GetPid(((unsigned char*)sbuf->data)+s_offset+1);
+	while(sbuf->size > s_offset) {
+		pid = GetPid(sptr + s_offset + 1);
+switch(pid) {
 		// PAT
-		if (0x0000 == pid)
+		case 0x0000:
-		{
 			// 巡回カウンタカウントアップ
-			if (0xFF == splitter->pat_count)
+			if(0xFF == splitter->pat_count) {
-			{
 				splitter->pat_count = splitter->pat[3];
-			} else
+			}
-			{
+			else {
-				splitter->pat_count = (splitter->pat_count)+1;
+				splitter->pat_count = (splitter->pat_count + 1) % 16;
-				if (0 == splitter->pat_count % 0x10)
-				{
-					splitter->pat_count = splitter->pat_count - 0x10;
-				}
 			}
 			splitter->pat[3] = splitter->pat_count;
-			memcpy(((unsigned char*)dbuf->buffer)+d_offset, splitter->pat, LENGTH_PACKET);
+			memcpy(dptr + d_offset, splitter->pat, LENGTH_PACKET);
 			d_offset += LENGTH_PACKET;
-			dbuf->size = dbuf->size + LENGTH_PACKET;
+			dbuf->size += LENGTH_PACKET;
-		}
+			break;
+		default:
-		// その他 PID
+			/* pids[pid] が 1 は残すパケットなので書き込む */
-		else
+			if(1 == splitter->pids[pid]) {
-		{
+				memcpy(dptr + d_offset, sptr + s_offset, LENGTH_PACKET);
-			/* pids[pid] が 0 は落とさないパケットなので書き込む */
-			if (0 == splitter->pids[pid])
-			{
-				memcpy(((unsigned char*)dbuf->buffer)+d_offset, ((unsigned char*)sbuf->data)+s_offset, LENGTH_PACKET);
 				d_offset += LENGTH_PACKET;
-				dbuf->size = dbuf->size + LENGTH_PACKET;
+				dbuf->size += LENGTH_PACKET;
 			}
-		}
+			break;
-		s_offset = s_offset + LENGTH_PACKET;
+		} /* switch */
+		s_offset += LENGTH_PACKET;
 	}
 	return(TSS_SUCCESS);
 }
 /**
 * PAT 解析処理
 *
 * PAT を解析し、出力対象チャンネルが含まれているかチェックを行い、PAT を再構築する
 */
-int AnalyzePat(
+static int AnalyzePat(
 	unsigned char* buf,					// [in]		読み込んだバッファ
 	unsigned char** pat,				// [out]	PAT 情報（再構築後）
 	unsigned char* pids,				// [out]	出力対象 PID 情報
 	char** sid_list,					// [in]		出力対象サービス ID のリスト
 	unsigned char* pmt_pids,			// [out]	サービス ID に対応する PMT の PID
-	int* pmt_drop
+	int* pmt_remain
 )						// [out]	落とすPMTの数
 {
 	int pos[MAX_PID];
 	int service_id;
 	int i, j=0, k;
 	int result = TSS_SUCCESS;
 	char **p;
 	int sid_found;
 	int avail_sids[MAX_SERVICES];
-	if (NULL == *pat)
+	if(*pat == NULL) {
-	{
 		/* 初期化 */
-		*pmt_drop = 0;
+		*pmt_remain = 0;
 		memset(pos, 0, sizeof(pos));
 		size = buf[7];
 		// 対象チャンネル判定
 		/* size + 8 = パケット全長 */
 		/* 最終 4 バイトはCRCなので飛ばす */
-		for (i = 17; i < (size + 8) - 4; i = i + 4)
+		for(i = 17; i < (size + 8) - 4; i = i + 4) {
-		{
 			sid_found = 0;
 			service_id = (buf[i] << 8) + buf[i + 1];
 			avail_sids[j] = service_id;
 			p = sid_list;
-			while(*p != NULL)
-			{
+			while(*p) {
-				if (service_id == atoi(*p))
+				if(service_id == atoi(*p)) {
-				{
 					/* 録画対象 = 落とす対象とはしないものなので、基本的に何もしない */
-					/* 録画対象の pmt_pids は 0 とする */
+					/* 録画対象の pmt_pids は 1 とする */
-					/* 録画対象の pmt の pids は 0 とする */
+					/* 録画対象の pmt の pids は 1 とする */
 					pid = GetPid(&buf[i + 2]);
-					*(pmt_pids+pid) = 0;
+					*(pmt_pids+pid) = 1;
-					*(pids+pid) = 0;
+					*(pids+pid) = 1;
 					pos[pid] = i;
 					sid_found = 1;
+					*pmt_remain += 1;
+//					fprintf(stderr, "sid=%d pid=%d\n", service_id, pid);
 					break;
 				}
-				else if (strstr(*p, "all") != NULL )
+				else if(strstr(*p, "all")) {
-				{
 					/* all指定時には全保存する */
 					pid = GetPid(&buf[i + 2]);
-					*(pmt_pids+pid) = 0;
+					*(pmt_pids+pid) = 1;
-					*(pids+pid) = 0;
+					*(pids+pid) = 1;
 					pos[pid] = i;
 					sid_found = 1;
+					*pmt_remain += 1;
+//					fprintf(stderr, "sid=%d pid=%d\n", service_id, pid);
 					break;
 				}
 				p++;
-			}
+			} /* while */
-			if ( ! sid_found )
-			{
-				/* sid_list を全部なめたが録画対象であると判定されないものを落とす */
-				/* 削除対象の pmt_pids は 1 とする */
-				/* 削除対象の pmt の pids は 1 とする */
-				pid = GetPid(&buf[i + 2]);
-				*(pmt_pids+pid) = 1;
-				*(pids+pid) = 1;
-				pos[pid] = i;
-				*(pmt_drop) = *(pmt_drop)+1;
-			}
 			j++;
 		}
+		/* print SIDs */
 		fprintf(stderr, "Available sid = ");
 		for(k=0; k<j; k++)
 			fprintf(stderr, "%d ", avail_sids[k]);
 		fprintf(stderr, "\n");
 		}
 		fprintf(stderr, "\n");
 		// PAT 再構築
 		result = RecreatePat(buf, pat, pids, pos);
+#if 0
+		int tc;
+		for(tc=0; tc<188; tc++)
+			fprintf(stderr, "%02x ", *(*pat+tc));
+#endif
 	}
 	return(result);
 }
 /**
 * PAT 再構築処理
 *
 * PMT から出力対象チャンネル以外のチャンネル情報を削除し、PAT を再構築する
 */
-int RecreatePat(
+static int RecreatePat(
 	unsigned char* buf,					// [in]		読み込んだバッファ
 	unsigned char** pat,				// [out]	PAT 情報（再構築後）
 	unsigned char* pids,				// [out]	出力対象 PID 情報
 	int *pos)							// [in]		取得対象 PMT のバッファ中の位置
 {
 }
 /**
 * PMT 解析処理
 *
-* PMT を解析し、削除対象の PID を特定する
+* PMT を解析し、保存対象の PID を特定する
 */
-int AnalyzePmt(
+static int AnalyzePmt(
 	unsigned char* buf,					// [in]		読み込んだバッファ
 	unsigned char* pids)				// [out]	出力対象 PID 情報
 {
 	unsigned char Nall;
 	unsigned char N;
 	Nall = ((buf[6] & 0x0F) << 4) + buf[7];
 	// PCR
 	pcr = GetPid(&buf[13]);
 	pids[pcr] = 1;
+//    fprintf(stderr, "pcr_pid:%5d", pcr);
 	N = ((buf[15] & 0x0F) << 4) + buf[16] + 16 + 1;
 	// ES PID
 	while (N < Nall + 8 - 4)
 	{
-		epid = GetPid(&buf[N + 1]);
+		// ストリーム種別が 0x0D（type D）は出力対象外
+		if (0x0D != buf[N])
-		pids[epid] = 1;
+		{
+			epid = GetPid(&buf[N + 1]);
+			pids[epid] = 1;
+//			fprintf(stderr, "%10d", epid);
+		}
 		N += 4 + (((buf[N + 3]) & 0x0F) << 4) + buf[N + 4] + 1;
 	}
 	return TSS_SUCCESS;
 }
 /**
 * CRC 計算
 */
-int GetCrc32(
+static int GetCrc32(
 	unsigned char* data,				// [in]		CRC 計算対象データ
 	int len)							// [in]		CRC 計算対象データ長
 {
 	int crc;
 	int i, j;
 }
 /**
 * PID 取得
 */
-int GetPid(
+static int GetPid(
 	unsigned char* data)				// [in]		取得対象データのポインタ
 {
 	return ((data[0] & 0x1F) << 8) + data[1];
 }

Mercurial > pt1

comparison recpt1/tssplitter_lite.c @ 99:3a3f15b063e1