subparser.cpp

 #include "subparser.h"

SubParser::SubParser(QWidget *parent)
    : QMainWindow(parent)
{
}

SubParser::~SubParser()
{
}

void SubParser::ParseSRT(QString filePath)
{
    QString start,end,comment;
    QString timeLine;
    int index =0, turn = 0,r=0;
    double startPos =0,endPos=0 ;
    
    /*
     * turn = 0 -> Add subtitle number
     * turn = 1 -> Add string to timeLine
     * turn > 1 -> Add string to completeLine
     */
    if (filePath.isEmpty())
        return;
    QFile srtFile(filePath);
    if (!srtFile.exists() || !srtFile.open(QIODevice::ReadOnly)){
        qDebug() << " Failed attempt on opening " << filePath;
        return;
    }
    QTextStream stream(&srtFile);
    QString line;
    qDebug() << " correct file  " << filePath;
    //SRT PARSING STARTS HERE
    while (stream.readLineInto(&line)) {
        line = line.simplified();
        qDebug() << " begin  " << line;
        if (line.compare("")){
            if (!turn){
                index=atoi(line.toStdString().c_str());
                turn++;
                continue;
            }
            if (line.contains("-->")){
                timeLine += line;
                QStringList srtTime;
                srtTime = timeLine.split(' ');
                start = srtTime[0];
                startPos= strToTime(start);
                end = srtTime[2];
                endPos= strToTime(end);
            }
            else{
                r++;
                if (comment != "")
                    comment += " ";
                if (r == 1)
                    comment+= line;
                else
                    comment = comment + "\r" +line;
            }
            turn++;
        }
        else {
            turn = 0;
            qDebug() << index << startPos << endPos << comment << "\n";
            //addSubtitle(index,startPos,endPos,comment));
            comment = timeLine = "";
            r = 0;
        }
    }
}

void SubParser::ParseSSA(QString filePath)
{
    QString start,end,comment,startTime,endTime,text;
    QString EventFormat, timeLine, container;
    int turn = 0, maxSplit =0;
    double startPos=0, endPos=0;
    if (filePath.isEmpty())
        return;
    if (filePath.contains(".ass"))
        qDebug()<< "ass File";
    QFile assFile(filePath);

    if (!assFile.exists() || !assFile.open(QIODevice::ReadOnly)){
        qDebug() << " Failed attempt on opening " << filePath;
        return;
    }
    QTextStream stream(&assFile);
    QString line;
    qDebug() << " correct ass file  " << filePath;
    //SSA PARSING STARTS HERE
    while (stream.readLineInto(&line)) {
        line = line.simplified();
        if (line.compare("")){
            if (!turn){
                qDebug() << " turn = 0  " << line;
                //check if it is script info, event,or v4+ style
                if(line.replace(" ","").contains("ScriptInfo")){
                    qDebug()<< "Script Info";
                    container = "Script Info";
                    turn++;
                    qDebug()<< "turn" << turn;
                    continue;
                }
                else if(line.contains("Styles")){
                    qDebug()<< "V4 Styles";
                    container = "V4 Styles";
                    turn++;
                    qDebug()<< "turn" << turn;
                    continue;
                }
                else{
                    turn++;
                    container = "Events";
                    qDebug()<< "turn" << turn;
                    continue;
                }
            }
            //qDebug() << "\n turn != 0  " << turn<< line;
            if(container.contains("Events")){
                //if it is event
                QStringList format;
                if(line.contains("Format:")){
                    EventFormat += line;
                    format = (EventFormat.split(": ")[1].replace(" ","")).split(',');
                    qDebug() << format << format.count();
                    maxSplit = format.count();
                    //TIME
                    start = format[1];
                    end = format[2];
                    // Text
                    text = format[9];
                    qDebug()<< start << end << text;
                }
                else{
                    QString EventDialogue;
                    QStringList dialogue;
                    EventDialogue += line;
                    dialogue = EventDialogue.split(": ")[1].split(',');
                    QString remainingStr = EventDialogue.split(": ")[1].section(',', maxSplit);
                    //qDebug()<< dialogue;
                    //TIME
                    startTime = dialogue[1];
                    startPos = strToTime(startTime);
                    endTime = dialogue[2];
                    endPos= strToTime(endTime);
                    // Text
                    text = dialogue[9]+ remainingStr;
                    qDebug()<<"Start: "<< startPos << "End: "<<endPos <<text;
                }
            }
            turn++;
        }
        else {
            turn = 0;
            text = startTime = endTime = "";
        }
    }
}

double SubParser::strToTime(QString str)
{
    QStringList total,secs;
    int hours=0, mins=0, seconds=0, ms=0;
    double total_sec=0;
    total = str.split(':');
    hours = atoi(total[0].toStdString().c_str());
    mins = atoi(total[1].toStdString().c_str());
    if(total[2].contains('.'))
        secs = total[2].split('.'); //ssa file
    else
        secs = total[2].split(','); //srt file
    seconds = atoi(secs[0].toStdString().c_str());
    ms = atoi(secs[1].toStdString().c_str());
    total_sec= hours *3600 + mins *60 + seconds + ms * 0.001 ;
    return total_sec;
}