const { flatten } = require('flatten.js'); const { applyOutlines } = require('./outline.js'); const { applyInfill } = require('./infill.js'); const { adjustDimensions } = require('./scaling.js'); /** * Processes SVG DOM according to specified parameters * @param {Document} svg_orig - Original SVG DOM * @param {string[]} colors - Array of colors to consider * @param {boolean} drawOutlines - Whether to add outlines * @param {boolean} flattenSVG - Whether to flatten SVG * @param {'None'|'Hatching'|'Interpolation'} infillType - Infill type * @param {number} scaling - Scaling factor * @param {number} maxHeight - Maximum output height * @param {number} maxWidth - Maximum output width * @returns {Document} Processed SVG DOM */ function processSVG(svg_orig, colors, drawOutlines, flattenSVG, infillType = 'None', scaling = 1, maxHeight, maxWidth) { // Clone original SVG to avoid mutation const parser = new DOMParser(); const serializer = new XMLSerializer(); const svgString = serializer.serializeToString(svg_orig); const svgPrepared = parser.parseFromString(svgString, 'image/svg+xml'); // Filter elements based on colors const elements = svgPrepared.documentElement.querySelectorAll('*'); elements.forEach(element => { const elementColor = getComputedStyle(element).fill; if (!colors.includes(elementColor)) { element.parentNode.removeChild(element); } }); // Apply outlines if requested if (drawOutlines) { applyOutlines(svgPrepared); } // Apply infill based on type if (infillType !== 'None') { applyInfill(svgPrepared, infillType); } // Flatten SVG if requested if (flattenSVG) { flatten(svgPrepared.documentElement); } // Apply scaling and size constraints adjustDimensions(svgPrepared, scaling, maxHeight, maxWidth); return svgPrepared; } module.exports = processSVG; /** * Applies outlines to SVG elements * @param {Document} svg - SVG DOM to modify */ function applyOutlines(svg) { const elements = svg.querySelectorAll('path, rect, circle, polygon'); elements.forEach(element => { element.setAttribute('stroke', 'black'); element.setAttribute('stroke-width', '0.5'); }); } module.exports = { applyOutlines }; /** * Applies specified infill pattern to SVG * @param {Document} svg - SVG DOM to modify * @param {'Hatching'|'Interpolation'} infillType - Type of infill to apply */ function applyInfill(svg, infillType) { const patterns = { Hatching: createHatchingPattern, Interpolation: createInterpolationPattern }; const defs = svg.createElementNS('http://www.w3.org/2000/svg', 'defs'); const pattern = patterns[infillType](); defs.appendChild(pattern); svg.documentElement.insertBefore(defs, svg.documentElement.firstChild); const elements = svg.querySelectorAll('path'); elements.forEach(element => { element.setAttribute('fill', `url(#${pattern.id})`); }); } function createHatchingPattern() { const pattern = document.createElementNS('http://www.w3.org/2000/svg', 'pattern'); pattern.setAttribute('id', 'hatching'); pattern.setAttribute('patternUnits', 'userSpaceOnUse'); pattern.setAttribute('width', '4'); pattern.setAttribute('height', '4'); const path = document.createElementNS('http://www.w3.org/2000/svg', 'path'); path.setAttribute('d', 'M-1,1 l2,-2 M0,4 l4,-4 M3,5 l2,-2'); path.setAttribute('stroke', 'currentColor'); path.setAttribute('stroke-width', '0.5'); pattern.appendChild(path); return pattern; } function createInterpolationPattern() { const pattern = document.createElementNS('http://www.w3.org/2000/svg', 'pattern'); pattern.setAttribute('id', 'interpolation'); pattern.setAttribute('patternUnits', 'userSpaceOnUse'); pattern.setAttribute('width', '8'); pattern.setAttribute('height', '8'); const circle = document.createElementNS('http://www.w3.org/2000/svg', 'circle'); circle.setAttribute('cx', '4'); circle.setAttribute('cy', '4'); circle.setAttribute('r', '2'); circle.setAttribute('fill', 'currentColor'); pattern.appendChild(circle); return pattern; } module.exports = { applyInfill }; /** * Adjusts SVG dimensions and scaling * @param {Document} svg - SVG DOM to modify * @param {number} scaling - Scaling factor * @param {number} maxHeight - Maximum height constraint * @param {number} maxWidth - Maximum width constraint */ function adjustDimensions(svg, scaling, maxHeight, maxWidth) { const svgElement = svg.documentElement; const viewBox = svgElement.getAttribute('viewBox')?.split(' ') || [0, 0, 100, 100]; let [,, width, height] = viewBox.map(Number); // Apply scaling width *= scaling; height *= scaling; // Apply size constraints if (maxWidth && width > maxWidth) { const ratio = maxWidth / width; width = maxWidth; height *= ratio; } if (maxHeight && height > maxHeight) { const ratio = maxHeight / height; height = maxHeight; width *= ratio; } svgElement.setAttribute('viewBox', `0 0 ${width} ${height}`); svgElement.setAttribute('width', width); svgElement.setAttribute('height', height); } module.exports = { adjustDimensions };