www.pudn.com > EblindDlc.rar > test.cpp
#include "stdio.h"
#include "iostream.h"
#include "stdlib.h"
//函数声明
unsigned char* ProdRandTem(FILE* f1);
unsigned char* WaterMarking(unsigned char* array1);
void SaveFile(unsigned char* arr);
void LoadFile();
void LoadWaterFile();
void CheckWater(unsigned char* arr1);
int CheckResult(unsigned char z);
//全局变量声明
unsigned char* array=new unsigned char [256*256];
unsigned char* CTarray=new unsigned char [256*256];
unsigned char Zlc;
FILE* f1;
FILE* f2;
unsigned char a[54];
void main()
{
unsigned char *array1,*array2;
int choice,result;
cout <<"请输入choice的值(1表示给Lean图象嵌入水印,2表示从嵌入水印的图象中提取水印):";
cin >>choice;
switch(choice)
{
case 1:
LoadFile();
array1=ProdRandTem(f1);
array2=WaterMarking(array1);
SaveFile(array2);
break;
case 2:
LoadWaterFile();
CheckWater(array);
result=CheckResult(Zlc);
if(result==0)
cout <<"经检测,图象中没有嵌入水印!"<>ch;
if((f1=fopen(ch,"rb+"))==NULL)
{
cout <<"can't open the file!"<0.7)
return 1;
if(z<-0.7)
return -1;
return 0;
}