Манипулируем деревьями в фокусе
Теперь, когда мы можем перемещаться вверх и вниз, давайте создадим функцию, изменяющую элемент в корне поддерева, на котором фокусируется застёжка.
>modify :: (a –> a) –> Zipper a –> Zipper a
>modify f (Node x l r, bs) = (Node (f x) l r, bs)
>modify f (Empty, bs) = (Empty, bs)
Если мы фокусируемся на узле, мы изменяем его корневой элемент с помощью функции >f. Фокусируясь на пустом дереве, мы оставляем его как есть. Теперь мы можем начать с дерева, перейти куда захотим и изменить элемент, одновременно сохраняя фокус на этом элементе, чтобы можно было легко переместиться далее вверх или вниз. Вот пример:
>ghci> let newFocus = modify (\_ –> 'P') (goRight (goLeft (freeTree, [])))
Мы идём влево, затем вправо, а потом изменяем корневой элемент, заменяя его на >'P'. Если мы используем оператор >–:, это будет читаться ещё лучше:
>ghci> let newFocus = (freeTree, []) –: goLeft –: goRight –: modify (\_ –> 'P')
Затем мы можем перейти вверх, если захотим, и заменить имеющийся там элемент таинственным символом >'X':
>ghci> let newFocus2 = modify (\_ –> 'X') (goUp newFocus)
Либо можем записать это, используя оператор >–: следующим образом:
>ghci> let newFocus2 = newFocus –: goUp –: modify (\_ –> 'X')
Перемещаться вверх просто, потому что «хлебные крошки», которые мы оставляем, формируют часть структуры данных, на которой мы не фокусируемся, но она вывернута наизнанку подобно носку. Вот почему когда мы хотим переместиться вверх, нам не нужно начинать с корня и пробираться вниз. Мы просто берём верхушку нашего вывернутого наизнанку дерева, при этом выворачивая обратно его часть и добавляя её в наш фокус.
Каждый узел имеет два поддерева, даже если эти поддеревья пусты. Поэтому, фокусируясь на пустом поддереве, мы по крайней мере можем сделать одну вещь: заменить его непустым поддеревом, таким образом прикрепляя дерево к листу. Код весьма прост:
>attach :: Tree a –> Zipper a –> Zipper a
>attach t (_, bs) = (t, bs)
Мы берём дерево и застёжку и возвращаем новую застёжку, фокус которой заменён переданным деревом. Можно не только расширять деревья, заменяя пустые поддеревья новыми, но и заменять существующие поддеревья. Давайте прикрепим дерево к дальнему левому краю нашего дерева >freeTree:
>ghci> let farLeft = (freeTree, []) –: goLeft –: goLeft –: goLeft –: goLeft