# -*- coding: utf-8 -*- # ! /usr/bin/env python3 # @Author : yangzihao # @CreateTime : 2024/04/17 """ This file is for PDF report generate. Refer to the CSDN: https://blog.csdn.net/cainiao_python/article/details/128928298 """ from reportlab.pdfbase import pdfmetrics # 注册字体 from reportlab.pdfbase.ttfonts import TTFont # 字体类 from reportlab.platypus import Table, SimpleDocTemplate, Paragraph, Image # 报告内容相关类 from reportlab.lib.pagesizes import A4 # 页面的标志尺寸(8.5*inch, 11*inch) from reportlab.lib.styles import getSampleStyleSheet # 文本样式 from reportlab.lib import colors # 颜色模块 from reportlab.graphics.charts.barcharts import VerticalBarChart # 图表类 from reportlab.graphics.charts.legends import Legend # 图例类 from reportlab.graphics.shapes import Drawing # 绘图工具 from reportlab.lib.units import cm # 单位：cm from reportlab.lib import utils import json import pandas as pd # 注册字体(提前准备好字体文件, 如果同一个文件需要多种字体可以注册多个) pdfmetrics.registerFont(TTFont('SimSun', 'SimSun.ttf')) class ReportPDF: """ This class is for PDF report generate. """ # 绘制标题 @staticmethod def draw_title(title: str): # 获取所有样式表 style = getSampleStyleSheet() # 拿到标题样式 ct = style['Heading4'] # 单独设置样式相关属性 ct.fontName = 'SimSun' # 字体名 ct.fontSize = 18 # 字体大小 ct.leading = 30 # 行间距 ct.textColor = colors.black # 字体颜色 ct.alignment = 1 # 居中 ct.bold = True # 创建标题对应的段落，并且返回 return Paragraph(title, ct) # 绘制小标题 @staticmethod def draw_little_title(title: str): # 获取所有样式表 style = getSampleStyleSheet() # 拿到标题样式 ct = style['Normal'] # 单独设置样式相关属性 ct.fontName = 'SimSun' # 字体名 ct.fontSize = 15 # 字体大小 # ct.leading = 15 # 行间距 ct.textColor = colors.darkblue # 字体颜色 # 创建标题对应的段落，并且返回 return Paragraph(title, ct) # 绘制普通段落内容 @staticmethod def draw_text(text: str): # 获取所有样式表 style = getSampleStyleSheet() # 获取普通样式 ct = style['Normal'] ct.fontName = 'SimSun' ct.fontSize = 10 # ct.wordWrap = 'CJK' # 设置自动换行 ct.alignment = 0 # 左对齐 # ct.firstLineIndent = 32 # 第一行开头空格 ct.leading = 15 return Paragraph(text, ct) # 绘制普通段落内容 @staticmethod def draw_line_feed(text: str): # 获取所有样式表 style = getSampleStyleSheet() # 获取普通样式 ct = style['Normal'] ct.fontName = 'SimSun' ct.fontSize = 10 ct.textColor = colors.white # ct.wordWrap = 'CJK' # 设置自动换行 ct.alignment = 0 # 左对齐 # ct.firstLineIndent = 32 # 第一行开头空格 ct.leading = 15 return Paragraph(text, ct) # 绘制表格 @staticmethod def draw_table(*args): # 列宽度 # col_width = 120 style = [ ('FONTNAME', (0, 0), (-1, -1), 'SimSun'), # 字体 ('FONTSIZE', (0, 0), (-1, 0), 9), # 第一行的字体大小 ('FONTSIZE', (0, 1), (-1, -1), 9), # 第二行到最后一行的字体大小 ('BACKGROUND', (0, 0), (-1, 0), '#d5dae6'), # 设置第一行背景颜色 ('ALIGN', (0, 0), (-1, -1), 'CENTER'), # 第一行水平居中 ('ALIGN', (0, 1), (-1, -1), 'CENTER'), # 第二行到最后一行左右左对齐 ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'), # 所有表格上下居中对齐 ('TEXTCOLOR', (0, 0), (-1, -1), colors.darkslategray), # 设置表格内文字颜色 ('GRID', (0, 0), (-1, -1), 0.5, colors.grey), # 设置表格框线为grey色，线宽为0.5 # ('SPAN', (0, 1), (0, 2)), # 合并第一列二三行 # ('SPAN', (0, 3), (0, 4)), # 合并第一列三四行 # ('SPAN', (0, 5), (0, 6)), # 合并第一列五六行 # ('SPAN', (0, 7), (0, 8)), # 合并第一列五六行 ] # table = Table(args, colWidths=col_width, style=style) table = Table(args, style=style) return table # 创建图表 @staticmethod def draw_bar(bar_data: list, ax: list, items: list): drawing = Drawing(500, 250) bc = VerticalBarChart() bc.x = 45 # 整个图表的x坐标 bc.y = 45 # 整个图表的y坐标 bc.height = 200 # 图表的高度 bc.width = 350 # 图表的宽度 bc.data = bar_data bc.strokeColor = colors.black # 顶部和右边轴线的颜色 bc.valueAxis.valueMin = 5000 # 设置y坐标的最小值 bc.valueAxis.valueMax = 26000 # 设置y坐标的最大值 bc.valueAxis.valueStep = 2000 # 设置y坐标的步长 bc.categoryAxis.labels.dx = 2 bc.categoryAxis.labels.dy = -8 bc.categoryAxis.labels.angle = 20 bc.categoryAxis.categoryNames = ax # 图示 leg = Legend() leg.fontName = 'SimSun' leg.alignment = 'right' leg.boxAnchor = 'ne' leg.x = 475 # 图例的x坐标 leg.y = 240 leg.dxTextSpace = 10 leg.columnMaximum = 3 leg.colorNamePairs = items drawing.add(leg) drawing.add(bc) return drawing # 绘制图片 @staticmethod def draw_img(path): img = Image(path) # 读取指定路径下的图片 img.drawWidth = 0.8 * cm # 设置图片的宽度 img.drawHeight = 0.8 * cm # 设置图片的高度 img.hAlign = 0 return img def get_image(path, width=400): img = utils.ImageReader(path) iw, ih = img.getSize() aspect = ih / float(iw) # 按照图片限宽锁定纵横比之后计算高度 height = (width * aspect) # 如果高度过高，报告可能超出此页，则限高，宽度按比例调整 HEIGHT_LIMIT = 500 if height > HEIGHT_LIMIT: height = HEIGHT_LIMIT width = height / aspect return Image(path, width=width, height=height) def report_generate(reportDict, reportPdf, trackPath): # with open(f'{reportJson}', 'r', encoding='utf-8') as f: # data = json.load(f) data = reportDict # dict # 创建PDF文档 styles = getSampleStyleSheet() # 提取数据 name = data["name"] algorithm_score = data["algorithmComprehensiveScore"] algorithm_level = data["algorithmLevel"] test_mileage = data["testMileage"] test_duration = data["testDuration"] algorithm_result_description = data["algorithmResultDescription"] efficient_description = data["details"]["efficient"]["description"] comfort_description = data["details"]["comfort"]["description"] efficient_description1 = data["details"]["efficient"]["description1"] efficient_description2 = data["details"]["efficient"]["description2"] comfort_description1 = data["details"]["comfort"]["description1"] comfort_description2 = data["details"]["comfort"]["description2"] # 创建表格数据 # table_data = [ # ["评价维度", "评价指标", "指标描述"], # ["高效性", "无障碍物停止", efficient_description1], # ["高效性", "速度", efficient_description2], # ["平顺性", "线加速度", comfort_description1], # ["平顺性", "角加速度", comfort_description2] # ] table_data = [ ("评价维度", "评价指标", "指标描述"), ("高效性", "无障碍物停止", efficient_description1), ("高效性", "速度", efficient_description2), ("平顺性", "线加速度", comfort_description1), ("平顺性", "角加速度", comfort_description2) ] # 创建内容对应的空列表 content = list() content.append(ReportPDF.draw_img('Pji.png')) content.append(ReportPDF.draw_title('朴津AMR算法评价报告')) # content.append(ReportPDF.draw_line_feed("换行")) # content.append(ReportPDF.draw_line_feed("换行")) infos = f"数据包: {name}, 行驶时长: {test_duration}, 行驶里程: {test_mileage}" content.append(ReportPDF.draw_text(infos)) content.append(ReportPDF.draw_text(algorithm_result_description)) content.append(ReportPDF.draw_text(efficient_description)) content.append(ReportPDF.draw_text(comfort_description)) # content.append(ReportPDF.draw_line_feed("换行")) content.append(ReportPDF.draw_table(*table_data)) # content.append(ReportPDF.draw_line_feed("换行")) # 添加图片到PDF文档 for key, value in data["graphPath"].items(): content.append(Paragraph(key, styles['Heading5'])) content.append(Image(value, width=480, height=120)) # content.append(ReportPDF.draw_line_feed("换行")) content.append(ReportPDF.draw_line_feed("换行")) content.append(ReportPDF.draw_line_feed("换行")) content.append(ReportPDF.draw_line_feed("换行")) # content.append(Image(trackPath, width=300, height=200)) content.append(Paragraph("trajectory", styles['Heading5'])) content.append(get_image(trackPath)) content.append(ReportPDF.draw_line_feed("换行")) if "track.png" in trackPath: content.append(Paragraph("The trajectory starts with the yellow dot, then the color deepened.", styles['Normal'])) else: content.append(Paragraph("The trajectory starts with the green dot and ends with the red dot.", styles['Normal'])) # content.append(ReportPDF.draw_line_feed("换行")) # content.append(ReportPDF.draw_line_feed("换行")) # 创建PDF文档 # pdf_file = "./report123.pdf" pdf_file = reportPdf doc = SimpleDocTemplate(pdf_file, pagesize=A4) doc.build(content) print("PDF报告已生成！") if __name__ == '__main__': reportJson = './report.json' reportPdf = "./report0422.pdf" trackPath = './track.png' with open(reportJson, 'r', encoding='utf-8') as f: reportJson = json.load(f) report_generate(reportJson, reportPdf, trackPath) print('over')