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},
        { "�鴻��若�����潟���, "3012.ontvjapan.com", 16529, 4, 200},
        { "BS11", "3013.ontvjapan.com", 16528, 4, 211},
        { "TwellV", "3014.ontvjapan.com", 16530, 4, 222},
};

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



static STATION csSta[] = {
        { "�鴻��種�鐔�����", "1002.ontvjapan.com", 24608, 6, 237},
        { "�ユ����絨��鐔��鐚�爾", "1086.ontvjapan.com", 24608, 6, 239},//��������SHD ��      { "��������汐鐚ワ叱鐚�, "306ch.epgdata.ontvjapan", 24608, 6, 306},
        { "�激��������潟���, "1059.ontvjapan.com", 24704, 6, 55},
        { "�吟��激���, "1217.ontvjapan.com", 24736, 6, 228},
        { "�鴻�����鰹示鐚わ�鐚��", "800ch.epgdata.ontvjapan", 24736, 6, 800},
        { "�鴻�����鰹�鐚��", "801ch.epgdata.ontvjapan", 24736, 6, 801},
        { "�鴻�����鰹�鐚��", "802ch.epgdata.ontvjapan", 24736, 6, 802},
        { "鐔�������, "100ch.epgdata.ontvjapan", 28736, 7, 100},
        { "�ゃ��帥�������鐚器雫", "194ch.epgdata.ontvjapan", 28736, 7, 194},
        { "鐚��������鐚ワ竺鐚逸汐", "1025.ontvjapan.com", 28736, 7, 256},
        { "鐚�鹿鐚�, "1016.ontvjapan.com", 28736, 7, 312},
        { "�鴻��若��激����鐚器雫", "1018.ontvjapan.com", 28736, 7, 322},
        { "�������若�������", "1046.ontvjapan.com", 28736, 7, 331},
        { "����若��祉��ｃ����", "1213.ontvjapan.com", 28736, 7, 334},
        { "�掩�����潟���, "1010.ontvjapan.com", 28768, 7, 221},
        { "茵�����", "1005.ontvjapan.com", 28768, 7, 222},
        { "����潟���汐鐚ワ治鐚�, "1008.ontvjapan.com", 28768, 7, 223},
        { "羇���������ｃ�", "1009.ontvjapan.com", 28768, 7, 224},
        { "�鴻��若�����激���, "1003.ontvjapan.com", 28768, 7, 238},
        { "��撮�������ｃ����", "1133.ontvjapan.com", 28768, 7, 292},
        { "�鴻���������, "1006.ontvjapan.com", 28768, 7, 310},
        { "鐚￥叱鐚�, "1014.ontvjapan.com", 28768, 7, 311},
        { "����с�����ｃ����", "1204.ontvjapan.com", 28768, 7, 343},
        { "����������帥�", "110ch.epgdata.ontvjapan", 28864, 7, 110},
        { "�眼�����ｃ����", "1028.ontvjapan.com", 28864, 7, 260},
        { "�������ｃ����", "1092.ontvjapan.com", 28864, 7, 303},
        { "鐚�軸鐚�, "1019.ontvjapan.com", 28864, 7, 323},
        { "����若�����祉���, "1024.ontvjapan.com", 28864, 7, 324},
        { "�������若��帥�", "1067.ontvjapan.com", 28864, 7, 352},
        { "鐚�滋鐚ｃ��若���, "1070.ontvjapan.com", 28864, 7, 353},
        { "鐚ｏ汐鐚��", "1069.ontvjapan.com", 28864, 7, 354},
        { "�吾��鴻��祉���, "361ch.epgdata.ontvjapan", 28864, 7, 361},
        { "鐚��������鐚�, "1041.ontvjapan.com", 28896, 7, 251},
        { "鐚��������鐚�, "1042.ontvjapan.com", 28896, 7, 252},
        { "鐚�������式鐔��鐔�示", "1043.ontvjapan.com", 28896, 7, 253},
        { "鐚э次鐚�鴫鐚�, "1026.ontvjapan.com", 28896, 7, 254},
        { "鐔��鐔��鐚＜�������", "1040.ontvjapan.com", 28896, 7, 255},
        { "絎������≪��ｃ����", "101ch.epgdata.ontvjapan", 28928, 7, 101},
        { "鐚鰹自鐚鴻�鐚鰹軸鐚￥磁鐚�, "1207.ontvjapan.com", 28928, 7, 290},
        { "����潟����羃�, "305ch.epgdata.ontvjapan", 28928, 7, 305},
        { "鐚￥軸-鐚�, "1201.ontvjapan.com", 28928, 7, 333},
        { "�������若��ｃ����", "1050.ontvjapan.com", 28928, 7, 342},
        { "�鴻�����鰹�鐚��", "803ch.epgdata.ontvjapan", 28928, 7, 803},
        { "�鴻�����鰹�鐚��", "804ch.epgdata.ontvjapan", 28928, 7, 804},
        { "����������刻示鐚�, "1007.ontvjapan.com", 28960, 7, 240},
        { "�眼������������, "1027.ontvjapan.com", 28960, 7, 262},
        { "鐚��鐚����示鐚�, "1074.ontvjapan.com", 28960, 7, 314},
        { "��������鹿鐚�璽", "1073.ontvjapan.com", 28992, 7, 307},//��������39��   { "��������軸鐚件鹿", "1072.ontvjapan.com", 28992, 7, 308},//��������21��   { "�≪�������", "1047.ontvjapan.com", 28992, 7, 332},
        { "����鴻������, "1062.ontvjapan.com", 28992, 7, 340},
        { "�≪������������, "1193.ontvjapan.com", 28992, 7, 341},
        { "鐚�鐚器滋鐚潟�������", "160ch.epgdata.ontvjapan", 29024, 7, 160},
        { "鐚縁雫鐚�, "1120.ontvjapan.com", 29024, 7, 161},
        { "����ゃ�鐚��鐚��鐚器雫", "185ch.epgdata.ontvjapan", 29024, 7, 185},
        { "�������弱���, "1015.ontvjapan.com", 29024, 7, 293},
        { "鐚器滋鐚潟��ｃ����", "3201.ontvjapan.com", 29024, 7, 301},
        { "����冴��若��ｃ����", "1090.ontvjapan.com", 29024, 7, 304},
        { "MUSIC ON! TV", "1022.ontvjapan.com", 29024, 7, 325},
        { "����冴�����激���D", "1045.ontvjapan.com", 29024, 7, 335},//HD��        { "鐚器滋鐚潟��ャ��鴻��若�", "1076.ontvjapan.com", 29024, 7, 351},
        { "鐚ｏ竺�ユ���������, "147ch.epgdata.ontvjapan", 29056, 7, 147},
        { "�ャ���磁鐚�, "1068.ontvjapan.com", 29056, 7, 257},
        { "fashion TV", "5004.ontvjapan.com", 29056, 7, 291},
        { "�ャ�������", "300ch.epgdata.ontvjapan", 29056, 7, 300},
        { "�������若����鐚器雫", "1023.ontvjapan.com", 29056, 7, 320},
        { "Music Japan TV", "1208.ontvjapan.com", 29056, 7, 321},
        { "�ャ���汐鐚ワ七鐚鰹�鐚�, "2002.ontvjapan.com", 29056, 7, 350},
};

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;
}