using System;
using System.Drawing;
using System.Drawing.Imaging;

public class DCTWatermarker
{
    public void EmbedWatermark(string imagePath, string outputImagePath, string watermarkText)
    {
        Bitmap image = new Bitmap(imagePath);
        int width = image.Width;
        int height = image.Height;

        // 將圖像分塊為8x8的DCT塊
        for (int y = 0; y < height; y += 8)
        {
            for (int x = 0; x < width; x += 8)
            {
                Rectangle blockRect = new Rectangle(x, y, 8, 8);
                Bitmap block = image.Clone(blockRect, PixelFormat.Format24bppRgb);

                // 對每個(gè)塊進(jìn)行DCT變換
                double[,] dcta = DCT(block);

                // 在DCT系數(shù)中嵌入水印（例如修改特定位置的系數(shù)）
                for (int i = 0; i < 8; i++)
                {
                    for (int j = 0; j < 8; j++)
                    {
                        if (i * j < watermarkText.Length) // 示例邏輯
                        {
                            dcta[i, j] += 5; // 調(diào)整系數(shù)值
                        }
                    }
                }

                // IDCT逆變換并覆蓋原塊
                Bitmap modifiedBlock = IDCT(dcta);
                Graphics g = Graphics.FromImage(image);
                g.DrawImage(modifiedBlock, x, y);
                g.Dispose();
                modifiedBlock.Dispose();
            }
        }

        image.Save(outputImagePath, ImageFormat.Jpeg);
        image.Dispose();
    }

    // 簡化的DCT和IDCT實(shí)現(xiàn)（需完整數(shù)學(xué)公式支持）
    private double[,] DCT(Bitmap block) { /* ... */ }
    private Bitmap IDCT(double[,] dcta) { /* ... */ }
}

public void EmbedLSBWatermark(string imagePath, string outputImagePath, byte[] watermark)
{
    Bitmap image = new Bitmap(imagePath);
    int index = 0;

    for (int y = 0; y < image.Height && index < watermark.Length; y++)
    {
        for (int x = 0; x < image.Width && index < watermark.Length; x++)
        {
            Color pixel = image.GetPixel(x, y);
            byte r = (byte)((pixel.R & 0xFE) | ((watermark[index] >> 7) & 0x01)); // 修改第1位
            byte g = (byte)((pixel.G & 0xFE) | ((watermark[index] >> 6) & 0x01)); // 修改第2位
            byte b = (byte)((pixel.B & 0xFE) | ((watermark[index] >> 5) & 0x01)); // 修改第3位
            index += 1;
            image.SetPixel(x, y, Color.FromArgb(r, g, b));
        }
    }

    image.Save(outputImagePath, ImageFormat.Png);
    image.Dispose();
}

// 調(diào)用外部Python服務(wù)生成水印
ProcessStartInfo psi = new ProcessStartInfo("python", "generate_watermark.py")
{
    RedirectStandardOutput = true,
    UseShellExecute = false,
    CreateNoWindow = true
};
Process process = Process.Start(psi);
string watermarkData = process.StandardOutput.ReadToEnd(); // 獲取水印數(shù)據(jù)

// 將水印嵌入圖像（結(jié)合DCT或其他算法）
EmbedWatermarkUsingDCT(imagePath, watermarkData);