// 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 tag a PDF document.

// Include Foxit SDK header files.
#include <time.h>
#include <iostream>
#include <map>
#if defined(_WIN32) || defined(_WIN64)
#include<direct.h>
#else
#include <sys/stat.h>
#endif
#include "../../../include/common/fs_common.h"
#include "../../../include/pdf/fs_pdfdoc.h"
#include "../../../include/pdf/fs_pdfpage.h"
#include "../../../include/addon/accessibility/fs_taggedpdf.h"

using namespace std;
using namespace foxit;
using namespace foxit::common;
using foxit::common::Library;
using namespace pdf;
using namespace foxit::addon::accessibility;

static const char* sn = "Rew1vB0TqoMIxX444UcpScARWdhgJpsJfzmyMLR32a0+EgrCjp2NaQ==";
static const char* 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=";

#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_;
};

static String GetReportCategoryString(TaggedPDFCallback::ReportCategory type) {
  String sFormat = "";
  switch (type) {
  case TaggedPDFCallback::e_ReportCategoryRegion:
    sFormat = "Region";
    break;
  case TaggedPDFCallback::e_ReportCategoryArtifact:
    sFormat = "Artifact";
    break;
  case TaggedPDFCallback::e_ReportCategoryParagraph:
    sFormat = "Paragraph";
    break;
  case TaggedPDFCallback::e_ReportCategoryListItem:
    sFormat = "List Item";
    break;
  case TaggedPDFCallback::e_ReportCategoryFigure:
    sFormat = "Figure";
    break;
  case TaggedPDFCallback::e_ReportCategoryTable:
    sFormat = "Table";
    break;
  case TaggedPDFCallback::e_ReportCategoryTableRow:
    sFormat = "Table Row";
    break;
  case TaggedPDFCallback::e_ReportCategoryTableHeader:
    sFormat = "Table Header";
    break;
  case TaggedPDFCallback::e_ReportCategoryTocItem:
    sFormat = "Toc Item";
    break;
  default:
    break;
  }
  return sFormat;
}

static String GetReportConfidenceString(TaggedPDFCallback::ReportConfidence type) {
  String sFormat = "";
  switch (type) {
  case TaggedPDFCallback::e_ReportConfidenceHigh:
    sFormat = "High";
    break;
  case TaggedPDFCallback::e_ReportConfidenceMediumHigh:
    sFormat = "Medium High";
    break;
  case TaggedPDFCallback::e_ReportConfidenceMedium:
    sFormat = "Medium";
    break;
  case TaggedPDFCallback::e_ReportConfidenceMediumLow:
    sFormat = "Medium Low";
    break;
  case TaggedPDFCallback::e_ReportConfidenceLow:
    sFormat = "Low";
    break;
  default:
    break;
  }
  return sFormat;
}

struct AutoTag_ReportElemRectCon {
  CFX_FloatRect rcRect;
  TaggedPDFCallback::ReportConfidence eConfidence;
};

typedef vector<AutoTag_ReportElemRectCon*> ReportElemRectConArray;
typedef map<TaggedPDFCallback::ReportCategory, ReportElemRectConArray*> ReportElemMap;
typedef map<int, ReportElemMap*> ReportResultPagesMap;
class TaggedPDFCallbackImpl : public TaggedPDFCallback {
public:
  TaggedPDFCallbackImpl() {}

  ~TaggedPDFCallbackImpl() {
    ResetResult();
  }
  virtual void Release() {
    delete this;
  }

  virtual void Report(ReportCategory category, ReportConfidence confidence, int page_index, const RectF& rect) {
    cout << "Page Index: " << page_index << ", ReportCategory: " << (FX_LPCSTR)GetReportCategoryString(category)
      << ", ReportConfidence: " << (FX_LPCSTR)GetReportConfidenceString(confidence) << ", Rect: [" << rect.left
      << ", " << rect.top << ", " << rect.right << ", " << rect.bottom << "]" << endl;

    AutoTag_ReportElemRectCon* pRectCon = new AutoTag_ReportElemRectCon();
    pRectCon->rcRect = rect;
    pRectCon->eConfidence = confidence;

    ReportElemMap* pElemMap = NULL;
    ReportElemRectConArray* pArray = NULL;

    ReportResultPagesMap::iterator it = result_map_.find(page_index);
    if (it == result_map_.end()) {
      pArray = new ReportElemRectConArray();
      pArray->push_back(pRectCon);
      pElemMap = new ReportElemMap();
      pElemMap->insert(ReportElemMap::value_type(category, pArray));
      result_map_.insert(ReportResultPagesMap::value_type(page_index, pElemMap));
    }
    else {
      pElemMap = it->second;
      ReportElemMap::iterator iter = pElemMap->find(category);
      if (iter != pElemMap->end()) {
        pArray = iter->second;
        pArray->push_back(pRectCon);
      }
      else {
        pArray = new ReportElemRectConArray();
        pArray->push_back(pRectCon);
        pElemMap->insert(ReportElemMap::value_type(category, pArray));
      }
    }
  }

  ReportResultPagesMap& GetResult() {
    return result_map_;
  }

  void ResetResult() {
    for (auto iter = result_map_.begin(); iter != result_map_.end(); ++iter) {
      auto item = iter->second;
      for (auto iter_category = item->begin(); iter_category != item->end(); ++iter_category) {
        delete iter_category->second;
      }
      item->clear();
      delete item;
    }
    result_map_.clear();
  }

private:
  ReportResultPagesMap result_map_;
};

int main(int argc, char *argv[])
{
  WString output_directory = output_path + L"taggedpdf/";
#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;
  WString input_file = input_path + L"AboutFoxit.pdf";
  SdkLibMgr sdk_lib_mgr;
  // Initialize library
  ErrorCode error_code = sdk_lib_mgr.Initialize();
  if (error_code != foxit::e_ErrSuccess) {
    return 1;
  }

  try {
    {
      WString output_file_path = output_directory + L"TaggedPdf_StartAutoTagged.pdf";

      PDFDoc pdfDoc(input_file);
      pdfDoc.Load();
      TaggedPDF taggedpdf(pdfDoc);
      Progressive progressive = taggedpdf.StartTagDocument(NULL);
      Progressive::State progressState = Progressive::e_ToBeContinued;
      while (Progressive::e_ToBeContinued == progressState)
        progressState = progressive.Continue();

      pdfDoc.SaveAs(output_file_path);
    }
    {//use callback
      WString output_file_path = output_directory + L"TaggedPdf_StartAutoTagged_SetCallback.pdf";

      PDFDoc pdfDoc(input_file);
      pdfDoc.Load();
      TaggedPDF taggedpdf(pdfDoc);
      TaggedPDFCallbackImpl* callback = new TaggedPDFCallbackImpl();
      taggedpdf.SetCallback(callback);
      Progressive progressive = taggedpdf.StartTagDocument(NULL);
      auto progressState = Progressive::e_ToBeContinued;
      while (Progressive::e_ToBeContinued == progressState)
        progressState = progressive.Continue();
      pdfDoc.SaveAs(output_file_path);
    }
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }
  catch(...)
  {
    cout << "Unknown Exception" << endl;
    err_ret = 1;
  }

  return err_ret;
}

