// 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 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 add watermarks (in different types)
// into PDF files.

// 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/pdf/fs_watermark.h"

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

static const char* sn = "fpj42y0uOGt8cNp2iJUJdW5LNQGpJDawgJ5niOHYLl+0Emd0mMDhtA==";
static const char* key = "8f3gFsNtvW8N+Vea402IXXgHXIciB/Uw21sIKl1s0+MWvuGiCwz8VVTEwrNsIr1uDlREGorTmUmz7lEC6UcgWUsLrWhG6ezQc5PcgosYOpKsVTaOCyFdxIVCIj7QY9pjCyDgAV1S24QKmBz/ayQSbCepuRh2MBANca27kAN9Frb1qZXYI6Bmrr08YesEz5YKDHFyC04BbHhHiCKBeoQQXzf3O8KST9cFlArWUdIIz2/c3cMjj8uuxFiqVskJrKA1wxvsW4TzZEnDldm9Aye/mZIgHOhIlf8Hg1gIvnA0GCnWhUOs2UpRSFd8YuSnm0UPI7WR1K6vRdgeT/1sYh5bctEzdQh/0bMoN0LpciklqYPwpIFR8HU/zxfleXVwkxWKYkaPW7HxBAE6apzSd18nwRZ8fTsZjcYhTo64q4Jzukyr7m4ODp3tnWgrTgJKnTQYDEEEqK2cRA0ZIsKYTJoQ45gbVMpB7u7wLZ8kAIJbCmXrRPRh72fobCrlh67yJeH98JF8tB7pljmdGX7AD/0IDqtbcXcTUBxLskyaFPQHCuTzzMpLdoCxzGIsqVyobq7E3XKBC7kYcmdH79GwcsHl3BrYma1jTnRobtVOAWg9AgdI2yIhcZJSPrCn7SiQ4+73kbKHpQ2ekoZCF7Cwy6mQ3I+45yg1SLNB0K4YTtoz0t7FMDU1sLf0WsZ4GGlI67WdrOCrYKiXmHK/xjO5+20eR1nSYQlm3H/gyfCLHGdJtEzCB0K/ZwncTJ8ChQTdMHjtU5/rGK7BpDwbIewU2QItE935S9+BHfTQqcSlIvdiDlMtakKkfTCFkr5DUJPCD7XTxoaFN2s8fr9FvQPhhduKufiyl4Ds//jjiT0gRGSN6zDPfrEfJehqsSJgdqDOPlPHDCCj0BK34I8khlWiE8GVxb8qOWUXUfHY9eGLJgMkDiC6UtODXYCwEwNnrWpXGsQjuwtFmUBX7rtNrFr1uVqC1hmvo17a+cnvkRvKMuW9AxiFyak0WMVKT5zvzm9zBsBpAslZP17COOVYIhsDWnmVKt/i5AGFTCwlTlyqxi4RABXOhXHGRpTog+9wV6R2Z9VFjPSLDogFTcO5SkChkmQIKml+anIAnbLjll5Tgwgs7pqsAiCjetSJN0v3fLfcEeyonSnyN3s4VZrxYS1LE4S604EcW0O9gOXAVC6cmZDZkUvSfMh+UVj8pBhF7BSL6qUd5HEfObkL1SZkhh8EHFfe48elqVidbY3Lj9pq52rBnR03oWtUeMNkTYWncYTSzeTTP3jEnlbduUkKQOLVPdXAzPAfQ0FlB31RXm3a1y6VZcgNMsHF/WmPmd6Hwgc48+TUFY7XzLaBlEwcfolyAvFCfVOxO/m0b9pxEfCBYN7lhaMU9+Vjyujj15uCNDSpn4wcs32F2+wfY03pvsv+XMCrYZxvYYDHtw==";

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

