Home | History | Annotate | Download | only in tools 
      1 /*
      2  * Copyright 2012 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 #include "picture_utils.h"
      9 #include "CopyTilesRenderer.h"
     10 #include "SkCanvas.h"
     11 #include "SkDevice.h"
     12 #include "SkImageEncoder.h"
     13 #include "SkPicture.h"
     14 #include "SkPixelRef.h"
     15 #include "SkRect.h"
     16 #include "SkString.h"
     17 
     18 namespace sk_tools {
     19     CopyTilesRenderer::CopyTilesRenderer(int x, int y)
     20     : fXTilesPerLargeTile(x)
     21     , fYTilesPerLargeTile(y) {
     22     }
     23     void CopyTilesRenderer::init(SkPicture* pict, const SkString* writePath,
     24                                  const SkString* mismatchPath, const SkString* inputFilename,
     25                                  bool useChecksumBasedFilenames) {
     26         // Do not call INHERITED::init(), which would create a (potentially large) canvas which is
     27         // not used by bench_pictures.
     28         SkASSERT(pict != NULL);
     29         // Only work with absolute widths (as opposed to percentages).
     30         SkASSERT(this->getTileWidth() != 0 && this->getTileHeight() != 0);
     31         fPicture.reset(pict)->ref();
     32         this->CopyString(&fWritePath, writePath);
     33         this->CopyString(&fMismatchPath, mismatchPath);
     34         this->CopyString(&fInputFilename, inputFilename);
     35         fUseChecksumBasedFilenames = useChecksumBasedFilenames;
     36         this->buildBBoxHierarchy();
     37         // In order to avoid allocating a large canvas (particularly important for GPU), create one
     38         // canvas that is a multiple of the tile size, and draw portions of the picture.
     39         fLargeTileWidth = fXTilesPerLargeTile * this->getTileWidth();
     40         fLargeTileHeight = fYTilesPerLargeTile * this->getTileHeight();
     41         fCanvas.reset(this->INHERITED::setupCanvas(fLargeTileWidth, fLargeTileHeight));
     42     }
     43 
     44     bool CopyTilesRenderer::render(SkBitmap** out) {
     45         int i = 0;
     46         bool success = true;
     47         SkBitmap dst;
     48         for (int x = 0; x < this->getViewWidth(); x += fLargeTileWidth) {
     49             for (int y = 0; y < this->getViewHeight(); y += fLargeTileHeight) {
     50                 SkAutoCanvasRestore autoRestore(fCanvas, true);
     51                 // Translate so that we draw the correct portion of the picture.
     52                 // Perform a postTranslate so that the scaleFactor does not interfere with the
     53                 // positioning.
     54                 SkMatrix mat(fCanvas->getTotalMatrix());
     55                 mat.postTranslate(SkIntToScalar(-x), SkIntToScalar(-y));
     56                 fCanvas->setMatrix(mat);
     57                 // Draw the picture
     58                 fCanvas->drawPicture(fPicture);
     59                 // Now extract the picture into tiles
     60                 const SkBitmap& baseBitmap = fCanvas->getDevice()->accessBitmap(false);
     61                 SkIRect subset;
     62                 for (int tileY = 0; tileY < fLargeTileHeight; tileY += this->getTileHeight()) {
     63                     for (int tileX = 0; tileX < fLargeTileWidth; tileX += this->getTileWidth()) {
     64                         subset.set(tileX, tileY, tileX + this->getTileWidth(),
     65                                    tileY + this->getTileHeight());
     66                         SkDEBUGCODE(bool extracted =)
     67                         baseBitmap.extractSubset(&dst, subset);
     68                         SkASSERT(extracted);
     69                         if (!fWritePath.isEmpty()) {
     70                             // Similar to write() in PictureRenderer.cpp, but just encodes
     71                             // a bitmap directly.
     72                             // TODO: Share more common code with write() to do this, to properly
     73                             // write out the JSON summary, etc.
     74                             SkString pathWithNumber = SkOSPath::SkPathJoin(fWritePath.c_str(),
     75                                                                            fInputFilename.c_str());
     76                             pathWithNumber.remove(pathWithNumber.size() - 4, 4);
     77                             pathWithNumber.appendf("%i.png", i++);
     78                             SkBitmap copy;
     79 #if SK_SUPPORT_GPU
     80                             if (isUsingGpuDevice()) {
     81                                 dst.pixelRef()->readPixels(&copy, &subset);
     82                             } else {
     83 #endif
     84                                 dst.copyTo(&copy);
     85 #if SK_SUPPORT_GPU
     86                             }
     87 #endif
     88                             success &= SkImageEncoder::EncodeFile(pathWithNumber.c_str(), copy,
     89                                                                   SkImageEncoder::kPNG_Type, 100);
     90                         }
     91                     }
     92                 }
     93             }
     94         }
     95         return success;
     96     }
     97 
     98     SkString CopyTilesRenderer::getConfigNameInternal() {
     99         return SkString("copy_tiles");
    100     }
    101 }
    102