// 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 call layout recognition related classes
// for PDF document.

// Include Foxit SDK header files.
#include <time.h>
#include <iostream>
#if defined(_WIN32) || defined(_WIN64)
#include <direct.h>
#else
#include <sys/stat.h>
#endif

#define _FSDK_LAYOUTRECOGNITION_

#include "../../../include/common/fs_common.h"
#include "../../../include/pdf/fs_pdfdoc.h"
#include "../../../include/pdf/fs_pdfpage.h"
#include "../../../include/addon/layoutrecognition/fs_layoutrecognition.h"

using namespace std;
using namespace foxit;
using namespace foxit::common;
using foxit::common::Library;
using namespace pdf;
using namespace annots;
using namespace actions;
using namespace interform;
using namespace foxit::addon::layoutrecognition;
using namespace foxit::pdf::objects;
using namespace foxit::pdf::graphics;

static const char* sn = "bPxThFlBAYifO3yiW0C51bKbjPap1zHaf5aG5sX378TjFvB7Q0802A==";
static const char* key = "8f3g1sGNtR8NAgfSNv+BvaJWy2fnrw36NZvzh8GUxYXvs+/YczAk4e34yqlOnmL4asgxGMO7+rlOcSHUSSKagx5g4G7VEXfkYLXnDDx77FOOedZPrFFcOCu1vT4+wOKzy1zkORq9cKtiFf44byH5sfRqf+OvELjep2teT9eF9yPCsSw2vRKljXlPy40DMP9XX1O5OTe+mSG3RvwKUMb0+RpG3YobM87j+/ynTDlJMMXw4vgwuTb1ZuAtlvkTdFz7HDLV81C6TYnn73ddHeGD3qYWBVsD4uJkDwyyH/SWr7QX0HSgVg8HSZqnab1VGdAUYxKY2vGWeOTJdu6qDGb5ZEb9O3kZZg+9LYAmUKzDfCYwfIsYZ8Y9LPC4wMx70kIq7JpNnzPvigrdshV8/ew1F0+qP/P5Z1EB53vDVrp2LVSuL873Wv4g3PzjZlNH+1qgT87hC+Buto4qsSUfP5Y2z0uApu+c/TfttCXbeMLSsw9TY1HQD3mkrZExMaQqRrAuXAcZWdvBnVuG3DCZmm8q0DLxJiCfYfViAU/AkbeKMIGaTJdrvIXBcssfde4nmU6qr0M4x4+Hs2ESddtY2QRtLAUBq/7h7GJlJHkEZr8ueaUy/6D/2vvAWz6wMLgmjlPbDTLqzfnCJylHCPtcJmu1SPPT9v1VreBX3+zNWcIT8HueyHEiUniqzwWkRKaotrnNb/zhnHLPpgaBSwKjyi8rOzzVwu/nY84G9ye4jsVjmkIRItKDrW7z22iS+QQuTwfqLVuMB0OD4jMkPCC2slzPfEeZnuh4+Rewln2qLRJVgI72zJ20OBWAelGSpO1Ii2Y3gOvojdmtUwx6R3Z1qFNWTTB25o9ytcux8BLOtTCkA8T8xaT4N339OTCsu/TmfahIDixKvr8cDa6F2WkBT8SOfv64W9PgiNv5B/BMOD2VWYaAl0N7yD0NkYYg/cDWRC4dfpdeip1jrr0k3mND0oSpIkxC+FicZbaYYwRdtJ9OSVWJ5OE0voLv8HgLze8mw6s4TrN3z+a2NQCveMa2VRN7YcEF7agPrvb/qU+otMwHae/wM7+f+HFAaOW+WTfzfxIVFB8/A1DN0NiEX+em4G7nO5F5ZcvfYPpaaioe8Nn33BbVLtMjXqsOBeM8AXiVfpD6YO75j7BqnBUPnivvD8BxJaaIQZvVSuQmvGm0ynt03IYwE8Rq+jYsunvsZ86jh5BiUR+Nm7sK0Qe9BdZn/7jdqohMv9czfMGUAscX+xBMunFbDpcRbWVpUADcm209L8Ndek2XY7YlCMxPgPVNvfWRF2jxehib9Tpd7zB5ZIDfAEH9zrpLfp29DLvCh2Yx8mH7xxdKG1O9Q0GdHphYHhej6smAcPm68R0GUQaTOLfXqNpueGJC1GAOI3Y4swjyt7jMTWcibnKLzwukWE5rnEDhn5n64JnyUM0ZFkRmtOiV";

