Первый запуск
This commit is contained in:
53
Generator/Models/Term.py
Normal file
53
Generator/Models/Term.py
Normal file
@@ -0,0 +1,53 @@
|
||||
from dataclasses import dataclass
|
||||
from typing import List
|
||||
|
||||
from Generator.Models.ConstLiteral import ConstLiteral
|
||||
from Generator.Models.Literal import Literal
|
||||
from Generator.Models.RenderResult import RenderResult
|
||||
from Generator.Models.VariableLiteral import VariableLiteral
|
||||
|
||||
|
||||
@dataclass
|
||||
class Term:
|
||||
literals: List[Literal]
|
||||
separator: str = " "
|
||||
|
||||
def render(self, chanse: float = 1) -> RenderResult:
|
||||
final_text = ""
|
||||
final_spans = []
|
||||
|
||||
for i, literal in enumerate(self.literals):
|
||||
res = literal.render(chanse)
|
||||
|
||||
current_offset = len(final_text)
|
||||
final_text += res.text
|
||||
|
||||
# Сдвигаем координаты с учетом позиции слова в строке
|
||||
for (start, end, label) in res.spans:
|
||||
final_spans.append((current_offset + start, current_offset + end, label))
|
||||
|
||||
# Добавляем разделитель, если это не последнее слово
|
||||
if i < len(self.literals) - 1:
|
||||
final_text += self.separator
|
||||
|
||||
return RenderResult(final_text, final_spans)
|
||||
|
||||
def structure(self) -> RenderResult:
|
||||
final_text = ""
|
||||
final_spans = []
|
||||
|
||||
for i, literal in enumerate(self.literals):
|
||||
res = literal.structure()
|
||||
|
||||
current_offset = len(final_text)
|
||||
final_text += res.text
|
||||
|
||||
# Сдвигаем координаты с учетом позиции слова в строке
|
||||
for (start, end, label) in res.spans:
|
||||
final_spans.append((current_offset + start, current_offset + end, label))
|
||||
|
||||
# Добавляем разделитель, если это не последнее слово
|
||||
if i < len(self.literals) - 1:
|
||||
final_text += self.separator
|
||||
|
||||
return RenderResult(final_text, final_spans)
|
||||
Reference in New Issue
Block a user