public DoubleImage ToDoubleImage(int width, int height)
{
DoubleImage image = new DoubleImage(width, height);
for (int i = 0; i < image.Height; i++)
{
for (int j = 0; j < image.Width; j++)
{
int index = i * image.Width + j;
//black
if (this[index] == 0)
{
image.SetPixel(i, j, 0, 0, 0);
}
//white
if (this[index] == 1)
{
image.SetPixel(i, j, 255, 255, 255);
}
}
}
return(image);
}
static void Main(string[] args)
{
var keyWord = "lena";
var inPicture = keyWord + ".jpg";
var changedPicture = keyWord + "_changed.jpg";
var qr = "QRcode_phone.png";
var qr_extracted = keyWord + "_QR.jpg";
DoubleImage initialImage, changedImage;
using (Bitmap imageBitmap = new Bitmap("landscape.jpg")) //lena_changed.jpg
{
initialImage = new DoubleImage(imageBitmap);
}
using (Bitmap imageBitmap = new Bitmap("landscape_changed_cr.png"))
{
changedImage = new DoubleImage(imageBitmap);
}
DoubleImage full = DoubleImage.FillWithSmallerImage(initialImage, changedImage);
Bitmap result = full.ToBitmap(1, 0);
result.Save("full.jpg");
Console.WriteLine("Done");
Console.ReadKey();
}
public int GetOptimalWatermarkLength(DoubleImage image)
{
Watermark watermark = new Watermark(image.Height * image.Width);
KIMembedComponent(image, watermark, DoubleImage.ColorComponent.Blue);
int result = p;
p = 0;
return(result);
}
public int GetMaxWatermarkLength(DoubleImage image)
{
Watermark watermark = new Watermark(image.Height * image.Width);
KIMembedComponent(image, watermark, DoubleImage.ColorComponent.Blue);
KIMembedComponent(image, watermark, DoubleImage.ColorComponent.Red);
KIMembedComponent(image, watermark, DoubleImage.ColorComponent.Green);
int maxWatermarkLength = p;
p = 0;
return(maxWatermarkLength);
}
public DoubleImage Copy()
{
DoubleImage copy = new DoubleImage(Width, Height);
for (int i = 0; i < Height; i++)
{
for (int j = 0; j < Width; j++)
{
DoublePixel pixel = this._pixels[i, j];
copy._pixels[i, j] = new DoublePixel(pixel.Red, pixel.Green, pixel.Blue);
}
}
return(copy);
}
public DoubleImage KIMextract(DoubleImage initialImage, DoubleImage changedImage, int widthWatermark, int heigthWatermark)
{
Watermark watermark = KIMextractComponent(initialImage, changedImage, widthWatermark, heigthWatermark, DoubleImage.ColorComponent.Blue);
if (p < widthWatermark * heigthWatermark)
{
int startIndex = p;
watermark = watermark.Append(KIMextractComponent(initialImage, changedImage, widthWatermark, heigthWatermark, DoubleImage.ColorComponent.Red), startIndex);
}
if (p < widthWatermark * heigthWatermark)
{
int startIndex = p;
watermark = watermark.Append(KIMextractComponent(initialImage, changedImage, widthWatermark, heigthWatermark, DoubleImage.ColorComponent.Green), startIndex);
}
p = 0;
return(watermark.ToDoubleImage(widthWatermark, heigthWatermark));
}
public DoubleImage UpdateColorComponent(ColorComponent component, double[,] colorComponent)
{
DoubleImage updatedImage = this;
switch (component)
{
case ColorComponent.Red:
for (int i = 0; i < this.Height; i++)
{
for (int j = 0; j < this.Width; j++)
{
DoublePixel initialPixel = this.GetPixel(i, j);
updatedImage.SetPixel(i, j, colorComponent[i, j], initialPixel.Green, initialPixel.Blue);
}
}
break;
case ColorComponent.Green:
for (int i = 0; i < this.Height; i++)
{
for (int j = 0; j < this.Width; j++)
{
DoublePixel initialPixel = this.GetPixel(i, j);
updatedImage.SetPixel(i, j, initialPixel.Red, colorComponent[i, j], initialPixel.Blue);
}
}
break;
case ColorComponent.Blue:
for (int i = 0; i < this.Height; i++)
{
for (int j = 0; j < this.Width; j++)
{
DoublePixel initialPixel = this.GetPixel(i, j);
updatedImage.SetPixel(i, j, initialPixel.Red, initialPixel.Green, colorComponent[i, j]);
}
}
break;
}
return(updatedImage);
}
public Watermark(DoubleImage image)
{
Length = image.Width * image.Height;
_bits = new int[Length];
for (int i = 0; i < image.Height; i++)
{
for (int j = 0; j < image.Width; j++)
{
//black
if (image.GetPixel(i, j).Blue == 0 && image.GetPixel(i, j).Red == 0 && image.GetPixel(i, j).Green == 0)
{
_bits[i * image.Width + j] = 0;
}
//white
if (image.GetPixel(i, j).Blue == 255 && image.GetPixel(i, j).Red == 255 && image.GetPixel(i, j).Green == 255)
{
_bits[i * image.Width + j] = 1;
}
}
}
}
public DoubleImage KIMembed(DoubleImage image, Watermark initialWatermark)
{
DoubleImage copy = image.Copy();
int maxWatermarkLength = GetMaxWatermarkLength(copy);
if (maxWatermarkLength < initialWatermark.Length)
{
throw new Exception("Размер ЦВЗ слишком большой для данного изображения. Проверьте, максимально возможный размер ЦВЗ.");
}
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);
}
/*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);
}
public DoubleImage KIMembed(DoubleImage image, Watermark initialWatermark)
{
DoubleImage result = KIMembedComponent(image, initialWatermark, DoubleImage.ColorComponent.Green);
return(result);
}
private Watermark KIMextractComponent(DoubleImage initialImage, DoubleImage changedImage, int widthWatermark, int heigthWatermark, DoubleImage.ColorComponent colorComponent)
{
// Get blue component from changed image
double[,] blueComponentChanged = changedImage.GetColorComponent(colorComponent);
// Transfrom blue component of changed image with wavelet transformation
double[,] waveletTransformedChangedComponent = Wavelet.Transform(blueComponentChanged, 3);
// Get coefficients of transformation
double[,] approx3 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Approximation, 3);
double[,] detailHor3 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Horizontal, 3);
double[,] detailVert3 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Vertical, 3);
double[,] detailDiag3 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Diagonal, 3);
double[,] detailHor2 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Horizontal, 2);
double[,] detailVert2 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Vertical, 2);
double[,] detailDiag2 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Diagonal, 2);
double[,] detailHor1 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Horizontal, 1);
double[,] detailVert1 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Vertical, 1);
double[,] detailDiag1 = Wavelet.GetCoefficient(waveletTransformedChangedComponent, Wavelet.Coefficients.Diagonal, 1);
// Get blue component from initial image
double[,] blueComponentInitial = initialImage.GetColorComponent(colorComponent);
// Transfrom blue component of initial image with wavelet transformation
double[,] waveletTransformedComponent = Wavelet.Transform(blueComponentInitial, 3);
// Get coefficients of transformation
double[,] approx3Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Approximation, 3);
double[,] detailHor3Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 3);
double[,] detailVert3Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 3);
double[,] detailDiag3Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 3);
double[,] detailHor2Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 2);
double[,] detailVert2Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 2);
double[,] detailDiag2Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 2);
double[,] detailHor1Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Horizontal, 1);
double[,] detailVert1Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Vertical, 1);
double[,] detailDiag1Initial = Wavelet.GetCoefficient(waveletTransformedComponent, Wavelet.Coefficients.Diagonal, 1);
// Get thresholds from initial image
double threshold1 = Threshold.GetThreshold(detailHor1Initial, detailVert1Initial, detailDiag1Initial);
double threshold2 = Threshold.GetThreshold(detailHor2Initial, detailVert2Initial, detailDiag2Initial);
double threshold3 = Threshold.GetThreshold(detailHor3Initial, detailVert3Initial, detailDiag3Initial, approx3Initial);
int watermarkSize = widthWatermark * heigthWatermark;
Watermark watermark = new Watermark(watermarkSize);
// Extract watermark from coefficients
//int p = 0;
p = ExtractFromCoef(coefApprox, approx3Initial, approx3, threshold3, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef3Level, detailHor3Initial, detailHor3, threshold3, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef3Level, detailVert3Initial, detailVert3, threshold3, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef3Level, detailDiag3Initial, detailDiag3, threshold3, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef2Level, detailHor2Initial, detailHor2, threshold2, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef2Level, detailVert2Initial, detailVert2, threshold2, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef2Level, detailDiag2Initial, detailDiag2, threshold2, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef1Level, detailHor1Initial, detailHor1, threshold1, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef1Level, detailVert1Initial, detailVert1, threshold1, p, watermarkSize, ref watermark);
p = ExtractFromCoef(coef1Level, detailDiag1Initial, detailDiag1, threshold1, p, watermarkSize, ref watermark);
watermark = UnTransformWatermark(watermark);
return(watermark);
//DoubleImage watermarkImage = watermark.ToDoubleImage(widthWatermark, heigthWatermark);
//return watermarkImage;
}
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();
}
public static DoubleImage FillWithSmallerImage(DoubleImage bigImage, DoubleImage smallImage)
{
if (smallImage.Height > bigImage.Height || smallImage.Width > bigImage.Width)
{
return(smallImage);
}
double[,] greenSmall = smallImage.GetColorComponent(ColorComponent.Red);
double[,] greenBig = bigImage.GetColorComponent(ColorComponent.Red);
int jEnd = greenSmall.GetLength(1) - 1;
int iEnd = greenSmall.GetLength(0) - 1;
int countOfChecking = Math.Min(smallImage.Height, smallImage.Width) / 2;
int iStart = 0;
int jStart = 0;
bool isFound = false;
for (int i = bigImage.Height - 1; i >= 0; i--)
{
if (isFound)
{
break;
}
for (int j = bigImage.Width - 1; j >= 0; j--)
{
if (greenBig[i, j] == greenSmall[iEnd, jEnd])
{
isFound = true;
for (int k = 1; k < countOfChecking; k++)
{
if (greenBig[i - k, j - k] != greenSmall[iEnd - k, jEnd - k])
{
isFound = false;
break;
}
}
if (isFound)
{
iStart = i - smallImage.Height;
jStart = j - smallImage.Width;
Console.WriteLine("iStart {0}", iStart);
Console.WriteLine("jStart {0}", jStart);
Console.WriteLine("iEnd {0}", iEnd);
Console.WriteLine("jEnd {0}", jEnd);
break;
}
}
}
}
DoubleImage updateSmallImage = new DoubleImage(bigImage.Width, bigImage.Height);
for (int i = 0; i < bigImage.Width; i++)
{
for (int j = 0; j < bigImage.Height; j++)
{
if (iStart <= i && i <= iStart + iEnd && jStart <= j && j <= jStart + jEnd)
{
DoublePixel pixel = smallImage.GetPixel(i - iStart, j - jStart);
updateSmallImage.SetPixel(i, j, pixel.Red, pixel.Green, pixel.Blue);
}
else
{
updateSmallImage.SetPixel(i, j, 255, 255, 255);
}
}
}
return(updateSmallImage);
}
