215 lines
8.3 KiB
C++
215 lines
8.3 KiB
C++
#pragma once
|
|
|
|
#ifndef igs_maxmin_getput_h
|
|
#define igs_maxmin_getput_h
|
|
|
|
#include <vector>
|
|
#include <limits> /* std::numeric_limits<->::max()
|
|
--> (std::numeric_limits<->::max)() */
|
|
#include "igs_ifx_common.h" // igs::color::ref_value(-)
|
|
|
|
namespace {
|
|
/* 画像配列の高さ位置を、実際の範囲内にclampし、scanlineの先頭を返す */
|
|
template <class T>
|
|
const T *csl_top_clamped_in_h_(const T *top, const int height, const int width,
|
|
const int channels, const int yy) {
|
|
if (height <= yy) {
|
|
return top + (channels * width * (height - 1));
|
|
} else if (yy < 0) {
|
|
return top;
|
|
}
|
|
return top + (channels * width * yy);
|
|
}
|
|
template <class T>
|
|
T *sl_out_clamped_in_h_(T *out, const int height, const int width,
|
|
const int channels, const int yy) {
|
|
if (height <= yy) {
|
|
return out + (channels * width * (height - 1));
|
|
} else if (yy < 0) {
|
|
return out;
|
|
}
|
|
return out + (channels * width * yy);
|
|
}
|
|
/* trackの余白を画像の端の色で塗りつぶす(余白部分をtracksの前後にコピー) */
|
|
template <class T>
|
|
void paint_margin_(const int margin, std::vector<T> &track) {
|
|
int xx;
|
|
/* 始点側の余白を塗りつぶす */
|
|
for (xx = 0; xx < margin; ++xx) {
|
|
track.at(xx) = track.at(margin);
|
|
}
|
|
/* 終端側の余白を塗りつぶす */
|
|
for (xx = 0; xx < margin; ++xx) {
|
|
track.at(track.size() - 1 - xx) = track.at(track.size() - 1 - margin);
|
|
}
|
|
}
|
|
/* 入力画像を計算用のtrackバッファーの真ん中に正規化して入れる */
|
|
template <class T>
|
|
void inn_to_track_(const T *sl, const int width, const int channels,
|
|
const double div_val, const int margin,
|
|
std::vector<double> &track) {
|
|
for (int xx = 0; xx < width; ++xx) {
|
|
track.at(margin + xx) = sl[xx * channels] / div_val;
|
|
}
|
|
}
|
|
/* 入力画像を正規化して、結果を置く場所(result)に初期値として入れる */
|
|
template <class T>
|
|
void inn_to_result_(
|
|
const T *inn, const int height, const int width, const int channels,
|
|
const int yy, const int zz, const double div_val,
|
|
std::vector<double> &result /* 元値をいれといて、結果を入れる */
|
|
) {
|
|
const T *ss = csl_top_clamped_in_h_(inn, height, width, channels, yy) + zz;
|
|
for (int xx = 0; xx < width; ++xx) {
|
|
result.at(xx) = ss[xx * channels] / div_val;
|
|
}
|
|
}
|
|
/* Scanlineで、効果を調節するデータ(alpha_ref)に、初期値を与える */
|
|
void alpha_ref_init_one_(const int width, std::vector<double> &alpha_ref) {
|
|
for (int xx = 0; xx < width; ++xx) {
|
|
alpha_ref.at(xx) = 1.0; /* 全面的に処理を加える値で埋める */
|
|
}
|
|
}
|
|
/* Scanlineで、効果を調節するデータ(alpha_ref)に、画像の影響を与える */
|
|
template <class RT>
|
|
void alpha_ref_mul_ref_(const RT *ref, const int height, const int width,
|
|
const int channels, const int yy, const int ref_mode,
|
|
std::vector<double> &alpha_ref) {
|
|
const int r_max = (std::numeric_limits<RT>::max)();
|
|
const RT *rr = csl_top_clamped_in_h_(ref, height, width, channels, yy);
|
|
for (int xx = 0; xx < width; ++xx) {
|
|
alpha_ref.at(xx) *=
|
|
igs::color::ref_value(&rr[xx * channels], channels, r_max, ref_mode);
|
|
}
|
|
}
|
|
/* Scanlineで、効果を調節するデータ(alpha_ref)に、
|
|
計算済み画像(out)のalphaチャンネル値の影響を与える */
|
|
template <class T>
|
|
void alpha_ref_mul_alpha_(const T *out, const int height, const int width,
|
|
const int channels, const int yy,
|
|
const double div_val,
|
|
std::vector<double> &alpha_ref) {
|
|
const T *dd = csl_top_clamped_in_h_(out, height, width, channels, yy) + 3;
|
|
for (int xx = 0; xx < width; ++xx) {
|
|
alpha_ref.at(xx) *= dd[xx * channels] / div_val;
|
|
}
|
|
}
|
|
}
|
|
//------------------------------------------------------------
|
|
namespace igs {
|
|
namespace maxmin {
|
|
namespace getput {
|
|
/* 1番初めのスキャンラインのセット
|
|
* IT is 'unsigned char' or 'unsigned short'
|
|
* alphaは先に処理して結果(out)に入れ、
|
|
その後rgbのために参照する
|
|
*/
|
|
template <class IT, class RT>
|
|
void get_first(
|
|
const IT *inn /* outと同じ高さ、幅、チャンネル数 */
|
|
,
|
|
const IT *out /* outの処理結果alpha値をinとして使用 */
|
|
,
|
|
const int hh, const int ww, const int ch,
|
|
const RT *ref /* outと同じ高さ、幅、チャンネル数 */
|
|
,
|
|
const int ref_mode /* 0=R,1=G,2=B,3=A,4=Luminance,5=Nothing */
|
|
,
|
|
const int yy, const int zz, const int margin, const bool add_blend_sw,
|
|
std::vector<std::vector<double>> &tracks /* sl影響範囲のpixel値 */
|
|
,
|
|
std::vector<double> &alpha_ref /* pixel毎の変化の割合 */
|
|
,
|
|
std::vector<double> &result /* 元値をいれといて、結果を入れる */
|
|
) {
|
|
const int t_max = (std::numeric_limits<IT>::max)();
|
|
const double div_val = static_cast<double>(t_max);
|
|
|
|
/* 計算範囲の画像値を計算バッファ(tracks)に入れる */
|
|
int ii = margin * 2;
|
|
for (int yp = -margin + yy; yp <= margin + yy; ++yp, --ii) {
|
|
const IT *sl = csl_top_clamped_in_h_(inn, hh, ww, ch, yp) + zz;
|
|
inn_to_track_(sl, ww, ch, div_val, margin, tracks.at(ii));
|
|
paint_margin_(margin, tracks.at(ii));
|
|
}
|
|
inn_to_result_(inn, hh, ww, ch, yy, zz, div_val, result);
|
|
if (alpha_ref.size() <= 0) {
|
|
return;
|
|
} /* alphaチャンネルを計算する場合 */
|
|
alpha_ref_init_one_(ww, alpha_ref);
|
|
if (ref != 0) {
|
|
alpha_ref_mul_ref_(ref, hh, ww, ch, yy, ref_mode, alpha_ref);
|
|
}
|
|
if (ch < 4) {
|
|
return;
|
|
} /* alphaチャンネルがない場合はここで終わり */
|
|
if (add_blend_sw) {
|
|
alpha_ref_mul_alpha_(out, hh, ww, ch, yy, div_val, alpha_ref);
|
|
}
|
|
}
|
|
/*--- 2番以後のスキャンラインのセット -------------------------------*/
|
|
template <class IT, class RT>
|
|
void get_next(const IT *inn /* outと同じ高さ、幅、チャンネル数 */
|
|
,
|
|
const IT *out /* outの処理結果alpha値をinとして使用 */
|
|
,
|
|
const int hh, const int ww, const int ch,
|
|
const RT *ref /* 求める画像(out)と同じ高さ、幅、チャンネル数 */
|
|
,
|
|
const int ref_mode /* 0=R,1=G,2=B,3=A,4=Luminance,5=Nothing */
|
|
,
|
|
const int yy, const int zz, const int margin,
|
|
const bool add_blend_sw,
|
|
std::vector<std::vector<double>> &tracks /* sl影響範囲のpixel値 */
|
|
,
|
|
std::vector<double> &alpha_ref /* pixel毎の変化の割合 */
|
|
,
|
|
std::vector<double> &result /* 元値をいれといて、結果を入れる */
|
|
) {
|
|
const int t_max = (std::numeric_limits<IT>::max)();
|
|
const double div_val = static_cast<double>(t_max);
|
|
|
|
const IT *sl = csl_top_clamped_in_h_(inn, hh, ww, ch, yy + margin) + zz;
|
|
inn_to_track_(sl, ww, ch, div_val, margin, tracks.at(0));
|
|
paint_margin_(margin, tracks.at(0));
|
|
|
|
inn_to_result_(inn, hh, ww, ch, yy, zz, div_val, result);
|
|
if (alpha_ref.size() <= 0) {
|
|
return;
|
|
} /* alphaチャンネルを計算する場合 */
|
|
alpha_ref_init_one_(ww, alpha_ref);
|
|
if (ref != 0) {
|
|
alpha_ref_mul_ref_(ref, hh, ww, ch, yy, ref_mode, alpha_ref);
|
|
}
|
|
if (ch < 4) {
|
|
return;
|
|
} /* alphaチャンネルがない場合はここで終わり */
|
|
if (add_blend_sw) {
|
|
alpha_ref_mul_alpha_(out, hh, ww, ch, yy, div_val, alpha_ref);
|
|
}
|
|
}
|
|
template <class T>
|
|
void copy(const T *inn, const int hh, const int ww, const int ch, const int yy,
|
|
const int zz, T *out) {
|
|
const T *ss = csl_top_clamped_in_h_(inn, hh, ww, ch, yy) + zz;
|
|
T *dd = sl_out_clamped_in_h_(out, hh, ww, ch, yy) + zz;
|
|
for (int xx = 0; xx < ww; ++xx) {
|
|
dd[ch * xx] = ss[ch * xx];
|
|
}
|
|
}
|
|
template <class T>
|
|
void put(const std::vector<double> &result, const int hh, const int ww,
|
|
const int ch, const int yy, const int zz, T *out) {
|
|
const int t_max = (std::numeric_limits<T>::max)();
|
|
const double mul_val = static_cast<double>(t_max) + 0.999999;
|
|
T *dd = sl_out_clamped_in_h_(out, hh, ww, ch, yy) + zz;
|
|
for (int xx = 0; xx < ww; ++xx) {
|
|
dd[ch * xx] = static_cast<T>(result.at(xx) * mul_val);
|
|
// std::cout << " (" << result.at(xx) << ")(" << (int)dd[ch * xx] << ")";
|
|
}
|
|
// std::cout << std::endl;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#endif /* !igs_maxmin_getput_h */
|