third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ApplyColorMap can be used with a user colormap

Browse Source
This commit is contained in:
LaurentBerger 2016-12-15 10:17:05 +01:00
parent 36b5abf6b7
commit 61b9484155
2 changed files with 42 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/imgproc/include/opencv2/imgproc.hpp
View File

@ -4097,9 +4097,17 @@ enum ColormapTypes
@param src The source image, grayscale or colored of type CV_8UC1 or CV_8UC3.
@param dst The result is the colormapped source image. Note: Mat::create is called on dst.
@param colormap The colormap to apply, see cv::ColormapTypes
*/
*/
CV_EXPORTS_W void applyColorMap(InputArray src, OutputArray dst, int colormap);
/** @brief Applies a user colormap on a given image.
@param src The source image, grayscale or colored of type CV_8UC1 or CV_8UC3.
@param dst The result is the colormapped source image. Note: Mat::create is called on dst.
@param userColor The colormap to apply of type CV_8UC1 or CV_8UC3 and size 256
*/
CV_EXPORTS_W void applyColorMap(InputArray src, OutputArray dst, InputArray userColor);
//! @} imgproc_colormap
//! @addtogroup imgproc_draw

33
modules/imgproc/src/colormap.cpp
View File

@ -490,6 +490,22 @@ namespace colormap
}
};
// UserColormap .
class UserColorMap : public ColorMap {
public:
UserColorMap(Mat c) : ColorMap() {
init(c);
}
void init(Mat c) {
this->_lut = c;
}
void init(int n) {
CV_Error(Error::StsAssert, "unused method in UserColormap.");
}
};
void ColorMap::operator()(InputArray _src, OutputArray _dst) const
{
CV_INSTRUMENT_REGION()
@ -546,4 +562,21 @@ namespace colormap
delete cm;
}
void applyColorMap(InputArray src, OutputArray dst, InputArray userColor)
{
if (userColor.total() != 256)
CV_Error(Error::StsAssert, "cv::LUT only supports tables of size 256.");
if (userColor.type() != CV_8UC1 && userColor.type() != CV_8UC3)
CV_Error(Error::StsAssert, "cv::LUT only supports tables CV_8UC1 or CV_8UC3.");
colormap::ColorMap* cm = (colormap::ColorMap*) (new colormap::UserColorMap(userColor.getMat()));
if (!cm)
CV_Error(Error::StsBadArg, "Unknown colormap id; use one of COLORMAP_*");
(*cm)(src, dst);
delete cm;
}
}