Skip to content

Commit 3fb4842

Browse files
authored
Merge pull request #95 from gcrowards/patch-1
Improved variable names and comments
2 parents 1192d8d + e56cd6f commit 3fb4842

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

src/nodelet/simple_flow_nodelet.cpp

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
7979
static bool need_config_update_;
8080
int scale_;
8181

82-
cv::Mat gray, prevGray;
82+
cv::Mat color, prevColor;
8383

8484
void reconfigureCallback(Config& new_config, uint32_t level)
8585
{
@@ -117,7 +117,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
117117
// Convert the image into something opencv can handle.
118118
cv::Mat frame_src = cv_bridge::toCvShare(msg, sensor_msgs::image_encodings::BGR8)->image;
119119

120-
/// Convert it to gray
120+
/// Convert it to 3 channel
121121
cv::Mat frame;
122122
if (frame_src.channels() > 1)
123123
{
@@ -128,18 +128,18 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
128128
cv::cvtColor(frame_src, frame, cv::COLOR_GRAY2BGR);
129129
}
130130

131-
cv::resize(frame, gray, cv::Size(frame.cols / (double)MAX(1, scale_), frame.rows / (double)MAX(1, scale_)), 0, 0,
131+
cv::resize(frame, color, cv::Size(frame.cols / (double)MAX(1, scale_), frame.rows / (double)MAX(1, scale_)), 0, 0,
132132
CV_INTER_AREA);
133-
if (prevGray.empty())
134-
gray.copyTo(prevGray);
133+
if (prevColor.empty())
134+
color.copyTo(prevColor);
135135

136-
if (gray.rows != prevGray.rows && gray.cols != prevGray.cols)
136+
if (color.rows != prevColor.rows && color.cols != prevColor.cols)
137137
{
138138
NODELET_WARN("Images should be of equal sizes");
139-
gray.copyTo(prevGray);
139+
color.copyTo(prevColor);
140140
}
141141

142-
if (frame.type() != 16)
142+
if (frame.type() != CV_8UC3)
143143
{
144144
NODELET_ERROR("Images should be of equal type CV_8UC3");
145145
}
@@ -164,9 +164,9 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
164164

165165
float start = (float)cv::getTickCount();
166166
#if CV_MAJOR_VERSION == 3
167-
cv::optflow::calcOpticalFlowSF(gray, prevGray,
167+
cv::optflow::calcOpticalFlowSF(color, prevColor,
168168
#else
169-
cv::calcOpticalFlowSF(gray, prevGray,
169+
cv::calcOpticalFlowSF(color, prevColor,
170170
#endif
171171
flow, 3, 2, 4, 4.1, 25.5, 18, 55.0, 25.5, 0.35, 18, 55.0, 25.5, 10);
172172
NODELET_INFO("calcOpticalFlowSF : %lf sec", (cv::getTickCount() - start) / cv::getTickFrequency());
@@ -210,7 +210,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
210210
int c = cv::waitKey(1);
211211
}
212212

213-
cv::swap(prevGray, gray);
213+
cv::swap(prevColor, color);
214214
// Publish the image.
215215
sensor_msgs::Image::Ptr out_img = cv_bridge::CvImage(msg->header, "bgr8", frame).toImageMsg();
216216
img_pub_.publish(out_img);

0 commit comments

Comments
 (0)