Mercurial > hg > forks > libbpg
diff x265/source/output/raw.cpp @ 0:772086c29cc7
Initial import.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Wed, 16 Nov 2016 11:16:33 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/x265/source/output/raw.cpp Wed Nov 16 11:16:33 2016 +0200 @@ -0,0 +1,80 @@ +/***************************************************************************** + * Copyright (C) 2013-2015 x265 project + * + * Authors: Steve Borho <steve@borho.org> + * Xinyue Lu <i@7086.in> + * + * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA. + * + * This program is also available under a commercial proprietary license. + * For more information, contact us at license @ x265.com. + *****************************************************************************/ + +#include "raw.h" + +using namespace X265_NS; +using namespace std; + +RAWOutput::RAWOutput(const char* fname, InputFileInfo&) +{ + b_fail = false; + if (!strcmp(fname, "-")) + { + ofs = &cout; + return; + } + ofs = new ofstream(fname, ios::binary | ios::out); + if (ofs->fail()) + b_fail = true; +} + +void RAWOutput::setParam(x265_param* param) +{ + param->bAnnexB = true; +} + +int RAWOutput::writeHeaders(const x265_nal* nal, uint32_t nalcount) +{ + uint32_t bytes = 0; + + for (uint32_t i = 0; i < nalcount; i++) + { + ofs->write((const char*)nal->payload, nal->sizeBytes); + bytes += nal->sizeBytes; + nal++; + } + + return bytes; +} + +int RAWOutput::writeFrame(const x265_nal* nal, uint32_t nalcount, x265_picture&) +{ + uint32_t bytes = 0; + + for (uint32_t i = 0; i < nalcount; i++) + { + ofs->write((const char*)nal->payload, nal->sizeBytes); + bytes += nal->sizeBytes; + nal++; + } + + return bytes; +} + +void RAWOutput::closeFile(int64_t, int64_t) +{ + if (ofs != &cout) + delete ofs; +}