// Copyright (C) 2003-2026, Foxit Software Inc..
// All Rights Reserved.
//
// http://www.foxitsoftware.com
//
// The following code is copyrighted and contains proprietary information and trade secrets of Foxit Software Inc..
// You cannot distribute any part of Foxit PDF SDK to any third party or general public,
// unless there is a separate license agreement with Foxit Software Inc. which explicitly grants you such rights.
//
// This file contains an example to demonstrate how to use Foxit PDF SDK to to call layout recognition related classes
// for PDF document.

import com.foxit.sdk.PDFException;
import com.foxit.sdk.common.Font;
import com.foxit.sdk.common.Image;
import com.foxit.sdk.common.Library;
import com.foxit.sdk.common.Path;
import com.foxit.sdk.common.fxcrt.Matrix2D;
import com.foxit.sdk.common.fxcrt.PointF;
import com.foxit.sdk.common.fxcrt.RectF;
import com.foxit.sdk.common.Progressive;
import com.foxit.sdk.common.Range;
import com.foxit.sdk.pdf.PDFDoc;
import com.foxit.sdk.pdf.PDFPage;
import com.foxit.sdk.pdf.graphics.*;
import com.foxit.sdk.pdf.objects.PDFDictionary;
import com.foxit.sdk.addon.layoutrecognition.*;

import java.io.File;
import java.io.FileWriter;
import java.util.ArrayList;
import java.io.IOException;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;

import static com.foxit.sdk.common.Constants.e_ErrSuccess;
import static com.foxit.sdk.common.Constants.e_FillModeAlternate;
import static com.foxit.sdk.common.Font.*;
import static com.foxit.sdk.pdf.PDFDoc.e_SaveFlagNormal;
import static com.foxit.sdk.pdf.PDFPage.e_ParsePageNormal;
import static com.foxit.sdk.pdf.graphics.GraphicsObject.*;
import static com.foxit.sdk.addon.layoutrecognition.LRElement.*;
import static com.foxit.sdk.addon.layoutrecognition.LRStructureElement.*;


public class layout_recognition {

    private static String key = "8f3g1sFttW8NAgfSNv+BvaJWy4Pp8flcm7ncNDkdvCCO4JwUFbGLyZdujReMaiplbYJwwxam3lr7uHtQaEf0xTSu8j2ZBdrZ8FS4OQkEUjOI9UEGb3Ic0GqmlP9T9OnSItOYqoZZA/OnYg5EaVP+c/ha32nUquOk+5riEdhYgINAZDdgJkPc+YSWsrfixOIROf6t34j3eRcyFnOY0MS7tSXi1AEPE0VZdyu+5bfmBSGvLI+WR5lQ+oOQR5hRGwqf990WEdBelNO7dxhZELVhEZ+/L60PpDf5y9A8qmrY6PiEU8nhqIyLpks0BxeIsYyZ3Cr82jjfbItPGyEARsIVzluBSgVMD4oGjojz8T6wGDEXUbmZH/pra94XAs8wZ28mzbmIEppWnhBqX0c6Ov8iXCfFfUZr4KNuPxMEgk4VOYTMOd70GEjGnylivw18hbn7ZpkWh2Zql+uU2HejvOgn+WHvT0HF9NhK0Xb5vH8aFkgBP92PJx8obIbxBVeeExczxvsqWufuPdXS2sPNIgMYZ2TtvqVVsiUToeGB2FQo09fzZOSH5Vi18u/siyvC3FmjeMf4+OYWPuDPNCcrb21NgcHam7XI8AAoGLTGtLLWR+fxynbBBcg2d2ZPcZ5CMFIfOBHRWCkcA0sDT3cl9nsBppHNJze/jVxHy+lREeQEA/etSIgohTGAWW3ZlpibRXqMxlmVWEPifCubbWTzQ6U54mRMpW2UghH3gvCNFNgdday3Av7EtlIJOdUuiMq3zmmd5J4Jt4C82CKnVewA0DtZ4EEtwRG9IYuLrfpuZlnjBAtoeXRV1lqUZ6BcCe0E3m9jO28sGkALuIeJQ74XG7X8lMC7lMWWQYiYD7D6G9QT6X3qV4vXJwNKRKpYc/rI+amEawTSnDYAMEaYPwtZ0rb/7BL+atsZZaI+ncebdfVfjDJZbCZAc+JnkuNlUn2xrNRuObw8h/G0lm2pNOEJ10S8R4+yplo5B1QmC2dCDhaTaW67QJPqnJPKbVmjw7i6KGBi+CDP3JnAKlOX/YUu8uk56JfeQUuGcQg/KaBMGctnchH2n1i3+oGbibhurQj22+OO66Q6mijKoCRP/QJm5WTGcO4c0iVX0B8dIlai/uv/pGU8Wps9MxysH5ViVIj56LWKuNfy1FNO9+wx4iHYnUxfW0Iccq3YJYAlOBkMv8VZIJl9aTEEFJ626LJEw/UbEBdR9CdlwnylPf4RuNW/lcGYMgq3xNVpp/bzYKwZivvQOFgMdX1tv8D33SHvAm3zrRCU+v94BJXnBi9k25fnKsc705IMwv+I6bueXA+mDnagj1oTqsWXZ0WnPAYI3WUO7sBf7XUH230wfYXnTSVhujJxiEP/2oWqi2813gtQkKwz9N2p6yT8Vsh3+BaAwwDvtOZMJKUFcabTBSL6NoouzmzwItJ4mIboT6KXaef66rs8jCU=";
    private static String sn = "Rew1vB0TqoMIxX444UcpScARWdhgJpsJfzmyMLR32a0+EgrCjp2NaQ==";
    private static String output_path = "../output_files/layout_recognition/";
    private static String input_path = "../input_files/";

