-module(gleam_synapses@data_preprocessor).
-compile(no_auto_import).

-export([init/2, encoded_datapoint/2, decoded_datapoint/2, to_json/1, of_json/1]).

-spec init(
    list({binary(), boolean()}),
    gleam@iterator:iterator(gleam@map:map_(binary(), binary()))
) -> gleam_zlists@interop:z_list(gleam_synapses@model@encoding@serialization:attribute()).
init(Keys_with_discrete_flags, Datapoints) ->
    Keys_with_flags = gleam_zlists:of_list(Keys_with_discrete_flags),
    Data = gleam_zlists:of_iterator(Datapoints),
    gleam_synapses@model@encoding@preprocessor:init(Keys_with_flags, Data).

-spec encoded_datapoint(
    gleam_zlists@interop:z_list(gleam_synapses@model@encoding@serialization:attribute()),
    gleam@map:map_(binary(), binary())
) -> list(float()).
encoded_datapoint(Data_preprocessor, Datapoint) ->
    gleam_zlists:to_list(
        gleam_synapses@model@encoding@preprocessor:encode(
            Data_preprocessor,
            Datapoint
        )
    ).

-spec decoded_datapoint(
    gleam_zlists@interop:z_list(gleam_synapses@model@encoding@serialization:attribute()),
    list(float())
) -> gleam@map:map_(binary(), binary()).
decoded_datapoint(Data_preprocessor, Encoded_values) ->
    Values = gleam_zlists:of_list(Encoded_values),
    gleam_synapses@model@encoding@preprocessor:decode(Data_preprocessor, Values).

-spec to_json(
    gleam_zlists@interop:z_list(gleam_synapses@model@encoding@serialization:attribute())
) -> binary().
to_json(Data_preprocessor) ->
    gleam_synapses@model@encoding@preprocessor:to_json(Data_preprocessor).

-spec of_json(binary()) -> gleam_zlists@interop:z_list(gleam_synapses@model@encoding@serialization:attribute()).
of_json(Json) ->
    gleam_synapses@model@encoding@preprocessor:of_json(Json).