src/third_party/WebKit/Source/platform/graphics/ImageFrameGenerator.h

   1 /*
   2  * Copyright (C) 2012 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef ImageFrameGenerator_h
  27 #define ImageFrameGenerator_h
  28
  29 #include "SkBitmap.h"
  30 #include "SkSize.h"
  31 #include "SkTypes.h"
  32 #include "platform/PlatformExport.h"
  33 #include "platform/graphics/ThreadSafeDataTransport.h"
  34 #include "wtf/PassOwnPtr.h"
  35 #include "wtf/PassRefPtr.h"
  36 #include "wtf/RefCounted.h"
  37 #include "wtf/RefPtr.h"
  38 #include "wtf/ThreadingPrimitives.h"
  39 #include "wtf/ThreadSafeRefCounted.h"
  40 #include "wtf/Vector.h"
  41
  42 namespace blink {
  43
  44 class ImageDecoder;
  45 class SharedBuffer;
  46
  47 class PLATFORM_EXPORT ImageDecoderFactory {
  48     WTF_MAKE_NONCOPYABLE(ImageDecoderFactory);
  49 public:
  50     ImageDecoderFactory() {}
  51     virtual ~ImageDecoderFactory() { }
  52     virtual PassOwnPtr<ImageDecoder> create() = 0;
  53 };
  54
  55 class PLATFORM_EXPORT ImageFrameGenerator : public ThreadSafeRefCounted<ImageFrameGenerator> {
  56     WTF_MAKE_NONCOPYABLE(ImageFrameGenerator);
  57 public:
  58     static PassRefPtr<ImageFrameGenerator> create(const SkISize& fullSize, PassRefPtr<SharedBuffer> data, bool allDataReceived, bool isMultiFrame = false)
  59     {
  60         return adoptRef(new ImageFrameGenerator(fullSize, data, allDataReceived, isMultiFrame));
  61     }
  62
  63     ImageFrameGenerator(const SkISize& fullSize, PassRefPtr<SharedBuffer>, bool allDataReceived, bool isMultiFrame);
  64     ~ImageFrameGenerator();
  65
  66     // Decodes and scales the specified frame indicated by |index|. Dimensions
  67     // and output format are specified in |info|. Decoded pixels are written
  68     // into |pixels| with a stride of |rowBytes|.
  69     //
  70     // Returns true if decoding was successful.
  71     bool decodeAndScale(const SkImageInfo&, size_t index, void* pixels, size_t rowBytes);
  72
  73     // Decodes YUV components directly into the provided memory planes.
  74     bool decodeToYUV(SkISize componentSizes[3], void* planes[3], size_t rowBytes[3]);
  75
  76     void setData(PassRefPtr<SharedBuffer>, bool allDataReceived);
  77
  78     // Creates a new SharedBuffer containing the data received so far.
  79     void copyData(RefPtr<SharedBuffer>*, bool* allDataReceived);
  80
  81     SkISize getFullSize() const { return m_fullSize; }
  82
  83     bool isMultiFrame() const { return m_isMultiFrame; }
  84
  85     // FIXME: Return alpha state for each frame.
  86     bool hasAlpha(size_t);
  87
  88     bool getYUVComponentSizes(SkISize componentSizes[3]);
  89
  90 private:
  91     class ExternalMemoryAllocator;
  92     friend class ImageFrameGeneratorTest;
  93     friend class DeferredImageDecoderTest;
  94     // For testing. |factory| will overwrite the default ImageDecoder creation logic if |factory->create()| returns non-zero.
  95     void setImageDecoderFactory(PassOwnPtr<ImageDecoderFactory> factory) { m_imageDecoderFactory = factory; }
  96
  97     void setHasAlpha(size_t index, bool hasAlpha);
  98
  99     // These methods are called while m_decodeMutex is locked.
 100     SkBitmap tryToResumeDecode(const SkISize& scaledSize, size_t index);
 101
 102     // Use the given decoder to decode. If a decoder is not given then try to create one.
 103     // Returns true if decoding was complete.
 104     bool decode(size_t index, ImageDecoder**, SkBitmap*);
 105
 106     SkISize m_fullSize;
 107     ThreadSafeDataTransport m_data;
 108     bool m_isMultiFrame;
 109     bool m_decodeFailedAndEmpty;
 110     Vector<bool> m_hasAlpha;
 111     int m_decodeCount;
 112     OwnPtr<ExternalMemoryAllocator> m_externalAllocator;
 113
 114     OwnPtr<ImageDecoderFactory> m_imageDecoderFactory;
 115
 116     // Prevents multiple decode operations on the same data.
 117     Mutex m_decodeMutex;
 118
 119     // Protect concurrent access to m_hasAlpha.
 120     Mutex m_alphaMutex;
 121 };
 122
 123 } // namespace blink
 124
 125 #endif