    // You can also use System.load("filename") instead. The filename argument must be an absolute path name.
    static {
      String os = System.getProperty("os.name").toLowerCase();
      String lib = "fsdk_java_";
      if (os.startsWith("win")) {
          lib += "win";
      } else if (os.startsWith("mac")) {
          lib += "mac";
      } else {
          lib += "linux";
      }
        
      if (System.getProperty("sun.arch.data.model").equals("64")) {
        if(System.getProperty("os.arch").equals("aarch64")){
          if(os.startsWith("mac")){
             lib += "arm";
           }
           else{
             lib += "armv8";
           }
         }
         else{
            lib += "64";
         }
      } else {
         if(System.getProperty("os.arch").equals("arm")){
            lib += "armv7";
         }
         else{
            lib += "32";
         }
      }
      System.loadLibrary(lib);
    }

    private static void createResultFolder(String output_path) {
        File myPath = new File(output_path);
        if (!myPath.exists()) {
            myPath.mkdir();
        }
    }

    private static void getChildFromElement(LRStructureElement pElement, ArrayList<LRElement> elemArray, int nType) throws PDFException {
        int nElemListSize = pElement.getChildCount();
        for (int i=0; i<nElemListSize; i++) {
            LRElement item = pElement.getChild(i);
            int item_type = item.getElementType();
            if (item_type == nType) 
                elemArray.add(item);
            if (item.isStructureElement()) {
                LRStructureElement srt_item = new LRStructureElement(item);
                getChildFromElement(srt_item, elemArray, nType);
            }
        }
    }

    public static void writeTextWithPrefix(OutputStreamWriter doc, int depth, String prefix, String content) throws IOException {
        for (int i = 0; i < depth; i++) {
            doc.write(prefix);
        }
        doc.write(content);
    } 

    private static String lr_Format(boolean val) {
        return val ? "True" : "False";
    }

    private static String lr_Format(int val) {
        return String.format("%d", val);
    }

    private static String lr_Format(float val) {
        return String.format("%.1f", val);
    }

    private static String lr_Format(long val) {
        return String.format("#%02X%02X%02X", (byte)(val >> 16), (byte)(val >> 8), (byte)(val));
    }

    private static String lr_FormatAttributeValueEnum(int val) {
        return LRStructureElement.stringifyEnumVariant(val);
    }

