bool TestExtFile::test_dir() {
Variant d = f_dir("test");
VS(d.toArray()["path"], "test");
Variant entry;
bool seen = false;
while (!same(entry = f_readdir(d.toArray()["handle"]), false)) {
if (same(entry, "test_ext_file.txt")) {
seen = true;
}
}
f_closedir(d);
VERIFY(seen);
return Count(true);
}
bool TestExtFile::test_dir() {
static const StaticString s_handle("handle");
static const StaticString s_path("path");
Variant d = f_dir("test/ext");
VS(d.toArray()[s_path], "test/ext");
Variant entry;
bool seen = false;
while (!same(entry = f_readdir(d.toArray()[s_handle]), false)) {
if (same(entry, "test_ext_file.txt")) {
seen = true;
}
}
f_closedir(d);
VERIFY(seen);
return Count(true);
}
int main()
{
cv::FileStorage fs("config.yml", cv::FileStorage::READ);
std::string folder_name = static_cast<std::string>(fs["folder_name"]);
double c = static_cast<double>(fs["f_edge_scale"]);
std::cout << c << " a" << folder_name << std::endl;
std::stringstream stream_result;
stream_result << folder_name << "/result.csv";
std::string fileName_result = stream_result.str();
std::ofstream ofs( fileName_result );
ofs << "thita" << "," << "fay" << "," << "f_edge" << "," << "f_grad" << "," << "f_var" << "," << "f_mean" << "," << "f_hist" << "," << "f_dir" << "," << "f_Q" << std::endl;
std::vector< std::vector <double> > f_e_all, f_g_all, f_v_all, f_m_all, f_h_all, f_q_all;
double min[6] = {1000000,1000000,1000000,1000000,1000000,1000000};
double max[6]= {-1000000,-1000000,-1000000,-1000000,-1000000,-1000000};
std::cout << "|----------|" << std::endl;
std::cout << " ";
for(int i = 0; i <= 9; ++i )
{
std::vector<double> f_e_tmp, f_g_tmp, f_v_tmp, f_m_tmp, f_h_tmp, f_q_tmp;
std::cout << "+";
for(int j = 0; j < 36; ++j)
{
if(j!=0 && i==0) break;
std::stringstream stream;
stream << folder_name << "/" << "sita" << i*10 << "fay" << j*10 << ".png";
std::string fileName = stream.str();
std::stringstream stream_pim;
stream_pim << folder_name << "/PIM.bmp";
std::string fileName_pim = stream_pim.str();
cv::Mat raw_img = cv::imread(fileName,0);
cv::Mat PIM = cv::imread(fileName_pim,0);
double f_e = f_edge(raw_img, PIM);
double f_g = f_grad(raw_img, PIM);
double f_v = f_var(raw_img, PIM);
double f_m = f_mean(raw_img, PIM);
double f_h = f_hist(raw_img, PIM);
double f_d = f_dir(raw_img, PIM);
double f_q = f_Q( f_g, f_e, f_v, f_m, f_h,f_d);
f_e_tmp.push_back(f_e);
f_g_tmp.push_back(f_g);
f_v_tmp.push_back(f_v);
f_m_tmp.push_back(f_m);
f_h_tmp.push_back(f_h);
f_q_tmp.push_back(f_q);
if(min[0] > f_e) min[0] = f_e;
if(min[1] > f_g) min[1] = f_g;
if(min[2] > f_v) min[2] = f_v;
if(min[3] > f_m) min[3] = f_m;
if(min[4] > f_h) min[4] = f_h;
if(min[5] > f_q) min[5] = f_q;
if(max[0] < f_e) max[0] = f_e;
if(max[1] < f_g) max[1] = f_g;
if(max[2] < f_v) max[2] = f_v;
if(max[3] < f_m) max[3] = f_m;
if(max[4] < f_h) max[4] = f_h;
if(max[5] < f_q) max[5] = f_q;
ofs << i*10 << "," << j*10 << "," << f_e << ","<< f_g << "," << f_v << ","<<
f_m << "," << f_h << "," << f_d << ","<< f_q << std::endl;
}
f_e_all.push_back(f_e_tmp);
f_g_all.push_back(f_g_tmp);
f_v_all.push_back(f_v_tmp);
f_m_all.push_back(f_m_tmp);
f_h_all.push_back(f_h_tmp);
f_q_all.push_back(f_q_tmp);
}
std::cout << std::endl;
std::vector<cv::Mat> results;
for(int k = 0; k < 6; ++k)
{
results.push_back(cv::Mat::zeros(600, 600, CV_8UC3));
int angle = 270;
cv::ellipse(results[k], cv::Point(300, 300), cv::Size(250, 250), angle, angle, angle+360, cv::Scalar(255,255,255), 3, CV_AA);
for( int i = 9; i >= 0; --i )
{
for( int j = 35; j >= 0; --j )
{
double color = 0.0;
/*if (k == 0) color = f_e_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_e_all[i][j]) / (max[k] - min[k]));
else if (k == 1) color = f_g_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_g_all[i][j]) / (max[k] - min[k]));
else if (k == 2) color = f_v_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_v_all[i][j]) / (max[k] - min[k]));
else if (k == 3) color = f_m_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_m_all[i][j]) / (max[k] - min[k]));
else if (k == 4) color = f_h_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_h_all[i][j]) / (max[k] - min[k]));
else if (k == 5) color = f_q_all[i][j] == 0 ? 0 : 255 - (255 * (max[k] - f_q_all[i][j]) / (max[k] - min[k]));
*/
if (k == 0) color = f_e_all[i][j] == 0 ? 0 : 255 - (255 * (f_e_all[i][j] - min[k]) / (max[k] - min[k]));
else if (k == 1) color = f_g_all[i][j] == 0 ? 0 : 255 - (255 * (f_g_all[i][j] - min[k]) / (max[k] - min[k]));
else if (k == 2) color = f_v_all[i][j] == 0 ? 0 : 255 - (255 * (f_v_all[i][j] - min[k]) / (max[k] - min[k]));
else if (k == 3) color = f_m_all[i][j] == 0 ? 0 : 255 - (255 * (f_m_all[i][j] - min[k]) / (max[k] - min[k]));
else if (k == 4) color = f_m_all[i][j] == 0 ? 0 : 255 - (255 * (f_h_all[i][j] - min[k]) / (max[k] - min[k]));
else if (k == 5) color = f_q_all[i][j] == 0 ? 0 : 255 - (255 * (f_q_all[i][j] - min[k]) / (max[k] - min[k]));
/*if (k == 0) color = f_e_all[i][j] == 0 ? 0 : (255 * (1 - f_e_all[i][j]) / (1 - 0));
else if (k == 1) color = f_g_all[i][j] == 0 ? 0 : (255 * (1 - f_g_all[i][j]) / (1 - 0));
else if (k == 2) color = f_v_all[i][j] == 0 ? 0 : (255 * (1 - f_v_all[i][j]) / (1 - 0));
else if (k == 3) color = f_m_all[i][j] == 0 ? 0 : (255 * (1 - f_m_all[i][j]) / (1 - 0));
else if (k == 4) color = f_m_all[i][j] == 0 ? 0 : (255 * (1 - f_h_all[i][j]) / (1 - 0));
else if (k == 5) color = f_q_all[i][j] == 0 ? 0 : (255 * (max[k] - f_q_all[i][j]) / (max[k] - min[k]));
*/
if(i==0 && j == 0){
cv::ellipse(results[k], cv::Point(300, 300), cv::Size(25, 25), angle, angle-90, angle-90+360, cv::Scalar(color,color,color), -1, CV_AA);
}
else if(i==0 && j != 0) ;
else cv::ellipse(results[k], cv::Point(300, 300), cv::Size(250*(i+1)/10.0, 250*(i+1)/10.0), angle, angle-90-10*(j), angle-90-10*(j+1), cv::Scalar(color,color,color), -1, CV_AA);
}
}
if(k==0) cv::imwrite("f_edge.jpg",results[k]);
else if(k==1) cv::imwrite("f_grad.jpg",results[k]);
else if(k==2) cv::imwrite("f_var.jpg",results[k]);
else if(k==3) cv::imwrite("f_mean.jpg",results[k]);
else if(k==4) cv::imwrite("f_hist.jpg",results[k]);
else if(k==5) cv::imwrite("f_Q.jpg",results[k]);
cv::waitKey( 0 );
}
return 0;
}
