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#ifndef STAGEFRIGHT_RECORDER_H_
18
19#define STAGEFRIGHT_RECORDER_H_
20
21#include <media/MediaAnalyticsItem.h>
22#include <media/MediaRecorderBase.h>
23#include <camera/CameraParameters.h>
24#include <utils/String8.h>
25
26#include <system/audio.h>
27
28#include <MetadataBufferType.h>
29
30namespace android {
31
32class Camera;
33class ICameraRecordingProxy;
34class CameraSource;
35class CameraSourceTimeLapse;
36struct MediaCodecSource;
37struct MediaSource;
38struct MediaWriter;
39class MetaData;
40struct AudioSource;
41class MediaProfiles;
42struct ALooper;
43
44struct StagefrightRecorder : public MediaRecorderBase {
45    explicit StagefrightRecorder(const String16 &opPackageName);
46    virtual ~StagefrightRecorder();
47    virtual status_t init();
48    virtual status_t setAudioSource(audio_source_t as);
49    virtual status_t setVideoSource(video_source vs);
50    virtual status_t setOutputFormat(output_format of);
51    virtual status_t setAudioEncoder(audio_encoder ae);
52    virtual status_t setVideoEncoder(video_encoder ve);
53    virtual status_t setVideoSize(int width, int height);
54    virtual status_t setVideoFrameRate(int frames_per_second);
55    virtual status_t setCamera(const sp<hardware::ICamera>& camera, const sp<ICameraRecordingProxy>& proxy);
56    virtual status_t setPreviewSurface(const sp<IGraphicBufferProducer>& surface);
57    virtual status_t setInputSurface(const sp<PersistentSurface>& surface);
58    virtual status_t setOutputFile(int fd);
59    virtual status_t setNextOutputFile(int fd);
60    virtual status_t setParameters(const String8 &params);
61    virtual status_t setListener(const sp<IMediaRecorderClient> &listener);
62    virtual status_t setClientName(const String16 &clientName);
63    virtual status_t prepare();
64    virtual status_t start();
65    virtual status_t pause();
66    virtual status_t resume();
67    virtual status_t stop();
68    virtual status_t close();
69    virtual status_t reset();
70    virtual status_t getMaxAmplitude(int *max);
71    virtual status_t getMetrics(Parcel *reply);
72    virtual status_t dump(int fd, const Vector<String16> &args) const;
73    // Querying a SurfaceMediaSourcer
74    virtual sp<IGraphicBufferProducer> querySurfaceMediaSource() const;
75
76private:
77    mutable Mutex mLock;
78    sp<hardware::ICamera> mCamera;
79    sp<ICameraRecordingProxy> mCameraProxy;
80    sp<IGraphicBufferProducer> mPreviewSurface;
81    sp<PersistentSurface> mPersistentSurface;
82    sp<IMediaRecorderClient> mListener;
83    String16 mClientName;
84    uid_t mClientUid;
85    pid_t mClientPid;
86    sp<MediaWriter> mWriter;
87    int mOutputFd;
88    sp<AudioSource> mAudioSourceNode;
89
90    MediaAnalyticsItem *mAnalyticsItem;
91    bool mAnalyticsDirty;
92    void resetMetrics();
93    void updateMetrics();
94
95    audio_source_t mAudioSource;
96    video_source mVideoSource;
97    output_format mOutputFormat;
98    audio_encoder mAudioEncoder;
99    video_encoder mVideoEncoder;
100    bool mUse64BitFileOffset;
101    int32_t mVideoWidth, mVideoHeight;
102    int32_t mFrameRate;
103    int32_t mVideoBitRate;
104    int32_t mAudioBitRate;
105    int32_t mAudioChannels;
106    int32_t mSampleRate;
107    int32_t mInterleaveDurationUs;
108    int32_t mIFramesIntervalSec;
109    int32_t mCameraId;
110    int32_t mVideoEncoderProfile;
111    int32_t mVideoEncoderLevel;
112    int32_t mMovieTimeScale;
113    int32_t mVideoTimeScale;
114    int32_t mAudioTimeScale;
115    int64_t mMaxFileSizeBytes;
116    int64_t mMaxFileDurationUs;
117    int64_t mTrackEveryTimeDurationUs;
118    int32_t mRotationDegrees;  // Clockwise
119    int32_t mLatitudex10000;
120    int32_t mLongitudex10000;
121    int32_t mStartTimeOffsetMs;
122    int32_t mTotalBitRate;
123
124    bool mCaptureFpsEnable;
125    double mCaptureFps;
126    int64_t mTimeBetweenCaptureUs;
127    sp<CameraSourceTimeLapse> mCameraSourceTimeLapse;
128
129    String8 mParams;
130
131    MetadataBufferType mMetaDataStoredInVideoBuffers;
132    MediaProfiles *mEncoderProfiles;
133
134    int64_t mPauseStartTimeUs;
135    int64_t mTotalPausedDurationUs;
136    sp<MediaCodecSource> mAudioEncoderSource;
137    sp<MediaCodecSource> mVideoEncoderSource;
138
139    bool mStarted;
140    // Needed when GLFrames are encoded.
141    // An <IGraphicBufferProducer> pointer
142    // will be sent to the client side using which the
143    // frame buffers will be queued and dequeued
144    sp<IGraphicBufferProducer> mGraphicBufferProducer;
145    sp<ALooper> mLooper;
146
147    static const int kMaxHighSpeedFps = 1000;
148
149    status_t prepareInternal();
150    status_t setupMPEG4orWEBMRecording();
151    void setupMPEG4orWEBMMetaData(sp<MetaData> *meta);
152    status_t setupAMRRecording();
153    status_t setupAACRecording();
154    status_t setupRawAudioRecording();
155    status_t setupRTPRecording();
156    status_t setupMPEG2TSRecording();
157    sp<MediaCodecSource> createAudioSource();
158    status_t checkVideoEncoderCapabilities();
159    status_t checkAudioEncoderCapabilities();
160    // Generic MediaSource set-up. Returns the appropriate
161    // source (CameraSource or SurfaceMediaSource)
162    // depending on the videosource type
163    status_t setupMediaSource(sp<MediaSource> *mediaSource);
164    status_t setupCameraSource(sp<CameraSource> *cameraSource);
165    status_t setupAudioEncoder(const sp<MediaWriter>& writer);
166    status_t setupVideoEncoder(const sp<MediaSource>& cameraSource, sp<MediaCodecSource> *source);
167
168    // Encoding parameter handling utilities
169    status_t setParameter(const String8 &key, const String8 &value);
170    status_t setParamAudioEncodingBitRate(int32_t bitRate);
171    status_t setParamAudioNumberOfChannels(int32_t channles);
172    status_t setParamAudioSamplingRate(int32_t sampleRate);
173    status_t setParamAudioTimeScale(int32_t timeScale);
174    status_t setParamCaptureFpsEnable(int32_t timeLapseEnable);
175    status_t setParamCaptureFps(double fps);
176    status_t setParamVideoEncodingBitRate(int32_t bitRate);
177    status_t setParamVideoIFramesInterval(int32_t seconds);
178    status_t setParamVideoEncoderProfile(int32_t profile);
179    status_t setParamVideoEncoderLevel(int32_t level);
180    status_t setParamVideoCameraId(int32_t cameraId);
181    status_t setParamVideoTimeScale(int32_t timeScale);
182    status_t setParamVideoRotation(int32_t degrees);
183    status_t setParamTrackTimeStatus(int64_t timeDurationUs);
184    status_t setParamInterleaveDuration(int32_t durationUs);
185    status_t setParam64BitFileOffset(bool use64BitFileOffset);
186    status_t setParamMaxFileDurationUs(int64_t timeUs);
187    status_t setParamMaxFileSizeBytes(int64_t bytes);
188    status_t setParamMovieTimeScale(int32_t timeScale);
189    status_t setParamGeoDataLongitude(int64_t longitudex10000);
190    status_t setParamGeoDataLatitude(int64_t latitudex10000);
191    void clipVideoBitRate();
192    void clipVideoFrameRate();
193    void clipVideoFrameWidth();
194    void clipVideoFrameHeight();
195    void clipAudioBitRate();
196    void clipAudioSampleRate();
197    void clipNumberOfAudioChannels();
198    void setDefaultProfileIfNecessary();
199    void setDefaultVideoEncoderIfNecessary();
200
201
202    StagefrightRecorder(const StagefrightRecorder &);
203    StagefrightRecorder &operator=(const StagefrightRecorder &);
204};
205
206}  // namespace android
207
208#endif  // STAGEFRIGHT_RECORDER_H_
209