new API for retrieving YUV data

src/core/SkImageGenerator.cpp

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkImageGenerator.h"
 
 bool SkImageGenerator::getInfo(SkImageInfo* info) {
@@ skipping 37 matching lines @@
 }
 
 bool SkImageGenerator::getPixels(const SkImageInfo& info, void* pixels, size_t rowBytes) {
     SkASSERT(kIndex_8_SkColorType != info.colorType());
     if (kIndex_8_SkColorType == info.colorType()) {
         return false;
     }
     return this->getPixels(info, pixels, rowBytes, NULL, NULL);
 }
 
+#ifdef SK_SUPPORT_LEGACY_IMAGEGENERATOR_YUV_API
 bool SkImageGenerator::getYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3],
                                      SkYUVColorSpace* colorSpace) {
 #ifdef SK_DEBUG
     // In all cases, we need the sizes array
     SkASSERT(sizes);
 
     bool isValidWithPlanes = (planes) && (rowBytes) &&
         ((planes[0]) && (planes[1]) && (planes[2]) &&
          (0  != rowBytes[0]) && (0  != rowBytes[1]) && (0  != rowBytes[2]));
     bool isValidWithoutPlanes =
@@ skipping 15 matching lines @@
               (sizes[1].fHeight >= 0) &&
               (sizes[2].fWidth  >= 0) &&
               (sizes[2].fHeight >= 0) &&
               (rowBytes[0] >= (size_t)sizes[0].fWidth) &&
               (rowBytes[1] >= (size_t)sizes[1].fWidth) &&
               (rowBytes[2] >= (size_t)sizes[2].fWidth)));
 #endif
 
     return this->onGetYUV8Planes(sizes, planes, rowBytes, colorSpace);
 }
+#endif
 
-bool SkImageGenerator::onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3]) {
-    return false;
+bool SkImageGenerator::queryYUV8(SkISize logicalSizes[3], SkISize optimalAllocationSizes[3]) {
+    SkISize logicalStorage[3], optimalStorage[3];
+    if (!logicalSizes) {
+        logicalSizes = logicalStorage;
+    }
+    if (!optimalAllocationSizes) {
+        optimalAllocationSizes = optimalStorage;
+    }
+    return this->onQueryYUV8(logicalSizes, optimalAllocationSizes);
 }
 
-bool SkImageGenerator::onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3],
-                                       SkYUVColorSpace* colorSpace) {
-    // In order to maintain compatibility with clients that implemented the original
-    // onGetYUV8Planes interface, we assume that the color space is JPEG.
-    // TODO(rileya): remove this and the old onGetYUV8Planes once clients switch over to
-    // the new interface.
-    if (colorSpace) {
-        *colorSpace = kJPEG_SkYUVColorSpace;
+bool SkImageGenerator::getYUV8(const SkISize allocationSizes[3], void* planes[3],
+                               SkYUVColorSpace* colorSpace) {
+    SkASSERT(allocationSizes);
+    SkASSERT(planes);
+
+    for (int i = 0; i < 3; ++i) {
+        if (allocationSizes[i].width() <= 0 || allocationSizes[i].height() <= 0) {
+            return false;
+        }
+        if (!planes[i]) {
+            return false;
+        }
     }
-    return this->onGetYUV8Planes(sizes, planes, rowBytes);
+
+    SkYUVColorSpace colorSpaceStorage;
+    if (!colorSpace) {
+        colorSpace = &colorSpaceStorage;
+    }
+    return this->onGetYUV8(allocationSizes, planes, colorSpace);
 }
 
 /////////////////////////////////////////////////////////////////////////////////////////////
 
 SkData* SkImageGenerator::onRefEncodedData() {
     return NULL;
 }
 
 bool SkImageGenerator::onGetInfo(SkImageInfo*) {
     return false;
 }
 
 bool SkImageGenerator::onGetPixels(const SkImageInfo&, void*, size_t, SkPMColor*, int*) {
     return false;
 }
+
+bool SkImageGenerator::onQueryYUV8(SkISize logicalSizes[3], SkISize optimalAllocationSizes[3]) {
+#ifdef SK_SUPPORT_LEGACY_IMAGEGENERATOR_YUV_API
+    if (this->getYUV8Planes(logicalSizes, NULL, NULL, NULL)) {
+        memcpy(optimalAllocationSizes, logicalSizes, sizeof(SkISize) * 3);
+        return true;
+    }
+#endif
+    return false;
+}
+
+bool SkImageGenerator::onGetYUV8(const SkISize allocationSizes[3], void* planes[3],
+                                 SkYUVColorSpace* colorSpace) {
+#ifdef SK_SUPPORT_LEGACY_IMAGEGENERATOR_YUV_API
+    SkISize tmpSizes[3];
+    size_t rowBytes[3];
+    for (int i = 0; i < 3; ++i) {
+        tmpSizes[i] = allocationSizes[i];
+        rowBytes[i] = allocationSizes[i].width();
+    }
+    if (this->getYUV8Planes(tmpSizes, planes, rowBytes, colorSpace)) {
+        return true;
+    }
+#endif
+    return false;
+}
+
+#ifdef SK_SUPPORT_LEGACY_IMAGEGENERATOR_YUV_API
+bool SkImageGenerator::onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3]) {
+    return false;
+}
+
+bool SkImageGenerator::onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3],
+                                       SkYUVColorSpace* colorSpace) {
+    // In order to maintain compatibility with clients that implemented the original
+    // onGetYUV8Planes interface, we assume that the color space is JPEG.
+    // TODO(rileya): remove this and the old onGetYUV8Planes once clients switch over to
+    // the new interface.
+    if (colorSpace) {
+        *colorSpace = kJPEG_SkYUVColorSpace;
+    }
+    return this->onGetYUV8Planes(sizes, planes, rowBytes);
+}
+#endif
+