Spaces:
Running
Running
| // 轻量 PDF 生成器（无依赖） | |
| // - 图纸页为纯矢量：珠子、色号、网格都是矢量指令，任意放大都清晰，文件小、生成快 | |
| // - 中文（页眉、信息页）以 JPEG 图片嵌入 | |
| // - 支持 1-bit 蒙版图片（用于矢量页上的半透明水印） | |
| const A4 = { w: 595.28, h: 841.89 } | |
| // Helvetica / Helvetica-Bold 字宽（AFM，1/1000 em），色号只含 A-Z 0-9 | |
| const W_BOLD = { ' ': 278, '-': 333, '/': 278 } | |
| const W_REG = { ' ': 278, '-': 333, '/': 278 } | |
| '0123456789'.split('').forEach((c) => { W_BOLD[c] = 556; W_REG[c] = 556 }) | |
| const UPPER_BOLD = [722, 722, 722, 722, 667, 611, 778, 722, 278, 556, 722, 611, 833, 722, 778, 667, 778, 722, 667, 611, 722, 667, 944, 667, 667, 611] | |
| const UPPER_REG = [667, 667, 722, 722, 667, 611, 778, 722, 278, 500, 667, 556, 833, 722, 778, 667, 778, 722, 667, 611, 722, 667, 944, 667, 667, 611] | |
| for (let i = 0; i < 26; i += 1) { | |
| W_BOLD[String.fromCharCode(65 + i)] = UPPER_BOLD[i] | |
| W_REG[String.fromCharCode(65 + i)] = UPPER_REG[i] | |
| } | |
| function textWidth(text, size, bold = true) { | |
| const table = bold ? W_BOLD : W_REG | |
| let w = 0 | |
| for (let i = 0; i < text.length; i += 1) w += table[text[i]] || 556 | |
| return (w / 1000) * size | |
| } | |
| const n2 = (v) => (Math.round(v * 100) / 100).toString() | |
| const esc = (s) => String(s).replace(/[\\()]/g, (m) => `\\${m}`).replace(/[^\x20-\x7e]/g, '?') | |
| function ascii(text) { | |
| const out = new Uint8Array(text.length) | |
| for (let i = 0; i < text.length; i += 1) out[i] = text.charCodeAt(i) & 0xff | |
| return out | |
| } | |
| class PdfDoc { | |
| constructor() { | |
| this.objects = [null, null, null] // 1: Catalog, 2: Pages（0 号占位） | |
| this.pageIds = [] | |
| this.fontRegular = this.add(['<< /Type /Font /Subtype /Type1 /BaseFont /Helvetica /Encoding /WinAnsiEncoding >>']) | |
| this.fontBold = this.add(['<< /Type /Font /Subtype /Type1 /BaseFont /Helvetica-Bold /Encoding /WinAnsiEncoding >>']) | |
| this.forms = {} | |
| this.gstates = {} | |
| } | |
| add(parts) { | |
| this.objects.push(parts) | |
| return this.objects.length - 1 | |
| } | |
| stream(dict, data) { | |
| const bytes = typeof data === 'string' ? ascii(data) : new Uint8Array(data) | |
| return this.add([`<< ${dict} /Length ${bytes.length} >>\nstream\n`, bytes, '\nendstream']) | |
| } | |
| addJpeg(buffer, width, height) { | |
| return this.stream(`/Type /XObject /Subtype /Image /Width ${width} /Height ${height} /ColorSpace /DeviceRGB /BitsPerComponent 8 /Filter /DCTDecode`, buffer) | |
| } | |
| // bits：每行按字节对齐，0 = 绘制 | |
| addImageMask(bits, width, height) { | |
| return this.stream(`/Type /XObject /Subtype /Image /Width ${width} /Height ${height} /ImageMask true /BitsPerComponent 1`, bits) | |
| } | |
| // 单位圆（直径 1）表单：不设颜色，继承外部填充色 | |
| form(kind) { | |
| if (this.forms[kind]) return this.forms[kind] | |
| const k = 0.5523 | |
| const circle = (r) => { | |
| const c = r * k | |
| return `${n2(0.5 + r)} 0.5 m ${n2(0.5 + r)} ${n2(0.5 + c)} ${n2(0.5 + c)} ${n2(0.5 + r)} 0.5 ${n2(0.5 + r)} c ` | |
| + `${n2(0.5 - c)} ${n2(0.5 + r)} ${n2(0.5 - r)} ${n2(0.5 + c)} ${n2(0.5 - r)} 0.5 c ` | |
| + `${n2(0.5 - r)} ${n2(0.5 - c)} ${n2(0.5 - c)} ${n2(0.5 - r)} 0.5 ${n2(0.5 - r)} c ` | |
| + `${n2(0.5 + c)} ${n2(0.5 - r)} ${n2(0.5 + r)} ${n2(0.5 - c)} ${n2(0.5 + r)} 0.5 c h ` | |
| } | |
| let body | |
| if (kind === 'round') body = `${circle(0.46)}f` | |
| else if (kind === 'hollow') body = `${circle(0.46)}${circle(0.2)}f*` | |
| else body = `${circle(0.14)}f` // peg 空钉位 | |
| const id = this.stream('/Type /XObject /Subtype /Form /BBox [0 0 1 1]', body) | |
| this.forms[kind] = id | |
| return id | |
| } | |
| gstate(alpha) { | |
| const key = String(alpha) | |
| if (!this.gstates[key]) this.gstates[key] = this.add([`<< /Type /ExtGState /ca ${alpha} /CA ${alpha} >>`]) | |
| return this.gstates[key] | |
| } | |
| addPage(content, { images = {} } = {}) { | |
| const contentId = this.stream('', content) | |
| const xobjects = Object.assign({}, images) | |
| Object.keys(this.forms).forEach((k) => { xobjects[`Fm_${k}`] = this.forms[k] }) | |
| const xo = Object.keys(xobjects).map((k) => `/${k} ${xobjects[k]} 0 R`).join(' ') | |
| const gs = Object.keys(this.gstates).map((k) => `/GS${k.replace('.', '_')} ${this.gstates[k]} 0 R`).join(' ') | |
| const id = this.add([`<< /Type /Page /Parent 2 0 R /MediaBox [0 0 ${A4.w} ${A4.h}] /Resources << /ProcSet [/PDF /Text /ImageB /ImageC] /Font << /F1 ${this.fontRegular} 0 R /F2 ${this.fontBold} 0 R >> /XObject << ${xo} >> /ExtGState << ${gs} >> >> /Contents ${contentId} 0 R >>`]) | |
| this.pageIds.push(id) | |
| return id | |
| } | |
| build() { | |
| this.objects[1] = ['<< /Type /Catalog /Pages 2 0 R >>'] | |
| this.objects[2] = [`<< /Type /Pages /Kids [${this.pageIds.map((id) => `${id} 0 R`).join(' ')}] /Count ${this.pageIds.length} >>`] | |
| const chunks = [] | |
| let pos = 0 | |
| const push = (part) => { | |
| const bytes = typeof part === 'string' ? ascii(part) : part | |
| chunks.push(bytes) | |
| pos += bytes.length | |
| } | |
| push('%PDF-1.4\n%\xe2\xe3\xcf\xd3\n') | |
| const offsets = [] | |
| for (let id = 1; id < this.objects.length; id += 1) { | |
| offsets[id] = pos | |
| push(`${id} 0 obj\n`) | |
| this.objects[id].forEach(push) | |
| push('\nendobj\n') | |
| } | |
| const xref = pos | |
| let table = `xref\n0 ${this.objects.length}\n0000000000 65535 f \n` | |
| for (let id = 1; id < this.objects.length; id += 1) table += `${String(offsets[id]).padStart(10, '0')} 00000 n \n` | |
| push(table) | |
| push(`trailer\n<< /Size ${this.objects.length} /Root 1 0 R >>\nstartxref\n${xref}\n%%EOF\n`) | |
| const out = new Uint8Array(pos) | |
| let cursor = 0 | |
| chunks.forEach((c) => { out.set(c, cursor); cursor += c.length }) | |
| return out.buffer | |
| } | |
| } | |
| // —— 矢量图纸 —— | |
| // 使用“左上角原点、y 向下”的坐标系（页面开头统一翻转），文字与图片局部再翻回来 | |
| function hexToPdf(hex) { | |
| const v = parseInt(hex.slice(1), 16) | |
| return `${n2(((v >> 16) & 255) / 255)} ${n2(((v >> 8) & 255) / 255)} ${n2((v & 255) / 255)}` | |
| } | |
| function lumaOf(hex) { | |
| const v = parseInt(hex.slice(1), 16) | |
| return (((v >> 16) & 255) * 0.299 + ((v >> 8) & 255) * 0.587 + (v & 255) * 0.114) / 255 | |
| } | |
| function vectorSheet(doc, artwork, { x0, y0, cell, region, style, codes, grid }) { | |
| const { size, cells, pal } = artwork | |
| const out = [] | |
| out.push(`${hexToPdf('#f1eee7')} rg ${n2(x0)} ${n2(y0)} ${n2(region.cols * cell)} ${n2(region.rows * cell)} re f`) | |
| const groups = new Map() | |
| const empty = [] | |
| for (let y = region.row; y < region.row + region.rows; y += 1) { | |
| for (let x = region.col; x < region.col + region.cols; x += 1) { | |
| const c = cells[y * size + x] | |
| const px = x0 + (x - region.col) * cell | |
| const py = y0 + (y - region.row) * cell | |
| if (c < 0) { empty.push([px, py]); continue } | |
| if (!groups.has(c)) groups.set(c, []) | |
| groups.get(c).push([px, py]) | |
| } | |
| } | |
| if (empty.length) { | |
| doc.form('peg') | |
| out.push(`${hexToPdf('#d9d3c7')} rg`) | |
| empty.forEach(([x, y]) => out.push(`q ${n2(cell)} 0 0 ${n2(cell)} ${n2(x)} ${n2(y)} cm /Fm_peg Do Q`)) | |
| } | |
| if (style !== 'square') doc.form(style) | |
| groups.forEach((list, c) => { | |
| out.push(`${hexToPdf(pal.hexes[c])} rg`) | |
| if (style === 'square') { | |
| // 略微外扩 0.05pt 消除阅读器抗锯齿造成的白缝 | |
| list.forEach(([x, y]) => out.push(`${n2(x - 0.05)} ${n2(y - 0.05)} ${n2(cell + 0.1)} ${n2(cell + 0.1)} re`)) | |
| out.push('f') | |
| } else { | |
| list.forEach(([x, y]) => out.push(`q ${n2(cell)} 0 0 ${n2(cell)} ${n2(x)} ${n2(y)} cm /Fm_${style} Do Q`)) | |
| } | |
| }) | |
| if (codes) { | |
| const fs = cell * (style === 'hollow' ? 0.27 : 0.34) | |
| const writeGroup = (ink) => { | |
| const lines = [] | |
| groups.forEach((list, c) => { | |
| const useInk = style === 'hollow' || lumaOf(pal.hexes[c]) > 0.56 | |
| if (useInk !== ink) return | |
| const id = pal.ids[c] | |
| const tw = textWidth(id, fs) | |
| list.forEach(([x, y]) => lines.push(`1 0 0 -1 ${n2(x + (cell - tw) / 2)} ${n2(y + cell / 2 + fs * 0.35)} Tm (${id}) Tj`)) | |
| }) | |
| if (!lines.length) return | |
| out.push(`${ink ? hexToPdf('#1d2b2a') : '1 1 1'} rg BT /F2 ${n2(fs)} Tf`) | |
| out.push(...lines) | |
| out.push('ET') | |
| } | |
| writeGroup(true) | |
| writeGroup(false) | |
| } | |
| if (grid && grid.enabled) { | |
| const w = region.cols * cell | |
| const h = region.rows * cell | |
| const color = hexToPdf(grid.color) | |
| const alpha = grid.opacity / 100 | |
| const thin = [] | |
| for (let i = 0; i <= region.cols; i += 1) thin.push(`${n2(x0 + i * cell)} ${n2(y0)} m ${n2(x0 + i * cell)} ${n2(y0 + h)} l`) | |
| for (let i = 0; i <= region.rows; i += 1) thin.push(`${n2(x0)} ${n2(y0 + i * cell)} m ${n2(x0 + w)} ${n2(y0 + i * cell)} l`) | |
| doc.gstate(Math.round(alpha * 30) / 100) | |
| doc.gstate(alpha) | |
| out.push(`q /GS${String(Math.round(alpha * 30) / 100).replace('.', '_')} gs ${color} RG ${n2(Math.max(0.2, cell * 0.02))} w`, ...thin, 'S Q') | |
| const thick = [] | |
| for (let i = 0; i <= region.cols; i += 1) { | |
| const g = region.col + i | |
| if (g % grid.interval && i && i !== region.cols) continue | |
| thick.push(`${n2(x0 + i * cell)} ${n2(y0)} m ${n2(x0 + i * cell)} ${n2(y0 + h)} l`) | |
| } | |
| for (let i = 0; i <= region.rows; i += 1) { | |
| const g = region.row + i | |
| if (g % grid.interval && i && i !== region.rows) continue | |
| thick.push(`${n2(x0)} ${n2(y0 + i * cell)} m ${n2(x0 + w)} ${n2(y0 + i * cell)} l`) | |
| } | |
| out.push(`q /GS${String(alpha).replace('.', '_')} gs ${color} RG ${n2(Math.max(0.6, cell * 0.06))} w`, ...thick, 'S Q') | |
| // 坐标数字 | |
| const fs = Math.min(7, Math.max(4, cell * 0.42)) | |
| const step = cell < 8 ? grid.interval : (cell < 13 ? 5 : 1) | |
| const labels = [] | |
| for (let i = 0; i < region.cols; i += 1) { | |
| const g = region.col + i + 1 | |
| if (g % step && g !== 1 && i !== 0) continue | |
| const t = String(g) | |
| labels.push(`1 0 0 -1 ${n2(x0 + (i + 0.5) * cell - textWidth(t, fs, false) / 2)} ${n2(y0 - fs * 0.5)} Tm (${t}) Tj`) | |
| } | |
| for (let i = 0; i < region.rows; i += 1) { | |
| const g = region.row + i + 1 | |
| if (g % step && g !== 1 && i !== 0) continue | |
| const t = String(g) | |
| labels.push(`1 0 0 -1 ${n2(x0 - textWidth(t, fs, false) - fs * 0.4)} ${n2(y0 + (i + 0.5) * cell + fs * 0.35)} Tm (${t}) Tj`) | |
| } | |
| out.push(`${color} rg BT /F1 ${n2(fs)} Tf`, ...labels, 'ET') | |
| } | |
| return out.join('\n') | |
| } | |
| // 页眉图片 + 水印蒙版的放置 | |
| function imageCmd(name, x, y, w, h) { | |
| return `q ${n2(w)} 0 0 ${n2(-h)} ${n2(x)} ${n2(y + h)} cm /${name} Do Q` | |
| } | |
| function watermarkCmd(mask) { | |
| if (!mask) return '' | |
| const h = 16 | |
| const w = h * mask.width / mask.height | |
| const out = [`q /GS0_08 gs ${hexToPdf('#1d2b2a')} rg`] | |
| const stepX = w + 90 | |
| const stepY = 120 | |
| for (let row = 0, y = -200; y < A4.h + 200; y += stepY, row += 1) { | |
| for (let x = -200; x < A4.w + 200; x += stepX) { | |
| // 旋转 -20°：cos=0.94 sin=0.342；y 轴已翻转，所以蒙版局部再翻回 | |
| out.push(`q 0.94 -0.342 0.342 0.94 ${n2(x + (row % 2) * stepX / 2)} ${n2(y)} cm ${n2(w)} 0 0 ${n2(-h)} 0 ${n2(h)} cm /WM Do Q`) | |
| } | |
| } | |
| out.push('Q') | |
| return out.join('\n') | |
| } | |
| const MARGIN = 28.35 // 10mm | |
| const HEADER_H = 34 | |
| const AXIS = 14 | |
| const TARGET_CELL = 18 // 约 6.4mm/格，打印后色号清晰可读 | |
| const AREA_W = A4.w - MARGIN * 2 | |
| const AREA_H = A4.h - MARGIN * 2 - HEADER_H | |
| /** | |
| * 规划图纸页：小图一页搞定；大图 = 总览页 + 若干分块页 | |
| * 返回 { tiled, fitCell, tilesX, tilesY, tileCols, tileRows, pages: [{ kind, tile, region }] } | |
| */ | |
| function planPatternPages(size) { | |
| const fitCell = Math.min((AREA_W - AXIS) / size, (AREA_H - AXIS) / size) | |
| const tiled = fitCell < 11 | |
| const full = { col: 0, row: 0, cols: size, rows: size } | |
| const pages = [{ kind: 'overview', tile: 0, region: full }] | |
| let tilesX = 1 | |
| let tilesY = 1 | |
| let tileCols = size | |
| let tileRows = size | |
| if (tiled) { | |
| tilesX = Math.ceil(size / Math.floor((AREA_W - AXIS) / TARGET_CELL)) | |
| tilesY = Math.ceil(size / Math.floor((AREA_H - AXIS) / TARGET_CELL)) | |
| tileCols = Math.ceil(size / tilesX) | |
| tileRows = Math.ceil(size / tilesY) | |
| let n = 1 | |
| for (let ty = 0; ty < tilesY; ty += 1) { | |
| for (let tx = 0; tx < tilesX; tx += 1) { | |
| const col = tx * tileCols | |
| const row = ty * tileRows | |
| if (col >= size || row >= size) continue | |
| pages.push({ kind: 'tile', tile: n, region: { col, row, cols: Math.min(tileCols, size - col), rows: Math.min(tileRows, size - row) } }) | |
| n += 1 | |
| } | |
| } | |
| } | |
| return { tiled, fitCell, tilesX, tilesY, tileCols, tileRows, pages } | |
| } | |
| /** | |
| * 生成图纸页（矢量）。 | |
| * opts: { plan, headers: [{ id, width, height }]（按页序）, style, grid, watermark } | |
| */ | |
| function addPatternPages(doc, artwork, opts) { | |
| const size = artwork.size | |
| const plan = opts.plan || planPatternPages(size) | |
| const headers = opts.headers || [] | |
| const flip = `1 0 0 -1 0 ${A4.h} cm` | |
| const wm = opts.watermark ? watermarkCmd(opts.watermark) : '' | |
| if (opts.watermark) doc.gstate(0.08) | |
| if (plan.tiled) doc.gstate(0.85) | |
| const images = (header) => { | |
| const map = {} | |
| if (header) map.HD = header.id | |
| if (opts.watermark) map.WM = opts.watermark.id | |
| return map | |
| } | |
| const headerCmd = (header) => (header ? imageCmd('HD', MARGIN, MARGIN, AREA_W, AREA_W * header.height / header.width) : '') | |
| plan.pages.forEach((page, i) => { | |
| const header = headers[i] || null | |
| const content = [flip, headerCmd(header)] | |
| if (page.kind === 'overview') { | |
| const fitCell = plan.fitCell | |
| const gridW = fitCell * size | |
| const x0 = MARGIN + AXIS + (AREA_W - AXIS - gridW) / 2 | |
| const y0 = MARGIN + HEADER_H + AXIS | |
| content.push(vectorSheet(doc, artwork, { | |
| x0, y0, cell: fitCell, region: page.region, style: opts.style, codes: !plan.tiled, grid: opts.grid, | |
| })) | |
| if (plan.tiled) { | |
| const lines = [`q ${hexToPdf('#e11d48')} RG 1.2 w`] | |
| const labels = [] | |
| plan.pages.forEach((tp) => { | |
| if (tp.kind !== 'tile') return | |
| const r = tp.region | |
| const cx = x0 + r.col * fitCell | |
| const cy = y0 + r.row * fitCell | |
| const cw = r.cols * fitCell | |
| const ch = r.rows * fitCell | |
| lines.push(`${n2(cx)} ${n2(cy)} ${n2(cw)} ${n2(ch)} re S`) | |
| const t = String(tp.tile) | |
| labels.push(`1 0 0 -1 ${n2(cx + cw / 2 - textWidth(t, 22) / 2)} ${n2(cy + ch / 2 + 8)} Tm (${t}) Tj`) | |
| }) | |
| lines.push('Q') | |
| content.push(lines.join('\n')) | |
| content.push(`q /GS0_85 gs ${hexToPdf('#e11d48')} rg BT /F2 22 Tf\n${labels.join('\n')}\nET Q`) | |
| } | |
| } else { | |
| const cell = Math.min((AREA_W - AXIS) / plan.tileCols, (AREA_H - AXIS) / plan.tileRows) | |
| content.push(vectorSheet(doc, artwork, { | |
| x0: MARGIN + AXIS, y0: MARGIN + HEADER_H + AXIS, cell, region: page.region, style: opts.style, codes: true, grid: opts.grid, | |
| })) | |
| } | |
| content.push(wm) | |
| doc.addPage(content.join('\n'), { images: images(header) }) | |
| }) | |
| return plan.pages.length | |
| } | |
| // 整页图片（信息页） | |
| function addImagePage(doc, jpegId, pixelW, pixelH) { | |
| const areaW = A4.w - MARGIN * 2 | |
| const areaH = A4.h - MARGIN * 2 | |
| const scale = Math.min(areaW / pixelW, areaH / pixelH) | |
| const w = pixelW * scale | |
| const h = pixelH * scale | |
| const x = (A4.w - w) / 2 | |
| const y = (A4.h - h) / 2 | |
| doc.addPage(`1 0 0 -1 0 ${A4.h} cm\n${imageCmd('IM', x, y, w, h)}`, { images: { IM: jpegId } }) | |
| } | |
| module.exports = { A4, PdfDoc, planPatternPages, addPatternPages, addImagePage, vectorSheet, textWidth, HEADER_H, MARGIN } | |