PlaybackSession.h revision 2aea9552aeba92bbaf9e56c666049ea2d14057b5 
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 PLAYBACK_SESSION_H_
18
19#define PLAYBACK_SESSION_H_
20
21#include "MediaSender.h"
22#include "VideoFormats.h"
23#include "WifiDisplaySource.h"
24
25namespace android {
26
27struct ABuffer;
28struct BufferQueue;
29struct IHDCP;
30struct IGraphicBufferProducer;
31struct MediaPuller;
32struct MediaSource;
33struct MediaSender;
34struct NuMediaExtractor;
35
36// Encapsulates the state of an RTP/RTCP session in the context of wifi
37// display.
38struct WifiDisplaySource::PlaybackSession : public AHandler {
39    PlaybackSession(
40            const sp<ANetworkSession> &netSession,
41            const sp<AMessage> &notify,
42            const struct in_addr &interfaceAddr,
43            const sp<IHDCP> &hdcp,
44            const char *path = NULL);
45
46    status_t init(
47            const char *clientIP,
48            int32_t clientRtp,
49            RTPSender::TransportMode rtpMode,
50            int32_t clientRtcp,
51            RTPSender::TransportMode rtcpMode,
52            bool enableAudio,
53            bool usePCMAudio,
54            bool enableVideo,
55            VideoFormats::ResolutionType videoResolutionType,
56            size_t videoResolutionIndex);
57
58    void destroyAsync();
59
60    int32_t getRTPPort() const;
61
62    int64_t getLastLifesignUs() const;
63    void updateLiveness();
64
65    status_t play();
66    status_t finishPlay();
67    status_t pause();
68
69    sp<IGraphicBufferProducer> getSurfaceTexture();
70
71    void requestIDRFrame();
72
73    enum {
74        kWhatSessionDead,
75        kWhatBinaryData,
76        kWhatSessionEstablished,
77        kWhatSessionDestroyed,
78    };
79
80protected:
81    virtual void onMessageReceived(const sp<AMessage> &msg);
82    virtual ~PlaybackSession();
83
84private:
85    struct Track;
86
87    enum {
88        kWhatMediaPullerNotify,
89        kWhatConverterNotify,
90        kWhatTrackNotify,
91        kWhatUpdateSurface,
92        kWhatPause,
93        kWhatResume,
94        kWhatMediaSenderNotify,
95        kWhatPullExtractorSample,
96    };
97
98    sp<ANetworkSession> mNetSession;
99    sp<AMessage> mNotify;
100    in_addr mInterfaceAddr;
101    sp<IHDCP> mHDCP;
102    AString mMediaPath;
103
104    sp<MediaSender> mMediaSender;
105    int32_t mLocalRTPPort;
106
107    bool mWeAreDead;
108    bool mPaused;
109
110    int64_t mLastLifesignUs;
111
112    sp<BufferQueue> mBufferQueue;
113
114    KeyedVector<size_t, sp<Track> > mTracks;
115    ssize_t mVideoTrackIndex;
116
117    int64_t mPrevTimeUs;
118
119    sp<NuMediaExtractor> mExtractor;
120    KeyedVector<size_t, size_t> mExtractorTrackToInternalTrack;
121    bool mPullExtractorPending;
122    int32_t mPullExtractorGeneration;
123    int64_t mFirstSampleTimeRealUs;
124    int64_t mFirstSampleTimeUs;
125
126    status_t setupMediaPacketizer(bool enableAudio, bool enableVideo);
127
128    status_t setupPacketizer(
129            bool enableAudio,
130            bool usePCMAudio,
131            bool enableVideo,
132            VideoFormats::ResolutionType videoResolutionType,
133            size_t videoResolutionIndex);
134
135    status_t addSource(
136            bool isVideo,
137            const sp<MediaSource> &source,
138            bool isRepeaterSource,
139            bool usePCMAudio,
140            size_t *numInputBuffers);
141
142    status_t addVideoSource(
143            VideoFormats::ResolutionType videoResolutionType,
144            size_t videoResolutionIndex);
145
146    status_t addAudioSource(bool usePCMAudio);
147
148    status_t onMediaSenderInitialized();
149
150    void notifySessionDead();
151
152    void schedulePullExtractor();
153    void onPullExtractor();
154
155    void onSinkFeedback(const sp<AMessage> &msg);
156
157    DISALLOW_EVIL_CONSTRUCTORS(PlaybackSession);
158};
159
160}  // namespace android
161
162#endif  // PLAYBACK_SESSION_H_
163
164