public CBitmap CheckError()
{
CBitmap dst = new CBitmap(cj.sof.width, cj.sof.height);
for (int i = 0; i < dst.width; i++)
{
for (int j = 0; j < dst.height; j++)
{
if (error[i] == 1)
{
dst.SetPixel(i * 8, (i * 8) + 7, j * 8, (j * 8) + 7, 255, 0, 0);
}
}
}
return dst;
}
public void GetPicture(CBitmap src)
{
this.pictureBox1.Image = src.ToBitmap();
}
public MainForm(CBitmap src)
{
InitializeComponent();
this.pictureBox1.Image = src.ToBitmap();
}
static void Main(string[] args)
{
Cjpeg cj = null;
CBitmap cbmp = null;
WaterMarking wm = null;
int embed_bits = 8;
int offset = 0;
int color = 3;
string path = @"c:\dst.jpg";
string passwd = "aaaaa";
string dst_path;
string dst_files = "";
Cjpeg cj_raw = null;
//CJpegDecoderT.HuffmanDecode(ref cj_raw);
FileInfo fi = null;
Cjpeg temp = null;
int[] err;
switch (Console.ReadLine())
{
case "r":
for (int i = 0; i < args.Length; i++)
{
cj_raw = new Cjpeg(args[i]);
CJpegDecoderT.HuffmanDecode(ref cj_raw);
fi = new FileInfo(args[i]);
//Directory.CreateDirectory(args[i]+"\\");
dst_files = args[i];
dst_files += " ";
for (int bits = 0; bits < 65; bits++)
{
temp = new Cjpeg(cj_raw);
WaterMarkingT.Embed(ref temp, passwd, bits, 0, 3);
dst_path = args[i].Substring(0, args[i].Length - 4);
dst_path += "\\b";
for (int j = 0; j < 2 - bits.ToString().Length; j++)
{
dst_path += "0";
}
dst_path += bits.ToString() + ".jpg";
CJpegEncoderT.WriteFile(ref temp, dst_path);
Console.WriteLine(dst_path + " wrote");
err = WaterMarkingT.Check(ref temp, passwd, bits, 0, 3);
dst_files += dst_path + " ";
}
//Process exec = Process.Start("PSNR.exe", dst_files);
}
break;
case "c":
int check_count = int.Parse(Console.ReadLine());
BinaryWriter bw_csv = new BinaryWriter(File.Open(args[0]+".csv", FileMode.Create));
for (int i = 0; i < args.Length; i++)
{
int emb = int.Parse(args[i].Substring(args[i].Length - 6, 2));
cj_raw = new Cjpeg(args[i]);
cbmp = new CBitmap(new Bitmap(args[i]));
CJpegDecoderT.HuffmanDecode(ref cj_raw);
err = WaterMarkingT.Check(ref cj_raw, passwd, emb, 0, 3);
cbmp.CheckError(ref cj_raw, err, check_count);
//string dst_name = args[i] + args[i].Substring(args[i].Length - 6, 2) + ".bmp";
string dst_name = args[i] + ".bmp";
bw_csv.Write("" + i + "," + CountError(err, check_count) + "\n");
cbmp.ToBitmap().Save(dst_name);
Console.WriteLine(args[i] + " checked");
}
bw_csv.Close();
Console.ReadLine();
break;
}
return;
while (true)
{
Console.Write(">");
switch (Console.ReadLine())
{
case "o":
Console.WriteLine("enter filename");
path = Console.ReadLine();
if (path == "")
{
path = @"h:\lenna512.jpg";
Console.WriteLine("read \"h:\\lenna512.jpg\"");
}
try
{
cj = new Cjpeg(path);
cbmp = new CBitmap(new Bitmap(path));
CJpegDecoderT.HuffmanDecode(ref cj);
}
catch
{
Console.WriteLine("read error");
}
break;
case "w":
Console.WriteLine("enter filename");
dst_path = Console.ReadLine();
if (dst_path == "")
{
dst_path = @"h:\dst.jpg";
Console.WriteLine("write \"h:\\dst.jpg\"");
}
CJpegEncoderT.WriteFile(ref cj, dst_path);
break;
case "c":
wm.Extract(embed_bits, offset, color);
break;
case "e":
//wm.Embed(10);
WaterMarkingT.Embed(ref cj, "aaaaa",embed_bits, offset, color);
break;
case "t":
Cjpeg test = new Cjpeg(cj);
cj = test;
break;
case "q":
return;
}
}
//cjd.HuffmanDecode();
////wm.Embed(0);
//wm.Extract(0);
//wm.CheckError(ref cbmp);
//cje.WriteFile();
//Application.EnableVisualStyles();
//Application.SetCompatibleTextRenderingDefault(false);
//Application.Run(new MainForm(cbmp));
}
