audiohook: add directional awareness

Add enum to allow setting optional direction. If set to only one
direction, only feed matching-direction frames to the associated
slin factory.

This prevents mangling the transcoder on non-mixed frames when the
READ and WRITE frames would have otherwise required it.  Also
removes the need to mute or discard the un-wanted frames as they
are no longer added in the first place.

res_stasis_snoop is changed to use this addition to set direction
on audiohook based on spy direction.

If no direction is set, the ast_audiohook_init will init this enum
to BOTH which maintains existing functionality.

ASTERISK-30252

Change-Id: If8716bad334562a5d812be4eeb2a92e4f3be28eb
This commit is contained in:
Mike Bradeen 2022-09-29 14:55:20 -06:00 committed by Friendly Automation
parent b397dc3ca3
commit f61dbd566b
3 changed files with 40 additions and 13 deletions

View File

@ -118,6 +118,7 @@ struct ast_audiohook {
ast_audiohook_manipulate_callback manipulate_callback; /*!< Manipulation callback */
struct ast_audiohook_options options; /*!< Applicable options */
unsigned int hook_internal_samp_rate; /*!< internal read/write sample rate on the audiohook.*/
enum ast_audiohook_direction direction; /*!< Intended audiohook direction, BOTH by default on init */
AST_LIST_ENTRY(ast_audiohook) list; /*!< Linked list information */
};
@ -140,6 +141,14 @@ int ast_audiohook_init(struct ast_audiohook *audiohook, enum ast_audiohook_type
*/
int ast_audiohook_destroy(struct ast_audiohook *audiohook);
/*! \brief Sets direction on audiohook
* \param audiohook
* \param direction In which direction should the audiohook feed frames, ie if we are snooping 'in', set direction to READ so that only the 'in' frames are fed to the slin factory
* \retval 0 on success
* \retval -1 on failure due to audiohook already in use or in shutdown. Can only set direction on NEW audiohooks
*/
int ast_audiohook_set_frame_feed_direction(struct ast_audiohook *audiohook, enum ast_audiohook_direction direction);
/*! \brief Writes a frame into the audiohook structure
* \param audiohook
* \param direction Direction the audio frame came from

View File

@ -109,6 +109,9 @@ int ast_audiohook_init(struct ast_audiohook *audiohook, enum ast_audiohook_type
audiohook->init_flags = init_flags;
/* Set direction to BOTH so that we feed frames in both directions */
audiohook->direction = AST_AUDIOHOOK_DIRECTION_BOTH;
/* initialize internal rate at 8khz, this will adjust if necessary */
audiohook_set_internal_rate(audiohook, DEFAULT_INTERNAL_SAMPLE_RATE, 0);
@ -144,6 +147,18 @@ int ast_audiohook_destroy(struct ast_audiohook *audiohook)
return 0;
}
int ast_audiohook_set_frame_feed_direction(struct ast_audiohook *audiohook, enum ast_audiohook_direction direction)
{
/* Only set the direction on new audiohooks */
if (audiohook->status != AST_AUDIOHOOK_STATUS_NEW) {
ast_debug(3, "Can not set direction on attached Audiohook %p\n", audiohook);
return -1;
}
audiohook->direction = direction;
return 0;
}
#define SHOULD_MUTE(hook, dir) \
((ast_test_flag(hook, AST_AUDIOHOOK_MUTE_READ) && (dir == AST_AUDIOHOOK_DIRECTION_READ)) || \
(ast_test_flag(hook, AST_AUDIOHOOK_MUTE_WRITE) && (dir == AST_AUDIOHOOK_DIRECTION_WRITE)) || \
@ -159,6 +174,13 @@ int ast_audiohook_write_frame(struct ast_audiohook *audiohook, enum ast_audiohoo
int other_factory_samples;
int other_factory_ms;
/* Don't feed the frame if we are set to read and this is a write frame or if set to
write and this is a read frame as we don't want it. Plus, it can cause mis-resampling
if the READ and WRITE frames have different bitrates */
if (audiohook->direction != AST_AUDIOHOOK_DIRECTION_BOTH && audiohook->direction != direction) {
return 0;
}
/* Update last feeding time to be current */
*rwtime = ast_tvnow();

View File

@ -188,21 +188,9 @@ static struct ast_frame *snoop_read(struct ast_channel *chan)
}
ast_audiohook_lock(&snoop->spy);
if (snoop->spy_direction != AST_AUDIOHOOK_DIRECTION_BOTH) {
/*
* When a singular direction is chosen frames are still written to the
* opposing direction's queue. Those frames must be read so the queue
* does not continue to grow, however since they are not needed for the
* selected direction they can be dropped.
*/
enum ast_audiohook_direction opposing_direction =
snoop->spy_direction == AST_AUDIOHOOK_DIRECTION_READ ?
AST_AUDIOHOOK_DIRECTION_WRITE : AST_AUDIOHOOK_DIRECTION_READ;
ast_frame_dtor(ast_audiohook_read_frame(&snoop->spy, snoop->spy_samples,
opposing_direction, snoop->spy_format));
}
frame = ast_audiohook_read_frame(&snoop->spy, snoop->spy_samples, snoop->spy_direction, snoop->spy_format);
ast_audiohook_unlock(&snoop->spy);
return frame ? frame : &snoop->silence;
@ -287,6 +275,14 @@ static int snoop_setup_audiohook(struct ast_channel *chan, enum ast_audiohook_ty
return -1;
}
/* Set the audiohook direction so we don't write unnecessary frames */
if (ast_audiohook_set_frame_feed_direction(audiohook, *direction)) {
/* If we are unable to set direction, the audiohook either failed to init
or someone else started using it already. If we don't bail here, we risk
feeding frames that will never be read */
return -1;
}
return ast_audiohook_attach(chan, audiohook);
}