1 /* 2 * Copyright 2012, 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 #ifndef CONVERTER_H_ 18 19 #define CONVERTER_H_ 20 21 #include "WifiDisplaySource.h" 22 23 #include <media/stagefright/foundation/AHandler.h> 24 25 namespace android { 26 27 struct ABuffer; 28 struct MediaCodec; 29 30 #define ENABLE_SILENCE_DETECTION 0 31 32 // Utility class that receives media access units and converts them into 33 // media access unit of a different format. 34 // Right now this'll convert raw video into H.264 and raw audio into AAC. 35 struct Converter : public AHandler { 36 Converter( 37 const sp<AMessage> ¬ify, 38 const sp<ALooper> &codecLooper, 39 const sp<AMessage> &format, 40 bool usePCMAudio); 41 42 status_t initCheck() const; 43 44 size_t getInputBufferCount() const; 45 46 sp<AMessage> getOutputFormat() const; 47 bool needToManuallyPrependSPSPPS() const; 48 49 void feedAccessUnit(const sp<ABuffer> &accessUnit); 50 void signalEOS(); 51 52 void requestIDRFrame(); 53 54 void dropAFrame(); 55 56 enum { 57 kWhatAccessUnit, 58 kWhatEOS, 59 kWhatError, 60 }; 61 62 enum { 63 kWhatDoMoreWork, 64 kWhatRequestIDRFrame, 65 kWhatShutdown, 66 kWhatMediaPullerNotify, 67 kWhatEncoderActivity, 68 kWhatDropAFrame, 69 }; 70 71 void shutdownAsync(); 72 73 int32_t getVideoBitrate() const; 74 void setVideoBitrate(int32_t bitrate); 75 76 static int32_t GetInt32Property(const char *propName, int32_t defaultValue); 77 78 protected: 79 virtual ~Converter(); 80 virtual void onMessageReceived(const sp<AMessage> &msg); 81 82 private: 83 status_t mInitCheck; 84 sp<AMessage> mNotify; 85 sp<ALooper> mCodecLooper; 86 sp<AMessage> mInputFormat; 87 bool mIsVideo; 88 bool mIsPCMAudio; 89 sp<AMessage> mOutputFormat; 90 bool mNeedToManuallyPrependSPSPPS; 91 92 sp<MediaCodec> mEncoder; 93 sp<AMessage> mEncoderActivityNotify; 94 95 Vector<sp<ABuffer> > mEncoderInputBuffers; 96 Vector<sp<ABuffer> > mEncoderOutputBuffers; 97 98 List<size_t> mAvailEncoderInputIndices; 99 100 List<sp<ABuffer> > mInputBufferQueue; 101 102 bool mDoMoreWorkPending; 103 104 #if ENABLE_SILENCE_DETECTION 105 int64_t mFirstSilentFrameUs; 106 bool mInSilentMode; 107 #endif 108 109 sp<ABuffer> mPartialAudioAU; 110 111 int32_t mPrevVideoBitrate; 112 113 int32_t mNumFramesToDrop; 114 115 status_t initEncoder(); 116 void releaseEncoder(); 117 118 status_t feedEncoderInputBuffers(); 119 120 void scheduleDoMoreWork(); 121 status_t doMoreWork(); 122 123 void notifyError(status_t err); 124 125 // Packetizes raw PCM audio data available in mInputBufferQueue 126 // into a format suitable for transport stream inclusion and 127 // notifies the observer. 128 status_t feedRawAudioInputBuffers(); 129 130 static bool IsSilence(const sp<ABuffer> &accessUnit); 131 132 DISALLOW_EVIL_CONSTRUCTORS(Converter); 133 }; 134 135 } // namespace android 136 137 #endif // CONVERTER_H_ 138