    private static void OutputLRStructureElement(LRStructureElement element, OutputStreamWriter doc, int depth) throws PDFException, IOException {
        if (element.isEmpty())
            return;

        LRStructureElement parentElement = element.getParentElement();
        boolean bIsEmpty = parentElement.isEmpty();
        
        int elemType = element.getElementType();
        String elementTypeStr = LRStructureElement.stringifyElementType(elemType);
        String outputStr = "< StructureElement: " + elementTypeStr + " >\r\n";
        writeTextWithPrefix(doc, depth, "\t", outputStr);
        
        int nSize = element.getSupportedAttributeCount();
        for (int i=0; i<nSize; i++) {
            int attrType = element.getSupportedAttribute(i);
            
            int attrValueType = element.getAttributeValueType(attrType);
            int nLength = 0;
            boolean bIsArray = LRStructureElement.isArrayAttributeValueType(attrValueType);
            nLength = element.getAttributeValueCount(attrType);
            if (attrValueType == LRStructureElement.e_AttributeValueTypeEmpty)
                continue;
        
            String szKey = LRStructureElement.stringifyAttributeType(attrType);
            String szVal = "";
            if (bIsArray) szVal += "[";
        
            for (int idx = 0; idx < nLength; idx++)
            {
                if (idx > 0) szVal += ", ";
        
                switch (attrValueType)
                {
                case LRStructureElement.e_AttributeValueTypeEnum:
                case LRStructureElement.e_AttributeValueTypeEnumArray:
                    szVal += lr_FormatAttributeValueEnum(element.getAttributeValueEnum(attrType, idx));
                    break;
                case LRStructureElement.e_AttributeValueTypeInt32:
                case LRStructureElement.e_AttributeValueTypeInt32Array:
                    szVal += lr_Format(element.getAttributeValueInt32(attrType, idx));
                    break;
                case LRStructureElement.e_AttributeValueTypeFloat:
                case LRStructureElement.e_AttributeValueTypeFloatArray:
                    szVal += lr_Format(element.getAttributeValueFloat(attrType, idx));
                    break;
                case LRStructureElement.e_AttributeValueTypeARGB:
                case LRStructureElement.e_AttributeValueTypeARGBArray:
                    szVal += lr_Format(element.getAttributeValueARGB(attrType, idx));
                    break;
                case LRStructureElement.e_AttributeValueTypeWStr:
                case LRStructureElement.e_AttributeValueTypeWStrArray:
                    szVal += element.getAttributeValueString(attrType, idx);
                    break;
                }
            }
            if (bIsArray) szVal += "]";
        
            outputStr = szKey + ": " + szVal + "\r\n";
            writeTextWithPrefix(doc, depth, "\t", outputStr);
        }
    }

    private static void OutputLRStructureElement(LRGraphicsObjectElement element, OutputStreamWriter doc, int depth) throws PDFException, IOException {
        if (element.isEmpty())
            return;
        
        String outputStr = "< LRGraphicsObjectElement: >\r\n";
        LRGraphicsObjectElement parentPageObj = element.getParentGraphicsObjectElement();
        boolean bIsEmpty = parentPageObj.isEmpty();
        writeTextWithPrefix(doc, depth, "\t", outputStr);
        GraphicsObject pageObj = element.getGraphicsObject();
        int type = 0;
		if (pageObj != null)
		  type = pageObj.getType();
        RectF rcBox = element.getBBox();
        PDFDictionary pDict = element.getDict();
        int objIndex = element.getGraphicsObjectIndex();
        Matrix2D matrix = element.getMatrix();
        String szVal = "";
        szVal = String.format("BBox: [%.1f,%.1f,%.1f,%.1f]\r\n", rcBox.getLeft(), rcBox.getTop(), rcBox.getRight(), rcBox.getBottom());
        writeTextWithPrefix(doc, depth, "\t", szVal);
        szVal = String.format("Matrix: [%.1f,%.1f,%.1f,%.1f,%.1f,%.1f]\r\n", matrix.getA(), matrix.getB(), matrix.getC(), matrix.getD(), matrix.getE(), matrix.getF());
        writeTextWithPrefix(doc, depth, "\t", szVal);
        szVal = String.format("PageObjectIndex: %d\r\n", objIndex);
        writeTextWithPrefix(doc, depth, "\t", szVal);

        String text = (type == (int)GraphicsObject.e_TypeText) ? pageObj.getTextObject().getText() : "";
        if (text.length() > 0) {
            szVal = "Text: ";
            szVal += text;
            szVal += "\r\n";
            writeTextWithPrefix(doc, depth, "\t", szVal);
        }
    }

