72 lines
1.6 KiB
C++
72 lines
1.6 KiB
C++
|
/**
|
||
|
|
* @file LinearTransforms.cpp
|
||
|
|
* @brief Simple program to change contrast and brightness
|
||
|
|
* @date Mon, June 6, 2011
|
||
|
|
* @author OpenCV team
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <cv.h>
|
||
|
|
#include <highgui.h>
|
||
|
|
|
||
|
|
using namespace cv;
|
||
|
|
|
||
|
|
/** Global Variables */
|
||
|
|
const int alpha_max = 5;
|
||
|
|
const int beta_max = 125;
|
||
|
|
int alpha; /**< Simple contrast control */
|
||
|
|
int beta; /**< Simple brightness control*/
|
||
|
|
|
||
|
|
/** Matrices to store images */
|
||
|
|
Mat image;
|
||
|
|
Mat new_image;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @function on_trackbar
|
||
|
|
* @brief Called whenever any of alpha or beta changes
|
||
|
|
*/
|
||
|
|
void on_trackbar( int, void* )
|
||
|
|
{
|
||
|
|
Mat new_image = Mat::zeros( image.size(), image.type() );
|
||
|
|
|
||
|
|
for( int y = 0; y < image.rows; y++ )
|
||
|
|
{ for( int x = 0; x < image.cols; x++ )
|
||
|
|
{ for( int c = 0; c < 3; c++ )
|
||
|
|
{
|
||
|
|
new_image.at<Vec3b>(y,x)[c] = saturate_cast<uchar>( alpha*( image.at<Vec3b>(y,x)[c] ) + beta );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
imshow("New Image", new_image);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @function main
|
||
|
|
* @brief Main function
|
||
|
|
*/
|
||
|
|
int main( int argc, char** argv )
|
||
|
|
{
|
||
|
|
/// Read image given by user
|
||
|
|
image = imread( argv[1] );
|
||
|
|
|
||
|
|
/// Initialize values
|
||
|
|
alpha = 1;
|
||
|
|
beta = 0;
|
||
|
|
|
||
|
|
/// Create Windows
|
||
|
|
namedWindow("Original Image", 1);
|
||
|
|
namedWindow("New Image", 1);
|
||
|
|
|
||
|
|
/// Create Trackbars
|
||
|
|
createTrackbar( "Contrast Trackbar", "New Image", &alpha, alpha_max, on_trackbar );
|
||
|
|
createTrackbar( "Brightness Trackbar", "New Image", &beta, beta_max, on_trackbar );
|
||
|
|
|
||
|
|
/// Show some stuff
|
||
|
|
imshow("Original Image", image);
|
||
|
|
imshow("New Image", image);
|
||
|
|
|
||
|
|
/// Wait until user press some key
|
||
|
|
waitKey();
|
||
|
|
return 0;
|
||
|
|
}
|