"""
Copyright 2024 Entropica Labs Pte Ltd
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
from loom.eka import Eka
from loom.interpreter.applicator import CodeApplicator
from .grow import grow
from .shrink import shrink
from .merge import merge
from .split import split
from ..code_factory import RepetitionCode
# pylint: disable=too-few-public-methods
[docs]
class RepetitionCodeApplicator(CodeApplicator):
"""
Contains the implementation logic for each operation, for the Repetition Code.
"""
def __init__(
self,
eka: Eka,
):
# Ensure that all blocks are typed RepetitionCode
if any(not isinstance(block, RepetitionCode) for block in eka.blocks):
raise ValueError("All blocks must be of type RepetitionCode.")
super().__init__(eka)
# Add the extra operations that are supported by the Repetition Code
self.supported_operations |= {
"Grow": grow,
"Shrink": shrink,
"Split": split,
"Merge": merge,
}