root/trunk/epgdumpr2/epgdump.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <getopt.h>
#include <iconv.h>
#include <time.h>

#include "ts.h"
#include "sdt.h"
#include "eit.h"
#include "ts_ctl.h"

typedef         struct  _ContentTYPE{
        char    *japanese ;
        char    *english ;
}CONTENT_TYPE;

#define         CAT_COUNT               16
static  CONTENT_TYPE    ContentCatList[CAT_COUNT] = {
        { "����若��糸���, "news" },
        { "�鴻��若�", "sports" },
        { "���", "information" },
        { "�����, "drama" },
        { "�恰ソ", "music" },
        { "��������, "variety" },
        { "���", "cinema" },
        { "�≪��＜��号�", "anime" },
        { "����ャ��潟�����紙�蕕�, "documentary" },
        { "羲��", "stage" },
        { "莇ｅ��糸���, "hobby" },
        { "胼霛�, "etc" },                        //胼霛�   { "篋��", "etc" }, //篋��
        { "篋��", "etc" }, //篋��
        { "篋��", "etc" }, //篋��
        { "���篁�, "etc" } //���篁�};
typedef struct _TAG_STATION
{
        char    *name;
        char    *ontv;
        int             tsId;           // OriginalNetworkID
        int             onId;           // TransportStreamID
        int             svId;           // ServiceID
} STATION;

static STATION bsSta[] = {
        { "NHK BS1", "3001.ontvjapan.com", 16625, 4, 101},
        { "NHK BS2", "3002.ontvjapan.com", 16625, 4, 102},//綮�� 2011/3/31 
        { "NHK BS��������, "3003.ontvjapan.com", 16626, 4, 103},
        { "BS�ャ���, "3004.ontvjapan.com", 16592, 4, 141},
        { "BS���", "3005.ontvjapan.com", 16400, 4, 151},
        { "BS-TBS", "3006.ontvjapan.com", 16401, 4, 161},
        { "BS�吾����", "3007.ontvjapan.com", 16433, 4, 171},
        { "BS���", "3008.ontvjapan.com", 16593, 4, 181},
//      { "WOWOW", "3009.ontvjapan.com", 16432, 4, 191},
//      { "WOWOW2", "3010.ontvjapan.com", 16432, 4, 192},
//      { "WOWOW3", "3011.ontvjapan.com", 16432, 4, 193},
        { "WOWOW����ゃ�", "3009.ontvjapan.com", 16432, 4, 191},
        { "WOWOW�����, "4192.epgdata.ontvjapan", 17488, 4, 192},
        { "WOWOW�激���, "4193.epgdata.ontvjapan", 17489, 4, 193},
//      { "�鴻��若�����潟���, "3012.ontvjapan.com", 16529, 4, 200},
        { "�鴻��若�����潟���", "3012.ontvjapan.com", 16529, 4, 200},
        { "�鴻��若�����潟���", "4201.epgdata.ontvjapan", 17520, 4, 201},
        { "�鴻��若�����潟���", "4202.epgdata.ontvjapan", 17520, 4, 202},
        { "BS11", "3013.ontvjapan.com", 16528, 4, 211},
        { "TwellV", "3014.ontvjapan.com", 16530, 4, 222},
        { "�冗�紊у�1", "4231.epgdata.ontvjapan", 18098, 4, 231},
        { "�冗�紊у�2", "4232.epgdata.ontvjapan", 18098, 4, 232},
        { "�冗�紊у�3", "4233.epgdata.ontvjapan", 18098, 4, 233},
        { "�違��若�����潟���, "4234.epgdata.ontvjapan", 18224, 4, 234},
        { "BS�≪�������", "1047.ontvjapan.com", 18033, 4, 236},
        { "FOX bs238", "4238.epgdata.ontvjapan", 18096, 4, 238},
        { "BS�鴻����鐚�, "4241.epgdata.ontvjapan", 18097, 4, 241},
        { "J SPORTS 1", "4242.epgdata.ontvjapan", 18225, 4, 242},
        { "J SPORTS 2", "4243.epgdata.ontvjapan", 18226, 4, 243},
        { "J SPORTS 3", "4244.epgdata.ontvjapan", 18257, 4, 244},
        { "J SPORTS 4", "4245.epgdata.ontvjapan", 18258, 4, 245},
        { "BS�ｃ�����с�", "4251.epgdata.ontvjapan", 18288, 4, 251},
        { "IMAGICA BS", "4252.epgdata.ontvjapan", 18256, 4, 252},
        { "�ユ����絨������潟���, "4255.epgdata.ontvjapan", 18289, 4, 255},
        { "����冴��若�����潟���, "1090.ontvjapan.com", 18034, 4, 256},
        { "D-Life", "4258.epgdata.ontvjapan", 18290, 4, 258},
        { "NHK膩�������吾��鰹��延根鐚�, "4291.epgdata.ontvjapan", 17168, 4, 291},
        { "NHK����������с�鐚��篋��", "4292.epgdata.ontvjapan", 17168, 4, 292},
        { "�ユ������, "4294.epgdata.ontvjapan", 17169, 4, 294},
        { "��������, "4295.epgdata.ontvjapan", 17169, 4, 295},
        { "TBS�����, "4296.epgdata.ontvjapan", 17169, 4, 296},
        { "������篋�, "4297.epgdata.ontvjapan", 17169, 4, 297},
        { "��������, "4298.epgdata.ontvjapan", 17168, 4, 298},
        { "�冗�紊у������, "4531.epgdata.ontvjapan", 18098, 4, 531},
        { "WNI", "4910.ontvjapan.com", 16626, 4, 910},
};

static int bsStaCount = sizeof(bsSta) / sizeof (STATION);



static STATION csSta[] = {
//ND2
        { "鐚器滋鐚潟��ｃ����鐚�, "3201.ontvjapan.com", 24608, 6, 296},//2012綛�����CS1/Ch.296�����     { "�������ｃ����鐚�爾", "1092.ontvjapan.com", 24608, 6, 298},//2012綛�����CS1/Ch.298�����   { "�������若��帥�鐚�爾", "1067.ontvjapan.com", 24608, 6, 299},//2012綛�����CS1/Ch.299�����
//ND4
        { "�鴻����鐚�����", "100ch.epgdata.ontvjapan", 28736, 7, 100},
        { "����潟���汐鐚ワ治鐚�, "1008.ontvjapan.com", 28736, 7, 223},
        { "�吟��激���, "1217.ontvjapan.com", 28736, 7, 227},//228��27
        { "鐔��鐔��鐚＜�������", "1040.ontvjapan.com", 28736, 7, 250},//2012綛���4��Ch.255���Ch.250�����     { "�������若��ｃ����", "1050.ontvjapan.com", 28736, 7, 342},
        { "�牙��糸�罍���ｃ����", "363ch.epgdata.ontvjapan", 28736, 7, 363},
                
//ND6
        { "���������鐚ｏ示", "294ch.epgdata.ontvjapan", 28768, 7, 294}, 
        { "鐚�軸鐚吟�鐚�爾", "1019.ontvjapan.com", 28768, 7, 323},
        { "罩��������", "329ch.epgdata.ontvjapan", 28768, 7, 329},
        { "����鴻������, "1062.ontvjapan.com", 28768, 7, 340},
        { "�≪������������, "1193.ontvjapan.com", 28768, 7, 341},
        { "鐚ｏ汐鐚��", "1069.ontvjapan.com", 28768, 7, 354},
//ND8
        { "�激��������潟���, "1059.ontvjapan.com", 24704, 6, 55},
        { "�掩�����潟���, "1010.ontvjapan.com", 24704, 6, 218},//2012/8/1���
        { "茵�����", "1005.ontvjapan.com", 24704, 6, 219},//2012/8/1���
        { "����若�����祉���, "1024.ontvjapan.com", 24704, 6, 326},//2012/8/1���
        { "����冴��若��ャ���, "339ch.epgdata.ontvjapan", 24704, 6, 339},
        { "�ャ���汐鐚ワ七鐚鰹�鐚�, "2002.ontvjapan.com", 24704, 6, 349},
//ND10
        { "�鴻�����鰹�", "800ch.epgdata.ontvjapan", 24736, 6, 800},
        { "�鴻�����鰹�", "801ch.epgdata.ontvjapan", 24736, 6, 801},
        { "�鴻�����鰹�", "802ch.epgdata.ontvjapan", 24736, 6, 802},
        { "�鴻�����鰹�", "805ch.epgdata.ontvjapan", 24736, 6, 805},
//ND12
        { "鐚э次鐚�鴫鐚�, "1026.ontvjapan.com", 28864, 7, 254},
        { "������鐚�示鐚�, "1022.ontvjapan.com", 28864, 7, 325},
        { "����冴�����激���, "1045.ontvjapan.com", 28864, 7, 330},//HD��2012/07/26�泣����330��Щ��/ND14
//ND14
        { "��撮������鐔�示鐚�, "1133.ontvjapan.com", 28896, 7, 292},
        { "�������弱��器示鐚�, "1015.ontvjapan.com", 28896, 7, 293},
        { "�鴻����鐚�����鐚�爾", "1006.ontvjapan.com", 28896, 7, 310},
//ND16
        { "鐚鰹自鐚鴻�鐚鰹軸鐚￥磁鐚�, "1207.ontvjapan.com", 28928, 7, 290},
        { "����潟����羃�, "305ch.epgdata.ontvjapan", 28928, 7, 305},
        { "鐚￥叱鐚�, "1014.ontvjapan.com", 28928, 7, 311},
        { "鐚￥軸��叱", "1201.ontvjapan.com", 28928, 7, 333},
        { "����с�����ｃ����", "1204.ontvjapan.com", 28928, 7, 343},
        { "鐚�滋鐚ｃ��若���, "1070.ontvjapan.com", 28928, 7, 353},
//ND18
        { "����������刻示鐚�, "1007.ontvjapan.com", 28960, 7, 240},
        { "�眼�������鐚�爾", "1027.ontvjapan.com", 28960, 7, 262},
        { "絅恰�鐔��鐚���鐚��", "1074.ontvjapan.com", 28960, 7, 314},
//ND20
        { "��������鹿鐚�璽", "1073.ontvjapan.com", 28992, 7, 307},//��������39��   { "��������軸鐚件鹿", "1072.ontvjapan.com", 28992, 7, 308},//��������21��   { "��������汐鐚ワ叱鐚�, "309ch.epgdata.ontvjapan", 28992, 6, 309},//306��09����ｃ����紊��
//ND22
        { "鐚縁雫鐚�, "1120.ontvjapan.com", 29024, 7, 161},
        { "鐚器滋鐚潟��ｃ����鐚�, "297ch.epgdata.ontvjapan", 29024, 7, 297},
        { "鐚�鹿鐚�, "1016.ontvjapan.com", 29024, 7, 312},
        { "�鴻��若��激����鐚器雫", "1018.ontvjapan.com", 29024, 7, 322},
        { "�������若�", "1046.ontvjapan.com", 29024, 7, 331},
        { "鐚器滋鐚潟��ャ��鴻��若�", "1076.ontvjapan.com", 29024, 7, 351},
//ND24
        { "鐚�鹿鐚吾��若���, "229ch.epgdata.ontvjapan", 29056, 7, 229},
        { "�ャ���磁鐚��鐚�爾", "1068.ontvjapan.com", 29056, 7, 257},//HD��   { "�ャ�������", "300ch.epgdata.ontvjapan", 29056, 7, 300},
        { "�鴻��激������, "321ch.epgdata.ontvjapan", 29056, 7, 321},
        { "����ｃ����", "1052.ontvjapan.com", 29056, 7, 362},
};

static int csStaCount = sizeof(csSta) / sizeof (STATION);
SVT_CONTROL     *svttop = NULL;
#define         SECCOUNT        4
char    title[1024];
char    subtitle[1024];
char    Category[1024];
char    ServiceName[1024];

/* prototype */
extern int strrep(char *buf, char *mae, char *ato);

void    xmlspecialchars(char *str)
{
        strrep(str, "&", "&amp;");
        strrep(str, "'", "&apos;");
        strrep(str, "\"", "&quot;");
        strrep(str, "<", "&lt;");
        strrep(str, ">", "&gt;");
}



void    GetSDT(FILE *infile, SVT_CONTROL *svttop, SECcache *secs, int count)
{
        SECcache  *bsecs;

        while((bsecs = readTS(infile, secs, count)) != NULL) {
                /* SDT */
                if((bsecs->pid & 0xFF) == 0x11) {
                        dumpSDT(bsecs->buf, svttop);
                }
        }
}
void    GetEIT(FILE *infile, FILE *outfile, STATION *psta, SECcache *secs, int count)
{
        SECcache  *bsecs;
        EIT_CONTROL     *eitcur ;
        EIT_CONTROL     *eitnext ;
        EIT_CONTROL     *eittop = NULL;
        time_t  l_time ;
        time_t  end_time ;
        struct  tm      tl ;
        struct  tm      *endtl ;
        char    cendtime[32];
        char    cstarttime[32];

        eittop = calloc(1, sizeof(EIT_CONTROL));
        eitcur = eittop ;
        fseek(infile, 0, SEEK_SET);
        while((bsecs = readTS(infile, secs, SECCOUNT)) != NULL) {
                /* EIT */
                if((bsecs->pid & 0xFF) == 0x12) {
                        dumpEIT(bsecs->buf, psta->svId, psta->onId, psta->tsId, eittop);
                }else if((bsecs->pid & 0xFF) == 0x26) {
                        dumpEIT(bsecs->buf, psta->svId, psta->onId, psta->tsId, eittop);
                }else if((bsecs->pid & 0xFF) == 0x27) {
                        dumpEIT(bsecs->buf, psta->svId, psta->onId, psta->tsId, eittop);
                }
        }
        eitcur = eittop ;
        while(eitcur != NULL){
                if(!eitcur->servid){
                        eitcur = eitcur->next ;
                        continue ;
                }
                if(eitcur->content_type > CAT_COUNT){
                        eitcur->content_type = CAT_COUNT -1 ;
                }
                memset(title, '\0', sizeof(title));
                strcpy(title, eitcur->title);
                xmlspecialchars(title);

                memset(subtitle, '\0', sizeof(subtitle));
                strcpy(subtitle, eitcur->subtitle);
                xmlspecialchars(subtitle);

                memset(Category, '\0', sizeof(Category));
                strcpy(Category, ContentCatList[eitcur->content_type].japanese);
                xmlspecialchars(Category);

                tl.tm_sec = eitcur->ss ;
                tl.tm_min = eitcur->hm ;
                tl.tm_hour = eitcur->hh ;
                tl.tm_mday = eitcur->dd ;
                tl.tm_mon = (eitcur->mm - 1);
                tl.tm_year = (eitcur->yy - 1900);
                tl.tm_wday = 0;
                tl.tm_isdst = 0;
                tl.tm_yday = 0;
                l_time = mktime(&tl);
                if((eitcur->ehh == 0) && (eitcur->emm == 0) && (eitcur->ess == 0)){
                        (void)time(&l_time);
                        end_time = l_time + (60 * 5);           // 鐚��緇��荐��
                endtl = localtime(&end_time);
                }else{
                        end_time = l_time + eitcur->ehh * 3600 + eitcur->emm * 60 + eitcur->ess;
                        endtl = localtime(&end_time);
                }
                memset(cendtime, '\0', sizeof(cendtime));
                memset(cstarttime, '\0', sizeof(cstarttime));
                strftime(cendtime, (sizeof(cendtime) - 1), "%Y%m%d%H%M%S", endtl);
                strftime(cstarttime, (sizeof(cstarttime) - 1), "%Y%m%d%H%M%S", &tl);
#if 1
                fprintf(outfile, "  <programme start=\"%s +0900\" stop=\"%s +0900\" channel=\"%s\">\n", 
                                cstarttime, cendtime, psta->ontv);
                fprintf(outfile, "    <title lang=\"ja_JP\">%s</title>\n", title);
                fprintf(outfile, "    <desc lang=\"ja_JP\">%s</desc>\n", subtitle);
                fprintf(outfile, "    <category lang=\"ja_JP\">%s</category>\n", Category);
                fprintf(outfile, "    <category lang=\"en\">%s</category>\n", ContentCatList[eitcur->content_type].english);
                fprintf(outfile, "  </programme>\n");
#else
                fprintf(outfile, "(%x:%x:%x)%s,%s,%s,%s,%s,%s\n",
                                        eitcur->servid, eitcur->table_id, eitcur->event_id,
                                        cstarttime, cendtime,
                                        title, subtitle,
                                        Category,
                                        ContentCatList[eitcur->content_type].english);
#endif
#if 0
                fprintf(outfile, "(%x:%x)%04d/%02d/%02d,%02d:%02d:%02d,%02d:%02d:%02d,%s,%s,%s,%s\n",
                                        eitcur->table_id, eitcur->event_id,
                                        eitcur->yy, eitcur->mm, eitcur->dd,
                                        eitcur->hh, eitcur->hm, eitcur->ss,
                                        eitcur->ehh, eitcur->emm, eitcur->ess,
                                        eitcur->title, eitcur->subtitle,
                                        ContentCatList[eitcur->content_type].japanese,
                                        ContentCatList[eitcur->content_type].english);
#endif
                eitnext = eitcur->next ;
                free(eitcur->title);
                free(eitcur->subtitle);
                free(eitcur);
                eitcur = eitnext ;
        }
        free(eittop);
        eittop = NULL;
}
int main(int argc, char *argv[])
{

        FILE *infile = stdin;
        FILE *outfile = stdout;
        char    *arg_onTV ;
        int             staCount ;
        char *file;
        int   inclose = 0;
        int   outclose = 0;
        SVT_CONTROL     *svtcur ;
        SVT_CONTROL     *svtsave ;
        SECcache   secs[SECCOUNT];
        int             lp ;
        STATION *pStas ;
        int             act ;

        /* ��������id���絎�*/
        memset(secs, 0,  sizeof(SECcache) * SECCOUNT);
        secs[0].pid = 0x11;
        secs[1].pid = 0x12;
        secs[2].pid = 0x26;
        secs[3].pid = 0x27;

        if(argc == 4){
                arg_onTV = argv[1];
                file = argv[2];
                if(strcmp(file, "-")) {
                        infile = fopen(file, "r");
                        inclose = 1;
                }
                if(strcmp(argv[3], "-")) {
                        outfile = fopen(argv[3], "w+");
                        outclose = 1;
                }
        }else{
                fprintf(stdout, "Usage : %s {/BS|/CS} <tsFile> <outfile>\n", argv[0]);
                fprintf(stdout, "Usage : %s <ontvcode> <tsFile> <outfile>\n", argv[0]);
                fprintf(stdout, "\n");
                fprintf(stdout, "  ontvcode   Channel identifier (ex. ****.ontvjapan.com)\n");
                fprintf(stdout, "  /BS        BS mode\n");
                fprintf(stdout, "               This mode reads the data of all BS TV stations\n");
                fprintf(stdout, "               from one TS data.\n");
                fprintf(stdout, "  /CS        CS mode\n");
                fprintf(stdout, "               This mode reads the data of two or more CS TV stations\n");
                fprintf(stdout, "               from one TS data.\n");
                return 0;
        }

        if(strcmp(arg_onTV, "/BS") == 0){
                pStas = bsSta;
                staCount = bsStaCount;
                act = 0 ;
        }else if(strcmp(arg_onTV, "/CS") == 0){
                pStas = csSta;
                staCount = csStaCount;
                act = 0 ;
        }else{
                act = 1 ;
                svttop = calloc(1, sizeof(SVT_CONTROL));
                GetSDT(infile, svttop, secs, SECCOUNT);
                svtcur = svttop->next ; //���
                if(svtcur == NULL){
                        free(svttop);
                        return 1;
                }

                pStas = calloc(1, sizeof(STATION));
                pStas->tsId = svtcur->transport_stream_id ;
                pStas->onId = svtcur->original_network_id ;
                pStas->svId = svtcur->event_id ;
                pStas->ontv = arg_onTV ;
                pStas->name = svtcur->servicename ;
                staCount = 1;
        }

        fprintf(outfile, "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
        fprintf(outfile, "<!DOCTYPE tv SYSTEM \"xmltv.dtd\">\n\n");
        fprintf(outfile, "<tv generator-info-name=\"tsEPG2xml\" generator-info-url=\"http://localhost/\">\n");

        for(lp = 0 ; lp < staCount ; lp++){
                memset(ServiceName, '\0', sizeof(ServiceName));
                strcpy(ServiceName, pStas[lp].name);
                xmlspecialchars(ServiceName);

                fprintf(outfile, "  <channel id=\"%s\">\n", pStas[lp].ontv);
                fprintf(outfile, "    <display-name lang=\"ja_JP\">%s</display-name>\n", ServiceName);
                fprintf(outfile, "  </channel>\n");
        }
        for(lp = 0 ; lp < staCount ; lp++){
                GetEIT(infile, outfile, &pStas[lp], secs, SECCOUNT);
        }
        fprintf(outfile, "</tv>\n");
        if(inclose) {
                fclose(infile);
        }

        if(outclose) {
                fclose(outfile);
        }
        if(act){
                free(pStas);
                svtcur = svttop ;       //���
                while(svtcur != NULL){
                        svtsave = svtcur->next ;
                        free(svtcur);
                        svtcur = svtsave ;
                }
        }

        return 0;
}