blob: 00e32d912476c5b484c10371bc41fbaa5215b14c [file] [log] [blame]
/*
* Copyright (C) 2012 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#define LOG_TAG "CpuConsumer_test"
//#define LOG_NDEBUG 0
//#define LOG_NNDEBUG 0
#ifdef LOG_NNDEBUG
#define ALOGVV(...) ALOGV(__VA_ARGS__)
#else
#define ALOGVV(...) ((void)0)
#endif
#include <gtest/gtest.h>
#include <gui/CpuConsumer.h>
#include <gui/Surface.h>
#include <ui/GraphicBuffer.h>
#include <utils/String8.h>
#include <utils/Thread.h>
#include <utils/Mutex.h>
#include <utils/Condition.h>
#include <thread>
#include <vector>
#define CPU_CONSUMER_TEST_FORMAT_RAW 0
#define CPU_CONSUMER_TEST_FORMAT_Y8 0
#define CPU_CONSUMER_TEST_FORMAT_Y16 0
#define CPU_CONSUMER_TEST_FORMAT_RGBA_8888 1
namespace android {
struct CpuConsumerTestParams {
uint32_t width;
uint32_t height;
int maxLockedBuffers;
PixelFormat format;
};
::std::ostream& operator<<(::std::ostream& os, const CpuConsumerTestParams& p) {
return os << "[ (" << p.width << ", " << p.height << "), B:"
<< p.maxLockedBuffers << ", F:0x"
<< ::std::hex << p.format << "]";
}
class CpuConsumerTest : public ::testing::TestWithParam<CpuConsumerTestParams> {
protected:
virtual void SetUp() {
const ::testing::TestInfo* const test_info =
::testing::UnitTest::GetInstance()->current_test_info();
CpuConsumerTestParams params = GetParam();
ALOGV("** Starting test %s (%d x %d, %d, 0x%x)",
test_info->name(),
params.width, params.height,
params.maxLockedBuffers, params.format);
sp<IGraphicBufferProducer> producer;
sp<IGraphicBufferConsumer> consumer;
BufferQueue::createBufferQueue(&producer, &consumer);
mCC = new CpuConsumer(consumer, params.maxLockedBuffers);
String8 name("CpuConsumer_Under_Test");
mCC->setName(name);
mSTC = new Surface(producer);
mANW = mSTC;
}
virtual void TearDown() {
mANW.clear();
mSTC.clear();
mCC.clear();
}
class FrameWaiter : public CpuConsumer::FrameAvailableListener {
public:
FrameWaiter():
mPendingFrames(0) {
}
void waitForFrame() {
Mutex::Autolock lock(mMutex);
while (mPendingFrames == 0) {
mCondition.wait(mMutex);
}
mPendingFrames--;
}
virtual void onFrameAvailable(const BufferItem&) {
Mutex::Autolock lock(mMutex);
mPendingFrames++;
mCondition.signal();
}
int mPendingFrames;
Mutex mMutex;
Condition mCondition;
};
// Note that SurfaceTexture will lose the notifications
// onBuffersReleased and onFrameAvailable as there is currently
// no way to forward the events. This DisconnectWaiter will not let the
// disconnect finish until finishDisconnect() is called. It will
// also block until a disconnect is called
class DisconnectWaiter : public BufferQueue::ConsumerListener {
public:
DisconnectWaiter () :
mWaitForDisconnect(false),
mPendingFrames(0) {
}
void waitForFrame() {
Mutex::Autolock lock(mMutex);
while (mPendingFrames == 0) {
mFrameCondition.wait(mMutex);
}
mPendingFrames--;
}
virtual void onFrameAvailable(const BufferItem&) {
Mutex::Autolock lock(mMutex);
mPendingFrames++;
mFrameCondition.signal();
}
virtual void onBuffersReleased() {
Mutex::Autolock lock(mMutex);
while (!mWaitForDisconnect) {
mDisconnectCondition.wait(mMutex);
}
}
void finishDisconnect() {
Mutex::Autolock lock(mMutex);
mWaitForDisconnect = true;
mDisconnectCondition.signal();
}
private:
Mutex mMutex;
bool mWaitForDisconnect;
Condition mDisconnectCondition;
int mPendingFrames;
Condition mFrameCondition;
};
sp<CpuConsumer> mCC;
sp<Surface> mSTC;
sp<ANativeWindow> mANW;
};
#define ASSERT_NO_ERROR(err, msg) \
ASSERT_EQ(NO_ERROR, err) << (msg) << strerror(-(err))
void checkPixel(const CpuConsumer::LockedBuffer &buf,
uint32_t x, uint32_t y, uint32_t r, uint32_t g=0, uint32_t b=0) {
// Ignores components that don't exist for given pixel
switch(buf.format) {
case HAL_PIXEL_FORMAT_RAW16: {
String8 msg;
uint16_t *bPtr = (uint16_t*)buf.data;
bPtr += y * buf.stride + x;
// GRBG Bayer mosaic; only check the matching channel
switch( ((y & 1) << 1) | (x & 1) ) {
case 0: // G
case 3: // G
EXPECT_EQ(g, *bPtr);
break;
case 1: // R
EXPECT_EQ(r, *bPtr);
break;
case 2: // B
EXPECT_EQ(b, *bPtr);
break;
}
break;
}
// ignores g,b
case HAL_PIXEL_FORMAT_Y8: {
uint8_t *bPtr = (uint8_t*)buf.data;
bPtr += y * buf.stride + x;
EXPECT_EQ(r, *bPtr) << "at x = " << x << " y = " << y;
break;
}
// ignores g,b
case HAL_PIXEL_FORMAT_Y16: {
// stride is in pixels, not in bytes
uint16_t *bPtr = ((uint16_t*)buf.data) + y * buf.stride + x;
EXPECT_EQ(r, *bPtr) << "at x = " << x << " y = " << y;
break;
}
case HAL_PIXEL_FORMAT_RGBA_8888: {
const int bytesPerPixel = 4;
uint8_t *bPtr = (uint8_t*)buf.data;
bPtr += (y * buf.stride + x) * bytesPerPixel;
EXPECT_EQ(r, bPtr[0]) << "at x = " << x << " y = " << y;
EXPECT_EQ(g, bPtr[1]) << "at x = " << x << " y = " << y;
EXPECT_EQ(b, bPtr[2]) << "at x = " << x << " y = " << y;
break;
}
default: {
ADD_FAILURE() << "Unknown format for check:" << buf.format;
break;
}
}
}
// Fill a YV12 buffer with a multi-colored checkerboard pattern
void fillYV12Buffer(uint8_t* buf, int w, int h, int stride);
// Fill a Y8/Y16 buffer with a multi-colored checkerboard pattern
template <typename T> // T == uint8_t or uint16_t
void fillGreyscaleBuffer(T* buf, int w, int h, int stride, int bpp) {
const int blockWidth = w > 16 ? w / 16 : 1;
const int blockHeight = h > 16 ? h / 16 : 1;
const int yuvTexOffsetY = 0;
ASSERT_TRUE(bpp == 8 || bpp == 16);
ASSERT_TRUE(sizeof(T)*8 == bpp);
// stride is in pixels, not in bytes
int yuvTexStrideY = stride;
for (int x = 0; x < w; x++) {
for (int y = 0; y < h; y++) {
int parityX = (x / blockWidth) & 1;
int parityY = (y / blockHeight) & 1;
T intensity = (parityX ^ parityY) ? 63 : 191;
buf[yuvTexOffsetY + (y * yuvTexStrideY) + x] = intensity;
}
}
}
inline uint8_t chooseColorRgba8888(int blockX, int blockY, uint8_t channel) {
const int colorVariations = 3;
uint8_t color = ((blockX % colorVariations) + (blockY % colorVariations))
% (colorVariations) == channel ? 191: 63;
return color;
}
// Fill a RGBA8888 buffer with a multi-colored checkerboard pattern
void fillRgba8888Buffer(uint8_t* buf, int w, int h, int stride)
{
const int blockWidth = w > 16 ? w / 16 : 1;
const int blockHeight = h > 16 ? h / 16 : 1;
const int bytesPerPixel = 4;
// stride is in pixels, not in bytes
for (int x = 0; x < w; ++x) {
for (int y = 0; y < h; ++y) {
int blockX = (x / blockWidth);
int blockY = (y / blockHeight);
uint8_t r = chooseColorRgba8888(blockX, blockY, 0);
uint8_t g = chooseColorRgba8888(blockX, blockY, 1);
uint8_t b = chooseColorRgba8888(blockX, blockY, 2);
buf[(y*stride + x)*bytesPerPixel + 0] = r;
buf[(y*stride + x)*bytesPerPixel + 1] = g;
buf[(y*stride + x)*bytesPerPixel + 2] = b;
buf[(y*stride + x)*bytesPerPixel + 3] = 255;
}
}
}
// Fill a RAW sensor buffer with a multi-colored checkerboard pattern.
// Assumes GRBG mosaic ordering. Result should be a grid in a 2x2 pattern
// of [ R, B; G, W]
void fillBayerRawBuffer(uint8_t* buf, int w, int h, int stride) {
ALOGVV("fillBayerRawBuffer: %p with %d x %d, stride %d", buf, w, h ,stride);
// Blocks need to be even-width/height, aim for 8-wide otherwise
const int blockWidth = (w > 16 ? w / 8 : 2) & ~0x1;
const int blockHeight = (h > 16 ? h / 8 : 2) & ~0x1;
for (int y = 0; y < h; y+=2) {
uint16_t *bPtr1 = ((uint16_t*)buf) + stride*y;
uint16_t *bPtr2 = bPtr1 + stride;
for (int x = 0; x < w; x+=2) {
int blockX = (x / blockWidth ) & 1;
int blockY = (y / blockHeight) & 1;
unsigned short r = (blockX == blockY) ? 1000 : 200;
unsigned short g = blockY ? 1000: 200;
unsigned short b = blockX ? 1000: 200;
// GR row
*bPtr1++ = g;
*bPtr1++ = r;
// BG row
*bPtr2++ = b;
*bPtr2++ = g;
}
}
}
template<typename T> // uint8_t or uint16_t
void checkGreyscaleBuffer(const CpuConsumer::LockedBuffer &buf) {
uint32_t w = buf.width;
uint32_t h = buf.height;
const int blockWidth = w > 16 ? w / 16 : 1;
const int blockHeight = h > 16 ? h / 16 : 1;
// Top-left square is bright
checkPixel(buf, 0, 0, 191);
checkPixel(buf, 1, 0, 191);
checkPixel(buf, 0, 1, 191);
checkPixel(buf, 1, 1, 191);
// One-right square is dark
checkPixel(buf, blockWidth, 0, 63);
checkPixel(buf, blockWidth + 1, 0, 63);
checkPixel(buf, blockWidth, 1, 63);
checkPixel(buf, blockWidth + 1, 1, 63);
// One-down square is dark
checkPixel(buf, 0, blockHeight, 63);
checkPixel(buf, 1, blockHeight, 63);
checkPixel(buf, 0, blockHeight + 1, 63);
checkPixel(buf, 1, blockHeight + 1, 63);
// One-diag square is bright
checkPixel(buf, blockWidth, blockHeight, 191);
checkPixel(buf, blockWidth + 1, blockHeight, 191);
checkPixel(buf, blockWidth, blockHeight + 1, 191);
checkPixel(buf, blockWidth + 1, blockHeight + 1, 191);
// Test bottom-right pixel
const int maxBlockX = ((w-1 + (blockWidth-1)) / blockWidth) & 0x1;
const int maxBlockY = ((h-1 + (blockHeight-1)) / blockHeight) & 0x1;
uint32_t pixelValue = ((maxBlockX % 2) == (maxBlockY % 2)) ? 191 : 63;
checkPixel(buf, w-1, h-1, pixelValue);
}
void checkRgba8888Buffer(const CpuConsumer::LockedBuffer &buf) {
uint32_t w = buf.width;
uint32_t h = buf.height;
const int blockWidth = w > 16 ? w / 16 : 1;
const int blockHeight = h > 16 ? h / 16 : 1;
// Top-left square is bright red
checkPixel(buf, 0, 0, 191, 63, 63);
checkPixel(buf, 1, 0, 191, 63, 63);
checkPixel(buf, 0, 1, 191, 63, 63);
checkPixel(buf, 1, 1, 191, 63, 63);
// One-right square is bright green
checkPixel(buf, blockWidth, 0, 63, 191, 63);
checkPixel(buf, blockWidth + 1, 0, 63, 191, 63);
checkPixel(buf, blockWidth, 1, 63, 191, 63);
checkPixel(buf, blockWidth + 1, 1, 63, 191, 63);
// One-down square is bright green
checkPixel(buf, 0, blockHeight, 63, 191, 63);
checkPixel(buf, 1, blockHeight, 63, 191, 63);
checkPixel(buf, 0, blockHeight + 1, 63, 191, 63);
checkPixel(buf, 1, blockHeight + 1, 63, 191, 63);
// One-diag square is bright blue
checkPixel(buf, blockWidth, blockHeight, 63, 63, 191);
checkPixel(buf, blockWidth + 1, blockHeight, 63, 63, 191);
checkPixel(buf, blockWidth, blockHeight + 1, 63, 63, 191);
checkPixel(buf, blockWidth + 1, blockHeight + 1, 63, 63, 191);
// Test bottom-right pixel
{
const int maxBlockX = ((w-1) / blockWidth);
const int maxBlockY = ((h-1) / blockHeight);
uint8_t r = chooseColorRgba8888(maxBlockX, maxBlockY, 0);
uint8_t g = chooseColorRgba8888(maxBlockX, maxBlockY, 1);
uint8_t b = chooseColorRgba8888(maxBlockX, maxBlockY, 2);
checkPixel(buf, w-1, h-1, r, g, b);
}
}
void checkBayerRawBuffer(const CpuConsumer::LockedBuffer &buf) {
uint32_t w = buf.width;
uint32_t h = buf.height;
const int blockWidth = (w > 16 ? w / 8 : 2) & ~0x1;
const int blockHeight = (h > 16 ? h / 8 : 2) & ~0x1;
// Top-left square is red
checkPixel(buf, 0, 0, 1000, 200, 200);
checkPixel(buf, 1, 0, 1000, 200, 200);
checkPixel(buf, 0, 1, 1000, 200, 200);
checkPixel(buf, 1, 1, 1000, 200, 200);
// One-right square is blue
checkPixel(buf, blockWidth, 0, 200, 200, 1000);
checkPixel(buf, blockWidth + 1, 0, 200, 200, 1000);
checkPixel(buf, blockWidth, 1, 200, 200, 1000);
checkPixel(buf, blockWidth + 1, 1, 200, 200, 1000);
// One-down square is green
checkPixel(buf, 0, blockHeight, 200, 1000, 200);
checkPixel(buf, 1, blockHeight, 200, 1000, 200);
checkPixel(buf, 0, blockHeight + 1, 200, 1000, 200);
checkPixel(buf, 1, blockHeight + 1, 200, 1000, 200);
// One-diag square is white
checkPixel(buf, blockWidth, blockHeight, 1000, 1000, 1000);
checkPixel(buf, blockWidth + 1, blockHeight, 1000, 1000, 1000);
checkPixel(buf, blockWidth, blockHeight + 1, 1000, 1000, 1000);
checkPixel(buf, blockWidth + 1, blockHeight + 1, 1000, 1000, 1000);
// Test bottom-right pixel
const int maxBlockX = ((w-1) / blockWidth) & 0x1;
const int maxBlockY = ((w-1) / blockHeight) & 0x1;
unsigned short maxR = (maxBlockX == maxBlockY) ? 1000 : 200;
unsigned short maxG = maxBlockY ? 1000: 200;
unsigned short maxB = maxBlockX ? 1000: 200;
checkPixel(buf, w-1, h-1, maxR, maxG, maxB);
}
void checkAnyBuffer(const CpuConsumer::LockedBuffer &buf, int format) {
switch (format) {
case HAL_PIXEL_FORMAT_RAW16:
checkBayerRawBuffer(buf);
break;
case HAL_PIXEL_FORMAT_Y8:
checkGreyscaleBuffer<uint8_t>(buf);
break;
case HAL_PIXEL_FORMAT_Y16:
checkGreyscaleBuffer<uint16_t>(buf);
break;
case HAL_PIXEL_FORMAT_RGBA_8888:
checkRgba8888Buffer(buf);
break;
}
}
// Configures the ANativeWindow producer-side interface based on test parameters
void configureANW(const sp<ANativeWindow>& anw,
const CpuConsumerTestParams& params,
int maxBufferSlack) {
status_t err;
err = native_window_api_connect(anw.get(), NATIVE_WINDOW_API_CPU);
ASSERT_NO_ERROR(err, "connect error: ");
err = native_window_set_buffers_dimensions(anw.get(),
params.width, params.height);
ASSERT_NO_ERROR(err, "set_buffers_dimensions error: ");
err = native_window_set_buffers_format(anw.get(), params.format);
ASSERT_NO_ERROR(err, "set_buffers_format error: ");
err = native_window_set_usage(anw.get(),
GRALLOC_USAGE_SW_WRITE_OFTEN);
ASSERT_NO_ERROR(err, "set_usage error: ");
int minUndequeuedBuffers;
err = anw.get()->query(anw.get(),
NATIVE_WINDOW_MIN_UNDEQUEUED_BUFFERS,
&minUndequeuedBuffers);
ASSERT_NO_ERROR(err, "query error: ");
ALOGVV("Setting buffer count to %d",
maxBufferSlack + 1 + minUndequeuedBuffers);
err = native_window_set_buffer_count(anw.get(),
maxBufferSlack + 1 + minUndequeuedBuffers);
ASSERT_NO_ERROR(err, "set_buffer_count error: ");
}
// Produce one frame of image data; assumes format and resolution configuration
// is already done.
void produceOneFrame(const sp<ANativeWindow>& anw,
const CpuConsumerTestParams& params,
int64_t timestamp, uint32_t *stride) {
status_t err;
ANativeWindowBuffer* anb;
ALOGVV("Dequeue buffer from %p", anw.get());
err = native_window_dequeue_buffer_and_wait(anw.get(), &anb);
ASSERT_NO_ERROR(err, "dequeueBuffer error: ");
ASSERT_TRUE(anb != nullptr);
sp<GraphicBuffer> buf(GraphicBuffer::from(anb));
*stride = buf->getStride();
uint8_t* img = nullptr;
ALOGVV("Lock buffer from %p for write", anw.get());
err = buf->lock(GRALLOC_USAGE_SW_WRITE_OFTEN, (void**)(&img));
ASSERT_NO_ERROR(err, "lock error: ");
switch (params.format) {
case HAL_PIXEL_FORMAT_YV12:
fillYV12Buffer(img, params.width, params.height, *stride);
break;
case HAL_PIXEL_FORMAT_RAW16:
fillBayerRawBuffer(img, params.width, params.height, buf->getStride());
break;
case HAL_PIXEL_FORMAT_Y8:
fillGreyscaleBuffer<uint8_t>(img, params.width, params.height,
buf->getStride(), /*bpp*/8);
break;
case HAL_PIXEL_FORMAT_Y16:
fillGreyscaleBuffer<uint16_t>((uint16_t*)img, params.width,
params.height, buf->getStride(),
/*bpp*/16);
break;
case HAL_PIXEL_FORMAT_RGBA_8888:
fillRgba8888Buffer(img, params.width, params.height, buf->getStride());
break;
default:
FAIL() << "Unknown pixel format under test!";
break;
}
ALOGVV("Unlock buffer from %p", anw.get());
err = buf->unlock();
ASSERT_NO_ERROR(err, "unlock error: ");
ALOGVV("Set timestamp to %p", anw.get());
err = native_window_set_buffers_timestamp(anw.get(), timestamp);
ASSERT_NO_ERROR(err, "set_buffers_timestamp error: ");
ALOGVV("Queue buffer to %p", anw.get());
err = anw->queueBuffer(anw.get(), buf->getNativeBuffer(), -1);
ASSERT_NO_ERROR(err, "queueBuffer error:");
};
// This test is disabled because the HAL_PIXEL_FORMAT_RAW16 format is not
// supported on all devices.
TEST_P(CpuConsumerTest, FromCpuSingle) {
status_t err;
CpuConsumerTestParams params = GetParam();
// Set up
ASSERT_NO_FATAL_FAILURE(configureANW(mANW, params, 1));
// Produce
const int64_t time = 12345678L;
uint32_t stride;
ASSERT_NO_FATAL_FAILURE(produceOneFrame(mANW, params, time,
&stride));
// Consume
CpuConsumer::LockedBuffer b;
err = mCC->lockNextBuffer(&b);
ASSERT_NO_ERROR(err, "getNextBuffer error: ");
ASSERT_TRUE(b.data != nullptr);
EXPECT_EQ(params.width, b.width);
EXPECT_EQ(params.height, b.height);
EXPECT_EQ(params.format, b.format);
EXPECT_EQ(stride, b.stride);
EXPECT_EQ(time, b.timestamp);
checkAnyBuffer(b, GetParam().format);
mCC->unlockBuffer(b);
}
// This test is disabled because the HAL_PIXEL_FORMAT_RAW16 format is not
// supported on all devices.
TEST_P(CpuConsumerTest, FromCpuManyInQueue) {
status_t err;
CpuConsumerTestParams params = GetParam();
const int numInQueue = 5;
// Set up
ASSERT_NO_FATAL_FAILURE(configureANW(mANW, params, numInQueue));
// Produce
const int64_t time[numInQueue] = { 1L, 2L, 3L, 4L, 5L};
uint32_t stride[numInQueue];
for (int i = 0; i < numInQueue; i++) {
ALOGV("Producing frame %d", i);
ASSERT_NO_FATAL_FAILURE(produceOneFrame(mANW, params, time[i],
&stride[i]));
}
// Consume
for (int i = 0; i < numInQueue; i++) {
ALOGV("Consuming frame %d", i);
CpuConsumer::LockedBuffer b;
err = mCC->lockNextBuffer(&b);
ASSERT_NO_ERROR(err, "getNextBuffer error: ");
ASSERT_TRUE(b.data != nullptr);
EXPECT_EQ(params.width, b.width);
EXPECT_EQ(params.height, b.height);
EXPECT_EQ(params.format, b.format);
EXPECT_EQ(stride[i], b.stride);
EXPECT_EQ(time[i], b.timestamp);
checkAnyBuffer(b, GetParam().format);
mCC->unlockBuffer(b);
}
}
// This test is disabled because the HAL_PIXEL_FORMAT_RAW16 format is not
// supported on all devices.
TEST_P(CpuConsumerTest, FromCpuLockMax) {
status_t err;
CpuConsumerTestParams params = GetParam();
// Set up
ASSERT_NO_FATAL_FAILURE(configureANW(mANW, params, params.maxLockedBuffers + 1));
// Produce
const int64_t time = 1234L;
uint32_t stride;
for (int i = 0; i < params.maxLockedBuffers + 1; i++) {
ALOGV("Producing frame %d", i);
ASSERT_NO_FATAL_FAILURE(produceOneFrame(mANW, params, time,
&stride));
}
// Consume
std::vector<CpuConsumer::LockedBuffer> b(params.maxLockedBuffers);
for (int i = 0; i < params.maxLockedBuffers; i++) {
ALOGV("Locking frame %d", i);
err = mCC->lockNextBuffer(&b[i]);
ASSERT_NO_ERROR(err, "getNextBuffer error: ");
ASSERT_TRUE(b[i].data != nullptr);
EXPECT_EQ(params.width, b[i].width);
EXPECT_EQ(params.height, b[i].height);
EXPECT_EQ(params.format, b[i].format);
EXPECT_EQ(stride, b[i].stride);
EXPECT_EQ(time, b[i].timestamp);
checkAnyBuffer(b[i], GetParam().format);
}
ALOGV("Locking frame %d (too many)", params.maxLockedBuffers);
CpuConsumer::LockedBuffer bTooMuch;
err = mCC->lockNextBuffer(&bTooMuch);
ASSERT_TRUE(err == NOT_ENOUGH_DATA) << "Allowing too many locks";
ALOGV("Unlocking frame 0");
err = mCC->unlockBuffer(b[0]);
ASSERT_NO_ERROR(err, "Could not unlock buffer 0: ");
ALOGV("Locking frame %d (should work now)", params.maxLockedBuffers);
err = mCC->lockNextBuffer(&bTooMuch);
ASSERT_NO_ERROR(err, "Did not allow new lock after unlock");
ASSERT_TRUE(bTooMuch.data != nullptr);
EXPECT_EQ(params.width, bTooMuch.width);
EXPECT_EQ(params.height, bTooMuch.height);
EXPECT_EQ(params.format, bTooMuch.format);
EXPECT_EQ(stride, bTooMuch.stride);
EXPECT_EQ(time, bTooMuch.timestamp);
checkAnyBuffer(bTooMuch, GetParam().format);
ALOGV("Unlocking extra buffer");
err = mCC->unlockBuffer(bTooMuch);
ASSERT_NO_ERROR(err, "Could not unlock extra buffer: ");
ALOGV("Locking frame %d (no more available)", params.maxLockedBuffers + 1);
err = mCC->lockNextBuffer(&b[0]);
ASSERT_EQ(BAD_VALUE, err) << "Not out of buffers somehow";
for (int i = 1; i < params.maxLockedBuffers; i++) {
mCC->unlockBuffer(b[i]);
}
}
TEST_P(CpuConsumerTest, FromCpuInvalid) {
status_t err = mCC->lockNextBuffer(nullptr);
ASSERT_EQ(BAD_VALUE, err) << "lockNextBuffer did not fail";
CpuConsumer::LockedBuffer b;
err = mCC->unlockBuffer(b);
ASSERT_EQ(BAD_VALUE, err) << "unlockBuffer did not fail";
}
TEST_P(CpuConsumerTest, FromCpuMultiThread) {
CpuConsumerTestParams params = GetParam();
ASSERT_NO_FATAL_FAILURE(configureANW(mANW, params, params.maxLockedBuffers + 1));
for (int i = 0; i < 10; i++) {
std::atomic<int> threadReadyCount(0);
auto lockAndUnlock = [&]() {
threadReadyCount++;
// busy wait
while (threadReadyCount < params.maxLockedBuffers + 1);
CpuConsumer::LockedBuffer b;
status_t err = mCC->lockNextBuffer(&b);
if (err == NO_ERROR) {
usleep(1000);
err = mCC->unlockBuffer(b);
ASSERT_NO_ERROR(err, "Could not unlock buffer: ");
} else if (err == NOT_ENOUGH_DATA) {
// there are params.maxLockedBuffers+1 threads so one of the
// threads might get this error
} else {
FAIL() << "Could not lock buffer";
}
};
// produce buffers
for (int j = 0; j < params.maxLockedBuffers + 1; j++) {
const int64_t time = 1234L;
uint32_t stride;
ASSERT_NO_FATAL_FAILURE(produceOneFrame(mANW, params, time, &stride));
}
// spawn threads
std::vector<std::thread> threads;
for (int j = 0; j < params.maxLockedBuffers + 1; j++) {
threads.push_back(std::thread(lockAndUnlock));
}
// join threads
for (auto& thread : threads) {
thread.join();
}
// we produced N+1 buffers, but the threads might only consume N
CpuConsumer::LockedBuffer b;
if (mCC->lockNextBuffer(&b) == NO_ERROR) {
mCC->unlockBuffer(b);
}
if (HasFatalFailure()) {
break;
}
}
}
CpuConsumerTestParams y8TestSets[] = {
{ 512, 512, 1, HAL_PIXEL_FORMAT_Y8},
{ 512, 512, 3, HAL_PIXEL_FORMAT_Y8},
{ 2608, 1960, 1, HAL_PIXEL_FORMAT_Y8},
{ 2608, 1960, 3, HAL_PIXEL_FORMAT_Y8},
{ 100, 100, 1, HAL_PIXEL_FORMAT_Y8},
{ 100, 100, 3, HAL_PIXEL_FORMAT_Y8},
};
CpuConsumerTestParams y16TestSets[] = {
{ 512, 512, 1, HAL_PIXEL_FORMAT_Y16},
{ 512, 512, 3, HAL_PIXEL_FORMAT_Y16},
{ 2608, 1960, 1, HAL_PIXEL_FORMAT_Y16},
{ 2608, 1960, 3, HAL_PIXEL_FORMAT_Y16},
{ 100, 100, 1, HAL_PIXEL_FORMAT_Y16},
{ 100, 100, 3, HAL_PIXEL_FORMAT_Y16},
};
CpuConsumerTestParams rawTestSets[] = {
{ 512, 512, 1, HAL_PIXEL_FORMAT_RAW16},
{ 512, 512, 3, HAL_PIXEL_FORMAT_RAW16},
{ 2608, 1960, 1, HAL_PIXEL_FORMAT_RAW16},
{ 2608, 1960, 3, HAL_PIXEL_FORMAT_RAW16},
{ 100, 100, 1, HAL_PIXEL_FORMAT_RAW16},
{ 100, 100, 3, HAL_PIXEL_FORMAT_RAW16},
};
CpuConsumerTestParams rgba8888TestSets[] = {
{ 512, 512, 1, HAL_PIXEL_FORMAT_RGBA_8888},
{ 512, 512, 3, HAL_PIXEL_FORMAT_RGBA_8888},
{ 2608, 1960, 1, HAL_PIXEL_FORMAT_RGBA_8888},
{ 2608, 1960, 3, HAL_PIXEL_FORMAT_RGBA_8888},
{ 100, 100, 1, HAL_PIXEL_FORMAT_RGBA_8888},
{ 100, 100, 3, HAL_PIXEL_FORMAT_RGBA_8888},
};
#if CPU_CONSUMER_TEST_FORMAT_Y8
INSTANTIATE_TEST_CASE_P(Y8Tests,
CpuConsumerTest,
::testing::ValuesIn(y8TestSets));
#endif
#if CPU_CONSUMER_TEST_FORMAT_Y16
INSTANTIATE_TEST_CASE_P(Y16Tests,
CpuConsumerTest,
::testing::ValuesIn(y16TestSets));
#endif
#if CPU_CONSUMER_TEST_FORMAT_RAW
INSTANTIATE_TEST_CASE_P(RawTests,
CpuConsumerTest,
::testing::ValuesIn(rawTestSets));
#endif
#if CPU_CONSUMER_TEST_FORMAT_RGBA_8888
INSTANTIATE_TEST_CASE_P(Rgba8888Tests,
CpuConsumerTest,
::testing::ValuesIn(rgba8888TestSets));
#endif
} // namespace android