1 /* 2 * Copyright (C) 2009 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #define LOG_TAG "OmxJpegDecoder" 18 #include <sys/time.h> 19 #include <utils/Log.h> 20 21 #include <stdlib.h> 22 #include <string.h> 23 #include <unistd.h> 24 25 #include <binder/IServiceManager.h> 26 #include <binder/ProcessState.h> 27 #include <media/IMediaPlayerService.h> 28 #include <media/stagefright/foundation/ADebug.h> 29 #include <media/stagefright/MediaSource.h> 30 #include <media/stagefright/MetaData.h> 31 #include <media/stagefright/OMXClient.h> 32 #include <media/stagefright/OMXCodec.h> 33 #include <SkImage.h> 34 #include <SkMallocPixelRef.h> 35 36 #include "omx_jpeg_decoder.h" 37 #include "SkOmxPixelRef.h" 38 #include "StreamSource.h" 39 40 using namespace android; 41 42 static void getJpegOutput(MediaBuffer* buffer, const char* filename) { 43 int size = buffer->range_length(); 44 int offset = buffer->range_offset(); 45 FILE *pFile = fopen(filename, "w+"); 46 47 if (pFile == NULL) { 48 printf("Error: cannot open %s.\n", filename); 49 } else { 50 char* data = (char*) buffer->data(); 51 data += offset; 52 while (size > 0) { 53 int numChars = fwrite(data, sizeof(char), 1024, pFile); 54 int numBytes = numChars * sizeof(char); 55 size -= numBytes; 56 data += numBytes; 57 } 58 fclose(pFile); 59 } 60 return; 61 } 62 63 extern int storeBitmapToFile(SkBitmap* bitmap, const char* filename) { 64 bitmap->lockPixels(); 65 uint8_t* data = (uint8_t *)bitmap->getPixels(); 66 int size = bitmap->getSize(); 67 FILE* fp = fopen(filename, "w+"); 68 69 if (NULL == fp) { 70 printf("Cannot open the output file! \n"); 71 return -1; 72 } else { 73 while (size > 0) { 74 int numChars = fwrite(data, sizeof(char), 1024, fp); 75 int numBytes = numChars * sizeof(char); 76 size -= numBytes; 77 data += numBytes; 78 } 79 fclose(fp); 80 } 81 return 0; 82 } 83 84 static int64_t getNowUs() { 85 struct timeval tv; 86 gettimeofday(&tv, NULL); 87 88 return (int64_t)tv.tv_usec + tv.tv_sec * 1000000; 89 } 90 91 OmxJpegImageDecoder::OmxJpegImageDecoder() { 92 status_t err = mClient.connect(); 93 CHECK_EQ(err, (status_t)OK); 94 } 95 96 OmxJpegImageDecoder::~OmxJpegImageDecoder() { 97 mClient.disconnect(); 98 } 99 100 bool OmxJpegImageDecoder::onDecode(SkStream* stream, 101 SkBitmap* bm, Mode mode) { 102 sp<MediaSource> source = prepareMediaSource(stream); 103 sp<MetaData> meta = source->getFormat(); 104 int width; 105 int height; 106 meta->findInt32(kKeyWidth, &width); 107 meta->findInt32(kKeyHeight, &height); 108 configBitmapSize(bm, getPrefConfig(k32Bit_SrcDepth, false), width, height); 109 110 // mode == DecodeBounds 111 if (mode == SkImageDecoder::kDecodeBounds_Mode) { 112 return true; 113 } 114 115 // mode == DecodePixels 116 if (!this->allocPixelRef(bm, NULL)) { 117 ALOGI("Cannot allocPixelRef()!"); 118 return false; 119 } 120 121 sp<MediaSource> decoder = getDecoder(&mClient, source); 122 return decodeSource(decoder, source, bm); 123 } 124 125 JPEGSource* OmxJpegImageDecoder::prepareMediaSource(SkStream* stream) { 126 DataSource::RegisterDefaultSniffers(); 127 sp<DataSource> dataSource = new StreamSource(stream); 128 return new JPEGSource(dataSource); 129 } 130 131 sp<MediaSource> OmxJpegImageDecoder::getDecoder( 132 OMXClient *client, const sp<MediaSource>& source) { 133 sp<MetaData> meta = source->getFormat(); 134 sp<MediaSource> decoder = OMXCodec::Create( 135 client->interface(), meta, false /* createEncoder */, source); 136 137 CHECK(decoder != NULL); 138 return decoder; 139 } 140 141 bool OmxJpegImageDecoder::decodeSource(sp<MediaSource> decoder, 142 const sp<MediaSource>& source, SkBitmap* bm) { 143 status_t rt = decoder->start(); 144 if (rt != OK) { 145 ALOGE("Cannot start OMX Decoder!"); 146 return false; 147 } 148 int64_t startTime = getNowUs(); 149 MediaBuffer *buffer; 150 151 // decode source 152 status_t err = decoder->read(&buffer, NULL); 153 int64_t duration = getNowUs() - startTime; 154 155 if (err != OK) { 156 CHECK(buffer == NULL); 157 } 158 printf("Duration in decoder->read(): %.1f (msecs). \n", 159 duration / 1E3 ); 160 161 /* Mark the code for now, since we attend to copy buffer to SkBitmap. 162 // Install pixelRef to Bitmap. 163 installPixelRef(buffer, decoder, bm);*/ 164 165 // Copy pixels from buffer to bm. 166 // May need to check buffer->rawBytes() == bm->rawBytes(). 167 CHECK_EQ(buffer->size(), bm->getSize()); 168 memcpy(bm->getPixels(), buffer->data(), buffer->size()); 169 buffer->release(); 170 decoder->stop(); 171 172 return true; 173 } 174 175 void OmxJpegImageDecoder::installPixelRef(MediaBuffer *buffer, sp<MediaSource> decoder, 176 SkBitmap* bm) { 177 178 // set bm's pixelref based on the data in buffer. 179 SkAutoLockPixels alp(*bm); 180 SkPixelRef* pr = new SkOmxPixelRef(NULL, buffer, decoder); 181 bm->setPixelRef(pr)->unref(); 182 bm->lockPixels(); 183 return; 184 } 185 186 void OmxJpegImageDecoder::configBitmapSize(SkBitmap* bm, SkBitmap::Config pref, 187 int width, int height) { 188 bm->setConfig(getColorSpaceConfig(pref), width, height, 0, kOpaque_SkAlphaType); 189 } 190 191 SkBitmap::Config OmxJpegImageDecoder::getColorSpaceConfig( 192 SkBitmap::Config pref) { 193 194 // Set the color space to ARGB_8888 for now 195 // because of limitation in hardware support. 196 return SkBitmap::kARGB_8888_Config; 197 } 198