#if defined(_WIN32) || defined(_WIN64)
static WString output_path = WString::FromLocal("../output_files/");
static WString input_path = WString::FromLocal("../input_files/");
#else
static WString output_path = WString::FromLocal("./output_files/");
static WString input_path = WString::FromLocal("./input_files/");
#endif

class SdkLibMgr {
 public:
  SdkLibMgr()
      : is_initialize_(false){};
  ErrorCode Initialize() {
    ErrorCode error_code = Library::Initialize(sn, key);
    if (error_code != foxit::e_ErrSuccess) {
      printf("Library Initialize Error: %d\n", error_code);
    } else {
      is_initialize_ = true;
    }
    return error_code;
  }
  ~SdkLibMgr() {
    if (is_initialize_)
      Library::Release();
  }

 private:
  bool is_initialize_;
};

class TextDoc {
 public:
  TextDoc(const String& file_name, const String& fill_mode);
  TextDoc(const WString& file_name, const WString& fill_mode);
  ~TextDoc();

  void Write(const char* text_content);
  void Write(const wchar_t* text_content);
  void Write(int count, const wchar_t* prefix, const wchar_t* text_content);

 private:
  FILE* file_;
};

TextDoc::TextDoc(const String& file_name, const String& file_mode) throw(Exception)
    : file_(NULL) {
#if defined(_WIN32) || defined(_WIN64)
  fopen_s(&file_, file_name, (const char*)file_mode);
#else
  file_ = fopen((const char*)file_name, (const char*)file_mode);
#endif

  if (!file_)
    throw Exception(__FILE__, __LINE__, __FUNCTION__, foxit::e_ErrFile);

  uint8 temp[] = {0xFF, 0xFE};
  fwrite(temp, sizeof(uint8), 2, file_);
  fseek(file_, 0, SEEK_END);
}

TextDoc::TextDoc(const WString& file_name, const WString& file_mode) throw(Exception)
    : file_(NULL) {
  String s_file_name = String::FromUnicode(file_name);
  String s_file_mode = String::FromUnicode(file_mode);

#if defined(_WIN32) || defined(_WIN64)
  fopen_s(&file_, (const char*)s_file_name, (const char*)s_file_mode);
#else
  file_ = fopen((const char*)s_file_name, (const char*)s_file_mode);
#endif
  if (!file_)
    throw Exception(__FILE__, __LINE__, __FUNCTION__, foxit::e_ErrFile);

  uint8 temp[] = {0xFF, 0xFE};
  fwrite(temp, sizeof(uint8), 2, file_);
  fseek(file_, 0, SEEK_END);
}

TextDoc::~TextDoc() {
  fclose(file_);
  file_ = NULL;
}

void TextDoc::Write(const char* text_content) {
  WString wide_str = WString::FromLocal(text_content);
  Write(wide_str);
}

void TextDoc::Write(const wchar_t* text_content) {
  WString wide_str(text_content);
  if (wide_str.IsEmpty())
    return;
  String utf16le_str = wide_str.UTF16LE_Encode(false);
  if (utf16le_str.IsEmpty())
    return;
  int length = utf16le_str.GetLength();
  fwrite((const char*)utf16le_str, sizeof(char), length, file_);
}

void TextDoc::Write(int count, const wchar_t* prefix, const wchar_t* text_content) {
  for (int i = 0; i < count; i++) {
    Write(prefix);
  }

  Write(text_content);
}

typedef CFX_ArrayTemplate<LRElement> LRElementArray;

void GetChildFromElement(LRStructureElement pElement, LRElementArray& elemArray,
                         LRStructureElement::ElementType nType) {
  int nElemListSize = pElement.GetChildCount();
  for (int i = 0; i < nElemListSize; i++) {
    LRElement item = pElement.GetChild(i);
    LRElement::ElementType item_type = item.GetElementType();
    if (item_type == nType)
      elemArray.Add(item);
    if (item.IsStructureElement()) {
      LRStructureElement srt_item(item);
      GetChildFromElement(srt_item, elemArray, nType);
    }
  }
}

WString LR_Format(bool val) {
  return val ? L"True" : L"False";
}

WString LR_Format(int32 val) {
  WString s;
  s.Format(L"%d", val);
  return s;
}

WString LR_Format(float val) {
  WString s;
  s.Format(L"%.1f", val);
  return s;
}

WString LR_Format(WString val) {
  return val;
}

