generated from membraneframework/membrane_template_plugin
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Adjust to incorporating Funnel into membrane_core (#17)
* Adjust to incorporating Funnel into membrane_core * Implement review comments
- Loading branch information
1 parent
5e97caf
commit 5fad92d
Showing
4 changed files
with
68 additions
and
59 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,65 +1,70 @@ | ||
defmodule Membrane.Funnel do | ||
@moduledoc """ | ||
Element that can be used for collecting data from multiple inputs and sending it through one | ||
output. When a new input connects in the `:playing` state, the funnel sends | ||
`Membrane.Funnel.NewInputEvent` via output. | ||
""" | ||
use Membrane.Filter | ||
module_name = Membrane.Funnel | ||
|
||
alias Membrane.Funnel | ||
if not Code.ensure_loaded?(module_name) do | ||
defmodule module_name do | ||
@moduledoc """ | ||
Element that can be used for collecting data from multiple inputs and sending it through one | ||
output. When a new input connects in the `:playing` state, the funnel sends | ||
`Membrane.Funnel.NewInputEvent` via output. | ||
""" | ||
use Membrane.Filter | ||
|
||
def_input_pad :input, accepted_format: _any, flow_control: :auto, availability: :on_request | ||
def_output_pad :output, accepted_format: _any, flow_control: :auto | ||
def_input_pad :input, accepted_format: _any, flow_control: :auto, availability: :on_request | ||
def_output_pad :output, accepted_format: _any, flow_control: :auto | ||
|
||
def_options end_of_stream: [spec: :on_last_pad | :on_first_pad | :never, default: :on_last_pad] | ||
def_options end_of_stream: [ | ||
spec: :on_last_pad | :on_first_pad | :never, | ||
default: :on_last_pad | ||
] | ||
|
||
@impl true | ||
def handle_init(_ctx, opts) do | ||
{[], %{end_of_stream: opts.end_of_stream}} | ||
end | ||
|
||
@impl true | ||
def handle_buffer(Pad.ref(:input, _id), buffer, _ctx, state) do | ||
{[buffer: {:output, buffer}], state} | ||
end | ||
|
||
@impl true | ||
def handle_pad_added(Pad.ref(:input, _id), %{playback_state: :playing}, state) do | ||
{[event: {:output, %Funnel.NewInputEvent{}}], state} | ||
end | ||
@impl true | ||
def handle_init(_ctx, opts) do | ||
{[], %{end_of_stream: opts.end_of_stream}} | ||
end | ||
|
||
@impl true | ||
def handle_pad_added(Pad.ref(:input, _id), _ctx, state) do | ||
{[], state} | ||
end | ||
@impl true | ||
def handle_buffer(Pad.ref(:input, _id), buffer, _ctx, state) do | ||
{[buffer: {:output, buffer}], state} | ||
end | ||
|
||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), _ctx, %{end_of_stream: :never} = state) do | ||
{[], state} | ||
end | ||
@impl true | ||
def handle_pad_added(Pad.ref(:input, _id), %{playback_state: :playing}, state) do | ||
{[event: {:output, %__MODULE__.NewInputEvent{}}], state} | ||
end | ||
|
||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), ctx, %{end_of_stream: :on_first_pad} = state) do | ||
if ctx.pads.output.end_of_stream? do | ||
@impl true | ||
def handle_pad_added(Pad.ref(:input, _id), _ctx, state) do | ||
{[], state} | ||
else | ||
{[end_of_stream: :output], state} | ||
end | ||
end | ||
|
||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), ctx, %{end_of_stream: :on_last_pad} = state) do | ||
if ctx |> inputs_data() |> Enum.all?(& &1.end_of_stream?) do | ||
{[end_of_stream: :output], state} | ||
else | ||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), _ctx, %{end_of_stream: :never} = state) do | ||
{[], state} | ||
end | ||
end | ||
|
||
defp inputs_data(ctx) do | ||
Enum.flat_map(ctx.pads, fn | ||
{Pad.ref(:input, _id), data} -> [data] | ||
_output -> [] | ||
end) | ||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), ctx, %{end_of_stream: :on_first_pad} = state) do | ||
if ctx.pads.output.end_of_stream? do | ||
{[], state} | ||
else | ||
{[end_of_stream: :output], state} | ||
end | ||
end | ||
|
||
@impl true | ||
def handle_end_of_stream(Pad.ref(:input, _id), ctx, %{end_of_stream: :on_last_pad} = state) do | ||
if ctx |> inputs_data() |> Enum.all?(& &1.end_of_stream?) do | ||
{[end_of_stream: :output], state} | ||
else | ||
{[], state} | ||
end | ||
end | ||
|
||
defp inputs_data(ctx) do | ||
Enum.flat_map(ctx.pads, fn | ||
{Pad.ref(:input, _id), data} -> [data] | ||
_output -> [] | ||
end) | ||
end | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,9 +1,13 @@ | ||
defmodule Membrane.Funnel.NewInputEvent do | ||
@moduledoc """ | ||
Event sent each time new element is linked (via funnel input pad) after playing pipeline. | ||
""" | ||
@derive Membrane.EventProtocol | ||
module_name = Membrane.Funnel.NewInputEvent | ||
|
||
@type t :: %__MODULE__{} | ||
defstruct [] | ||
if not Code.ensure_loaded?(module_name) do | ||
defmodule module_name do | ||
@moduledoc """ | ||
Event sent each time new element is linked (via funnel input pad) after playing pipeline. | ||
""" | ||
@derive Membrane.EventProtocol | ||
|
||
@type t :: %__MODULE__{} | ||
defstruct [] | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters