spu.cc

来自「Linux下比较早的基于命令行的DVD播放器」· CC 代码 · 共 153 行

//
// Copyright (c) 2003 by Istv醤 V醨adi
//
// This file is part of dxr3Player, a DVD player written specifically 
// for the DXR3 (aka Hollywood+) decoder card.

// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

//------------------------------------------------------------------------------

#include "SPU.h"
#include "SPUBuffer.h"
#include "ControlSequence.h"

#include "defs.h"
#include <vector>

//------------------------------------------------------------------------------

using dvd::spu::SPU;

using std::vector;
using std::pair;
using std::make_pair;

//------------------------------------------------------------------------------

SPU::SPU(const SPUBuffer& spuBuffer)
{
    SPUBufferBitReader reader(spuBuffer);
    decode(reader);
}

//------------------------------------------------------------------------------

SPU::~SPU()
{
    for(controlSequences_t::iterator i = controlSequences.begin();
        i!=controlSequences.end(); ++i)
    {
        delete *i;
    }
}

//------------------------------------------------------------------------------

void SPU::appendControlSequencesTo(controlSequences_t& to)
{
    to.splice(to.end(), controlSequences);
}

//------------------------------------------------------------------------------

void SPU::decode(SPUBufferBitReader& reader)
{
    reader.read16(); // length

    size_t nextOffset = reader.read16();
    size_t offset;
    do {
        offset = nextOffset;
        reader.setOffset(offset);

        ControlSequence* ctrlSeq = new ControlSequence();

        nextOffset = ctrlSeq->decode(reader);

        controlSequences.push_back(ctrlSeq);
    } while(nextOffset!=offset);
}

//------------------------------------------------------------------------------

void SPU::encode(DynamicBufferBitWriter& writer) const
{
    size_t startOffset = writer.getOffset();

    writer.write16(0);  // length
    writer.write16(0);  // offset of control sequence table
    
    vector< pair<size_t, size_t>, DefaultAllocatorTemplate< pair<size_t, size_t> > >  fieldOffsets;

    for(controlSequences_t::const_iterator i = controlSequences.begin();
        i!=controlSequences.end(); ++i) 
    {
        size_t topFieldLength = 0;
        size_t pictureLength = 
            (*i)->encodePixelData(writer, topFieldLength);

        fieldOffsets.push_back(make_pair(writer.getOffset()-pictureLength,
                                         writer.getOffset()-pictureLength+topFieldLength));
    }
    
    size_t csOffset = writer.getOffset();

    writer.setOffset(startOffset+2);
    writer.write16(csOffset-startOffset);
    writer.setOffset(csOffset);

    size_t controlSequenceNumber = 0;
    for(controlSequences_t::const_iterator i = controlSequences.begin();
        i!=controlSequences.end(); ++i, ++controlSequenceNumber) 
    {
        (*i)->encodeCommands(writer,
                             fieldOffsets[controlSequenceNumber].first,
                             fieldOffsets[controlSequenceNumber].second,
                             (*i)==controlSequences.back());
    }

    size_t endOffset = writer.getOffset();

    writer.setOffset(startOffset);
    writer.write16(endOffset-startOffset);
    writer.setOffset(endOffset);
}

//------------------------------------------------------------------------------

void SPU::setPTS(pts_t basePTS)
{
    for(controlSequences_t::iterator i = controlSequences.begin();
        i!=controlSequences.end(); ++i)
    {
        basePTS = (*i)->setPTS(basePTS);
    }
}

//------------------------------------------------------------------------------

void SPU::reset()
{
    for(controlSequences_t::iterator i = controlSequences.begin();
        i!=controlSequences.end(); ++i)
    {
        delete *i;
    }
    controlSequences.clear();
}

//------------------------------------------------------------------------------