WString LR_Format(ARGB val) {
  WString s;
  s.Format(L"#%02X%02X%02X", (uint8)(val >> 16), (uint8)(val >> 8), (uint8)(val));
  return s;
}

WString LR_Format(LRStructureElement::AttributeValueEnum val) {
  return WString::FromLocal(LRStructureElement::StringifyEnumVariant(val));
}

void OutputLRStructureElement(LRStructureElement element, TextDoc& doc, int depth) {
  if (depth > 32)
    return;
  if (element.IsEmpty())
    return;

  if (element.IsEmpty())
    return;

  LRStructureElement parentElement = element.GetParentElement();
  bool bIsEmpty = parentElement.IsEmpty();

  LRStructureElement::ElementType elemType = element.GetElementType();
  String elementTypeStr = LRElement::StringifyElementType(elemType);
  WString outputStr = L"< StructureElement: " + WString::FromLocal(elementTypeStr) + L" >\r\n";
  doc.Write(depth, L"\t", outputStr);
  int32 nSize = element.GetSupportedAttributeCount();
  for (int i = 0; i < nSize; i++) {
    LRStructureElement::AttributeType attrType = element.GetSupportedAttribute(i);

    LRStructureElement::AttributeValueType attrValueType = element.GetAttributeValueType(attrType);
    int nLength = 0;
    bool bIsArray = LRStructureElement::IsArrayAttributeValueType(attrValueType);
    nLength = element.GetAttributeValueCount(attrType);
    if (attrValueType == LRStructureElement::e_AttributeValueTypeEmpty)
      continue;

    WString szKey = WString::FromLocal(LRStructureElement::StringifyAttributeType(attrType));
    WString szVal = L"";
    if (bIsArray)
      szVal += L"[";

    for (int idx = 0; idx < nLength; idx++) {
      if (idx)
        szVal += L", ";

      switch (attrValueType) {
        case LRStructureElement::e_AttributeValueTypeEnum:
        case LRStructureElement::e_AttributeValueTypeEnumArray:
          szVal += LR_Format(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 += LR_Format(element.GetAttributeValueString(attrType, idx));
          break;
        default:
          break;
      }
    }
    if (bIsArray)
      szVal += L"]";

    outputStr = szKey + L": " + szVal + L"\r\n";
    doc.Write(depth, L"\t", outputStr);
  }
}

void OutputLRStructureElement(LRGraphicsObjectElement element, TextDoc& doc, int depth) {
  if (element.IsEmpty())
    return;

  WString outputStr = L"< LRGraphicsObjectElement: >\r\n";
  LRGraphicsObjectElement parentPageObj = element.GetParentGraphicsObjectElement();
  bool bIsEmpty = parentPageObj.IsEmpty();
  doc.Write(depth, L"\t", outputStr);
  GraphicsObject* pageObj = element.GetGraphicsObject();
  GraphicsObject::Type type = GraphicsObject::e_TypeAll;
  if (pageObj != NULL)
    type = pageObj->GetType();
  RectF rcBox = element.GetBBox();
  PDFDictionary* pDict = element.GetDict();
  int objIndex = element.GetGraphicsObjectIndex();
  Matrix matrix = element.GetMatrix();
  WString szVal = L"";
  szVal.Format(L"BBox: [%.1f,%.1f,%.1f,%.1f]\r\n", rcBox.left, rcBox.top, rcBox.right, rcBox.bottom);
  doc.Write(depth, L"\t", szVal);
  szVal.Format(L"Matrix: [%.1f,%.1f,%.1f,%.1f,%.1f,%.1f]\r\n", matrix.a, matrix.b, matrix.c, matrix.d, matrix.e,
               matrix.f);
  doc.Write(depth, L"\t", szVal);
  szVal.Format(L"PageObjectIndex: %d\r\n", objIndex);
  doc.Write(depth, L"\t", szVal);

  GraphicsObject* pObject = element.GetGraphicsObject();
  WString text = (pObject && pObject->GetType() == GraphicsObject::e_TypeText)
                     ? element.GetGraphicsObject()->GetTextObject()->GetText()
                     : L"";
  if (!text.IsEmpty()) {
    szVal = L"Text: ";
    szVal += text;
    szVal += L"\r\n";

    doc.Write(depth, L"\t", szVal);
  }
}

void OutputLRStructureElement(LRContentElement element, TextDoc& doc, int depth) {
  if (element.IsEmpty())
    return;

  LRGraphicsObjectElement pageObj = element.GetGraphicsObjectElement();
  bool bIsEmpty = pageObj.IsEmpty();
  LRStructureElement parentPageObj = element.GetParentElement();
  bIsEmpty = parentPageObj.IsEmpty();

  LRElement::ElementType elemType = element.GetElementType();
  String elementTypeStr = LRStructureElement::StringifyElementType(elemType);
  WString outputStr = L"< LRContentElement: " + WString::FromLocal(elementTypeStr) + L" >\r\n";
  doc.Write(depth, L"\t", outputStr);

  RectF rcBox = element.GetBBox();
  int32 nStartPos = 0, nLength = 0;
  common::Range range = element.GetGraphicsObjectRange();
  if (!range.IsEmpty()) {
    nStartPos = range.GetSegmentStart(0);
    nLength = range.GetSegmentEnd(0) - nStartPos + 1;
  }
  Matrix matrix = element.GetMatrix();

  WString szVal = L"";
  szVal.Format(L"BBox: [%.1f,%.1f,%.1f,%.1f]\r\n", rcBox.left, rcBox.top, rcBox.right, rcBox.bottom);
  doc.Write(depth, L"\t", szVal);
  szVal.Format(L"Matrix: [%.1f,%.1f,%.1f,%.1f,%.1f,%.1f]\r\n", matrix.a, matrix.b, matrix.c, matrix.d, matrix.e,
               matrix.f);
  doc.Write(depth, L"\t", szVal);
  szVal.Format(L"StartPos: %d\r\n", nStartPos);
  doc.Write(depth, L"\t", szVal);
  szVal.Format(L"Length: %d\r\n", nLength);
  doc.Write(depth, L"\t", szVal);

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

void ShowLRElementInfo(LRStructureElement element, TextDoc& doc, int depth) {
  OutputLRStructureElement(element, doc, depth);
  int nElemListSize = element.GetChildCount();
  for (int i = 0; i < nElemListSize; i++) {
    LRElement item = element.GetChild(i);
    uint32 item_type = item.GetElementType();
    if (item.IsStructureElement()) {
      LRStructureElement srt_item(item);
      ShowLRElementInfo(srt_item, doc, depth + 1);
    } else if (item.IsContentElement()) {
      LRContentElement srt_item(item);
      OutputLRStructureElement(srt_item, doc, depth);
    } else if (item.IsGraphicsObjectElement()) {
      LRGraphicsObjectElement srt_item(item);
      OutputLRStructureElement(srt_item, doc, depth);
    }
  }
}

void OutputAllLRElement(LRStructureElement element, const WString& info_path) {
  TextDoc text_doc(info_path, L"w+b");
  if (!element.IsEmpty()) {
    ShowLRElementInfo(element, text_doc, 0);
  } else {
    text_doc.Write("No layout recognition information!\r\n");
  }
}

int main(int argc, char* argv[]) {
  WString output_directory = output_path + L"layout_recognition/";
#if defined(_WIN32) || defined(_WIN64)
  _mkdir(String::FromUnicode(output_directory));
#else
  mkdir(String::FromUnicode(output_directory), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
#endif

  int err_ret = 0;
  SdkLibMgr sdk_lib_mgr;
  // Initialize library
  ErrorCode error_code = sdk_lib_mgr.Initialize();
  if (error_code != foxit::e_ErrSuccess) {
    return 1;
  }

  try {
    WString input_file = input_path + L"AboutFoxit.pdf";
    WString info_file = output_directory + L"layout_recognition_info.txt";

    PDFDoc doc(input_file);
    ErrorCode error_code = doc.Load();
    if (error_code != foxit::e_ErrSuccess) {
      printf("The Doc [%s] Error: %d\n", (const char*)String::FromUnicode(input_file), error_code);
      return 1;
    }

    LRContext context(doc.GetPage(0));

    Progressive progressive = context.StartParse(NULL);

    LRStructureElement root = context.GetRootElement();
    LRStructureElement::ElementType type = root.GetElementType();
    int nElemListSize = root.GetChildCount();
    cout << "RootElement GetChildCount: " << nElemListSize << endl;

    OutputAllLRElement(root, info_file);

    LRElementArray element_array;
    GetChildFromElement(root, element_array, LRStructureElement::e_ElementTypeSpan);
    cout << "LRElement::e_ElementTypeSpan count: " << element_array.GetSize() << endl;

    cout << "Layout recognition test." << endl;
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  } catch (...) {
    cout << "Unknown Exception" << endl;
    err_ret = 1;
  }

  return err_ret;
}
