239 lines
7 KiB
C++
239 lines
7 KiB
C++
#pragma once
|
|
|
|
// YOMBParam.h: interface for the CYOMBParam class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#if !defined(AFX_YOMBPARAM_H__41D42152_F2EE_11D5_B92D_0040F674BE6A__INCLUDED_)
|
|
#define AFX_YOMBPARAM_H__41D42152_F2EE_11D5_B92D_0040F674BE6A__INCLUDED_
|
|
|
|
#include <vector>
|
|
#include "STColSelPic.h"
|
|
#include "SDef.h"
|
|
#include "InputParam.h"
|
|
#include "YOMBInputParam.h"
|
|
#include "BlurMatrix.h"
|
|
//#pragma warning(disable: 4786)
|
|
//#include "tmsg.h"
|
|
|
|
#define P(d) tmsg_info(" - %d -\n", d)
|
|
|
|
class CYOMBParam {
|
|
public:
|
|
bool m_isRandomSampling;
|
|
bool m_isShowSelection;
|
|
bool m_isStopAtContour;
|
|
bool m_isBlurOnPair;
|
|
double m_dSample;
|
|
int m_nbSample;
|
|
double m_dA, m_dAB;
|
|
std::string m_name;
|
|
|
|
bool m_isCM;
|
|
// for RGBM color groups
|
|
std::vector<I_PIXEL> m_color;
|
|
// for CMAP color indides
|
|
COLOR_INDEX_LIST m_ink, m_paint;
|
|
|
|
CYOMBParam()
|
|
: m_isRandomSampling(false)
|
|
, m_isShowSelection(false)
|
|
, m_isStopAtContour(false)
|
|
, m_isBlurOnPair(false)
|
|
, m_dSample(0.0)
|
|
, m_nbSample(0)
|
|
, m_dA(0.0)
|
|
, m_dAB(0.0)
|
|
, m_name("")
|
|
, m_isCM(false)
|
|
, m_color(0) {
|
|
m_ink.nb = m_paint.nb = 0;
|
|
};
|
|
|
|
CYOMBParam(const CYOMBParam &bp)
|
|
: m_color(bp.m_color)
|
|
, m_isRandomSampling(bp.m_isRandomSampling)
|
|
, m_isShowSelection(bp.m_isShowSelection)
|
|
, m_isStopAtContour(bp.m_isStopAtContour)
|
|
, m_isBlurOnPair(bp.m_isBlurOnPair)
|
|
, m_dSample(bp.m_dSample)
|
|
, m_nbSample(bp.m_nbSample)
|
|
, m_dA(bp.m_dA)
|
|
, m_dAB(bp.m_dAB)
|
|
, m_name(bp.m_name)
|
|
, m_isCM(bp.m_isCM) {
|
|
if (m_isCM) {
|
|
m_ink.nb = bp.m_ink.nb;
|
|
int i = 0;
|
|
for (i = 0; i < m_ink.nb; i++) m_ink.ci[i] = bp.m_ink.ci[i];
|
|
m_paint.nb = bp.m_paint.nb;
|
|
for (i = 0; i < m_paint.nb; i++) m_paint.ci[i] = bp.m_paint.ci[i];
|
|
}
|
|
};
|
|
|
|
virtual ~CYOMBParam(){};
|
|
|
|
void print();
|
|
void null();
|
|
void read(const CInputParam &ip);
|
|
#ifdef _WIN32
|
|
bool read(std::basic_ifstream<char> &in);
|
|
#endif
|
|
void makeColorsUS();
|
|
void makeItUS();
|
|
static void adjustToMatte(I_PIXEL &p);
|
|
|
|
// void blurPixel(CSTColSelPic<UC_PIXEL>& pic, const int xx, const int yy,
|
|
// const CBlurMatrix& bm, UC_PIXEL& col,const
|
|
// UCHAR
|
|
// osel);
|
|
// bool isContourOnPath(const int xx, const int yy,
|
|
// std::vector<BLURSECTION>::const_iterator
|
|
// pBS,
|
|
// CSTColSelPic<UC_PIXEL>& pic);
|
|
// void addPixel(I_PIXEL& p, const UC_PIXEL* pic);
|
|
// void addPixel(I_PIXEL& p, const CSTColSelPic<UC_PIXEL>& pic,
|
|
// const int xxyy, const int xy );
|
|
void addPixel(I_PIXEL &p, const I_PIXEL &pp);
|
|
int getColorIndex(const UCHAR c);
|
|
void scale(const double d);
|
|
|
|
template <class P>
|
|
void addPixel(I_PIXEL &p, const P *pic) {
|
|
p.r += ((int)pic->r);
|
|
p.g += ((int)pic->g);
|
|
p.b += ((int)pic->b);
|
|
p.m += ((int)pic->m);
|
|
}
|
|
|
|
// Checks the path from blurred pixel to the sample pixel.
|
|
template <class P>
|
|
bool isContourOnPath(const int xx, const int yy,
|
|
std::vector<BLURSECTION>::const_iterator pBS,
|
|
CSTColSelPic<P> &pic) {
|
|
for (std::vector<SXYD>::const_iterator p = pBS->begin(); p != pBS->end();
|
|
++p) {
|
|
int x = xx + p->x;
|
|
int y = yy + p->y;
|
|
int xy = y * pic.m_lX + x;
|
|
if (x >= 0 && y >= 0 && x < pic.m_lX && y < pic.m_lY)
|
|
if (*(pic.m_sel + xy) == (UCHAR)0) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template <class P>
|
|
void addPixel(I_PIXEL &ip, const CSTColSelPic<P> &pic, const int xxyy,
|
|
const int xy) {
|
|
if (!m_isBlurOnPair) {
|
|
addPixel(ip, pic.m_pic + xy);
|
|
} else {
|
|
int i = getColorIndex(*(pic.m_sel + xy));
|
|
int ii = getColorIndex(*(pic.m_sel + xxyy));
|
|
int l = m_color.size() / 2;
|
|
|
|
if (i == ii || abs(ii - i) == l) {
|
|
addPixel(ip, pic.m_pic + xy);
|
|
} else {
|
|
if (ii < l) {
|
|
if (i < l) {
|
|
addPixel(ip, m_color[ii]);
|
|
} else
|
|
addPixel(ip, m_color[ii + l]);
|
|
} else {
|
|
if (i < l) {
|
|
addPixel(ip, m_color[ii - l]);
|
|
} else
|
|
addPixel(ip, m_color[ii]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
template <class P>
|
|
void blurPixel(CSTColSelPic<P> &pic, const int xx, const int yy,
|
|
const CBlurMatrix &bm, I_PIXEL &col, const int iBm) {
|
|
I_PIXEL p = {0, 0, 0, 0};
|
|
int nb = 0;
|
|
|
|
int xxyy = yy * pic.m_lX + xx;
|
|
for (std::vector<BLURSECTION>::const_iterator pBS = bm.m_m[iBm].begin();
|
|
pBS != bm.m_m[iBm].end(); ++pBS) {
|
|
// const SXYD* xyd=pBS->begin();
|
|
BLURSECTION::const_iterator xyd = pBS->begin();
|
|
|
|
int x = xx + xyd->x;
|
|
int y = yy + xyd->y;
|
|
if (x >= 0 && y >= 0 && x < pic.m_lX && y < pic.m_lY) {
|
|
int xy = y * pic.m_lX + x;
|
|
if (*(pic.m_sel + xy) > (UCHAR)0) {
|
|
if (bm.m_isSAC) {
|
|
if (!isContourOnPath(xx, yy, pBS, pic)) {
|
|
addPixel(p, pic, xxyy, xy);
|
|
nb++;
|
|
}
|
|
} else {
|
|
addPixel(p, pic, xxyy, xy);
|
|
nb++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (nb > 0) {
|
|
double r = (double)p.r / (double)nb;
|
|
double g = (double)p.g / (double)nb;
|
|
double b = (double)p.b / (double)nb;
|
|
double m = (double)p.m / (double)nb;
|
|
col.r = I_ROUND(r);
|
|
col.g = I_ROUND(g);
|
|
col.b = I_ROUND(b);
|
|
col.m = I_ROUND(m);
|
|
} else {
|
|
P *pPic = pic.m_pic + yy * pic.m_lX + xx;
|
|
col.r = (int)(pPic->r);
|
|
col.g = (int)(pPic->g);
|
|
col.b = (int)(pPic->b);
|
|
col.m = (int)(pPic->m);
|
|
}
|
|
}
|
|
|
|
template <class P>
|
|
bool isRealMixture(CSTColSelPic<P> &pic, const int xx, const int yy,
|
|
const CBlurMatrix &bm, const int iBm,
|
|
const UCHAR osel) const {
|
|
for (std::vector<BLURSECTION>::const_iterator pBS = bm.m_m[iBm].begin();
|
|
pBS != bm.m_m[iBm].end(); ++pBS) {
|
|
// const SXYD* xyd= pBS->begin();
|
|
BLURSECTION::const_iterator xyd = pBS->begin();
|
|
int x = xx + xyd->x;
|
|
int y = yy + xyd->y;
|
|
if (x >= 0 && x < pic.m_lX && y >= 0 && y < pic.m_lY) {
|
|
int xy = y * pic.m_lX + x;
|
|
UCHAR *pSel = pic.m_sel + xy;
|
|
if (*pSel > (UCHAR)0 && *pSel != osel) return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template <class P>
|
|
void doIt(CSTColSelPic<P> &ipic, CSTColSelPic<P> &opic) {
|
|
CBlurMatrix bm(m_dSample, m_nbSample, m_isStopAtContour,
|
|
m_isRandomSampling);
|
|
UCHAR *pSel = ipic.m_sel;
|
|
for (int y = 0; y < ipic.m_lY; y++)
|
|
for (int x = 0; x < ipic.m_lX; x++, pSel++)
|
|
if (*pSel > (UCHAR)0) {
|
|
I_PIXEL col;
|
|
int iBm = bm.m_isRS ? rand() % NBRS : 0;
|
|
if (isRealMixture(ipic, x, y, bm, iBm, *pSel))
|
|
blurPixel(ipic, x, y, bm, col, iBm);
|
|
else
|
|
ipic.getPixel(x, y, col);
|
|
opic.setRGBM(x, y, col);
|
|
}
|
|
}
|
|
};
|
|
|
|
#endif // !defined(AFX_YOMBPARAM_H__41D42152_F2EE_11D5_B92D_0040F674BE6A__INCLUDED_)
|