C++随机马赛克图程序
效果：
或者灰度，cell大小可调
代码：
#include <opencv2\opencv.hpp>
#include <Windows.h>
struct parameters
{
parameters()
{
size=cv::Size(640,480);
fps=25;
nF=50;
save="";
crossSize= cv::Size(15, 5);
crossColor = 0;
show=true;
seed=cv::getTickCount();
color=false;
cellSize = cv::Size(5, 5);
}
cv::Size size;
int fps;
int nF;
std::string save;
cv::Size crossSize;
uchar crossColor;
bool show;
int64 seed;
bool color;
cv::Size cellSize;
void setParas(
cv::Size _size,
int _fps,
int _nF,
std::string _save,
cv::Size _crossSize,
uchar _crossColor,
bool _show,
int64 _seed,
bool _color,
cv::Size _cellSize)
{
size = _size ;
fps = _fps ;
nF = _nF ;
save = _save ;
crossSize = _crossSize ;
crossColor = _crossColor ;
show = _show ;
seed = _seed ;
color = _color ;
cellSize = _cellSize ;
}
void showParas()
{
std::cout << "size: " << size << std::endl;
std::cout << "fps: " << fps << std::endl;
std::cout << "nF: " << nF << std::endl;
std::cout << "save: " << save << std::endl;
std::cout << "crossSize: " << crossSize << std::endl;
std::cout << "crossColor: " << crossColor << std::endl;
std::cout << "show: " << (show? "TRUE" : "FALSE") << std::endl;
std::cout << "seed: " << seed << std::endl;
std::cout << "color: " << color << std::endl;
std::cout << "cellSize: " << cellSize << std::endl;
}
};
void help()
{
std::cout << "* -help : .\n"
<< " randomVideo.exe -help | to see how to use. \n"
<< " randomVideo.exe -size=(640,580) | to set video size. \n"
<< " randomVideo.exe -fps=25 | to set video fps(default is 25). \n"
<< " randomVideo.exe -nF=50 | to set number of frames. "
<< "If fps is 25, 50 nF means the video will last 2 seconds (50/25). \n"
<< " randomVideo.exe -save=example1 | "
<< "to set video name, only *.avi support, "
<< "you do not need input \".avi\". "
<< "If empty, the default name is local-system-time.avi, your previous files are safe. \n"
<< " randomVideo.exe -crossSize=(15,5) | "
<< "to set the size of cross."
<< " (height, width) you can see the showing window. \n"
<< " randomVideo.exe -crossColor=0 | Recommd not to set it. "
<< "If = 0, black(default); if = 255, white.\n\n"
<< " randomVideo.exe -show=TRUE | Recommd to be TRUE. "
<<"If FALSE, the window to show the video is stopped. \n"
<< " randomVideo.exe -seed=15 | Recommd not to set it. "
<<"If empty, the local time will be used, then,"
<<" you will get different video each time. \n\n"
<< " { NEW } randomVideo.exe -color=TRUE | If FALSE, the video is gray. \n"
<< " { NEW } randomVideo.exe -cellSize=(2,3) | To control the size of pixel size. "
<< "If cellSize=(2,3), the image width will be seperated into 2 parts, "
<< "the image height will be seperated into 3 parts. In defalut mode pixel size as 1 will be used!"
<< "In order to get exact pixel size, you need to calculate it by yourself. \n\n"
<< "example:\n"
<< " randomVideo.exe -size=(640,580) -fps=25 -nF=50 -crossSize=(15,5) -color=1 -cellSize=(5,5)\n"
<< " The other parameters using default is fine. \n"
<< std::endl;
}
void getParas(int argn, char** argv, parameters & _parameters)
{
if (argn <= 1)
{
std::cout << "Using default parameter to generate video.\n"
<< "How to use? Please check the usage by input \" randomVideo.exe -help\" \n"
<< std::endl;
help();
exit(0);
}
else
{
std::map<std::string, std::string> table;
for (int i = 1; i<argn; ++i)
{
std::string sub(argv[i]);
std::string name = sub.substr(1, sub.find("=")-1);
table.insert({name, sub.substr(sub.find("=")+1, sub.length()-sub.find("=")-1)});
}
if (table.count("help") == 1)
{
help();
exit(0);
}
if (table.count("size") == 1)
{
cv::Size _size(atoi(table["size"].substr(1, table["size"].find(",")-1).c_str()),
atoi(table["size"].substr(
table["size"].find(",")+1, table["size"].length()-table["size"].find(",")-2
).c_str()));
_parameters.size = _size;
}
else
_parameters.size = cv::Size(640, 480);
if (table.count("crossSize") == 1)
{
cv::Size _crossSize(atoi(table["crossSize"].substr(1, table["crossSize"].find(",") - 1).c_str()),
atoi(table["crossSize"].substr(
table["crossSize"].find(",") + 1, table["crossSize"].length() - table["crossSize"].find(",") - 2
).c_str()));
_parameters.crossSize = _crossSize;
}
else
_parameters.crossSize = cv::Size(15, 5);
if (table.count("crossColor") == 1)
_parameters.crossColor = atoi(table["crossColor"].c_str());
else
_parameters.crossColor = 0;
if (table.count("fps") == 1)
_parameters.fps = atoi(table["fps"].c_str());
else
_parameters.fps = 25;
if (table.count("nF") == 1)
_parameters.nF = atoi(table["nF"].c_str());
else
_parameters.nF = 50;
if (table.count("save") == 1)
_parameters.save = table["save"] + ".avi";
else
{
SYSTEMTIME sys;
GetLocalTime(&sys);
char fname[100] = {};
sprintf_s(fname, 100, "%4d-%02d-%02d_%02d_%02d_%02d.avi", \
sys.wYear, sys.wMonth, sys.wDay, sys.wHour, sys.wMinute, sys.wSecond);
_parameters.save = fname;
}
if (table.count("show") == 1)
_parameters.show = (table["show"] == "TRUE");
else
_parameters.show = true;
if (table.count("seed") == 1)
_parameters.seed = atoi(table["seed"].c_str());
else
{
SYSTEMTIME sys;
GetLocalTime(&sys);
_parameters.seed = sys.wYear + sys.wMonth + sys.wDay + sys.wHour + sys.wMinute + sys.wSecond;
}
/*if (table.count("color") == 1)
{
cv::Scalar _color(
atoi(table["color"].substr(1, table["color"].find(",") - 1).c_str()),
atoi(table["color"].substr(table["color"].find(",") + 1,
table["color"].find_last_of(",") - table["color"].find(",") - 2).c_str()),
atoi(table["color"].substr(table["color"].find_last_of(",") + 1,
table["color"].find_last_of(")") - table["color"].find_last_of(",") - 1).c_str()));
_parameters.color = _color;
}
else
_parameters.size = cv::Size(640, 480);*/
if (table.count("color") == 1)
_parameters.color = (table["color"] == "TRUE");
else
_parameters.color = true;
if (table.count("cellSize") == 1)
{
cv::Size _cellSize(atoi(table["cellSize"].substr(1, table["cellSize"].find(",") - 1).c_str()),
atoi(table["cellSize"].substr(
table["cellSize"].find(",") + 1, table["cellSize"].length() - table["cellSize"].find(",") - 2
).c_str()));
_parameters.cellSize = _cellSize;
}
else
_parameters.cellSize = cv::Size(2, 2);
}
_parameters.showParas();
}
void randMatrix(cv::Mat & img, int64 seed, cv::Size cell=cv::Size(0,0))
{
cv::RNG rng(seed);
if (cell == cv::Size(0, 0) || cell.width > img.cols || cell.height > img.rows)
{
for (size_t j = 0; j < img.rows; ++j)
{
uchar* pdata = img.ptr<uchar>(j);
for (size_t i = 0; i < img.cols; ++i)
{
pdata[i] = rng.uniform(0, 255);
}
}
}
else
{
size_t stepX = img.cols / cell.width;
size_t stepY = img.rows / cell.height;
for (size_t j = 0; j < img.rows; j+=stepY)
{
size_t lastY(j + stepY);
if (lastY >= img.rows)
lastY = img.rows - 1;
for (size_t i = 0; i < img.cols; i += stepX)
{
size_t lastX(i+stepX);
if (lastX >= img.cols)
lastX = img.cols - 1;
img.rowRange(j, lastY).colRange(i, lastX).setTo(rng.uniform(0,255));
}
}
}
}
void addCross(cv::Mat & img, cv::Size cSzie, cv::Scalar scalar)
{
int x_center = (img.cols - 1) / 2, y_center = (img.rows - 1) / 2;
cv::Point
P1_tl(x_center - (cSzie.width - 1) / 2, y_center - (cSzie.height - 1) / 2),
P1_bd(x_center + (cSzie.width - 1) / 2, y_center + (cSzie.height - 1) / 2),
P2_tl(x_center - (cSzie.height - 1) / 2, y_center - (cSzie.width - 1) / 2),
P2_bd(x_center + (cSzie.height - 1) / 2, y_center + (cSzie.width - 1) / 2);
cv::rectangle(img, P1_tl, P1_bd, scalar, -1);
cv::rectangle(img, P2_tl, P2_bd, scalar, -1);
}
int main(int argn, char** argv)
{
parameters _pa;
getParas(argn, argv, _pa);
cv::VideoWriter writer(_pa.save, CV_FOURCC('M', 'J','P','G'),
double(_pa.fps), _pa.size);
cv::Mat frame0 = cv::Mat::zeros(_pa.size, CV_8UC1);
cv::Mat frame1 = cv::Mat::zeros(_pa.size, CV_8UC1);
cv::Mat frame2 = cv::Mat::zeros(_pa.size, CV_8UC1);
for (int i=0; i<_pa.nF; ++i)
{
randMatrix(frame0, _pa.seed + i, _pa.cellSize);
addCross(frame0, _pa.crossSize, _pa.crossColor);
randMatrix(frame1, 0.3*(_pa.seed + i), _pa.cellSize);
addCross(frame1, _pa.crossSize, _pa.crossColor);
randMatrix(frame2, 0.7*(_pa.seed + i), _pa.cellSize);
addCross(frame2, _pa.crossSize, _pa.crossColor);
cv::Mat f3;
if(_pa.color)
cv::merge(std::vector<cv::Mat>{frame0,frame1,frame2}, f3);
else
cv::merge(std::vector<cv::Mat>{frame0, frame0, frame0}, f3);
writer << f3;
if (_pa.show)
cv::imshow("video", f3);
cv::waitKey(10);
}
return 0;
}
使用说明：
见help()
