| /* |
| * 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(); |
| ALOGD("** Starting parameterized 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++) { |
| ALOGD("Producing frame %d", i); |
| ASSERT_NO_FATAL_FAILURE(produceOneFrame(mANW, params, time[i], |
| &stride[i])); |
| } |
| |
| // Consume |
| |
| for (int i = 0; i < numInQueue; i++) { |
| ALOGD("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++) { |
| ALOGD("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++) { |
| ALOGD("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); |
| } |
| |
| ALOGD("Locking frame %d (too many)", params.maxLockedBuffers); |
| CpuConsumer::LockedBuffer bTooMuch; |
| err = mCC->lockNextBuffer(&bTooMuch); |
| ASSERT_TRUE(err == NOT_ENOUGH_DATA) << "Allowing too many locks"; |
| |
| ALOGD("Unlocking frame 0"); |
| err = mCC->unlockBuffer(b[0]); |
| ASSERT_NO_ERROR(err, "Could not unlock buffer 0: "); |
| |
| ALOGD("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); |
| |
| ALOGD("Unlocking extra buffer"); |
| err = mCC->unlockBuffer(bTooMuch); |
| ASSERT_NO_ERROR(err, "Could not unlock extra buffer: "); |
| |
| ALOGD("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 |