void AddTextWatermark(PDFDoc doc, PDFPage page) {
  WatermarkSettings settings;
  settings.flags = WatermarkSettings::e_FlagASPageContents | WatermarkSettings::e_FlagOnTop;
  settings.offset_x = 0;
  settings.offset_y = 0;
  settings.opacity = 90;
  settings.position = common::e_PosTopRight;
  settings.rotation = -45.f;
  settings.scale_x = 1.f;
  settings.scale_y = 1.f;

  WatermarkTextProperties text_properties;
  text_properties.alignment = CommonDefines::e_AlignmentCenter;
  text_properties.color = 0xF68C21;
  text_properties.font_style = WatermarkTextProperties::e_FontStyleNormal;
  text_properties.line_space = 1;
  text_properties.font_size = 12.f;
  text_properties.font = Font(Font::e_StdIDTimesB);

  Watermark watermark(doc, L"Foxit PDF SDK\nwww.foxitsoftware.com", text_properties, settings);
  watermark.InsertToPage(page);

}

void AddBitmapWatermark(PDFDoc doc, PDFPage page, WString bitmap_file) {
  WatermarkSettings settings;
  settings.flags = WatermarkSettings::e_FlagASPageContents | WatermarkSettings::e_FlagOnTop;
  settings.offset_x = 0.f;
  settings.offset_y = 0.f;
  settings.opacity = 60;
  settings.position = common::e_PosCenterLeft;
  settings.rotation = 90.f;

  Image image(bitmap_file);
  Bitmap bitmap = image.GetFrameBitmap(0);
  settings.scale_x = page.GetHeight() * 1.0f / bitmap.GetWidth();
  settings.scale_y = settings.scale_x;
  Watermark watermark(doc, bitmap, settings);
  watermark.InsertToPage(page);
}

void AddImageWatermark(PDFDoc doc, PDFPage page, WString image_file) {
  WatermarkSettings settings;
  settings.flags = WatermarkSettings::e_FlagASPageContents | WatermarkSettings::e_FlagOnTop;
  settings.offset_x = 0.f;
  settings.offset_y = 0.f;
  settings.opacity = 20;
  settings.position = common::e_PosCenter;
  settings.rotation = 0.0f;

  Image image(image_file);
  Bitmap bitmap = image.GetFrameBitmap(0);
  settings.scale_x = page.GetWidth() * 0.618f / bitmap.GetWidth();
  settings.scale_y = settings.scale_x;

  Watermark watermark(doc, image, 0, settings);
  watermark.InsertToPage(page);
}

void AddSingleWatermark(PDFDoc doc, PDFPage page) {
  WatermarkSettings settings;
  settings.flags = WatermarkSettings::e_FlagASPageContents|WatermarkSettings::e_FlagOnTop;
  settings.offset_x = 0.f;
  settings.offset_y = 0.f;
  settings.opacity = 90;
  settings.position = common::e_PosBottomRight;
  settings.rotation = 0.0f;
  settings.scale_x = 0.1f;
  settings.scale_y = 0.1f;

  Watermark watermark(doc, page, settings);
  watermark.InsertToPage(page);
}

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

  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_file), error_code);
      return 1;
    }
    int page_count = doc.GetPageCount();
    for(int i=0; i<page_count; i++)
    {
      PDFPage page = doc.GetPage(i);
      page.StartParse(PDFPage::e_ParsePageNormal, NULL, false);

      AddTextWatermark(doc, page);
      WString wm_bmp = input_path + L"watermark.bmp";
      AddBitmapWatermark(doc, page,wm_bmp );
      WString wm_image = input_path + L"sdk.png";
      AddImageWatermark(doc, page, wm_image);
      AddSingleWatermark(doc, page);
    }

    doc.SaveAs(output_file, PDFDoc::e_SaveFlagNoOriginal);
    cout << "Add watermarks to PDF file." << endl;

    int nCount = doc.GetPageCount();
    for (int i = 0; i < nCount; i++)
    {
      PDFPage page = doc.GetPage(i);
      page.StartParse(PDFPage::e_ParsePageNormal, NULL, false);
      if (page.HasWatermark()) {
        page.RemoveAllWatermarks();
      }
    }

    doc.SaveAs(output_remove_file, PDFDoc::e_SaveFlagNoOriginal);
    cout << "Remove watermarks from PDF file." << endl;

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

  return err_ret;
}
