天天看点

POI高亮word文档中指定文本

    public static void main(String[] args) throws Exception {

        InputStream is = new FileInputStream(new File("D:\\test.docx"));

        XWPFDocument doc = new XWPFDocument(is);

        for (XWPFParagraph paragraph : doc.getParagraphs()) {

            if (paragraph.getText().contains("想要高亮的内容")) {

                highLightParagraph(paragraph, "想要高亮的内容");

            }

        }

        File file = new File("D:\\test2.docx");

        FileOutputStream out = new FileOutputStream(file);

        doc.write(out);

        out.close();

        doc.close();

    }

    public CTRPr getRunCTRPr(XWPFParagraph p, XWPFRun pRun) {

        CTRPr pRpr;

        if (pRun.getCTR() != null) {

            pRpr = pRun.getCTR().getRPr();

            if (pRpr == null) {

                pRpr = pRun.getCTR().addNewRPr();

            }

        } else {

            pRpr = p.getCTP().addNewR().addNewRPr();

        }

        return pRpr;

    }

    private void highLightParagraph(XWPFParagraph p, String content) {

        TextSegment segment = p.searchText(content, new PositionInParagraph());

        int beginRunIndex = segment.getBeginRun();

        int endRunIndex = segment.getEndRun();

        List<XWPFRun> runs = p.getRuns();

        if (beginRunIndex == endRunIndex) {

            XWPFRun run = runs.get(beginRunIndex);

            String runText = run.getText(0);

            if (runText.equals(content)) {

                highLight(p, run);

            } else {

                containsMatch(p, content, run, beginRunIndex);

            }

        } else {

            begin = runs.get(beginRunIndex);

            end = runs.get(endRunIndex);

            // 获得第一个run标签,匹配内容

            String beginText = getBeginString(begin.text(), content);

            // 获得最后一个run标签,匹配内容

            String endText = getEndString(end.text(), content);

            // 高亮中间的文本

            for (int i = beginRunIndex + 1; i < endRunIndex; i++) {

                XWPFRun run = runs.get(i);

                highLight(p, run);

            }

        }

    }

    private XWPFRun beginMatch(XWPFParagraph p, String content, XWPFRun run, int beginRunIndex) {

        String runText = run.text();

        if (runText.equals(content)) {

            highLight(p, run);

            return run;

        }

        run.setText(runText.replace(content, ""), 0);

        XWPFRun newRun = p.insertNewRun(beginRunIndex + 1);

        newRun.setText(content);

        highLight(p, newRun);

        return newRun;

    }

    private XWPFRun endMatch(XWPFParagraph p, String content, XWPFRun run, int endRunIndex) {

        String runText = run.text();

        if (runText.equals(content)) {

            highLight(p, run);

            return run;

        }

        run.setText(runText.replace(content, ""), 0);

        XWPFRun newRun = p.insertNewRun(endRunIndex);

        newRun.setText(content);

        highLight(p, newRun);

        return newRun;

    }

    private XWPFRun containsMatch(XWPFParagraph p, String content, XWPFRun run, int runIndex) {

        String runText = run.text();

        String[] arr = runText.split(content);

        run.setText(arr[0], 0);

        XWPFRun newRun = p.insertNewRun(runIndex + 1);

        newRun.setText(content);

        highLight(p, newRun);

        p.insertNewRun(runIndex + 2).setText(arr[1]);

        return newRun;

    }

    private void highLight(XWPFParagraph p, XWPFRun run) {

        CTRPr pRpr = getRunCTRPr(p, run);

        CTHighlight highlight = pRpr.isSetHighlight() ? pRpr

                .getHighlight() : pRpr.addNewHighlight();

        highlight.setVal(STHighlightColor.YELLOW);

    }

    public static String getBeginString(String begin, String content) {

        for (int i = 1; i < begin.length(); i++) {

            if (content.startsWith(begin)) {

                return begin;

            }

            begin = begin.substring(1);

        }

        return null;

    }

    public static String getEndString(String end, String content) {

        for (int i = end.length() - 1; i > 0; i--) {

            if (content.endsWith(end)) {

                return end;

            }

            end = end.substring(0, i);

        }

        return null;

    }

继续阅读