Как перевернуть один конкретный бит в цепочке битов Эрланга?

Question

У меня есть следующий модуль:

-module (bitutils).
-export ([get_bits/1, get_bit/2, set_bit/2]).

get_bits(B) ->
    << <<X>> || <<X:1>> <= B >>.

get_bit(Bin, N) ->
    << _:N/bitstring, Bit:1, _/bitstring >> = Bin,
    Bit.

set_bit(Bin, N) ->
    << A:N, _:1, B/bitstring >> = Bin,
    << A:N, 1:1, B >>.

Erlang REPL session (пример):

Eshell V9.3  (abort with ^G)

1> c(bitutils).
{ok,bitutils}

2> B = <<0:3>>.
<<0:3>>

3> bitutils:get_bits(B).
<<0,0,0>>

4> bit_size(B).
3

5> byte_size(B).
1

6> bitutils:get_bit(B, 0).
0

7> bitutils:get_bit(B, 1).
0

8> bitutils:get_bit(B, 2).
0

9> bitutils:set_bit(B, 0).
** exception error: bad argument
     in function  bitutils:setBit/2 (bitutils.erl, line 12)

Вопрос:

Я не могу понять, почему set_bit Функция выдает исключение.

В приведенном выше примере, когда я звоню set_bit(B, 0)Я хочу, чтобы функция возвращала новую цепочку битов (например, NewB) где первый бит (с индексом 0) был переключен с 0 на 1.

Такой, что если бы я позвонил get_bits(NewB), это вернуло бы: <<1,0,0>>

Как я могу перевернуть один конкретный бит в цепочке битов Erlang?

Answer 1

Simpler

set_bit(Bin, N) ->
        << A:N/bits, _:1, B/bits >> = Bin,
        << A/bits, 1:1, B/bits >>.

Answer 2

Вы можете сделать это следующим образом:

set_bit(Bin, N) ->
        R = bit_size(Bin) - N - 1, 
        << A:N, _:1, B:R >> = Bin,
        << A:N, 1:1, B:R >>.