BitStreamerJPEGTest.cpp

Go to the documentation of this file.

/*
    RawSpeed - RAW file decoder.
 
    Copyright (C) 2018 Roman Lebedev
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.
 
    This library 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
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
 
#include "bitstreams/BitStreamerJPEG.h"
#include "adt/Array1DRef.h"
#include "adt/Casts.h"
#include "bitstreams/BitStreamerTest.h"
#include "io/Buffer.h"
#include "io/ByteStream.h"
#include "io/Endianness.h"
#include <array>
#include <cstdint>
#include <gtest/gtest.h>
 
using rawspeed::BitStreamerJPEG;
using rawspeed::Buffer;
using rawspeed::ByteStream;
using rawspeed::DataBuffer;
using rawspeed::Endianness;
 
namespace rawspeed_test {
 
namespace {
 
struct InvOnesTag;
struct OnesTag;
struct SaturatedTag;
 
template <>
const std::array<uint8_t, 8> Pattern<BitStreamerJPEG, OnesTag>::Data = {
    {/* [Byte0 Byte1 Byte2 Byte3] */
     /* Byte: [Bit0 .. Bit7] */
     0b10100100, 0b01000010, 0b00001000, 0b00011111}};
template <> uint32_t Pattern<BitStreamerJPEG, OnesTag>::data(int index) {
  const auto set = GenOnesBE(1, 0);
  return set[index];
}
 
template <>
const std::array<uint8_t, 8> Pattern<BitStreamerJPEG, InvOnesTag>::Data = {
    {0b11010010, 0b00100001, 0b00000100, 0b00001111}};
template <> uint32_t Pattern<BitStreamerJPEG, InvOnesTag>::data(int index) {
  const auto set = GenOnesBE(0, -1);
  return set[index];
}
 
// If 0xFF0x00 byte sequence is found, it is just 0xFF, i.e. 0x00 is ignored.
// So if we want 0xFF, we need to append 0x00 byte
template <>
const std::array<uint8_t, 8> Pattern<BitStreamerJPEG, SaturatedTag>::Data{
    {uint8_t(~0U), 0, uint8_t(~0U), 0, uint8_t(~0U), 0, uint8_t(~0U), 0}};
 
INSTANTIATE_TYPED_TEST_SUITE_P(JPEG, BitStreamerTest,
                               Patterns<BitStreamerJPEG>);
 
TEST(BitStreamerJPEGTest, 0xFF0x00Is0xFFTest) {
  // If 0xFF0x00 byte sequence is found, it is just 0xFF, i.e. 0x00 is ignored.
  static const std::array<uint8_t, 2 + 8> data{
      {0xFF, 0x00, 0b10100100, 0b01000010, 0b00001000, 0b00011111}};
 
  const rawspeed::Array1DRef<const uint8_t> input(
      data.data(), rawspeed::implicit_cast<int>(data.size()));
 
  BitStreamerJPEG p(input);
 
  ASSERT_EQ(p.getBits(8), 0xFF);
 
  for (int len = 1; len <= 7; len++)
    ASSERT_EQ(p.getBits(len), 1) << "     Where len: " << len;
}
 
TEST(BitStreamerJPEGTest, 0xFF0xXXIsTheEndTest) {
  // If 0xFF0xXX byte sequence is found, where XX != 0, then it is the end.
  for (uint8_t end = 0x01; end < 0xFF; end++) {
    static const std::array<uint8_t, 2 + 8> data{
        {0xFF, end, 0xFF, 0xFF, 0xFF, 0xFF}};
 
    const rawspeed::Array1DRef<const uint8_t> input(
        data.data(), rawspeed::implicit_cast<int>(data.size()));
 
    BitStreamerJPEG p(input);
 
    for (int cnt = 0; cnt <= 64 + 32 - 1; cnt++)
      ASSERT_EQ(p.getBits(1), 0);
  }
}
 
} // namespace
 
} // namespace rawspeed_test