newtx/imgproc/imgprc_mgr.cpp

#include "imgprc_mgr.h"

#include <json/gb_json.h>
#include <sane_opt_json/device_opt.h>
#include <huagao/hgscanner_error.h>
#include <base/packet.h>


static std::string device_opt_json[] = {
	"{\"is-multiout\":{\"cat\":\"base\",\"group\":\"base\",\"title\":\"\\u591a\\u6d41\\u8f93\\u51fa\",\"desc\":\"\\u540c\\u65f6\\u8f93\\u51fa\\u591a\\u79cd\\u989c\\u8272\\u6a21\\u5f0f\\u7684\\u56fe\\u50cf\",\"type\":\"bool\",\"fix-id\":34817,\"ui-pos\":10,\"auth\":0,\"size\":4,\"cur\":false,\"default\":false},\"multiout-type\":{\"cat\":\"base\",\"group\":\"base\",\"title\":\"\\u591a\\u6d41\\u8f93\\u51fa\\u7c7b\\u578b\",\"desc\":\"\\u9009\\u62e9\\u591a\\u6d41\\u8f93\\u51fa\\u7684\\u7c7b\\u578b\",\"type\":\"string\",\"fix-id\":34818,\"ui-pos\":11,\"auth\":0,\"enabled\":false,\"size\":66,\"cur\":\"\\u5f69\\u8272+\\u7070\\u5ea6+\\u9ed1\\u767d\",\"default\":\"\\u5f69\\u8272+\\u7070\\u5ea6+\\u9ed1\\u767d\",\"range\":[\"\\u5f69\\u8272+\\u7070\\u5ea6+\\u9ed1\\u767d\",\"\\u5f69\\u8272+\\u7070\\u5ea6\",\"\\u5f69\\u8272+\\u9ed1\\u767d\",\"\\u7070\\u5ea6+\\u9ed1\\u767d\"],\"depend\":\"is-multiout==true\"},\"mode\":{\"cat\":\"base\",\"group\":\"base\",\"title\":\"\\u989c\\u8272\\u6a21\\u5f0f\",\"desc\":\"\\u9009\\u62e9\\u8272\\u5f69\\u6a21\\u5f0f\",\"type\":\"string\",\"fix-id\":34819,\"ui-pos\":15,\"auth\":0,\"size\":24,\"cur\":\"24\\u4f4d\\u5f69\\u8272\",\"default\":\"24\\u4f4d\\u5f69\\u8272\",\"range\":[\"24\\u4f4d\\u5f69\\u8272\",\"256\\u7ea7\\u7070\\u5ea6\",\"\\u9ed1\\u767d\",\"\\u989c\\u8272\\u81ea\\u52a8\\u8bc6\\u522b\"],\"depend\":\"is-multiout!=true\"},\"resolution\":{\"cat\":\"base\",\"group\":\"base\",\"title\":\"\\u5206\\u8fa8\\u7387\",\"desc\":\"\\u8bbe\\u7f6e\\u626b\\u63cf\\u56fe\\u50cf\\u7684\\u5206\\u8fa8\\u7387\",\"type\":\"int\",\"fix-id\":34840,\"ui-pos\":20,\"auth\":0,\"size\":4,\"cur\":200,\"default\":200,\"range\":{\"min\":100,\"max\":{\"default\":600,\"paper==\\u6700\\u5927\\u626b\\u63cf\\u5c3a\\u5bf8\\u81ea\\u52a8\\u88c1\\u5207 || paper==\\u6700\\u5927\\u626b\\u63cf\\u5c3a\\u5bf8 || paper==\\u4e09\\u8054\\u8bd5\\u5377\":500},\"step\":1}},\"dump-img\":{\"cat\":\"base\",\"group\":\"\\u9ad8\\u7ea7\\u8bbe\\u7f6e\",\"title\":\"\\u8f93\\u51fa\\u4e2d\\u95f4\\u56fe\\u50cf\",\"desc\":\"\\u8f93\\u51fa\\u5404\\u7b97\\u6cd5\\u4e2d\\u95f4\\u7ed3\\u679c\\u56fe\\u50cf\",\"type\":\"bool\",\"ui-pos\":10,\"auth\":0,\"affect\":2,\"size\":4,\"cur\":false,\"default\":false}}"
};

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// imgproc_mgr

imgproc_mgr::imgproc_mgr(std::function<void(data_source_ptr)> sender
						, device_option* devopts
						)
						: img_sender_(sender), opts_(devopts), prc_que_("prcimg")
{
	std::string	text("");
	gb_json		*jsn = new gb_json();

	for (auto& v : device_opt_json)
		text += v;

	if (jsn->attach_text(&text[0]))
	{
		text = jsn->to_string();
	}
	jsn->release();

	set_opt_json_text(&text[0]);

	if (opts_)
		opts_->add_ref();
	else
		opts_ = new device_option();

	auto thrd = [&](void) -> void
	{
		thread_worker();
	};
	workers_.start(thrd, "imgproc_mgr::thread_worker", (void*)&imgproc_mgr::thread_worker);
}
imgproc_mgr::~imgproc_mgr()
{
	clear();
	opts_->release();
}

