// 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 do optimize in PDF documents under a folder.

// Include Foxit SDK header files.
#include <time.h>
#include <iostream>
#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/optimization/fs_optimization.h"

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

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/optimization/");
#else
static WString output_path = WString::FromLocal("./output_files/");
static WString input_path = WString::FromLocal("./input_files/optimization/");
#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 Optimization_Pause : public PauseCallback
{
public:
    Optimization_Pause(int pause_count_limit = 0, bool always_pause = false)
        :pause_count_(0)
        ,pause_count_limit_(pause_count_limit)
        ,always_pause_(always_pause)
    {

    }

    virtual FX_BOOL NeedToPauseNow()
    {
        if (always_pause_) return true;
        if (pause_count_< pause_count_limit_)
        {
            pause_count_ ++;
            return 1;
        }
        else{
            pause_count_ = 0;
            return 0; // This is to test a case: valid PauseCallback but needParseNow() will always return FALSE.
        }
    }

    void  ClearCount()
    {
        pause_count_ = 0;
    }

private:
    int pause_count_limit_;
    int pause_count_;
    bool always_pause_;
};


int main(int argc, char *argv[])
{
  WString output_directory = output_path + L"optimization/";
#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;
  }
  cout << "Optimized Start : Image Compression." << endl;
  WString input_file = input_path +  L"[Optimize]ImageCompression.pdf";
  try
  {
      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_path), error_code);
          return 1;
      }
      Optimization_Pause pause(0, true);
      addon::optimization::OptimizerSettings settings;
      settings.SetOptimizerOptions(foxit::addon::optimization::OptimizerSettings::e_OptimizerCompressImages);
      common::Progressive progressive = addon::optimization::Optimizer::Optimize(doc, settings, &pause);
      Progressive::State progress_state = Progressive::e_ToBeContinued;
      while (Progressive::e_ToBeContinued == progress_state) {
          progress_state = progressive.Continue();
          int percent = progressive.GetRateOfProgress();
          String res_string;
          res_string.Format("Optimize progress percent: %d %", percent);
          std::cout << res_string << std::endl;
      }
      if (Progressive::e_Finished == progress_state)
      {
          doc.SaveAs(output_directory + L"ImageCompression_Optimized.pdf", foxit::pdf::PDFDoc::e_SaveFlagRemoveRedundantObjects);
      }
  } catch (...) {
    cout << "Unknown Exception" << endl;
    return 1;
  }
  cout << "Optimized Finsih : Image Compression." << endl;

  cout << "Optimized Start : Clean up." << endl;
  input_file = input_path + L"[Optimize]Cleanup.pdf";
  try {
    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_path), error_code);
      return 1;
    }
    Optimization_Pause pause(0, true);
    addon::optimization::OptimizerSettings settings;
    settings.SetOptimizerOptions(foxit::addon::optimization::OptimizerSettings::e_OptimizerCleanUp);
    settings.SetCleanUpOptions(optimization::OptimizerSettings::e_CleanUpRemoveInvalidBookmarks | optimization::OptimizerSettings::e_CleanUpRemoveInvalidLinks
      | optimization::OptimizerSettings::e_CleanUpUseFlateForNonEncodedStream | optimization::OptimizerSettings::e_CleanUpUseFlateInsteadOfLZW);
    common::Progressive progressive = addon::optimization::Optimizer::Optimize(doc, settings, &pause);
    Progressive::State progress_state = Progressive::e_ToBeContinued;
    while (Progressive::e_ToBeContinued == progress_state) {
      progress_state = progressive.Continue();
      int percent = progressive.GetRateOfProgress();
      String res_string;
      res_string.Format("Optimize progress percent: %d %", percent);
      std::cout << res_string << std::endl;
    }
    if (Progressive::e_Finished == progress_state) {
      doc.SaveAs(output_directory + L"Cleanup_Optimized.pdf", foxit::pdf::PDFDoc::e_SaveFlagRemoveRedundantObjects);
    }
  } catch (...) {
    cout << "Unknown Exception" << endl;
    return 1;
  }
  cout << "Optimized Finsih : Clean up." << endl;

  cout << "Optimized Start : Discard Objects." << endl;
  input_file = input_path + L"[Optimize]DiscardObjects.pdf";
  try {
      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_path), error_code);
          return 1;
      }
      Optimization_Pause pause(0, true);
      addon::optimization::OptimizerSettings settings;
      settings.SetOptimizerOptions(foxit::addon::optimization::OptimizerSettings::e_OptimizerDiscardObjects);
      settings.SetDiscardObjectsOptions(optimization::OptimizerSettings::e_DiscardObjectsBookmarks | optimization::OptimizerSettings::e_DiscardObjectsEmbeddedPageThumbnails
          | optimization::OptimizerSettings::e_DiscardObjectsEmbeddedPrintSettings | optimization::OptimizerSettings::e_DiscardObjectsFlattenFormFields 
          | optimization::OptimizerSettings::e_DiscardObjectsFormActions | optimization::OptimizerSettings::e_DiscardObjectsJavaScriptActions);

      common::Progressive progressive = addon::optimization::Optimizer::Optimize(doc, settings, &pause);
      Progressive::State progress_state = Progressive::e_ToBeContinued;
      while (Progressive::e_ToBeContinued == progress_state) {
          progress_state = progressive.Continue();
          int percent = progressive.GetRateOfProgress();
          String res_string;
          res_string.Format("Optimize progress percent: %d %", percent);
          std::cout << res_string << std::endl;
      }
      if (Progressive::e_Finished == progress_state) {
          doc.SaveAs(output_directory + L"DiscardObjects_Optimized.pdf", foxit::pdf::PDFDoc::e_SaveFlagRemoveRedundantObjects);
      }
  }
  catch (...) {
      cout << "Unknown Exception" << endl;
      return 1;
  }
  cout << "Optimized Finsih : Discard Objects." << endl;
  return err_ret;
}