ConstructionCost/modules/reports/util/rpt_pdf_util.js

/**
 * Created by chen on 2017/8/16.
 */
/**
 * Created by zhang on 2017/8/14.
 */

let pdf = require('pdfkit');
let fs = require('fs');
let jpcCmnHelper = require('../rpt_component/helper/jpc_helper_common');
let DPI = jpcCmnHelper.getScreenDPI()[0];
let JV = require('../rpt_component/jpc_value_define');
const uuidV1 = require('uuid/v1');

let fontUtil = require('./rpt_font_util');

// 目前不支持下划线

module.exports = {
    export_pdf_file:export_pdf_file
}

function export_pdf_file (pageData, paperSize, fName, callback) {
    let offsetX= 10;
    let offsetY=10;
    let doc = new pdf({autoFirstPage: false});
    // let newName = '' + (new Date()).valueOf();
    let newName = uuidV1();
    let stream = doc.pipe(fs.createWriteStream(__dirname.slice(0, __dirname.length - 21) + '/tmp/'+newName+'.pdf'));
    let pageObj = pageData;
    // doc.rect(5,5,1190,890).lineWidth(1).strokeColor('black').stroke();//边框
    let paperSizeIdx = JV.PAGES_SIZE_STR.indexOf(paperSize);
    let size = JV.PAGES_SIZE[paperSizeIdx];

    function private_getIniPageMergeBorder(mergedBand) {
        let rst = {};
        rst[JV.PROP_LEFT] = mergedBand[JV.PROP_LEFT];
        rst[JV.PROP_RIGHT] = mergedBand[JV.PROP_RIGHT];
        rst[JV.PROP_TOP] = mergedBand[JV.PROP_TOP];
        rst[JV.PROP_BOTTOM] = mergedBand[JV.PROP_BOTTOM];
        rst[JV.PROP_STYLE] = mergedBand[JV.PROP_STYLE];
        return rst;
    }

    let newPageMergeBand = private_getIniPageMergeBorder(pageObj[JV.BAND_PROP_MERGE_BAND]);
    if (pageObj && pageObj.items.length > 0 ) {
        for(let i=0;i<pageObj.items.length;i++){
            if (pageData[JV.NODE_PAGE_INFO][JV.NODE_PAGE_SIZE][0] > pageData[JV.NODE_PAGE_INFO][JV.NODE_PAGE_SIZE][1]) {
                doc.addPage({size:[size[1]*DPI,size[0]*DPI]});
            } else {
                doc.addPage({size:[size[0]*DPI,size[1]*DPI]});
            }
            let page = pageObj.items[i],
                fonts = pageObj[JV.NODE_FONT_COLLECTION],
                styles = pageObj[JV.NODE_STYLE_COLLECTION],
                controls = pageObj[JV.NODE_CONTROL_COLLECTION];

            if (page[JV.PROP_PAGE_MERGE_BORDER]) {
                newPageMergeBand[JV.PROP_LEFT] = page[JV.PROP_PAGE_MERGE_BORDER][JV.PROP_LEFT];
                newPageMergeBand[JV.PROP_RIGHT] = page[JV.PROP_PAGE_MERGE_BORDER][JV.PROP_RIGHT];
                newPageMergeBand[JV.PROP_TOP] = page[JV.PROP_PAGE_MERGE_BORDER][JV.PROP_TOP];
                newPageMergeBand[JV.PROP_BOTTOM] = page[JV.PROP_PAGE_MERGE_BORDER][JV.PROP_BOTTOM];
            }
            for (let j = 0; j < page.cells.length; j++) {
                let cell = page.cells[j];
                private_drawCell(cell, fonts, styles, controls, newPageMergeBand);
            }
        }
    }
    doc.end();
    stream.on('finish',function () {
        console.log(newName + ".pdf was written.");
        callback(newName);
    });

    function private_chkIfInMergedBand(mergedBand, cell) {
        let rst = false;
        if (mergedBand && cell) {
            rst = mergedBand[JV.PROP_TOP] <= cell[JV.PROP_AREA][JV.PROP_TOP] && mergedBand[JV.PROP_BOTTOM] >= cell[JV.PROP_AREA][JV.PROP_BOTTOM] &&
                mergedBand[JV.PROP_LEFT] <= cell[JV.PROP_AREA][JV.PROP_LEFT] && mergedBand[JV.PROP_RIGHT] >= cell[JV.PROP_AREA][JV.PROP_RIGHT];
        }
        return rst;
    }

    function private_drawCell(cell, fonts, styles, controls, mergedBand) {
        doc.save();
        //doc.translate(0.5,0.5);
        let style = styles[cell[JV.PROP_STYLE]];
        if (style) {
            let isNeedMergeBand = private_chkIfInMergedBand(mergedBand, cell);
            private_drawLine(cell, doc, style, JV.PROP_TOP, [JV.PROP_LEFT, JV.PROP_TOP],[JV.PROP_RIGHT, JV.PROP_TOP], mergedBand, styles, isNeedMergeBand);
            private_drawLine(cell, doc, style, JV.PROP_RIGHT, [JV.PROP_RIGHT, JV.PROP_TOP],[JV.PROP_RIGHT, JV.PROP_BOTTOM], mergedBand, styles, isNeedMergeBand);
            private_drawLine(cell, doc, style, JV.PROP_BOTTOM, [JV.PROP_RIGHT, JV.PROP_BOTTOM],[JV.PROP_LEFT, JV.PROP_BOTTOM], mergedBand, styles, isNeedMergeBand);
            private_drawLine(cell, doc, style, JV.PROP_LEFT, [JV.PROP_LEFT, JV.PROP_BOTTOM],[JV.PROP_LEFT, JV.PROP_TOP], mergedBand, styles, isNeedMergeBand);
        }
        private_drawCellText(cell, fonts, controls);
        doc.restore();

    }

    function private_drawLine(cell, doc, style, styleBorderDest, startP, destP, mergedBand, styles, isNeedMergeBand) {
        //doc.beginPath();
        let destStyle = style;
        if (mergedBand) {
            if (isNeedMergeBand && parseFloat(mergedBand[styleBorderDest]) === parseFloat(cell[JV.PROP_AREA][styleBorderDest])) {
                destStyle = styles[mergedBand[JV.PROP_STYLE][JV.PROP_ID]];
            }
        }
        doc.moveTo(cell[JV.PROP_AREA][startP[0]] + offsetX, cell[JV.PROP_AREA][startP[1]] + offsetY);
        if (destStyle[styleBorderDest] && parseFloat(destStyle[styleBorderDest][JV.PROP_LINE_WEIGHT]) !== 0) {
            doc.lineWidth(1.0 * destStyle[styleBorderDest][JV.PROP_LINE_WEIGHT]);
            doc.lineTo(cell[JV.PROP_AREA][destP[0]] + offsetX, cell[JV.PROP_AREA][destP[1]] + offsetY);
            doc.strokeColor(destStyle[styleBorderDest][JV.PROP_COLOR]);
        }
        doc.stroke();
    }
    function private_drawCellText(cell, fonts, controls) {
        if (cell[JV.PROP_VALUE]) {
            let values = ("" + cell[JV.PROP_VALUE]).split('|');
            // let font = fonts[cell[JV.PROP_FONT]];
            let font = null;
            if (typeof cell[JV.PROP_FONT] === "string") {
                font = fonts[cell[JV.PROP_FONT]];
            } else {
                font = cell[JV.PROP_FONT];
            }
            // let control = controls[cell[JV.PROP_CONTROL]];
            let control = null;
            if (typeof cell[JV.PROP_CONTROL] === "string") {
                control = controls[cell[JV.PROP_CONTROL]];
            } else {
                control = cell[JV.PROP_CONTROL];
            }
            let height = cell[JV.PROP_AREA][JV.PROP_BOTTOM] - cell[JV.PROP_AREA][JV.PROP_TOP];
            let area = [cell[JV.PROP_AREA][JV.PROP_LEFT] + offsetX, cell[JV.PROP_AREA][JV.PROP_TOP] + offsetY, cell[JV.PROP_AREA][JV.PROP_RIGHT] + offsetX, cell[JV.PROP_AREA][JV.PROP_BOTTOM] + offsetY];
            for (let i = 0; i < values.length; i++) {
                area[JV.IDX_TOP] = cell[JV.PROP_AREA][JV.PROP_TOP] + i * (height / values.length) + offsetY;
                area[JV.IDX_BOTTOM] = cell[JV.PROP_AREA][JV.PROP_TOP] + (i + 1) * (height / values.length) + offsetY;
                private_drawText(values[i], area, font, control);
            }
        }
    }

    function private_drawText(val, area, font, control) {
        let dftFontHeight = 12;
        let output = [];
        let fontFile = __dirname + '/pdf_base_files/simkai.ttf';
        if (font) {
            dftFontHeight = 1 * font[JV.FONT_PROPS[1]];
            fontFile = __dirname + '/pdf_base_files/' + fontUtil.getActualFont(font[JV.FONT_PROPS[0]], (font[JV.FONT_PROPS[3]] === 'T'), (font[JV.FONT_PROPS[4]] === 'T')) + '.ttf';
            doc.fontSize(dftFontHeight);
        }
        doc.font(fontFile);
        let options={};
        if (control) {
            private_setupAreaH(area, control.Horizon, font.FontAngle, dftFontHeight, output,options);
            private_setupAreaV(area, control.Vertical, font.FontAngle, dftFontHeight, output);
        } else {
            private_setupAreaH(area, "left", font.FontAngle, dftFontHeight, output,options);
            private_setupAreaV(area, "bottom", font.FontAngle, dftFontHeight, output);
        }
        let w = area[JV.IDX_RIGHT] - JV.OUTPUT_OFFSET[JV.OFFSET_IDX_RIGHT] - area[JV.IDX_LEFT] - JV.OUTPUT_OFFSET[JV.OFFSET_IDX_RIGHT];
        if (parseInt(font.FontAngle) !== 0) {
            w = area[JV.IDX_BOTTOM] - JV.OUTPUT_OFFSET[JV.OFFSET_IDX_BOTTOM] - area[JV.IDX_TOP] - JV.OUTPUT_OFFSET[JV.OFFSET_IDX_TOP];
        }
        // doc.save();
        //doc.translate(output[0], output[1]);
        if (w >= doc.widthOfString(val)) {
            options.width = w;
            options.height = dftFontHeight;
        } else {
            while (true) {
                dftFontHeight--;
                doc.fontSize(dftFontHeight);
                if (w >= doc.widthOfString(val) || dftFontHeight < 6) {
                    options.width = w;
                    options.height = dftFontHeight;
                    break;
                }
            }
        }
        function private_drawUnderline() {
            //A. 暂不支持角度; B. PDF输出时，坐标没有translate
            let ctx = doc;
            //1. 计算下划线的相关坐标
            let width = ctx.widthOfString(val);
            let height = dftFontHeight;
            let startX = area[JV.IDX_LEFT], startY = area[JV.IDX_TOP], endX = area[JV.IDX_RIGHT], endY = area[JV.IDX_BOTTOM];
            // let startX = 0, startY = 0, endX = width, endY = startY;
            if (control.Horizon === "left") {
                startX = Math.round(area[JV.IDX_LEFT] + JV.OUTPUT_OFFSET[JV.IDX_LEFT]);
            } else if (control.Horizon === "right") {
                startX = Math.round(area[JV.IDX_RIGHT] - width - JV.OUTPUT_OFFSET[JV.IDX_RIGHT]);
            } else {
                startX = Math.round( area[JV.IDX_LEFT] + (area[JV.IDX_RIGHT] - area[JV.IDX_LEFT] - width) / 2);
            }
            endX = Math.round(startX + width);

            if (control.Vertical === "top") {
                startY = Math.round(area[JV.IDX_TOP] + JV.OUTPUT_OFFSET[JV.IDX_TOP] + JV.OUTPUT_OFFSET[JV.IDX_BOTTOM] + height);
            } else if (control.Vertical === "bottom") {
                startY = Math.round(area[JV.IDX_BOTTOM] + JV.OUTPUT_OFFSET[JV.IDX_BOTTOM]);
            } else {
                startY = Math.round( area[JV.IDX_TOP] + (area[JV.IDX_BOTTOM] - area[JV.IDX_TOP] + height) / 2) + JV.OUTPUT_OFFSET[JV.IDX_TOP] + JV.OUTPUT_OFFSET[JV.IDX_BOTTOM];
            }
            endY = Math.round(startY);
            //2. 画线
            // ctx.save();
            if ( output[1] !== Math.round(output[1])) {
                ctx.translate(0,0.5);
            }
            // ctx.beginPath();
            ctx.moveTo(startX, startY);
            ctx.lineWidth(1);
            ctx.strokeStyle = "BLACK";
            ctx.lineTo(endX, endY);
            ctx.stroke();
            // ctx.restore();
        }

        let rotateOptions;
        if (font[JV.FONT_PROPS[5]] === 'T' && parseInt(font.FontAngle) === 0) {
            private_drawUnderline();
        }
        if (parseInt(font.FontAngle) !== 0) {
            if (control){
                rotateOptions = private_setupAreaRotateOption(area,w,control.Vertical,dftFontHeight, output);
            } else {
                rotateOptions = private_setupAreaRotateOption(area,w,"bottom",dftFontHeight, output);
            }
            doc.rotate(font.FontAngle,rotateOptions);
        }
        doc.text(val,output[0], output[1], options);
        doc.font(__dirname + '/pdf_base_files/simhei_bold_italic.ttf');
        // doc.restore();
    }

    function private_setupAreaH(area, type, fontAngle, dftFontHeight, outputPoint,options) {
        let lType = type;
        if (type !== "left" && type !== "right" && type !== "center") lType = "left";
        options.align=lType;
        outputPoint[0]=1 * area[JV.IDX_LEFT]+ JV.OUTPUT_OFFSET[JV.OFFSET_IDX_LEFT];
    }

    function private_setupAreaV(area, type, fontAngle, dftFontHeight, outputPoint) {
        let lType = type;
        if (type !== "top" && type !== "bottom" && type !== "center") lType = "top";
        switch (lType) {
            case "top":
                outputPoint[1] = 1 * area[JV.IDX_TOP] + JV.OUTPUT_OFFSET[JV.OFFSET_IDX_TOP];
                break;
            case "bottom":
                outputPoint[1] = 1 * area[JV.IDX_BOTTOM] - JV.OUTPUT_OFFSET[JV.OFFSET_IDX_BOTTOM]-dftFontHeight;
                break;
            case "center":
                outputPoint[1] = 1 * area[JV.IDX_TOP] + (1 * area[JV.IDX_BOTTOM]-1 * area[JV.IDX_TOP]- dftFontHeight) / 2;
                break;
        }
    }

    function private_setupAreaRotateOption(area,w, type="top",dftFontHeight,outputPoint){
        let x = (area[JV.IDX_RIGHT] - area[JV.IDX_LEFT])/2+area[JV.IDX_LEFT];
        let y =(area[JV.IDX_BOTTOM] - area[JV.IDX_TOP])/2+ area[JV.IDX_TOP];
        let rotateOptions = {origin:[x,y]};
        let h = area[JV.IDX_RIGHT] - area[JV.IDX_LEFT];
        outputPoint[0]=x-w/2+JV.OUTPUT_OFFSET[JV.OFFSET_IDX_LEFT];
        switch (type) {
            case "top":
                outputPoint[1] = y- h/2+ JV.OUTPUT_OFFSET[JV.OFFSET_IDX_TOP];
                break;
            case "bottom":
                outputPoint[1] = y+ h/2-JV.OUTPUT_OFFSET[JV.OFFSET_IDX_BOTTOM]-dftFontHeight;
                break;
            case "center":
                outputPoint[1] = y+dftFontHeight/2;
                break;
        }
        return rotateOptions;
    }

}