109 lines
4.4 KiB
C++
109 lines
4.4 KiB
C++
#include "color_correct.h"
|
|
|
|
|
|
#include <huagao/hgscanner_error.h>
|
|
#include <sane/sane_ex.h>
|
|
#include <opencv2/imgproc/hal/hal.hpp>
|
|
#include "ImageProcess_Public.h"
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
static std::string device_opt_json[] = {
|
|
"{\"clr-correct\":{\"cat\":\"imgp\",\"group\":\"advance\",\"title\":\"\\u989c\\u8272\\u6821\\u6b63\",\"desc\":\"\\u6839\\u636e\\u6821\\u6b63\\u53c2\\u6570\\uff0c\\u8c03\\u6574\\u4eceCIS\\u51fa\\u6765\\u7684\\u539f\\u59cb\\u56fe\\u50cf\\u989c\\u8272\",\"type\":\"bool\",\"pos\":24,\"ui-pos\":15,\"auth\":100,\"size\":4,\"cur\":true,\"default\":true},\"lut-file\":{\"cat\":\"imgp\",\"group\":\"advance\",\"title\":\"\\u6821\\u6b63\\u6a21\\u677f\",\"desc\":\"\\u989c\\u8272\\u6821\\u6b63\\u4f7f\\u7528\\u7684\\u57fa\\u51c6\\u56fe\\u50cf\\u6570\\u636e\",\"type\":\"string\",\"pos\":25,\"ui-pos\":16,\"auth\":100,\"bind\":true,\"size\":80,\"default\":{\"cis-mode==\\u5f69\\u8272&&resolution>200&&resolution<400\":\"\\/usr\\/local\\/huago\\/Textlut300clr.bmp\",\"cis-mode==\\u5f69\\u8272&&resolution>=400\":\"\\/usr\\/local\\/huago\\/Textlut600clr.bmp\",\"cis-mode==\\u7070\\u5ea6&&resolution<=200\":\"\\/usr\\/local\\/huago\\/Textlut200gray.bmp\",\"cis-mode==\\u7070\\u5ea6&&resolution>200&&resolution<400\":\"\\/usr\\/local\\/huago\\/Textlut300gray.bmp\",\"cis-mode==\\u7070\\u5ea6&&resolution>=400\":\"\\/usr\\/local\\/huago\\/Textlut600gray.bmp\",\"default\":\"\\/usr\\/local\\/huago\\/Textlut200clr.bmp\"}},\"cis-mode\":{\"cat\":\"none\",\"group\":\"CIS\",\"title\":\"CIS\\u989c\\u8272\\u6a21\\u5f0f\",\"desc\":\"\\u9009\\u62e9\\u955c\\u5934\\u8272\\u5f69\\u5de5\\u4f5c\\u6a21\\u5f0f\",\"type\":\"string\",\"pos\":1000,\"ui-pos\":10,\"auth\":0,\"bind\":true,\"size\":12,\"default\":{\"(mode.enabled&&(mode==256\\u7ea7\\u7070\\u5ea6||mode==\\u9ed1\\u767d)) || (multiout-type.enabled&&multiout-type==\\u7070\\u5ea6+\\u9ed1\\u767d)\":\"\\u7070\\u5ea6\",\"default\":\"\\u5f69\\u8272\"},\"range\":[\"\\u5f69\\u8272\",\"\\u7070\\u5ea6\"]}}"
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
color_correct::color_correct(bool weaker) : image_processor("color_correct")
|
|
{
|
|
if(weaker)
|
|
return;
|
|
|
|
ADD_THIS_JSON();
|
|
lut_ = hg::loadLUT(lut_path_);
|
|
}
|
|
color_correct::~color_correct()
|
|
{}
|
|
|
|
|
|
int color_correct::set_value(const char* name/*nullptr for all options*/, void* val/*nullptr for restore*/)
|
|
{
|
|
int ret = SCANNER_ERR_OK;
|
|
|
|
if(strcmp(name, SANE_OPT_NAME(COLOR_CORRECT)) == 0)
|
|
enabled_ = correct_ = *(bool*)val;
|
|
else if(strcmp(name, SANE_OPT_NAME(LUT_FILE)) == 0)
|
|
{
|
|
if(lut_path_ != (char*)val)
|
|
{
|
|
lut_path_ = (char*)val;
|
|
lut_ = hg::loadLUT(lut_path_);
|
|
}
|
|
}
|
|
else if(strcmp(name, SANE_OPT_NAME(CIS_MODE)) == 0)
|
|
{
|
|
bool pre = clr_;
|
|
|
|
clr_ = strcmp((char*)val, WORDS_COLOR_COLOR) == 0;
|
|
}
|
|
else
|
|
ret = SCANNER_ERR_DEVICE_NOT_SUPPORT;
|
|
|
|
return ret;
|
|
}
|
|
image_processor* color_correct::copy_weaker(void)
|
|
{
|
|
color_correct *weaker = new color_correct(true);
|
|
|
|
weaker->pos_ = pos_;
|
|
weaker->enabled_ = enabled_;
|
|
weaker->correct_ = correct_;
|
|
weaker->clr_ = clr_;
|
|
weaker->lut_ = lut_.clone();
|
|
|
|
return weaker;
|
|
}
|
|
int color_correct::process(std::vector<PROCIMGINFO>& in, std::vector<PROCIMGINFO>& out)
|
|
{
|
|
int ret = SCANNER_ERR_OK;
|
|
|
|
if(correct_)
|
|
{
|
|
for(auto& v: in)
|
|
{
|
|
PROCIMGINFO o = v;
|
|
chronograph watch;
|
|
|
|
// hg::correctColor(o.img, o.info.resolution_x, 1, true);
|
|
cv::Mat image_temp(o.img.rows, o.img.cols * o.img.channels() / lut_.channels(), CV_8UC(lut_.channels()), o.img.data);
|
|
|
|
for (size_t i = 0; i < image_temp.cols; i++)
|
|
cv::LUT(image_temp(cv::Rect(i, 0, 1, image_temp.rows)), lut_(cv::Rect(0, i, 256, 1)), image_temp(cv::Rect(i, 0, 1, image_temp.rows)));
|
|
|
|
o.info.prc_time = watch.elapse_ms();
|
|
o.info.prc_stage = get_position();
|
|
o.info.width = o.img.cols;
|
|
o.info.height = o.img.rows;
|
|
out.push_back(o);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
out = in;
|
|
for(auto& v: out)
|
|
{
|
|
v.info.prc_stage = get_position();
|
|
v.info.prc_time = 0;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|