Home | History | Annotate | Download | only in tests 
      1 /*
      2  * Copyright 2015 Google Inc.
      3  *
      4  * Use of this source code is governed by a BSD-style license that can be
      5  * found in the LICENSE file.
      6  */
      7 
      8 // This is a GPU-backend specific test. It relies on static intializers to work
      9 
     10 #include "SkTypes.h"
     11 
     12 #if SK_SUPPORT_GPU && defined(SK_VULKAN)
     13 
     14 #include "GrContextFactory.h"
     15 #include "GrContextPriv.h"
     16 #include "GrProxyProvider.h"
     17 #include "GrSurfaceProxy.h"
     18 #include "GrTest.h"
     19 #include "SkGr.h"
     20 #include "Test.h"
     21 #include "vk/GrVkGpu.h"
     22 
     23 using sk_gpu_test::GrContextFactory;
     24 
     25 void fill_pixel_data(int width, int height, GrColor* data) {
     26 
     27     // build red-green gradient
     28     for (int j = 0; j < height; ++j) {
     29         for (int i = 0; i < width; ++i) {
     30             unsigned int red = (unsigned int)(256.f*(i / (float)width));
     31             unsigned int green = (unsigned int)(256.f*(j / (float)height));
     32             data[i + j*width] = GrColorPackRGBA(red - (red>>8), green - (green>>8), 0xff, 0xff);
     33         }
     34     }
     35 }
     36 
     37 bool does_full_buffer_contain_correct_color(GrColor* srcBuffer,
     38                                             GrColor* dstBuffer,
     39                                             int width,
     40                                             int height) {
     41     GrColor* srcPtr = srcBuffer;
     42     GrColor* dstPtr = dstBuffer;
     43     for (int j = 0; j < height; ++j) {
     44         for (int i = 0; i < width; ++i) {
     45             if (srcPtr[i] != dstPtr[i]) {
     46                 return false;
     47             }
     48         }
     49         srcPtr += width;
     50         dstPtr += width;
     51     }
     52     return true;
     53 }
     54 
     55 void basic_texture_test(skiatest::Reporter* reporter, GrContext* context, GrPixelConfig config,
     56                         bool renderTarget) {
     57     GrProxyProvider* proxyProvider = context->contextPriv().proxyProvider();
     58 
     59     const int kWidth = 16;
     60     const int kHeight = 16;
     61     SkAutoTMalloc<GrColor> srcBuffer(kWidth*kHeight);
     62     SkAutoTMalloc<GrColor> dstBuffer(kWidth*kHeight);
     63 
     64     fill_pixel_data(kWidth, kHeight, srcBuffer.get());
     65 
     66     GrSurfaceDesc surfDesc;
     67     surfDesc.fFlags = renderTarget ? kRenderTarget_GrSurfaceFlag : kNone_GrSurfaceFlags;
     68     surfDesc.fOrigin = kTopLeft_GrSurfaceOrigin;
     69     surfDesc.fWidth = kWidth;
     70     surfDesc.fHeight = kHeight;
     71     surfDesc.fConfig = config;
     72     surfDesc.fSampleCnt = 1;
     73 
     74     SkColorType ct;
     75     SkAssertResult(GrPixelConfigToColorType(config, &ct));
     76 
     77     sk_sp<GrTextureProxy> proxy = proxyProvider->createTextureProxy(surfDesc, SkBudgeted::kNo,
     78                                                                     srcBuffer, 0);
     79     REPORTER_ASSERT(reporter, proxy);
     80     if (proxy) {
     81         sk_sp<GrSurfaceContext> sContext = context->contextPriv().makeWrappedSurfaceContext(proxy);
     82 
     83         SkImageInfo dstInfo = SkImageInfo::Make(kWidth, kHeight, ct, kOpaque_SkAlphaType);
     84 
     85         bool result = sContext->readPixels(dstInfo, dstBuffer, 0, 0, 0);
     86         REPORTER_ASSERT(reporter, result);
     87         REPORTER_ASSERT(reporter, does_full_buffer_contain_correct_color(srcBuffer,
     88                                                                          dstBuffer,
     89                                                                          kWidth,
     90                                                                          kHeight));
     91 
     92         dstInfo = SkImageInfo::Make(10, 2, ct, kOpaque_SkAlphaType);
     93         result = sContext->writePixels(dstInfo, srcBuffer, 0, 2, 10);
     94         REPORTER_ASSERT(reporter, result);
     95 
     96         memset(dstBuffer, 0, kWidth*kHeight*sizeof(GrColor));
     97 
     98         result = sContext->readPixels(dstInfo, dstBuffer, 0, 2, 10);
     99         REPORTER_ASSERT(reporter, result);
    100 
    101         REPORTER_ASSERT(reporter, does_full_buffer_contain_correct_color(srcBuffer,
    102                                                                          dstBuffer,
    103                                                                          10,
    104                                                                          2));
    105     }
    106 
    107     surfDesc.fOrigin = kBottomLeft_GrSurfaceOrigin;
    108 
    109     proxy = proxyProvider->createTextureProxy(surfDesc, SkBudgeted::kNo, srcBuffer, 0);
    110     REPORTER_ASSERT(reporter, proxy);
    111     if (proxy) {
    112         sk_sp<GrSurfaceContext> sContext = context->contextPriv().makeWrappedSurfaceContext(proxy);
    113 
    114         SkImageInfo dstInfo = SkImageInfo::Make(kWidth, kHeight, ct, kOpaque_SkAlphaType);
    115 
    116         bool result = sContext->readPixels(dstInfo, dstBuffer, 0, 0, 0);
    117         REPORTER_ASSERT(reporter, result);
    118         REPORTER_ASSERT(reporter, does_full_buffer_contain_correct_color(srcBuffer,
    119                                                                          dstBuffer,
    120                                                                          kWidth,
    121                                                                          kHeight));
    122 
    123         dstInfo = SkImageInfo::Make(4, 5, ct, kOpaque_SkAlphaType);
    124         result = sContext->writePixels(dstInfo, srcBuffer, 0, 5, 4);
    125         REPORTER_ASSERT(reporter, result);
    126 
    127         memset(dstBuffer, 0, kWidth*kHeight*sizeof(GrColor));
    128 
    129         result = sContext->readPixels(dstInfo, dstBuffer, 0, 5, 4);
    130         REPORTER_ASSERT(reporter, result);
    131 
    132         REPORTER_ASSERT(reporter, does_full_buffer_contain_correct_color(srcBuffer,
    133                                                                          dstBuffer,
    134                                                                          4,
    135                                                                          5));
    136 
    137     }
    138 }
    139 
    140 DEF_GPUTEST_FOR_VULKAN_CONTEXT(VkUploadPixelsTests, reporter, ctxInfo) {
    141     // RGBA
    142     basic_texture_test(reporter, ctxInfo.grContext(), kRGBA_8888_GrPixelConfig, false);
    143     basic_texture_test(reporter, ctxInfo.grContext(), kRGBA_8888_GrPixelConfig, true);
    144 
    145     // BGRA
    146     basic_texture_test(reporter, ctxInfo.grContext(), kBGRA_8888_GrPixelConfig, false);
    147     basic_texture_test(reporter, ctxInfo.grContext(), kBGRA_8888_GrPixelConfig, true);
    148 }
    149 
    150 #endif
    151