some tweaks to samples

This commit is contained in:
yao
2013-06-28 15:08:39 +08:00
parent c1a59b8d80
commit 6982ea5a66
4 changed files with 134 additions and 7 deletions

View File

@@ -49,7 +49,7 @@ struct App
return ss.str();
}
private:
bool running;
bool running, write_once;
Mat left_src, right_src;
Mat left, right;
@@ -115,6 +115,7 @@ App::App(CommandLineParser& cmd)
cout << "stereo_match_ocl sample\n";
cout << "\nControls:\n"
<< "\tesc - exit\n"
<< "\to - save output image once\n"
<< "\tp - print current parameters\n"
<< "\tg - convert source images into gray\n"
<< "\tm - change stereo match method\n"
@@ -132,6 +133,7 @@ App::App(CommandLineParser& cmd)
else cout << "unknown method!\n";
ndisp = cmd.get<int>("n");
out_img = cmd.get<string>("o");
write_once = false;
}
@@ -161,10 +163,8 @@ void App::run()
printParams();
running = true;
bool written = false;
while (running)
{
// Prepare disparity map of specified type
Mat disp;
oclMat d_disp;
@@ -192,19 +192,21 @@ void App::run()
csbp(d_left, d_right, d_disp);
break;
}
// Show results
d_disp.download(disp);
workEnd();
if (method != BM)
{
disp.convertTo(disp, 0);
}
putText(disp, text(), Point(5, 25), FONT_HERSHEY_SIMPLEX, 1.0, Scalar::all(255));
imshow("disparity", disp);
if(!written)
if(write_once)
{
imwrite(out_img, disp);
written = true;
write_once = false;
}
handleKey((char)waitKey(3));
}
@@ -378,6 +380,10 @@ void App::handleKey(char key)
cout << "level_count: " << csbp.levels << endl;
}
break;
case 'o':
case 'O':
write_once = true;
break;
}
}