/*
    Antenna House PDF Tool API V8.0
    Java Interface sample program

    概要：テキスト検索：正規表現での検索

    Copyright 2025- Antenna House, Inc.
*/
package Sample;

import jp.co.antenna.ptl.PtlAnnotHighlight;
import jp.co.antenna.ptl.PtlAnnots;
import jp.co.antenna.ptl.PtlColorDeviceRGB;
import jp.co.antenna.ptl.PtlException;
import jp.co.antenna.ptl.PtlPDFDocument;
import jp.co.antenna.ptl.PtlPage;
import jp.co.antenna.ptl.PtlPages;
import jp.co.antenna.ptl.PtlParamInput;
import jp.co.antenna.ptl.PtlParamOutput;
import jp.co.antenna.ptl.PtlParamSearchText;
import jp.co.antenna.ptl.PtlQuadPoint;
import jp.co.antenna.ptl.PtlQuadPoints;
import jp.co.antenna.ptl.PtlSearchTextResult;
import jp.co.antenna.ptl.PtlSearchTextResultDetail;
import jp.co.antenna.ptl.PtlSearchTextResultDetails;
import jp.co.antenna.ptl.PtlSearchTextResults;

public class SearchTextRegEx {

	public static void main(String[] args) {
		// TODO 自動生成されたメソッド・スタブ
        if (args.length < 3)
        {
            System.out.println("usage: java SearchTextAndHighlight in-pdf-file out-pdf-file keyword");
            return;
        }

        try (PtlParamInput inputFile = new PtlParamInput(args[0]);
             PtlParamOutput outputFile = new PtlParamOutput(args[1]);
             PtlPDFDocument doc = new PtlPDFDocument())
        {
            // PDFファイルをロード
            doc.load(inputFile);

            try (PtlPages pages = doc.getPages()) // ページコンテナの取得
            {
                // ページコンテナが空かどうか
                if (pages.isEmpty())
                {
                    System.out.println("ページコンテナが空\n");
                    return;
                }

                try (PtlParamSearchText paramSearchText = new PtlParamSearchText();
                     PtlColorDeviceRGB colorRed = new PtlColorDeviceRGB(1.0f, 0.0f, 0.0f))
                {
                    // 検索設定
            		paramSearchText.appendTextRegEx(args[2]);
            		paramSearchText.setTextType(PtlParamSearchText.TEXT_TYPE.TEXT_RAW);

                    // 検索実行して検索結果コンテナ取得
                    try (PtlSearchTextResults results = doc.searchText(paramSearchText))
                    {
                        int numResults = results.getCount();
                        for (int i = 0; i < numResults; ++i)
                        {
                            // 検索結果取得
                            try (PtlSearchTextResult result = results.get(i))
                            {
                                // 検索に使用したキーワード
                                String keyword = result.getKeyword();

                                // ヒットしたページ番号
                                int pageNumber = result.getPageNumber();
                                try (PtlPage page = pages.get(pageNumber);
                                     PtlAnnots annots = page.getAnnots())
                                {
                                    // 検索結果詳細コンテナ取得
                                    try (PtlSearchTextResultDetails details = result.getResultDetails())
                                    {
                                        int numDetails = details.getCount();
                                        for (int j = 0; j < numDetails; ++j)
                                        {
                                            // 検索結果詳細取得
                                            try (PtlSearchTextResultDetail resultDetail = details.get(j))
                                            {
                                                // 検索に使用したキーワードの部分文字列
                                                String keywordD = resultDetail.getKeyword();

                                                try (PtlAnnotHighlight annot = new PtlAnnotHighlight();
                                                     PtlQuadPoints quads = annot.getQuadPoints())
                                                {
                                                    // 検索に使用したキーワードの部分文字列を囲むQuadPoint
                                                    try (PtlQuadPoint quad = resultDetail.getQuadPoint())
                                                    {
                                                        quads.append(quad);
                                                    }
                                                    // 検索に使用したキーワードの部分文字列が縦書きか
                                                    boolean verticalWriting = resultDetail.isVerticalWriting();
                                                    annot.setVerticalDirection(verticalWriting);

                                                    annot.setColor(colorRed);
                                                    annot.setMarkUpCA(0.5f);

                                                    annots.append(annot);
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }

            // ファイルに保存します。
            doc.save(outputFile);
        }
        catch (PtlException pex) {
             System.out.println("PtlException : ErrorCode = " + pex.getErrorCode() + "\n  " + pex.getErrorMessage());
        }
        catch (Exception ex) {
            System.out.println(ex.getMessage());
            ex.printStackTrace();
        }
        catch (Error ex) {
            System.out.println(ex.getMessage());
            ex.printStackTrace();
        }
        finally {
            System.out.println("-- 完了 --");
        }

	}

}