    private static void OutputLRStructureElement(LRContentElement element, OutputStreamWriter doc, int depth) throws PDFException, IOException {
        if (element.isEmpty())
            return;
        
        LRGraphicsObjectElement pageObj = element.getGraphicsObjectElement();
        boolean bIsEmpty = pageObj.isEmpty();
        LRStructureElement parentPageObj = element.getParentElement();
        bIsEmpty = parentPageObj.isEmpty();
        
        int elemType = element.getElementType();
        String elementTypeStr = LRElement.stringifyElementType(elemType);
        String outputStr = "< LRContentElement: " + elementTypeStr + " >\r\n";
        writeTextWithPrefix(doc, depth, "\t", outputStr);
        
        RectF rcBox = element.getBBox();
        int nStartPos = 0, nLength = 0;
        Range range = element.getGraphicsObjectRange();
        if (!range.isEmpty()) {
          nStartPos = range.getSegmentStart(0);
          nLength = range.getSegmentEnd(0) - nStartPos + 1;
        }
        Matrix2D matrix = element.getMatrix();
        
        String szVal = "";
        szVal = String.format("BBox: [%.1f,%.1f,%.1f,%.1f]\r\n", rcBox.getLeft(), rcBox.getTop(), rcBox.getRight(), rcBox.getBottom());
        writeTextWithPrefix(doc, depth, "\t", szVal);
        szVal = String.format("Matrix: [%.1f,%.1f,%.1f,%.1f,%.1f,%.1f]\r\n", matrix.getA(), matrix.getB(), matrix.getC(), matrix.getD(), matrix.getE(), matrix.getF());
        writeTextWithPrefix(doc, depth, "\t", szVal);
        szVal = String.format("StartPos: %d\r\n", nStartPos);
        writeTextWithPrefix(doc, depth, "\t", szVal);
        szVal = String.format("Length: %d\r\n", nLength);
        writeTextWithPrefix(doc, depth, "\t", szVal);

        if (!bIsEmpty)
            OutputLRStructureElement(pageObj, doc, depth + 1);
    }

    private static void showLRElementInfo(LRStructureElement element, OutputStreamWriter doc, int depth) throws PDFException, IOException {
        int nElemListSize = element.getChildCount();
        OutputLRStructureElement(element, doc, depth);
        for (int i=0; i<nElemListSize; i++)
        {
            LRElement item = element.getChild(i);
            int item_type = item.getElementType();
            if (item.isStructureElement())
            {
                LRStructureElement srt_item = new LRStructureElement(item);
                showLRElementInfo(srt_item, doc, depth + 1);
            } else if (item.isContentElement())
            {
                LRContentElement srt_item = new LRContentElement(item);
                OutputLRStructureElement(srt_item, doc, depth);
            } else if (item.isGraphicsObjectElement())
            {
                LRGraphicsObjectElement srt_item = new LRGraphicsObjectElement(item);
                OutputLRStructureElement(srt_item, doc, depth);
            }
        }
    }

    private static void outputAllLRElement(LRStructureElement element, String info_path) throws PDFException, IOException {
        FileOutputStream fos = new FileOutputStream(info_path);
        OutputStreamWriter text_doc = new OutputStreamWriter(fos, "UTF-8");
        if (!element.isEmpty()) {
            showLRElementInfo(element, text_doc, 0);
        } else {
            text_doc.write("No layout recognition information!\r\n");
        }
        text_doc.flush();
        text_doc.close();
        fos.close();
    }

    public static void main(String[] args) throws PDFException {
        createResultFolder(output_path);
        // Initialize library
        int error_code = Library.initialize(sn, key);
        if (error_code != e_ErrSuccess) {
            System.out.println(String.format("Library Initialize Error: %d\n", error_code));
            return;
        }

        try {
            String input_file = input_path + "AboutFoxit.pdf";
            String info_file = output_path + "layout_recognition_info.txt";
            PDFDoc doc = new PDFDoc(input_file);
            error_code = doc.load(null);

            if (error_code != e_ErrSuccess) {
                System.out.println(String.format("The Doc [%s] Error: %d\n", input_file, error_code));
                return;
            }

            PDFPage original_page = doc.getPage(0);
            original_page.startParse(e_ParsePageNormal, null, false);
            LRContext context = new LRContext(original_page);

            Progressive progressive = context.startParse(null);

            LRStructureElement root = context.getRootElement();
            int type = root.getElementType();
            int nElemListSize = root.getChildCount();
            System.out.println(String.format("RootElement getChildCount: %d", nElemListSize));
            outputAllLRElement(root, info_file);

            ArrayList<LRElement> element_array = new ArrayList<LRElement>();
            getChildFromElement(root, element_array, (int)LRElement.e_ElementTypeSpan);
            System.out.println(String.format("LRElement::e_ElementTypeSpan count: %d\n", element_array.size()));

            System.out.println("Layout recognition test.");

        } catch (Exception e) {
            e.printStackTrace();
            return;
        }

        Library.release();
    }
}
