[C++][opencv]基于opencv实现photoshop算法色阶调整

【测试环境】

vs2019

opencv==4.8.0

【效果演示】

【核心实现代码】

Levels.hpp

复制代码
#ifndef OPENCV2_PS_LEVELS_HPP_
#define OPENCV2_PS_LEVELS_HPP_

#include "opencv2/core.hpp"
#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
using namespace cv;

namespace cv {

/**
 * Class of Level for one channel
 */
class Level {
public:
	int   Shadow;  //输入色阶黑点值
	float Midtones; //输入色阶灰点值(注意是浮点数)
	int   Highlight; //输入色阶白点值

	int   OutputShadow; //输出色阶黑点值
	int   OutputHighlight; //输出色阶白点值

	Level();
	virtual ~Level();

	bool createColorTable(uchar * colorTable);
	void clear();
};

/**
 * Class of Levels for all channels
 */
class Levels {
protected:
	bool createColorTables(uchar colorTables[][256]);

public:
	Level RGBChannel;  //RGB整体调整
	Level RedChannel;  //红色通道
	Level GreenChannel; //绿色通道
	Level BlueChannel; //蓝色通道

	Levels();
	virtual ~Levels();

	int adjust(InputArray src, OutputArray dst); //实施色阶调整
};


} /* namespace cv */

#endif /* OPENCV2_PS_LEVELS_HPP_ */

Levels.cpp

复制代码
#include "Levels.hpp"

namespace cv {

Level::Level() {
	clear();
}

Level::~Level() {

}

void Level::clear() {
	Shadow = OutputShadow = 0;
	Highlight = OutputHighlight = 255;
	Midtones = 1.0;
}

//create color table for a channel
bool Level::createColorTable(uchar * colorTable)
{
    int diff = (int)(Highlight - Shadow);
    int outDiff = (int)(OutputHighlight - OutputShadow);

    if (!((Highlight <= 255 && diff <= 255 && diff >= 2) ||
        (OutputShadow <= 255 && OutputHighlight <= 255 && outDiff < 255) ||
        (!(Midtones > 9.99 && Midtones > 0.1) && Midtones != 1.0)))
        return false;

    double coef = 255.0 / diff;
    double outCoef = outDiff / 255.0;
    double exponent = 1.0 / Midtones;

    for (int i = 0; i < 256; i ++)
    {
        int v;
        // calculate black field and white field of input level
        if ( colorTable[i] <= (uchar)Shadow ) {
            v = 0;
        } else {
            v = (int)((colorTable[i] - Shadow) * coef + 0.5);
            if (v > 255) v = 255;
        }
        // calculate midtone field of input level
        v = (int)( pow(v / 255.0, exponent) * 255.0 + 0.5 );
        // calculate output level
        colorTable[i] = (uchar)( v * outCoef + OutputShadow + 0.5 );
    }

    return true;
}

//==================================================================
// Levels

Levels::Levels() {
}

Levels::~Levels() {

}

bool Levels::createColorTables(uchar colorTables[][256])
{
    bool result = false;
    int i, j;

    //initialize color table
    for (i = 0; i < 3; i ++) {
        for (j = 0; j < 256; j ++)
            colorTables[i][j] = (uchar)j;
    }

    //create color table for each channel
    result = BlueChannel.createColorTable( colorTables[0]);
    result = GreenChannel.createColorTable( colorTables[1]);
    result = RedChannel.createColorTable( colorTables[2]);

    result = RGBChannel.createColorTable( colorTables[0]);
    result = RGBChannel.createColorTable( colorTables[1]);
    result = RGBChannel.createColorTable( colorTables[2]);

    return result;
}


int Levels::adjust(InputArray src, OutputArray dst)
{
	Mat input = src.getMat();
	if( input.empty() ) {
		return -1;
	}

	dst.create(src.size(), src.type());
	Mat output = dst.getMat();

	const uchar *in;
	uchar *out;
	int width = input.cols;
	int height = input.rows;
	int channels = input.channels();

	uchar colorTables[3][256];

	//create color tables
	if ( ! createColorTables( colorTables ) )  {
		//error create color table"
		return 1;
	}

	//adjust each pixel
	#ifdef HAVE_OPENMP
	#pragma omp parallel for
	#endif
	for (int y = 0; y < height; y ++) {
		in = input.ptr<uchar>(y);
		out = output.ptr<uchar>(y);
		for (int x = 0; x < width; x ++) {
			for (int c = 0; c < 3; c++) {
				*out++ = colorTables[c][*in++];
			}
			for (int c = 0; c < channels - 3; c++) {
				*out++ = *in++;
			}
		}
	}

	return 0;
}


} /* namespace cv */

【完整演示代码下载】

https://download.csdn.net/download/FL1623863129/89632958

相关推荐
workflower1 分钟前
活动图描述场景
开发语言·软件工程·需求分析·软件需求·敏捷流程
梦想的初衷~4 分钟前
基于现代R语言【Tidyverse、Tidymodel】的机器学习方法
开发语言·机器学习·r语言
香蕉可乐荷包蛋7 分钟前
Python学习之路(十三)-常用函数的使用,及优化
开发语言·python·学习
chian-ocean10 分钟前
零基础入门:用C++从零实现TCP Socket网络小工具
网络·c++·tcp/ip
惜.己15 分钟前
使用python的读取xml文件,简单的处理成元组数组
xml·开发语言·python·测试工具
apihz38 分钟前
域名WHOIS信息查询免费API使用指南
android·开发语言·数据库·网络协议·tcp/ip
珹洺1 小时前
C++算法竞赛篇:DevC++ 如何进行debug调试
java·c++·算法
coding随想1 小时前
掌控网页的魔法之书:JavaScript DOM的奇幻之旅
开发语言·javascript·ecmascript
爱吃烤鸡翅的酸菜鱼1 小时前
IDEA高效开发:Database Navigator插件安装与核心使用指南
java·开发语言·数据库·编辑器·intellij-idea·database
心情好的小球藻2 小时前
Python应用进阶DAY9--类型注解Type Hinting
开发语言·python