ccm.hpp

// This file is part of OpenCV project.




// It is subject to the license terms in the LICENSE file found in the top-level directory




// of this distribution and at http://opencv.org/license.html.




//




//




//                       License Agreement




//              For Open Source Computer Vision Library




//




// Copyright(C) 2020, Huawei Technologies Co.,Ltd. All rights reserved.




// Third party copyrights are property of their respective owners.




//




// Licensed under the Apache License, Version 2.0 (the "License");




// you may not use this file except in compliance with the License.




// You may obtain a copy of the License at




//




//             http://www.apache.org/licenses/LICENSE-2.0




//




// Unless required by applicable law or agreed to in writing, software




// distributed under the License is distributed on an "AS IS" BASIS,




// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.




// See the License for the specific language governing permissions and




// limitations under the License.




//




// Author: Longbu Wang <wanglongbu@huawei.com.com>




//         Jinheng Zhang <zhangjinheng1@huawei.com>




//         Chenqi Shan <shanchenqi@huawei.com>








#ifndef __OPENCV_MCC_CCM_HPP__




#define __OPENCV_MCC_CCM_HPP__








#include <opencv2/core.hpp>




#include <opencv2/imgproc.hpp>








namespace

cv



{



namespace
ccm


{



enum
CCM_TYPE



{



CCM_3x3,




CCM_4x3,



};







enum
INITIAL_METHOD_TYPE



{



INITIAL_METHOD_WHITE_BALANCE,




INITIAL_METHOD_LEAST_SQUARE,



};



enum
CONST_COLOR
{



COLORCHECKER_Macbeth,




COLORCHECKER_Vinyl,




COLORCHECKER_DigitalSG,



};



enum
COLOR_SPACE
{



COLOR_SPACE_sRGB,




COLOR_SPACE_sRGBL,




COLOR_SPACE_AdobeRGB,




COLOR_SPACE_AdobeRGBL,




COLOR_SPACE_WideGamutRGB,




COLOR_SPACE_WideGamutRGBL,




COLOR_SPACE_ProPhotoRGB,




COLOR_SPACE_ProPhotoRGBL,




COLOR_SPACE_DCI_P3_RGB,




COLOR_SPACE_DCI_P3_RGBL,




COLOR_SPACE_AppleRGB,




COLOR_SPACE_AppleRGBL,




COLOR_SPACE_REC_709_RGB,




COLOR_SPACE_REC_709_RGBL,




COLOR_SPACE_REC_2020_RGB,




COLOR_SPACE_REC_2020_RGBL,




COLOR_SPACE_XYZ_D65_2,




COLOR_SPACE_XYZ_D65_10,




COLOR_SPACE_XYZ_D50_2,




COLOR_SPACE_XYZ_D50_10,




COLOR_SPACE_XYZ_A_2,




COLOR_SPACE_XYZ_A_10,




COLOR_SPACE_XYZ_D55_2,




COLOR_SPACE_XYZ_D55_10,




COLOR_SPACE_XYZ_D75_2,




COLOR_SPACE_XYZ_D75_10,




COLOR_SPACE_XYZ_E_2,




COLOR_SPACE_XYZ_E_10,




COLOR_SPACE_Lab_D65_2,




COLOR_SPACE_Lab_D65_10,




COLOR_SPACE_Lab_D50_2,




COLOR_SPACE_Lab_D50_10,




COLOR_SPACE_Lab_A_2,




COLOR_SPACE_Lab_A_10,




COLOR_SPACE_Lab_D55_2,




COLOR_SPACE_Lab_D55_10,




COLOR_SPACE_Lab_D75_2,




COLOR_SPACE_Lab_D75_10,




COLOR_SPACE_Lab_E_2,




COLOR_SPACE_Lab_E_10,



};







enum
LINEAR_TYPE



{







LINEARIZATION_IDENTITY,




LINEARIZATION_GAMMA,




LINEARIZATION_COLORPOLYFIT,




LINEARIZATION_COLORLOGPOLYFIT,




LINEARIZATION_GRAYPOLYFIT,




LINEARIZATION_GRAYLOGPOLYFIT



};







enum
DISTANCE_TYPE



{



DISTANCE_CIE76,




DISTANCE_CIE94_GRAPHIC_ARTS,




DISTANCE_CIE94_TEXTILES,



DISTANCE_CIE2000,



DISTANCE_CMC_1TO1,




DISTANCE_CMC_2TO1,



DISTANCE_RGB,




DISTANCE_RGBL



};







class
CV_EXPORTS_W
ColorCorrectionModel



{



public:



CV_WRAP
ColorCorrectionModel(const
Mat& src,
CONST_COLOR
constcolor);







CV_WRAP
ColorCorrectionModel(const
Mat& src,
Mat
colors,
COLOR_SPACE
ref_cs);







CV_WRAP
ColorCorrectionModel(const
Mat& src,
Mat
colors,
COLOR_SPACE
ref_cs,
Mat
colored);







CV_WRAP
void
setColorSpace(COLOR_SPACE
cs);







CV_WRAP
void
setCCM_TYPE(CCM_TYPE
ccm_type);







CV_WRAP
void
setDistance(DISTANCE_TYPE
distance);







CV_WRAP
void
setLinear(LINEAR_TYPE
linear_type);







CV_WRAP
void
setLinearGamma(const
double& gamma);







CV_WRAP
void
setLinearDegree(const
int& deg);







CV_WRAP
void
setSaturatedThreshold(const
double& lower,
const
double& upper);







CV_WRAP
void
setWeightsList(const
Mat& weights_list);







CV_WRAP
void
setWeightCoeff(const
double& weights_coeff);







CV_WRAP
void
setInitialMethod(INITIAL_METHOD_TYPE
initial_method_type);







CV_WRAP
void
setMaxCount(const
int& max_count);







CV_WRAP
void
setEpsilon(const
double& epsilon);







CV_WRAP
void
run();







CV_WRAP
Mat
getCCM()
const;



CV_WRAP
double
getLoss()
const;



CV_WRAP
Mat
get_src_rgbl()
const;



CV_WRAP
Mat
get_dst_rgbl()
const;



CV_WRAP
Mat
getMask()
const;



CV_WRAP
Mat
getWeights()
const;







CV_WRAP
Mat
infer(const
Mat& img,
bool
islinear =
false);







class
Impl;



private:



std::shared_ptr<Impl> p;


};






}
// namespace ccm



}
// namespace cv








#endif