1 /* 2 * Copyright (C) 2014 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_NDEBUG 0 18 #define LOG_TAG "NdkMediaFormat" 19 20 #include <inttypes.h> 21 22 #include "NdkMediaFormat.h" 23 24 #include <utils/Log.h> 25 #include <utils/StrongPointer.h> 26 #include <media/stagefright/foundation/ABuffer.h> 27 #include <media/stagefright/foundation/AMessage.h> 28 #include <media/stagefright/MetaData.h> 29 #include <android_runtime/AndroidRuntime.h> 30 #include <android_util_Binder.h> 31 32 #include <jni.h> 33 34 using namespace android; 35 36 struct AMediaFormat { 37 sp<AMessage> mFormat; 38 String8 mDebug; 39 KeyedVector<String8, String8> mStringCache; 40 }; 41 42 extern "C" { 43 44 // private functions for conversion to/from AMessage 45 AMediaFormat* AMediaFormat_fromMsg(const void* data) { 46 ALOGV("private ctor"); 47 AMediaFormat* mData = new AMediaFormat(); 48 mData->mFormat = *((sp<AMessage>*)data); 49 if (mData->mFormat == NULL) { 50 ALOGW("got NULL format"); 51 mData->mFormat = new AMessage; 52 } 53 return mData; 54 } 55 56 void AMediaFormat_getFormat(const AMediaFormat* mData, void* dest) { 57 *((sp<AMessage>*)dest) = mData->mFormat; 58 } 59 60 61 /* 62 * public function follow 63 */ 64 EXPORT 65 AMediaFormat *AMediaFormat_new() { 66 ALOGV("ctor"); 67 sp<AMessage> msg = new AMessage(); 68 return AMediaFormat_fromMsg(&msg); 69 } 70 71 EXPORT 72 media_status_t AMediaFormat_delete(AMediaFormat *mData) { 73 ALOGV("dtor"); 74 delete mData; 75 return AMEDIA_OK; 76 } 77 78 79 EXPORT 80 const char* AMediaFormat_toString(AMediaFormat *mData) { 81 sp<AMessage> f = mData->mFormat; 82 String8 ret; 83 int num = f->countEntries(); 84 for (int i = 0; i < num; i++) { 85 if (i != 0) { 86 ret.append(", "); 87 } 88 AMessage::Type t; 89 const char *name = f->getEntryNameAt(i, &t); 90 ret.append(name); 91 ret.append(": "); 92 switch (t) { 93 case AMessage::kTypeInt32: 94 { 95 int32_t val; 96 f->findInt32(name, &val); 97 ret.appendFormat("int32(%" PRId32 ")", val); 98 break; 99 } 100 case AMessage::kTypeInt64: 101 { 102 int64_t val; 103 f->findInt64(name, &val); 104 ret.appendFormat("int64(%" PRId64 ")", val); 105 break; 106 } 107 case AMessage::kTypeSize: 108 { 109 size_t val; 110 f->findSize(name, &val); 111 ret.appendFormat("size_t(%zu)", val); 112 break; 113 } 114 case AMessage::kTypeFloat: 115 { 116 float val; 117 f->findFloat(name, &val); 118 ret.appendFormat("float(%f)", val); 119 break; 120 } 121 case AMessage::kTypeDouble: 122 { 123 double val; 124 f->findDouble(name, &val); 125 ret.appendFormat("double(%f)", val); 126 break; 127 } 128 case AMessage::kTypeString: 129 { 130 AString val; 131 f->findString(name, &val); 132 ret.appendFormat("string(%s)", val.c_str()); 133 break; 134 } 135 case AMessage::kTypeBuffer: 136 { 137 ret.appendFormat("data"); 138 break; 139 } 140 default: 141 { 142 ret.appendFormat("unknown(%d)", t); 143 break; 144 } 145 } 146 } 147 ret.append("}"); 148 mData->mDebug = ret; 149 return mData->mDebug.string(); 150 } 151 152 EXPORT 153 bool AMediaFormat_getInt32(AMediaFormat* format, const char *name, int32_t *out) { 154 return format->mFormat->findInt32(name, out); 155 } 156 157 EXPORT 158 bool AMediaFormat_getInt64(AMediaFormat* format, const char *name, int64_t *out) { 159 return format->mFormat->findInt64(name, out); 160 } 161 162 EXPORT 163 bool AMediaFormat_getFloat(AMediaFormat* format, const char *name, float *out) { 164 return format->mFormat->findFloat(name, out); 165 } 166 167 EXPORT 168 bool AMediaFormat_getSize(AMediaFormat* format, const char *name, size_t *out) { 169 return format->mFormat->findSize(name, out); 170 } 171 172 EXPORT 173 bool AMediaFormat_getBuffer(AMediaFormat* format, const char *name, void** data, size_t *outsize) { 174 sp<ABuffer> buf; 175 if (format->mFormat->findBuffer(name, &buf)) { 176 *data = buf->data() + buf->offset(); 177 *outsize = buf->size(); 178 return true; 179 } 180 return false; 181 } 182 183 EXPORT 184 bool AMediaFormat_getString(AMediaFormat* mData, const char *name, const char **out) { 185 186 for (size_t i = 0; i < mData->mStringCache.size(); i++) { 187 if (strcmp(mData->mStringCache.keyAt(i).string(), name) == 0) { 188 mData->mStringCache.removeItemsAt(i, 1); 189 break; 190 } 191 } 192 193 AString tmp; 194 if (mData->mFormat->findString(name, &tmp)) { 195 String8 ret(tmp.c_str()); 196 mData->mStringCache.add(String8(name), ret); 197 *out = ret.string(); 198 return true; 199 } 200 return false; 201 } 202 203 EXPORT 204 void AMediaFormat_setInt32(AMediaFormat* format, const char *name, int32_t value) { 205 format->mFormat->setInt32(name, value); 206 } 207 208 EXPORT 209 void AMediaFormat_setInt64(AMediaFormat* format, const char *name, int64_t value) { 210 format->mFormat->setInt64(name, value); 211 } 212 213 EXPORT 214 void AMediaFormat_setFloat(AMediaFormat* format, const char* name, float value) { 215 format->mFormat->setFloat(name, value); 216 } 217 218 EXPORT 219 void AMediaFormat_setString(AMediaFormat* format, const char* name, const char* value) { 220 // AMessage::setString() makes a copy of the string 221 format->mFormat->setString(name, value, strlen(value)); 222 } 223 224 EXPORT 225 void AMediaFormat_setBuffer(AMediaFormat* format, const char* name, void* data, size_t size) { 226 // the ABuffer(void*, size_t) constructor doesn't take ownership of the data, so create 227 // a new buffer and copy the data into it 228 sp<ABuffer> buf = new ABuffer(size); 229 memcpy(buf->data(), data, size); 230 buf->setRange(0, size); 231 // AMessage::setBuffer() increases the refcount of the buffer 232 format->mFormat->setBuffer(name, buf); 233 } 234 235 236 EXPORT const char* AMEDIAFORMAT_KEY_AAC_PROFILE = "aac-profile"; 237 EXPORT const char* AMEDIAFORMAT_KEY_BIT_RATE = "bitrate"; 238 EXPORT const char* AMEDIAFORMAT_KEY_CHANNEL_COUNT = "channel-count"; 239 EXPORT const char* AMEDIAFORMAT_KEY_CHANNEL_MASK = "channel-mask"; 240 EXPORT const char* AMEDIAFORMAT_KEY_COLOR_FORMAT = "color-format"; 241 EXPORT const char* AMEDIAFORMAT_KEY_DURATION = "durationUs"; 242 EXPORT const char* AMEDIAFORMAT_KEY_FLAC_COMPRESSION_LEVEL = "flac-compression-level"; 243 EXPORT const char* AMEDIAFORMAT_KEY_FRAME_RATE = "frame-rate"; 244 EXPORT const char* AMEDIAFORMAT_KEY_HEIGHT = "height"; 245 EXPORT const char* AMEDIAFORMAT_KEY_IS_ADTS = "is-adts"; 246 EXPORT const char* AMEDIAFORMAT_KEY_IS_AUTOSELECT = "is-autoselect"; 247 EXPORT const char* AMEDIAFORMAT_KEY_IS_DEFAULT = "is-default"; 248 EXPORT const char* AMEDIAFORMAT_KEY_IS_FORCED_SUBTITLE = "is-forced-subtitle"; 249 EXPORT const char* AMEDIAFORMAT_KEY_I_FRAME_INTERVAL = "i-frame-interval"; 250 EXPORT const char* AMEDIAFORMAT_KEY_LANGUAGE = "language"; 251 EXPORT const char* AMEDIAFORMAT_KEY_MAX_HEIGHT = "max-height"; 252 EXPORT const char* AMEDIAFORMAT_KEY_MAX_INPUT_SIZE = "max-input-size"; 253 EXPORT const char* AMEDIAFORMAT_KEY_MAX_WIDTH = "max-width"; 254 EXPORT const char* AMEDIAFORMAT_KEY_MIME = "mime"; 255 EXPORT const char* AMEDIAFORMAT_KEY_PUSH_BLANK_BUFFERS_ON_STOP = "push-blank-buffers-on-shutdown"; 256 EXPORT const char* AMEDIAFORMAT_KEY_REPEAT_PREVIOUS_FRAME_AFTER = "repeat-previous-frame-after"; 257 EXPORT const char* AMEDIAFORMAT_KEY_SAMPLE_RATE = "sample-rate"; 258 EXPORT const char* AMEDIAFORMAT_KEY_WIDTH = "width"; 259 EXPORT const char* AMEDIAFORMAT_KEY_STRIDE = "stride"; 260 261 262 } // extern "C" 263 264 265