Update asymmetric-tiling-sd-webui-3.0/scripts/improved_tiling_functions.py

asymmetric-tiling-sd-webui-3.0/scripts/improved_tiling_functions.py

@@ -328,4 +328,63 @@ def validate_multires_params(params):
         'transition_start': float(params.get('multires_start', 0.0)),
         'transition_end': float(params.get('multires_end', 0.3)),
         'sharpness': float(params.get('multires_sharpness', 1.0))
-    }
+    }
+
+# =======================================================================
+# 4. ВСПОМОГАТЕЛЬНЫЕ ФУНКЦИИ (Возвращены для совместимости с тестами)
+# =======================================================================
+
+def create_circular_mask(h, w, center_x=0.5, center_y=0.5, radius=0.5, device='cpu'):
+    """
+    Создает круговую маску (белый круг на черном фоне).
+    Используется для BlendMode.CIRCULAR_REFLECT и тестов.
+    """
+    # Создаем сетку координат
+    y, x = torch.meshgrid(
+        torch.linspace(-1, 1, h, device=device),
+        torch.linspace(-1, 1, w, device=device),
+        indexing='ij'
+    )
+    
+    # Сдвигаем центр
+    # Координаты от -1 до 1. Если центр 0.5, то смещение 0.
+    # center_x=0.5 -> offset=0.0
+    x = x - (center_x - 0.5) * 2
+    y = y - (center_y - 0.5) * 2
+    
+    # Считаем расстояние от центра
+    dist = torch.sqrt(x*x + y*y)
+    
+    # Создаем мягкую маску (сглаживание краев 0.1)
+    # 1.0 внутри радиуса, 0.0 снаружи
+    mask = 1.0 - torch.clamp((dist - (radius - 0.1)) / 0.2, 0, 1)
+    
+    # Добавляем размерность каналов и батча (1, 1, H, W) для совместимости
+    if len(mask.shape) == 2:
+        mask = mask.unsqueeze(0).unsqueeze(0)
+        
+    return mask
+
+def create_fade_to_black_mask(h, w, strength=0.1, device='cpu'):
+    """
+    Создает виньетку (затемнение к краям).
+    """
+    y, x = torch.meshgrid(
+        torch.linspace(-1, 1, h, device=device),
+        torch.linspace(-1, 1, w, device=device),
+        indexing='ij'
+    )
+    dist = torch.sqrt(x*x + y*y)
+    
+    # Нормализуем, чтобы углы были 1.0 (максимальное расстояние ~1.41)
+    dist = dist / 1.4142
+    
+    # Инвертируем: центр белый (1), края черные (0)
+    # strength регулирует, как быстро начинается затемнение
+    threshold = 1.0 - strength
+    mask = 1.0 - torch.clamp((dist - threshold) / strength, 0, 1)
+    
+    if len(mask.shape) == 2:
+        mask = mask.unsqueeze(0).unsqueeze(0)
+        
+    return mask