清理无用的二维码代码
This commit is contained in:
parent
d53d5be8d1
commit
f1080cca23
206
face_rec.py
206
face_rec.py
@ -508,176 +508,8 @@ class FaceRecognitionSystem:
|
||||
# Composite
|
||||
image.alpha_composite(shadow_layer)
|
||||
|
||||
def _build_qrcode_instruction_canvas(self, qr_image: np.ndarray, canvas_size: Tuple[int, int]) -> Optional[np.ndarray]:
|
||||
try:
|
||||
canvas_width, canvas_height = canvas_size
|
||||
if canvas_width <= 0 or canvas_height <= 0:
|
||||
return None
|
||||
|
||||
# Colors
|
||||
PRIMARY_COLOR = (23, 43, 77) # Dark Blue for headings
|
||||
SECONDARY_COLOR = (94, 108, 132) # Grey for secondary text
|
||||
ACCENT_COLOR = (0, 82, 204) # Bright Blue for highlights/numbers
|
||||
|
||||
# 1. Background
|
||||
bg = self._create_gradient_background((canvas_width, canvas_height), (245, 247, 250), (223, 225, 230)).convert("RGBA")
|
||||
draw = ImageDraw.Draw(bg, "RGBA")
|
||||
|
||||
# Layout Constants
|
||||
MARGIN = 50
|
||||
GUTTER = 40
|
||||
|
||||
# Left Panel (QR Code) - 35% width approx
|
||||
left_width = int((canvas_width - 2 * MARGIN - GUTTER) * 0.35)
|
||||
# Ensure minimum width for QR code
|
||||
left_width = max(left_width, 400)
|
||||
right_width = canvas_width - 2 * MARGIN - GUTTER - left_width
|
||||
|
||||
left_box = (MARGIN, MARGIN, MARGIN + left_width, canvas_height - MARGIN)
|
||||
right_box = (MARGIN + left_width + GUTTER, MARGIN, canvas_width - MARGIN, canvas_height - MARGIN)
|
||||
|
||||
# --- Draw Left Panel ---
|
||||
# Shadow
|
||||
self._draw_shadow(bg, left_box, radius=30, offset=(0, 10), blur=20, shadow_color=(0,0,0,30))
|
||||
# Card
|
||||
self._draw_rounded_rect(draw, left_box, radius=30, fill=(255, 255, 255, 255))
|
||||
|
||||
# Left Content
|
||||
cx = (left_box[0] + left_box[2]) // 2
|
||||
cy_top = left_box[1] + 120
|
||||
|
||||
# Title
|
||||
text = "访客登记"
|
||||
try:
|
||||
w = draw.textlength(text, font=self.font_qr_title)
|
||||
except:
|
||||
w, _ = draw.textsize(text, font=self.font_qr_title)
|
||||
draw.text((cx - w/2, cy_top), text, font=self.font_qr_title, fill=PRIMARY_COLOR)
|
||||
|
||||
# Subtitle
|
||||
text = "Visitor Registration"
|
||||
try:
|
||||
w = draw.textlength(text, font=self.font_qr_subtitle)
|
||||
except:
|
||||
w, _ = draw.textsize(text, font=self.font_qr_subtitle)
|
||||
draw.text((cx - w/2, cy_top + 70), text, font=self.font_qr_subtitle, fill=SECONDARY_COLOR)
|
||||
|
||||
# QR Code
|
||||
qr_size = min(left_width - 100, 500)
|
||||
qr_y = cy_top + 180
|
||||
|
||||
# Resize QR
|
||||
qr_pil = Image.fromarray(cv2.cvtColor(qr_image, cv2.COLOR_BGR2RGB))
|
||||
qr_pil = qr_pil.resize((qr_size, qr_size), Image.LANCZOS)
|
||||
bg.paste(qr_pil, (cx - qr_size//2, qr_y))
|
||||
|
||||
# Scan Hint
|
||||
hint_y = qr_y + qr_size + 50
|
||||
hint_text = "请使用微信扫码登记"
|
||||
try:
|
||||
w = draw.textlength(hint_text, font=self.font_qr_step_title)
|
||||
except:
|
||||
w, _ = draw.textsize(hint_text, font=self.font_qr_step_title)
|
||||
|
||||
# Draw a pill background for hint
|
||||
pill_padding = 20
|
||||
pill_box = (cx - w/2 - pill_padding, hint_y - pill_padding, cx + w/2 + pill_padding, hint_y + 40 + pill_padding)
|
||||
self._draw_rounded_rect(draw, pill_box, radius=30, fill=(240, 242, 245), outline=None)
|
||||
draw.text((cx - w/2, hint_y), hint_text, font=self.font_qr_step_title, fill=ACCENT_COLOR)
|
||||
|
||||
# --- Draw Right Panel ---
|
||||
|
||||
# Right Title Area - Compacted
|
||||
rt_y = MARGIN + 20
|
||||
draw.text((right_box[0], rt_y), self.qrcode_instruction_title, font=self.font_qr_title, fill=PRIMARY_COLOR)
|
||||
draw.text((right_box[0], rt_y + 60), self.qrcode_instruction_subtitle, font=self.font_qr_subtitle, fill=SECONDARY_COLOR)
|
||||
|
||||
# Separator Line
|
||||
sep_y = rt_y + 110
|
||||
draw.line((right_box[0], sep_y, right_box[2], sep_y), fill=(200, 200, 200), width=2)
|
||||
|
||||
# Grid Configuration
|
||||
grid_y_start = sep_y + 40
|
||||
grid_width = right_width
|
||||
cols = 2
|
||||
col_gap = 30
|
||||
row_gap = 30
|
||||
|
||||
col_width = (grid_width - (cols - 1) * col_gap) // cols
|
||||
|
||||
# Pre-calculate text layout to find uniform height
|
||||
max_lines = 0
|
||||
processed_steps = []
|
||||
|
||||
padding = 30
|
||||
badge_size = 50
|
||||
text_left_margin = badge_size + 20
|
||||
|
||||
# Calculate available width for text inside a card
|
||||
text_max_width = col_width - padding * 2 - text_left_margin
|
||||
|
||||
for i, step in enumerate(self.qrcode_instruction_steps):
|
||||
# Remove "第一步：" etc prefix if present to make it cleaner, we have badges
|
||||
clean_step = step.split("：", 1)[-1] if "：" in step else step
|
||||
|
||||
lines = self._wrap_text_for_width(clean_step, self.font_qr_body, text_max_width, draw)
|
||||
processed_steps.append(lines)
|
||||
max_lines = max(max_lines, len(lines))
|
||||
|
||||
# Calculate Card Height
|
||||
# Padding top + max_lines * line_height + Padding bottom
|
||||
line_height = self.font_qr_body.size + 10
|
||||
card_content_height = max(badge_size, max_lines * line_height)
|
||||
uniform_card_height = int(padding * 2 + card_content_height)
|
||||
|
||||
# Draw Grid
|
||||
for idx, lines in enumerate(processed_steps):
|
||||
row = idx // cols
|
||||
col = idx % cols
|
||||
|
||||
x = right_box[0] + col * (col_width + col_gap)
|
||||
y = grid_y_start + row * (uniform_card_height + row_gap)
|
||||
|
||||
# Ensure we don't go out of bounds
|
||||
if y + uniform_card_height > canvas_height:
|
||||
break
|
||||
|
||||
card_box = (x, y, x + col_width, y + uniform_card_height)
|
||||
|
||||
# Card Shadow
|
||||
self._draw_shadow(bg, card_box, radius=20, offset=(0, 4), blur=12, shadow_color=(0,0,0,15))
|
||||
# Card Body
|
||||
self._draw_rounded_rect(draw, card_box, radius=20, fill=(255, 255, 255))
|
||||
|
||||
# Badge (Step Number)
|
||||
bx = x + padding
|
||||
by = y + padding
|
||||
draw.ellipse((bx, by, bx + badge_size, by + badge_size), fill=ACCENT_COLOR)
|
||||
|
||||
num_text = str(idx + 1)
|
||||
try:
|
||||
nw = draw.textlength(num_text, font=self.font_qr_badge)
|
||||
except:
|
||||
nw, _ = draw.textsize(num_text, font=self.font_qr_badge)
|
||||
# Center number
|
||||
draw.text((bx + (badge_size - nw)/2, by + (badge_size - self.font_qr_badge.size)/2 - 2),
|
||||
num_text, font=self.font_qr_badge, fill=(255, 255, 255))
|
||||
|
||||
# Text
|
||||
tx = x + padding + text_left_margin
|
||||
ty = y + padding
|
||||
|
||||
for line in lines:
|
||||
draw.text((tx, ty), line, font=self.font_qr_body, fill=PRIMARY_COLOR)
|
||||
ty += line_height
|
||||
|
||||
return cv2.cvtColor(np.array(bg.convert("RGB")), cv2.COLOR_RGB2BGR)
|
||||
except Exception as e:
|
||||
self.logger.error(f"渲染二维码指引面板失败: {e}")
|
||||
return None
|
||||
|
||||
def _show_qrcode(self):
|
||||
"""显示二维码图片"""
|
||||
"""显示二维码图片 - 直接全屏铺满"""
|
||||
try:
|
||||
if not os.path.exists(self.qrcode_image_path):
|
||||
self.logger.error(f"二维码图片不存在: {self.qrcode_image_path}")
|
||||
@ -689,48 +521,20 @@ class FaceRecognitionSystem:
|
||||
self.logger.error(f"无法读取二维码图片: {self.qrcode_image_path}")
|
||||
return False
|
||||
|
||||
# 获取二维码图片尺寸
|
||||
qr_height, qr_width = qr_image.shape[:2]
|
||||
|
||||
# 使用屏幕尺寸，按比例缩放图片
|
||||
# 直接铺满全屏
|
||||
canvas_width = self.screen_width
|
||||
canvas_height = self.screen_height
|
||||
|
||||
# 等比例缩放：保持图片宽高比
|
||||
scale = min(canvas_width / qr_width, canvas_height / qr_height)
|
||||
scaled_width = int(qr_width * scale)
|
||||
scaled_height = int(qr_height * scale)
|
||||
|
||||
# 缩放二维码图片
|
||||
qr_image_scaled = cv2.resize(qr_image, (scaled_width, scaled_height), interpolation=cv2.INTER_LINEAR)
|
||||
qr_image_scaled = cv2.resize(qr_image, (canvas_width, canvas_height), interpolation=cv2.INTER_LINEAR)
|
||||
|
||||
# 创建二维码窗口
|
||||
cv2.namedWindow(self.qrcode_window_name, cv2.WINDOW_NORMAL | cv2.WINDOW_GUI_NORMAL)
|
||||
|
||||
x_pos = 0
|
||||
y_pos = 0
|
||||
|
||||
rendered = self._build_qrcode_instruction_canvas(qr_image_scaled, (canvas_width, canvas_height))
|
||||
if rendered is None:
|
||||
# 回退到原二维码展示
|
||||
resized_qr = qr_image_scaled
|
||||
|
||||
# 创建背景画布
|
||||
rendered = np.zeros((canvas_height, canvas_width, 3), dtype=np.uint8)
|
||||
|
||||
# 计算居中位置
|
||||
x_offset = (canvas_width - scaled_width) // 2
|
||||
y_offset = (canvas_height - scaled_height) // 2
|
||||
|
||||
rendered[y_offset:y_offset+scaled_height, x_offset:x_offset+scaled_width] = resized_qr
|
||||
|
||||
cv2.resizeWindow(self.qrcode_window_name, canvas_width, canvas_height)
|
||||
cv2.moveWindow(self.qrcode_window_name, x_pos, y_pos)
|
||||
cv2.moveWindow(self.qrcode_window_name, 0, 0)
|
||||
|
||||
# 设置窗口全屏显示
|
||||
cv2.setWindowProperty(self.qrcode_window_name, cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)
|
||||
|
||||
cv2.imshow(self.qrcode_window_name, rendered)
|
||||
cv2.imshow(self.qrcode_window_name, qr_image_scaled)
|
||||
|
||||
# 设置窗口置顶
|
||||
cv2.setWindowProperty(self.qrcode_window_name, cv2.WND_PROP_TOPMOST, 1)
|
||||
|
||||
@ -1,333 +0,0 @@
|
||||
import math
|
||||
import os
|
||||
import cv2
|
||||
import yaml
|
||||
import numpy as np
|
||||
from PIL import Image, ImageDraw, ImageFont, ImageFilter
|
||||
|
||||
# Canvas Configuration
|
||||
CANVAS_SIZE = (1920, 1080)
|
||||
BG_COLOR_START = (240, 242, 245)
|
||||
BG_COLOR_END = (200, 210, 230)
|
||||
|
||||
# Colors
|
||||
PRIMARY_COLOR = (23, 43, 77) # Dark Blue for headings
|
||||
SECONDARY_COLOR = (94, 108, 132) # Grey for secondary text
|
||||
ACCENT_COLOR = (0, 82, 204) # Bright Blue for highlights/numbers
|
||||
CARD_BG_COLOR = (255, 255, 255)
|
||||
CARD_SHADOW_COLOR = (9, 30, 66, 40) # Shadow color
|
||||
|
||||
INSTRUCTION_TITLE = "访客预约流程"
|
||||
INSTRUCTION_SUBTITLE = "Visitor Registration Process"
|
||||
INSTRUCTION_STEPS = [
|
||||
"第一步：扫码关注“康达新材”公众号。",
|
||||
"第二步：点击【关于我们】→【我是访客】进入“访客注册”界面，填写并上传相应信息并点击“提交”。",
|
||||
"第三步：将第二步信息提交完毕后在“访客预约”界面选择右下角的“+”号按钮，填写预约信息。",
|
||||
"第四步：请仔细阅读安全告知书，点击我知道了。",
|
||||
"第五步：填写被访人信息及来访事由等内容并提交。",
|
||||
"第六步：显示提交成功，并且手机、微信会收到预约相关短信通知。",
|
||||
]
|
||||
|
||||
def load_config(path="config.yaml"):
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
return yaml.safe_load(f)
|
||||
except FileNotFoundError:
|
||||
return {}
|
||||
|
||||
def resolve_font(size: int, preferred: str | None = None) -> ImageFont.FreeTypeFont:
|
||||
candidates = [preferred] if preferred else []
|
||||
candidates += [
|
||||
"C:/Windows/Fonts/msyhbd.ttc", # Microsoft YaHei Bold
|
||||
"C:/Windows/Fonts/msyh.ttc", # Microsoft YaHei
|
||||
"C:/Windows/Fonts/simhei.ttf",
|
||||
"/usr/share/fonts/truetype/noto/NotoSansCJK-Bold.ttc",
|
||||
"/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",
|
||||
"/usr/share/fonts/truetype/noto/NotoMono-Regular.ttf",
|
||||
]
|
||||
for path in candidates:
|
||||
if path and os.path.exists(path):
|
||||
try:
|
||||
return ImageFont.truetype(path, size)
|
||||
except OSError:
|
||||
continue
|
||||
return ImageFont.load_default()
|
||||
|
||||
def wrap_text(text: str, font: ImageFont.FreeTypeFont, max_width: int, draw: ImageDraw.ImageDraw):
|
||||
lines = []
|
||||
current = ""
|
||||
for ch in text:
|
||||
tentative = current + ch
|
||||
# Pillow 10+ uses textlength, older uses textsize
|
||||
try:
|
||||
width = draw.textlength(tentative, font=font)
|
||||
except AttributeError:
|
||||
width, _ = draw.textsize(tentative, font=font)
|
||||
|
||||
if width <= max_width:
|
||||
current = tentative
|
||||
else:
|
||||
if current:
|
||||
lines.append(current)
|
||||
current = ch
|
||||
if current:
|
||||
lines.append(current)
|
||||
return lines
|
||||
|
||||
def create_gradient_background(size: tuple[int, int], start_color: tuple[int, int, int], end_color: tuple[int, int, int]) -> Image.Image:
|
||||
width, height = size
|
||||
# Vertical gradient
|
||||
base = Image.new('RGB', size, start_color)
|
||||
top = Image.new('RGB', size, end_color)
|
||||
mask = Image.new('L', size)
|
||||
mask_data = np.tile(np.linspace(0, 255, height, dtype=np.uint8), (width, 1)).T
|
||||
mask = Image.fromarray(mask_data, 'L')
|
||||
return Image.composite(top, base, mask)
|
||||
|
||||
def draw_rounded_rect(draw, box, radius, fill, outline=None, width=0):
|
||||
draw.rounded_rectangle(box, radius=radius, fill=fill, outline=outline, width=width)
|
||||
|
||||
def draw_shadow(image, box, radius, offset=(0, 4), blur=10, shadow_color=(0,0,0,50)):
|
||||
# Create a separate shadow layer
|
||||
shadow_layer = Image.new('RGBA', image.size, (0,0,0,0))
|
||||
shadow_draw = ImageDraw.Draw(shadow_layer)
|
||||
|
||||
sx0, sy0, sx1, sy1 = box
|
||||
dx, dy = offset
|
||||
|
||||
shadow_box = (sx0 + dx, sy0 + dy, sx1 + dx, sy1 + dy)
|
||||
shadow_draw.rounded_rectangle(shadow_box, radius=radius, fill=shadow_color)
|
||||
|
||||
# Blur the shadow
|
||||
shadow_layer = shadow_layer.filter(ImageFilter.GaussianBlur(blur))
|
||||
|
||||
# Composite
|
||||
image.alpha_composite(shadow_layer)
|
||||
|
||||
def build_canvas(qr_image: np.ndarray, title_font, subtitle_font, body_font, step_title_font, badge_font) -> np.ndarray:
|
||||
canvas_width, canvas_height = CANVAS_SIZE
|
||||
|
||||
# 1. Background
|
||||
bg = create_gradient_background(CANVAS_SIZE, (245, 247, 250), (223, 225, 230)).convert("RGBA")
|
||||
|
||||
draw = ImageDraw.Draw(bg, "RGBA")
|
||||
|
||||
# Layout Constants
|
||||
MARGIN = 50
|
||||
GUTTER = 40
|
||||
|
||||
# Left Panel (QR Code) - 35% width approx
|
||||
left_width = int((canvas_width - 2 * MARGIN - GUTTER) * 0.35)
|
||||
right_width = canvas_width - 2 * MARGIN - GUTTER - left_width
|
||||
|
||||
left_box = (MARGIN, MARGIN, MARGIN + left_width, canvas_height - MARGIN)
|
||||
right_box = (MARGIN + left_width + GUTTER, MARGIN, canvas_width - MARGIN, canvas_height - MARGIN)
|
||||
|
||||
# --- Draw Left Panel ---
|
||||
# Shadow
|
||||
draw_shadow(bg, left_box, radius=30, offset=(0, 10), blur=20, shadow_color=(0,0,0,30))
|
||||
# Card
|
||||
draw_rounded_rect(draw, left_box, radius=30, fill=(255, 255, 255, 255))
|
||||
|
||||
# Left Content
|
||||
cx = (left_box[0] + left_box[2]) // 2
|
||||
cy_top = left_box[1] + 120
|
||||
|
||||
# Title
|
||||
text = "访客登记"
|
||||
try:
|
||||
w = draw.textlength(text, font=title_font)
|
||||
except:
|
||||
w, _ = draw.textsize(text, font=title_font)
|
||||
draw.text((cx - w/2, cy_top), text, font=title_font, fill=PRIMARY_COLOR)
|
||||
|
||||
# Subtitle
|
||||
text = "Visitor Registration"
|
||||
try:
|
||||
w = draw.textlength(text, font=subtitle_font)
|
||||
except:
|
||||
w, _ = draw.textsize(text, font=subtitle_font)
|
||||
draw.text((cx - w/2, cy_top + 70), text, font=subtitle_font, fill=SECONDARY_COLOR)
|
||||
|
||||
# QR Code
|
||||
qr_size = min(left_width - 100, 500)
|
||||
qr_y = cy_top + 180
|
||||
|
||||
# Resize QR
|
||||
qr_pil = Image.fromarray(cv2.cvtColor(qr_image, cv2.COLOR_BGR2RGB))
|
||||
qr_pil = qr_pil.resize((qr_size, qr_size), Image.LANCZOS)
|
||||
bg.paste(qr_pil, (cx - qr_size//2, qr_y))
|
||||
|
||||
# Scan Hint
|
||||
hint_y = qr_y + qr_size + 50
|
||||
hint_text = "请使用微信扫码登记"
|
||||
try:
|
||||
w = draw.textlength(hint_text, font=step_title_font)
|
||||
except:
|
||||
w, _ = draw.textsize(hint_text, font=step_title_font)
|
||||
|
||||
# Draw a pill background for hint
|
||||
pill_padding = 20
|
||||
pill_box = (cx - w/2 - pill_padding, hint_y - pill_padding, cx + w/2 + pill_padding, hint_y + 40 + pill_padding)
|
||||
draw.rounded_rectangle(pill_box, radius=30, fill=(240, 242, 245), outline=None)
|
||||
draw.text((cx - w/2, hint_y), hint_text, font=step_title_font, fill=ACCENT_COLOR)
|
||||
|
||||
# --- Draw Right Panel ---
|
||||
|
||||
# Right Title Area - Compacted
|
||||
rt_y = MARGIN + 20
|
||||
draw.text((right_box[0], rt_y), INSTRUCTION_TITLE, font=title_font, fill=PRIMARY_COLOR)
|
||||
draw.text((right_box[0], rt_y + 60), INSTRUCTION_SUBTITLE, font=subtitle_font, fill=SECONDARY_COLOR)
|
||||
|
||||
# Separator Line
|
||||
sep_y = rt_y + 110
|
||||
draw.line((right_box[0], sep_y, right_box[2], sep_y), fill=(200, 200, 200), width=2)
|
||||
|
||||
# Grid Configuration
|
||||
grid_y_start = sep_y + 40
|
||||
grid_width = right_width
|
||||
cols = 2
|
||||
col_gap = 30
|
||||
row_gap = 30
|
||||
|
||||
col_width = (grid_width - (cols - 1) * col_gap) // cols
|
||||
|
||||
# Pre-calculate text layout to find uniform height
|
||||
max_lines = 0
|
||||
processed_steps = []
|
||||
|
||||
padding = 30
|
||||
badge_size = 50
|
||||
text_left_margin = badge_size + 20
|
||||
|
||||
# Calculate available width for text inside a card
|
||||
text_max_width = col_width - padding * 2 - text_left_margin
|
||||
|
||||
for i, step in enumerate(INSTRUCTION_STEPS):
|
||||
# Remove "第一步：" etc prefix if present to make it cleaner, we have badges
|
||||
clean_step = step.split("：", 1)[-1] if "：" in step else step
|
||||
|
||||
lines = wrap_text(clean_step, body_font, text_max_width, draw)
|
||||
processed_steps.append(lines)
|
||||
max_lines = max(max_lines, len(lines))
|
||||
|
||||
# Calculate Card Height
|
||||
# Padding top + max_lines * line_height + Padding bottom
|
||||
line_height = body_font.size + 10
|
||||
card_content_height = max(badge_size, max_lines * line_height)
|
||||
uniform_card_height = int(padding * 2 + card_content_height)
|
||||
|
||||
# Check if we overflow canvas height
|
||||
total_grid_height = 3 * uniform_card_height + 2 * row_gap
|
||||
if grid_y_start + total_grid_height > canvas_height - MARGIN:
|
||||
print(f"Warning: Content might overflow vertically. Required: {grid_y_start + total_grid_height}, Available: {canvas_height - MARGIN}")
|
||||
# Dynamic adjustment if needed (e.g. reduce gaps) but for now we rely on the font resizing done in main()
|
||||
|
||||
# Draw Grid
|
||||
for idx, lines in enumerate(processed_steps):
|
||||
row = idx // cols
|
||||
col = idx % cols
|
||||
|
||||
x = right_box[0] + col * (col_width + col_gap)
|
||||
y = grid_y_start + row * (uniform_card_height + row_gap)
|
||||
|
||||
card_box = (x, y, x + col_width, y + uniform_card_height)
|
||||
|
||||
# Card Shadow
|
||||
draw_shadow(bg, card_box, radius=20, offset=(0, 4), blur=12, shadow_color=(0,0,0,15))
|
||||
# Card Body
|
||||
draw_rounded_rect(draw, card_box, radius=20, fill=(255, 255, 255))
|
||||
|
||||
# Badge (Step Number)
|
||||
bx = x + padding
|
||||
by = y + padding
|
||||
draw.ellipse((bx, by, bx + badge_size, by + badge_size), fill=ACCENT_COLOR)
|
||||
|
||||
num_text = str(idx + 1)
|
||||
try:
|
||||
nw = draw.textlength(num_text, font=badge_font)
|
||||
except:
|
||||
nw, _ = draw.textsize(num_text, font=badge_font)
|
||||
# Center number
|
||||
draw.text((bx + (badge_size - nw)/2, by + (badge_size - badge_font.size)/2 - 2),
|
||||
num_text, font=badge_font, fill=(255, 255, 255))
|
||||
|
||||
# Text
|
||||
tx = x + padding + text_left_margin
|
||||
ty = y + padding
|
||||
|
||||
for line in lines:
|
||||
draw.text((tx, ty), line, font=body_font, fill=PRIMARY_COLOR)
|
||||
ty += line_height
|
||||
|
||||
return cv2.cvtColor(np.array(bg.convert("RGB")), cv2.COLOR_RGB2BGR)
|
||||
|
||||
def create_mock_qrcode(size: int = 640) -> np.ndarray:
|
||||
img = np.full((size, size, 3), 255, dtype=np.uint8) # White background
|
||||
|
||||
# Draw some patterns
|
||||
block = size // 15
|
||||
for y in range(0, size, block):
|
||||
for x in range(0, size, block):
|
||||
if (x // block + y // block) % 2 == 0:
|
||||
color = (0, 0, 0)
|
||||
# Corner markers
|
||||
if (x < 3*block and y < 3*block) or (x > size-4*block and y < 3*block) or (x < 3*block and y > size-4*block):
|
||||
color = (0, 0, 0)
|
||||
elif np.random.rand() > 0.3:
|
||||
color = (0, 0, 0)
|
||||
else:
|
||||
color = (255, 255, 255)
|
||||
|
||||
cv2.rectangle(img, (x, y), (x + block, y + block), color, -1)
|
||||
|
||||
# Borders for markers
|
||||
marker_len = 3 * block
|
||||
thickness = block
|
||||
# Top Left
|
||||
cv2.rectangle(img, (0,0), (marker_len, marker_len), (0,0,0), thickness)
|
||||
# Top Right
|
||||
cv2.rectangle(img, (size-marker_len,0), (size, marker_len), (0,0,0), thickness)
|
||||
# Bottom Left
|
||||
cv2.rectangle(img, (0,size-marker_len), (marker_len, size), (0,0,0), thickness)
|
||||
|
||||
return img
|
||||
|
||||
def main():
|
||||
config = load_config()
|
||||
preferred_font = config.get("display", {}).get("font_path") if config else None
|
||||
|
||||
# Define Fonts - Adjusted sizes for better fit
|
||||
# Title for left/right headers
|
||||
title_font = resolve_font(56, preferred_font)
|
||||
# Subtitles
|
||||
subtitle_font = resolve_font(30, preferred_font)
|
||||
# Step text - Reduced to ensure fit
|
||||
body_font = resolve_font(24, preferred_font)
|
||||
# Step titles or emphasis
|
||||
step_title_font = resolve_font(28, preferred_font)
|
||||
# Badge numbers
|
||||
badge_font = resolve_font(30, preferred_font)
|
||||
|
||||
# Mock data
|
||||
mock_qr = create_mock_qrcode(500)
|
||||
|
||||
# Build
|
||||
canvas = build_canvas(mock_qr, title_font, subtitle_font, body_font, step_title_font, badge_font)
|
||||
|
||||
# Display
|
||||
window_name = "Visitor Registration Preview"
|
||||
cv2.namedWindow(window_name, cv2.WINDOW_NORMAL)
|
||||
cv2.resizeWindow(window_name, 1280, 720)
|
||||
cv2.imshow(window_name, canvas)
|
||||
|
||||
print("Displaying preview. Press any key to exit.")
|
||||
cv2.waitKey(0)
|
||||
cv2.destroyAllWindows()
|
||||
|
||||
# Save for verification
|
||||
cv2.imwrite("preview_qrcode_sota.png", canvas)
|
||||
print("Saved preview to preview_qrcode_sota.png")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Reference in New Issue
Block a user