paulxstretch/Source/PS_Source/ProcessedStretch.h

486 lines
13 KiB
C++

/*
Copyright (C) 2011 Nasca Octavian Paul
Author: Nasca Octavian Paul
This program is free software; you can redistribute it and/or modify
it under the terms of version 2 of the GNU General Public License
as published by the Free Software Foundation.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License (version 2) for more details.
You should have received a copy of the GNU General Public License (version 2)
along with this program; if not, write to the Free Software Foundation,
Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#pragma once
#include "FreeEdit.h"
#include "Stretch.h"
struct ProcessParameters
{
ProcessParameters()
{
pitch_shift.enabled=false;
pitch_shift.cents=0;
octave.enabled=false;
octave.om2=octave.om1=octave.o1=octave.o15=octave.o2=0.0f;
octave.o0=1.0f;
freq_shift.enabled=false;
freq_shift.Hz=0;
compressor.enabled=false;
compressor.power=0.0f;
filter.enabled=false;
filter.stop=false;
filter.low=0.0f;
filter.high=22000.0f;
filter.hdamp=0.0f;
harmonics.enabled=false;
harmonics.freq=440.0f;
harmonics.bandwidth=25.0f;
harmonics.nharmonics=10;
harmonics.gauss=false;
spread.enabled=false;
spread.bandwidth=0.3f;
tonal_vs_noise.enabled=false;
tonal_vs_noise.preserve=0.5f;
tonal_vs_noise.bandwidth=0.9f;
};
~ProcessParameters(){
};
struct{
bool enabled;
int cents;
}pitch_shift;
struct{
bool enabled;
REALTYPE om2,om1,o0,o1,o15,o2;
}octave;
struct{
bool enabled;
int Hz;
}freq_shift;
struct{
bool enabled;
REALTYPE power;
}compressor;
struct{
bool enabled;
REALTYPE low,high;
REALTYPE hdamp;
bool stop;
}filter;
struct{
bool enabled;
REALTYPE freq;
REALTYPE bandwidth;
int nharmonics;
bool gauss;
}harmonics;
struct{
bool enabled;
REALTYPE bandwidth;
}spread;
struct{
bool enabled;
REALTYPE preserve;
REALTYPE bandwidth;
}tonal_vs_noise;
//FreeEdit free_filter;
//FreeEdit stretch_multiplier;
/*
auto getMembers() const
{
return std::make_tuple(pitch_shift.enabled,
pitch_shift.cents,
octave.enabled,
octave.o0,
octave.o1,
octave.o15,
octave.o2,
octave.om1,
octave.om2,
spread.enabled,
spread.bandwidth,
tonal_vs_noise.enabled,
tonal_vs_noise.bandwidth,
tonal_vs_noise.preserve,
freq_shift.enabled,
freq_shift.Hz,
compressor.enabled,
compressor.power,
harmonics.bandwidth,
harmonics.enabled,
harmonics.freq,
harmonics.gauss,
harmonics.nharmonics,
filter.enabled,
filter.hdamp,
filter.high,
filter.low,
filter.stop);
}
bool operator == (const ProcessParameters& other) const
{
return getMembers() == other.getMembers();
}
*/
bool operator == (const ProcessParameters& other) const noexcept
{
return pitch_shift.enabled == other.pitch_shift.enabled &&
pitch_shift.cents == other.pitch_shift.cents &&
octave.enabled == other.octave.enabled &&
octave.o0 == other.octave.o0 &&
octave.o1 == other.octave.o1 &&
octave.o15 == other.octave.o15 &&
octave.o2 == other.octave.o2 &&
octave.om1 == other.octave.om1 &&
octave.om2 == other.octave.om2 &&
spread.enabled == other.spread.enabled &&
spread.bandwidth == other.spread.bandwidth &&
tonal_vs_noise.enabled == other.tonal_vs_noise.enabled &&
tonal_vs_noise.bandwidth == other.tonal_vs_noise.bandwidth &&
tonal_vs_noise.preserve == other.tonal_vs_noise.preserve &&
freq_shift.enabled == other.freq_shift.enabled &&
freq_shift.Hz == other.freq_shift.Hz &&
compressor.enabled == other.compressor.enabled &&
compressor.power == other.compressor.power &&
harmonics.bandwidth == other.harmonics.bandwidth &&
harmonics.enabled == other.harmonics.enabled &&
harmonics.freq == other.harmonics.freq &&
harmonics.gauss == other.harmonics.gauss &&
harmonics.nharmonics == other.harmonics.nharmonics &&
filter.enabled == other.filter.enabled &&
filter.hdamp == other.filter.hdamp &&
filter.high == other.filter.high &&
filter.low == other.filter.low &&
filter.stop == other.filter.stop;
}
};
inline REALTYPE profile(REALTYPE fi, REALTYPE bwi) {
REALTYPE x = fi / bwi;
x *= x;
if (x>14.71280603) return 0.0;
return exp(-x);///bwi;
};
inline void spectrum_copy(int nfreq, REALTYPE* freq1, REALTYPE* freq2)
{
for (int i = 0; i<nfreq; i++) freq2[i] = freq1[i];
};
inline void spectrum_spread(int nfreq, double samplerate,
std::vector<REALTYPE>& tmpfreq1,
REALTYPE *freq1, REALTYPE *freq2, REALTYPE spread_bandwidth) {
//convert to log spectrum
REALTYPE minfreq = 20.0f;
REALTYPE maxfreq = 0.5f*samplerate;
REALTYPE log_minfreq = log(minfreq);
REALTYPE log_maxfreq = log(maxfreq);
for (int i = 0; i<nfreq; i++) {
REALTYPE freqx = i / (REALTYPE)nfreq;
REALTYPE x = exp(log_minfreq + freqx * (log_maxfreq - log_minfreq)) / maxfreq * nfreq;
REALTYPE y = 0.0f;
int x0 = (int)floor(x); if (x0 >= nfreq) x0 = nfreq - 1;
int x1 = x0 + 1; if (x1 >= nfreq) x1 = nfreq - 1;
REALTYPE xp = x - x0;
if (x<nfreq) {
y = freq1[x0] * (1.0f - xp) + freq1[x1] * xp;
};
tmpfreq1[i] = y;
};
//increase the bandwidth of each harmonic (by smoothing the log spectrum)
int n = 2;
REALTYPE bandwidth = spread_bandwidth;
REALTYPE a = 1.0f - pow(2.0f, -bandwidth * bandwidth*10.0f);
a = pow(a, 8192.0f / nfreq * n);
for (int k = 0; k<n; k++) {
tmpfreq1[0] = 0.0f;
for (int i = 1; i<nfreq; i++) {
tmpfreq1[i] = tmpfreq1[i - 1] * a + tmpfreq1[i] * (1.0f - a);
};
tmpfreq1[nfreq - 1] = 0.0f;
for (int i = nfreq - 2; i>0; i--) {
tmpfreq1[i] = tmpfreq1[i + 1] * a + tmpfreq1[i] * (1.0f - a);
};
};
freq2[0] = 0;
REALTYPE log_maxfreq_d_minfreq = log(maxfreq / minfreq);
for (int i = 1; i<nfreq; i++) {
REALTYPE freqx = i / (REALTYPE)nfreq;
REALTYPE x = log((freqx*maxfreq) / minfreq) / log_maxfreq_d_minfreq * nfreq;
REALTYPE y = 0.0;
if ((x>0.0) && (x<nfreq)) {
int x0 = (int)floor(x); if (x0 >= nfreq) x0 = nfreq - 1;
int x1 = x0 + 1; if (x1 >= nfreq) x1 = nfreq - 1;
REALTYPE xp = x - x0;
y = tmpfreq1[x0] * (1.0f - xp) + tmpfreq1[x1] * xp;
};
freq2[i] = y;
};
};
inline void spectrum_do_compressor(const ProcessParameters& pars, int nfreq, REALTYPE *freq1, REALTYPE *freq2) {
REALTYPE rms = 0.0;
for (int i = 0; i<nfreq; i++) rms += freq1[i] * freq1[i];
rms = sqrt(rms / nfreq)*0.1f;
if (rms<1e-3f) rms = 1e-3f;
REALTYPE _rap = pow(rms, -pars.compressor.power);
for (int i = 0; i<nfreq; i++) freq2[i] = freq1[i] * _rap;
};
inline void spectrum_do_tonal_vs_noise(const ProcessParameters& pars, int nfreq, double samplerate,
std::vector<REALTYPE>& tmpfreq1,
REALTYPE *freq1, REALTYPE *freq2) {
spectrum_spread(nfreq, samplerate, tmpfreq1, freq1, tmpfreq1.data(), pars.tonal_vs_noise.bandwidth);
if (pars.tonal_vs_noise.preserve >= 0.0) {
REALTYPE mul = (pow(10.0f, pars.tonal_vs_noise.preserve) - 1.0f);
for (int i = 0; i<nfreq; i++) {
REALTYPE x = freq1[i];
REALTYPE smooth_x = tmpfreq1[i] + 1e-6f;
REALTYPE result = 0.0f;
result = x - smooth_x * mul;
if (result<0.0f) result = 0.0f;
freq2[i] = result;
};
}
else {
REALTYPE mul = (pow(5.0f, 1.0f + pars.tonal_vs_noise.preserve) - 1.0f);
for (int i = 0; i<nfreq; i++) {
REALTYPE x = freq1[i];
REALTYPE smooth_x = tmpfreq1[i] + 1e-6f;
REALTYPE result = 0.0f;
result = x - smooth_x * mul + 0.1f*mul;
if (result<0.0f) result = x;
else result = 0.0f;
freq2[i] = result;
};
};
};
inline void spectrum_do_harmonics(const ProcessParameters& pars, std::vector<REALTYPE>& tmpfreq1, int nfreq, double samplerate, REALTYPE *freq1, REALTYPE *freq2) {
REALTYPE freq = pars.harmonics.freq;
REALTYPE bandwidth = pars.harmonics.bandwidth;
int nharmonics = pars.harmonics.nharmonics;
if (freq<10.0) freq = 10.0;
REALTYPE *amp = tmpfreq1.data();
for (int i = 0; i<nfreq; i++) amp[i] = 0.0;
for (int nh = 1; nh <= nharmonics; nh++) {//for each harmonic
REALTYPE bw_Hz;//bandwidth of the current harmonic measured in Hz
REALTYPE bwi;
REALTYPE fi;
REALTYPE f = nh * freq;
if (f >= samplerate / 2) break;
bw_Hz = (pow(2.0f, bandwidth / 1200.0f) - 1.0f)*f;
bwi = bw_Hz / (2.0f*samplerate);
fi = f / samplerate;
REALTYPE sum = 0.0f;
REALTYPE max = 0.0f;
for (int i = 1; i<nfreq; i++) {//todo: optimize here
REALTYPE hprofile;
hprofile = profile((i / (REALTYPE)nfreq*0.5f) - fi, bwi);
amp[i] += hprofile;
if (max<hprofile) max = hprofile;
sum += hprofile;
};
};
REALTYPE max = 0.0;
for (int i = 1; i<nfreq; i++) {
if (amp[i]>max) max = amp[i];
};
if (max<1e-8f) max = 1e-8f;
for (int i = 1; i<nfreq; i++) {
//REALTYPE c,s;
REALTYPE a = amp[i] / max;
if (!pars.harmonics.gauss) a = (a<0.368f ? 0.0f : 1.0f);
freq2[i] = freq1[i] * a;
};
};
inline void spectrum_add(int nfreq, REALTYPE *freq2, REALTYPE *freq1, REALTYPE a) {
for (int i = 0; i<nfreq; i++) freq2[i] += freq1[i] * a;
};
inline void spectrum_zero(int nfreq,REALTYPE *freq1) {
for (int i = 0; i<nfreq; i++) freq1[i] = 0.0;
};
inline void spectrum_do_freq_shift(const ProcessParameters& pars, int nfreq, double samplerate, REALTYPE *freq1, REALTYPE *freq2) {
spectrum_zero(nfreq, freq2);
int ifreq = (int)(pars.freq_shift.Hz / (samplerate*0.5)*nfreq);
for (int i = 0; i<nfreq; i++) {
int i2 = ifreq + i;
if ((i2>0) && (i2<nfreq)) freq2[i2] = freq1[i];
};
};
inline void spectrum_do_pitch_shift(const ProcessParameters& pars, int nfreq, REALTYPE *freq1, REALTYPE *freq2, REALTYPE _rap) {
spectrum_zero(nfreq,freq2);
if (_rap<1.0) {//down
for (int i = 0; i<nfreq; i++) {
int i2 = (int)(i*_rap);
if (i2 >= nfreq) break;
freq2[i2] += freq1[i];
};
};
if (_rap >= 1.0) {//up
_rap = 1.0f / _rap;
for (int i = 0; i<nfreq; i++) {
freq2[i] = freq1[(int)(i*_rap)];
};
};
};
inline void spectrum_do_octave(const ProcessParameters& pars, int nfreq, double samplerate,
std::vector<REALTYPE>& sumfreq,
std::vector<REALTYPE>& tmpfreq1,
REALTYPE *freq1, REALTYPE *freq2) {
spectrum_zero(nfreq,sumfreq.data());
if (pars.octave.om2>1e-3) {
spectrum_do_pitch_shift(pars,nfreq, freq1, tmpfreq1.data(), 0.25);
spectrum_add(nfreq, sumfreq.data(), tmpfreq1.data(), pars.octave.om2);
};
if (pars.octave.om1>1e-3) {
spectrum_do_pitch_shift(pars,nfreq, freq1, tmpfreq1.data(), 0.5);
spectrum_add(nfreq,sumfreq.data(), tmpfreq1.data(), pars.octave.om1);
};
if (pars.octave.o0>1e-3) {
spectrum_add(nfreq,sumfreq.data(), freq1, pars.octave.o0);
};
if (pars.octave.o1>1e-3) {
spectrum_do_pitch_shift(pars,nfreq, freq1, tmpfreq1.data(), 2.0);
spectrum_add(nfreq,sumfreq.data(), tmpfreq1.data(), pars.octave.o1);
};
if (pars.octave.o15>1e-3) {
spectrum_do_pitch_shift(pars,nfreq, freq1, tmpfreq1.data(), 3.0);
spectrum_add(nfreq,sumfreq.data(), tmpfreq1.data(), pars.octave.o15);
};
if (pars.octave.o2>1e-3) {
spectrum_do_pitch_shift(pars, nfreq, freq1, tmpfreq1.data(), 4.0);
spectrum_add(nfreq,sumfreq.data(), tmpfreq1.data(), pars.octave.o2);
};
REALTYPE sum = 0.01f + pars.octave.om2 + pars.octave.om1 + pars.octave.o0 + pars.octave.o1 + pars.octave.o15 + pars.octave.o2;
if (sum<0.5f) sum = 0.5f;
for (int i = 0; i<nfreq; i++) freq2[i] = sumfreq[i] / sum;
};
inline void spectrum_do_filter(const ProcessParameters& pars, int nfreq, double samplerate, REALTYPE *freq1, REALTYPE *freq2) {
REALTYPE low = 0, high = 0;
if (pars.filter.low<pars.filter.high) {//sort the low/high freqs
low = pars.filter.low;
high = pars.filter.high;
}
else {
high = pars.filter.low;
low = pars.filter.high;
};
int ilow = (int)(low / samplerate * nfreq*2.0f);
int ihigh = (int)(high / samplerate * nfreq*2.0f);
REALTYPE dmp = 1.0;
REALTYPE dmprap = 1.0f - pow(pars.filter.hdamp*0.5f, 4.0f);
for (int i = 0; i<nfreq; i++) {
REALTYPE a = 0.0f;
if ((i >= ilow) && (i<ihigh)) a = 1.0f;
if (pars.filter.stop) a = 1.0f - a;
freq2[i] = freq1[i] * a*dmp;
dmp *= dmprap + 1e-8f;
};
};
class SpectrumProcess
{
public:
SpectrumProcess() {}
SpectrumProcess(String name, int index) :
m_name(name), m_index(index) {}
String m_name;
int m_index = -1;
private:
};
std::vector<SpectrumProcess> make_spectrum_processes();
class ProcessedStretch final : public Stretch
{
public:
//stereo_mode: 0=mono,1=left,2=right
ProcessedStretch(REALTYPE rap_,int in_bufsize_,FFTWindow w=W_HAMMING,bool bypass_=false,REALTYPE samplerate_=44100.0f,int stereo_mode=0);
~ProcessedStretch();
void set_parameters(ProcessParameters *ppar);
std::vector<int> m_spectrum_processes;
void setBufferSize(int sz) override;
private:
REALTYPE get_stretch_multiplier(REALTYPE pos_percents) override;
// void process_output(REALTYPE *smps,int nsmps);
void process_spectrum(REALTYPE *freq) override;
//void copy(const realvector& freq1,realvector& freq2);
void copy(REALTYPE* freq1, REALTYPE* freq2);
void add(REALTYPE *freq2,REALTYPE *freq1,REALTYPE a=1.0);
void mul(REALTYPE *freq1,REALTYPE a);
void zero(REALTYPE *freq1);
void update_free_filter();
int nfreq=0;
std::vector<REALTYPE> free_filter_freqs;
ProcessParameters pars;
std::vector<REALTYPE> infreq,sumfreq,tmpfreq1,tmpfreq2;
//REALTYPE *fbfreq;
};