/*public DoubleImage KIMembed(DoubleImage image, Watermark initialWatermark)
* {
* DoubleImage copy = image.Copy();
* int maxWatermarkLength = GetOptimalWatermarkLength(copy);
* if (maxWatermarkLength < initialWatermark.Length)
* throw new Exception("Watermark length is too big to embed it in the image.");
*
* DoubleImage result = KIMembedComponent(image, initialWatermark, DoubleImage.ColorComponent.Blue);
* if (p < initialWatermark.Length)
* result = KIMembedComponent(result, initialWatermark, DoubleImage.ColorComponent.Red);
* if (p < initialWatermark.Length)
* result = KIMembedComponent(result, initialWatermark, DoubleImage.ColorComponent.Green);
* p = 0;
* return result;
* }*/
private DoubleImage KIMembedComponent(DoubleImage image, Watermark initialWatermark, DoubleImage.ColorComponent colorComponent)
{
// Get blue component from image
double[,] blueComponent = image.GetColorComponent(colorComponent);
// Transfrom blue component with wavelet transformation
double[,] waveletTransformedComponent = Wavelet.Transform(blueComponent, 3);
// Get coefficients of transformation
double[,] approx3 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Approximation, 3);
double[,] detailHor3 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 3);
double[,] detailVert3 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 3);
double[,] detailDiag3 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 3);
double[,] detailHor2 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 2);
double[,] detailVert2 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 2);
double[,] detailDiag2 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 2);
double[,] detailHor1 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 1);
double[,] detailVert1 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 1);
double[,] detailDiag1 = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 1);
// Get thresholds
double threshold1 = Threshold.GetThreshold(detailHor1, detailVert1, detailDiag1);
double threshold2 = Threshold.GetThreshold(detailHor2, detailVert2, detailDiag2);
double threshold3 = Threshold.GetThreshold(detailHor3, detailVert3, detailDiag3, approx3);
Watermark watermark = TransformWatermark(initialWatermark);
// Embed watermark into coefficients
//int p = 0;
p = EmbedInCoef(coefApprox, ref approx3, threshold3, watermark, p);
p = EmbedInCoef(coef3Level, ref detailHor3, threshold3, watermark, p);
p = EmbedInCoef(coef3Level, ref detailVert3, threshold3, watermark, p);
p = EmbedInCoef(coef3Level, ref detailDiag3, threshold3, watermark, p);
p = EmbedInCoef(coef2Level, ref detailHor2, threshold2, watermark, p);
p = EmbedInCoef(coef2Level, ref detailVert2, threshold2, watermark, p);
p = EmbedInCoef(coef2Level, ref detailDiag2, threshold2, watermark, p);
p = EmbedInCoef(coef1Level, ref detailHor1, threshold1, watermark, p);
p = EmbedInCoef(coef1Level, ref detailVert1, threshold1, watermark, p);
p = EmbedInCoef(coef1Level, ref detailDiag1, threshold1, watermark, p);
Console.WriteLine("watermark length {0}", watermark.Length);
Console.WriteLine("p {0}", p);
// Restore blue component from coefficients
double[,] componentWithWatermark = new double[blueComponent.GetLength(0), blueComponent.GetLength(1)];
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Approximation, 3, approx3);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Horizontal, 3, detailHor3);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Vertical, 3, detailVert3);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Diagonal, 3, detailDiag3);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Horizontal, 2, detailHor2);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Vertical, 2, detailVert2);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Diagonal, 2, detailDiag2);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Horizontal, 1, detailHor1);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Vertical, 1, detailVert1);
componentWithWatermark = Wavelet.SetCoefficient(componentWithWatermark, Wavelet.Coefficients.Diagonal, 1, detailDiag1);
double[,] untrasformedComponent = Wavelet.Untransfrom(componentWithWatermark, 3);
// Embed blue component into initial image
image = image.UpdateColorComponent(colorComponent, untrasformedComponent);
return(image);
}
static void Main(string[] args)
{
var keyWord = "roses";
var inPicture = keyWord + ".jpg";
var changedPicture = keyWord + "_changed.jpg";
var qr = "QRcode_phone.png";
var qr_extracted = keyWord + "_QR.jpg";
DoubleImage image;
using (Bitmap imageBM = new Bitmap(inPicture)) //lena_changed.jpg
{
image = new DoubleImage(imageBM);
}
/*Watermark watermark;
* using (Bitmap imageBM = new Bitmap(qr)) //lena_changed.jpg
* {
* DoubleImage imageTemp = new DoubleImage(imageBM);
* watermark = new Watermark(imageTemp);
* }
*
* JRKimAlgorithm algorithm = new JRKimAlgorithm();
* DoubleImage changed = algorithm.KIMembed(image, watermark);
* Bitmap result = changed.ToBitmap(1, 0);
* result.Save(changedPicture); */
DoubleImage black;
using (Bitmap imageBM = new Bitmap("black_cr.jpg")) //lena_changed.jpg
{
black = new DoubleImage(imageBM);
}
double[,] blue = image.GetColorComponent(DoubleImage.ColorComponent.Blue);
double[,] red = image.GetColorComponent(DoubleImage.ColorComponent.Red);
double[,] green = image.GetColorComponent(DoubleImage.ColorComponent.Green);
/*double[,] changed_blue = Wavelet.Transform(blue, 1);
* double[,] changed_green = Wavelet.Transform(green, 1);
* double[,] changed_red = Wavelet.Transform(red, 1);*/
//black.UpdateColorComponent(DoubleImage.ColorComponent.Red, red);
black.UpdateColorComponent(DoubleImage.ColorComponent.Red, red);
// black.UpdateColorComponent(DoubleImage.ColorComponent.Green, green);
Bitmap result = black.ToBitmap(1, 0);
result.Save("roses_red.jpg");
/*DoubleImage initialImage, changedImage;
*
* using (Bitmap imageBitmap = new Bitmap(inPicture)) //lena_changed.jpg
* {
* initialImage = new DoubleImage(imageBitmap);
* }
*
* using (Bitmap imageBitmap = new Bitmap(changedPicture))
* {
* changedImage = new DoubleImage(imageBitmap);
* }
*
* JRKimAlgorithm algorithm = new JRKimAlgorithm();
* DoubleImage watermark = algorithm.KIMextract(initialImage, changedImage, 100, 100);
* Bitmap result = watermark.ToBitmap(1,0);
* result.Save(qr_extracted);*/
Console.WriteLine("Done");
Console.ReadKey();
}
