1 /* Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
2  * Use of this source code is governed by a BSD-style license that can be
3  * found in the LICENSE file.
4  */
5 
6 /*
7  * Remote Stream - An audio steam from/to a client.
8  */
9 #ifndef CRAS_RSTREAM_H_
10 #define CRAS_RSTREAM_H_
11 
12 #include "buffer_share.h"
13 #include "cras_apm_list.h"
14 #include "cras_shm.h"
15 #include "cras_types.h"
16 #include "cras_rstream_config.h"
17 #include "ewma_power.h"
18 
19 struct cras_connect_message;
20 struct cras_rclient;
21 struct dev_mix;
22 
23 /* Holds informations about the main active device.
24  * Members:
25  *    dev_id - id of the main device.
26  *    dev_ptr - pointer to the main device.
27  */
28 struct main_dev_info {
29 	int dev_id;
30 	void *dev_ptr;
31 };
32 
33 /* cras_rstream is used to manage an active audio stream from
34  * a client.  Each client can have any number of open streams for
35  * playing or recording.
36  * Members:
37  *    stream_id - identifier for this stream.
38  *    stream_type - not used.
39  *    client_type - The client type of this stream, like Chrome, ARC++.
40  *    direction - input or output.
41  *    flags - Indicative of what special handling is needed.
42  *    fd - Socket for requesting and sending audio buffer events.
43  *    buffer_frames - Buffer size in frames.
44  *    cb_threshold - Callback client when this much is left.
45  *    main_dev_info - The info of the main device this stream attaches to.
46  *    is_draining - The stream is draining and waiting to be removed.
47  *    client - The client who uses this stream.
48  *    shm - shared memory
49  *    audio_area - space for playback/capture audio
50  *    format - format of the stream
51  *    next_cb_ts - Next callback time for this stream.
52  *    sleep_interval_ts - Time between audio callbacks.
53  *    last_fetch_ts - The time of the last stream fetch.
54  *    longest_fetch_interval_ts - Longest interval between two fetches.
55  *    start_ts - The time when the stream started.
56  *    first_missed_cb_ts - The time when the first missed callback happens.
57  *    buf_state - State of the buffer from all devices for this stream.
58  *    apm_list - List of audio processing module instances.
59  *    ewma - The ewma instance to calculate stream volume.
60  *    num_attached_devs - Number of iodevs this stream has attached to.
61  *    num_missed_cb - Number of callback schedules have been missed.
62  *    queued_frames - Cached value of the number of queued frames in shm.
63  *    is_pinned - True if the stream is a pinned stream, false otherwise.
64  *    pinned_dev_idx - device the stream is pinned, 0 if none.
65  *    triggered - True if already notified TRIGGER_ONLY stream, false otherwise.
66  */
67 struct cras_rstream {
68 	cras_stream_id_t stream_id;
69 	enum CRAS_STREAM_TYPE stream_type;
70 	enum CRAS_CLIENT_TYPE client_type;
71 	enum CRAS_STREAM_DIRECTION direction;
72 	uint32_t flags;
73 	int fd;
74 	size_t buffer_frames;
75 	size_t cb_threshold;
76 	int is_draining;
77 	struct main_dev_info main_dev;
78 	struct cras_rclient *client;
79 	struct cras_audio_shm *shm;
80 	struct cras_audio_area *audio_area;
81 	struct cras_audio_format format;
82 	struct timespec next_cb_ts;
83 	struct timespec sleep_interval_ts;
84 	struct timespec last_fetch_ts;
85 	struct timespec longest_fetch_interval;
86 	struct timespec start_ts;
87 	struct timespec first_missed_cb_ts;
88 	struct buffer_share *buf_state;
89 	struct cras_apm_list *apm_list;
90 	struct ewma_power ewma;
91 	int num_attached_devs;
92 	int num_missed_cb;
93 	int queued_frames;
94 	int is_pinned;
95 	uint32_t pinned_dev_idx;
96 	int triggered;
97 	struct cras_rstream *prev, *next;
98 };
99 
100 /* Creates an rstream.
101  * Args:
102  *    config - Params for configuration of the new rstream. It's a mutable
103  *             borrow.
104  *    stream_out - Filled with the newly created stream pointer.
105  * Returns:
106  *    0 on success, EINVAL if an invalid argument is passed, or ENOMEM if out of
107  *    memory.
108  */
109 int cras_rstream_create(struct cras_rstream_config *config,
110 			struct cras_rstream **stream_out);
111 
112 /* Destroys an rstream. */
113 void cras_rstream_destroy(struct cras_rstream *stream);
114 
115 /* Gets the id of the stream */
116 static inline cras_stream_id_t
cras_rstream_id(const struct cras_rstream * stream)117 cras_rstream_id(const struct cras_rstream *stream)
118 {
119 	return stream->stream_id;
120 }
121 
122 /* Gets the total buffer size in frames for the given client stream. */
123 static inline size_t
cras_rstream_get_buffer_frames(const struct cras_rstream * stream)124 cras_rstream_get_buffer_frames(const struct cras_rstream *stream)
125 {
126 	return stream->buffer_frames;
127 }
128 
129 /* Gets the callback threshold in frames for the given client stream. */
130 static inline size_t
cras_rstream_get_cb_threshold(const struct cras_rstream * stream)131 cras_rstream_get_cb_threshold(const struct cras_rstream *stream)
132 {
133 	return stream->cb_threshold;
134 }
135 
136 /* Gets the max write size for the stream. */
137 static inline size_t
cras_rstream_get_max_write_frames(const struct cras_rstream * stream)138 cras_rstream_get_max_write_frames(const struct cras_rstream *stream)
139 {
140 	if (stream->flags & BULK_AUDIO_OK)
141 		return cras_rstream_get_buffer_frames(stream);
142 	return cras_rstream_get_cb_threshold(stream);
143 }
144 
145 /* Gets the stream type of this stream. */
146 static inline enum CRAS_STREAM_TYPE
cras_rstream_get_type(const struct cras_rstream * stream)147 cras_rstream_get_type(const struct cras_rstream *stream)
148 {
149 	return stream->stream_type;
150 }
151 
152 /* Gets the direction (input/output/loopback) of the stream. */
153 static inline enum CRAS_STREAM_DIRECTION
cras_rstream_get_direction(const struct cras_rstream * stream)154 cras_rstream_get_direction(const struct cras_rstream *stream)
155 {
156 	return stream->direction;
157 }
158 
159 /* Gets the format for the stream. */
cras_rstream_set_format(struct cras_rstream * stream,const struct cras_audio_format * fmt)160 static inline void cras_rstream_set_format(struct cras_rstream *stream,
161 					   const struct cras_audio_format *fmt)
162 {
163 	stream->format = *fmt;
164 }
165 
166 /* Sets the format for the stream. */
cras_rstream_get_format(const struct cras_rstream * stream,struct cras_audio_format * fmt)167 static inline int cras_rstream_get_format(const struct cras_rstream *stream,
168 					  struct cras_audio_format *fmt)
169 {
170 	*fmt = stream->format;
171 	return 0;
172 }
173 
174 /* Gets the fd to be used to poll this client for audio. */
cras_rstream_get_audio_fd(const struct cras_rstream * stream)175 static inline int cras_rstream_get_audio_fd(const struct cras_rstream *stream)
176 {
177 	return stream->fd;
178 }
179 
180 /* Gets the is_draning flag. */
181 static inline int
cras_rstream_get_is_draining(const struct cras_rstream * stream)182 cras_rstream_get_is_draining(const struct cras_rstream *stream)
183 {
184 	return stream->is_draining;
185 }
186 
187 /* Sets the is_draning flag. */
cras_rstream_set_is_draining(struct cras_rstream * stream,int is_draining)188 static inline void cras_rstream_set_is_draining(struct cras_rstream *stream,
189 						int is_draining)
190 {
191 	stream->is_draining = is_draining;
192 }
193 
194 /* Gets the shm fds used for the stream shm */
cras_rstream_get_shm_fds(const struct cras_rstream * stream,int * header_fd,int * samples_fd)195 static inline int cras_rstream_get_shm_fds(const struct cras_rstream *stream,
196 					   int *header_fd, int *samples_fd)
197 {
198 	if (!header_fd || !samples_fd)
199 		return -EINVAL;
200 
201 	*header_fd = stream->shm->header_info.fd;
202 	*samples_fd = stream->shm->samples_info.fd;
203 
204 	return 0;
205 }
206 
207 /* Gets the size of the shm area used for samples for this stream. */
208 static inline size_t
cras_rstream_get_samples_shm_size(const struct cras_rstream * stream)209 cras_rstream_get_samples_shm_size(const struct cras_rstream *stream)
210 {
211 	return cras_shm_samples_size(stream->shm);
212 }
213 
214 /* Gets shared memory region for this stream. */
215 static inline struct cras_audio_shm *
cras_rstream_shm(struct cras_rstream * stream)216 cras_rstream_shm(struct cras_rstream *stream)
217 {
218 	return stream->shm;
219 }
220 
221 /* Checks if the stream uses an output device. */
stream_uses_output(const struct cras_rstream * s)222 static inline int stream_uses_output(const struct cras_rstream *s)
223 {
224 	return cras_stream_uses_output_hw(s->direction);
225 }
226 
227 /* Checks if the stream uses an input device. */
stream_uses_input(const struct cras_rstream * s)228 static inline int stream_uses_input(const struct cras_rstream *s)
229 {
230 	return cras_stream_uses_input_hw(s->direction);
231 }
232 
stream_is_server_only(const struct cras_rstream * s)233 static inline int stream_is_server_only(const struct cras_rstream *s)
234 {
235 	return s->flags & SERVER_ONLY;
236 }
237 
238 /* Gets the enabled effects of this stream. */
239 unsigned int cras_rstream_get_effects(const struct cras_rstream *stream);
240 
241 /* Gets the format of data after stream specific processing. */
242 struct cras_audio_format *
243 cras_rstream_post_processing_format(const struct cras_rstream *stream,
244 				    void *dev_ptr);
245 
246 /* Checks how much time has passed since last stream fetch and records
247  * the longest fetch interval. */
248 void cras_rstream_record_fetch_interval(struct cras_rstream *rstream,
249 					const struct timespec *now);
250 
251 /* Requests min_req frames from the client. */
252 int cras_rstream_request_audio(struct cras_rstream *stream,
253 			       const struct timespec *now);
254 
255 /* Tells a capture client that count frames are ready. */
256 int cras_rstream_audio_ready(struct cras_rstream *stream, size_t count);
257 
258 /* Let the rstream know when a device is added or removed. */
259 void cras_rstream_dev_attach(struct cras_rstream *rstream, unsigned int dev_id,
260 			     void *dev_ptr);
261 void cras_rstream_dev_detach(struct cras_rstream *rstream, unsigned int dev_id);
262 
cras_rstream_dev_ptr(struct cras_rstream * rstream,unsigned int dev_id)263 static inline void *cras_rstream_dev_ptr(struct cras_rstream *rstream,
264 					 unsigned int dev_id)
265 {
266 	return buffer_share_get_data(rstream->buf_state, dev_id);
267 }
268 
269 /* A device using this stream has read or written samples. */
270 void cras_rstream_dev_offset_update(struct cras_rstream *rstream,
271 				    unsigned int frames, unsigned int dev_id);
272 
273 void cras_rstream_update_input_write_pointer(struct cras_rstream *rstream);
274 void cras_rstream_update_output_read_pointer(struct cras_rstream *rstream);
275 
276 unsigned int cras_rstream_dev_offset(const struct cras_rstream *rstream,
277 				     unsigned int dev_id);
278 
cras_rstream_level(struct cras_rstream * rstream)279 static inline unsigned int cras_rstream_level(struct cras_rstream *rstream)
280 {
281 	const struct cras_audio_shm *shm = cras_rstream_shm(rstream);
282 	return cras_shm_frames_written(shm);
283 }
284 
cras_rstream_input_level_met(struct cras_rstream * rstream)285 static inline int cras_rstream_input_level_met(struct cras_rstream *rstream)
286 {
287 	const struct cras_audio_shm *shm = cras_rstream_shm(rstream);
288 	return cras_shm_frames_written(shm) >= rstream->cb_threshold;
289 }
290 
291 /* Updates the number of queued frames in shm. The queued frames should be
292  * updated everytime before calling cras_rstream_playable_frames.
293  */
294 void cras_rstream_update_queued_frames(struct cras_rstream *rstream);
295 
296 /* Returns the number of playable samples in shm for the given device id. */
297 unsigned int cras_rstream_playable_frames(struct cras_rstream *rstream,
298 					  unsigned int dev_id);
299 
300 /* Returns the volume scaler for this stream. */
301 float cras_rstream_get_volume_scaler(struct cras_rstream *rstream);
302 
303 /* Returns a pointer to readable frames, fills frames with the number of frames
304  * available. */
305 uint8_t *cras_rstream_get_readable_frames(struct cras_rstream *rstream,
306 					  unsigned int offset, size_t *frames);
307 
308 /* Returns non-zero if the stream is muted. */
309 int cras_rstream_get_mute(const struct cras_rstream *rstream);
310 
311 /*
312  * Returns non-zero if the stream is pending a reply from client.
313  * - For playback, stream is waiting for AUDIO_MESSAGE_DATA_READY message from
314  *   client.
315  * - For capture, stream is waiting for AUDIO_MESSAGE_DATA_CAPTURED message
316  *   from client.
317  */
318 int cras_rstream_is_pending_reply(const struct cras_rstream *stream);
319 
320 /*
321  * Reads any pending audio message from the socket.
322  */
323 int cras_rstream_flush_old_audio_messages(struct cras_rstream *stream);
324 
325 #endif /* CRAS_RSTREAM_H_ */
326