bool imgproc_mgr::sort_processor_by_pos(image_processor* l, image_processor* r)
{
	return l->get_position() < r->get_position();
}
bool imgproc_mgr::sort_image_packet(image_packet_ptr l, image_packet_ptr r)
{
	return l->get_paper_index() < r->get_paper_index();
}
data_source_ptr imgproc_mgr::scan_finished_packet(uint32_t scanid, uint32_t err)
{
	dyn_mem_ptr reply = dyn_mem::memory(sizeof(PACK_BASE));
	BASE_PACKET_REPLY(*((LPPACK_BASE)reply->ptr()), PACK_CMD_SCAN_FINISHED_ROGER, scanid, err);
	reply->set_len(sizeof(PACK_BASE));

	return reply;
}
image_packet_ptr imgproc_mgr::image_sent_packet(LPPACKIMAGE head, dyn_mem_ptr img, uint32_t scanid, void* info, size_t info_l)
{
	image_packet_ptr	pimg = new image_packet(head, img, scanid, info, info_l);

	return pimg;
}

uint32_t imgproc_mgr::add_busy_worker(int inc)
{
	SIMPLE_LOCK(working_cnt_lock_);
	working_cnt_ += inc;

	return working_cnt_;
}
void imgproc_mgr::thread_worker(void)
{
	RAWIMG	img;
	while(prc_que_.take(img, true))
	{
		add_busy_worker();

		process(&img);
		if(img.img)
			img.data->release();

		add_busy_worker(-1);
	}
}
void imgproc_mgr::process(RAWIMG* img)
{
	if(img->img)
	{
		std::vector<PROCIMGINFO>	in, out;
		PROCIMGINFO	ele;

		ele.ext_info = "";
		ele.info = img->info;
		ele.img = cv::Mat(ele.info.height, ele.info.width, CV_8UC1, (void*)img->data->ptr(), img->data->get_rest());
		in.push_back(ele);

		if(dump_img_)
		{
			for(auto& v: processors_)
			{
				send_image(in, false);
				v->process(in, out);
				in = std::move(out);
			}
		}
		else
		{
			for(auto& v: processors_)
			{
				v->process(in, out);
				in = std::move(out);
			}
		}
		send_image(in, true);
	}
	else
	{
		data_source_ptr	ptr = imgproc_mgr::scan_finished_packet(scan_id_, img->info.reserve);

		ptr->set_session_id(session_id_);
		img_sender_(ptr);
		ptr->release();
	}
}
void imgproc_mgr::send_image(std::vector<PROCIMGINFO>& imgs, bool clear_after_send)
{
	for(auto& v: imgs)
	{
		uint32_t	size = v.img.total() * v.img.channels();
		dyn_mem_ptr mem(dyn_mem::memory(size));
		printf("image bits: %p + %u\n", mem->ptr(), size);
		mem->put(v.img.data, size);
		// if(clear_after_send)
		// 	v.img.deallocate();

		image_packet_ptr ptr = imgproc_mgr::image_sent_packet(&v.info, mem, scan_id_, &v.ext_info[0], v.ext_info.length());

		ptr->set_session_id(session_id_);
		printf("sending image packet: %p + %u\n", ptr->ptr(), ptr->get_rest());
		img_sender_(ptr);
		ptr->release();
		mem->release();
	}
}

int imgproc_mgr::set_value(const char* name, void* val)
{
	int	ret = SCANNER_ERR_OK;

	if(strcmp(name, SANE_FULL_NAME(DUMP_IMG)) == 0)
		dump_img_ = *(bool*)val;
	else
		ret = SCANNER_ERR_DEVICE_NOT_SUPPORT;

	return ret;
}

int imgproc_mgr::load_processor(const char* path)
{
	int	ret = SCANNER_ERR_OK;

#define ADD_IMG_PROCESSOR(cls)	\
	{							\
		cls *obj = new cls();	\
		opts_->add(obj);		\
		obj->release();			\
	}

	std::sort(processors_.begin(), processors_.end(), &imgproc_mgr::sort_processor_by_pos);

	return ret;
}
int imgproc_mgr::clear(void)
{
	for (auto& v : processors_)
		v->release();
	processors_.clear();

	return 0;
}

int imgproc_mgr::process(LPPACKIMAGE info, dyn_mem_ptr data, bool img)
{
	RAWIMG	ri;
	int		ret = SCANNER_ERR_OK;

	ri.data = data;
	if(img)
		ri.info = *info;
	else
		ri.info.reserve = (uint32_t)(long)info;
	ri.img = img;
	prc_que_.save(ri, true);

	return ret;
}

void imgproc_mgr::stop(void)
{
	run_ = false;
	prc_que_.trigger();

	workers_.stop(nullptr);
}
bool imgproc_mgr::is_busy(void)
{
	SIMPLE_LOCK(working_cnt_lock_);

	return working_cnt_;
}
void imgproc_mgr::start_new_turn(uint32_t scanid, uint32_t sessionid)
{
	scan_id_ = scanid;
	sent_ind_ = 0;
	session_id_ = sessionid;
}