aboutsummaryrefslogtreecommitdiff
path: root/externals/grill/fftease/src/taint~.cpp
blob: b9bf9cf949104898aa89bea7cbfe3d7c193a16eb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*

FFTease - A set of Live Spectral Processors
Originally written by Eric Lyon and Christopher Penrose for the Max/MSP platform

Copyright (c)Thomas Grill (xovo@gmx.net)
For information on usage and redistribution, and for a DISCLAIMER OF ALL
WARRANTIES, see the file, "license.txt," in this distribution.  

*/

#include "main.h"
#include <stdlib.h>

class taint:
	public fftease
{
	FLEXT_HEADER_S(taint,fftease,setup)
	
public:
	taint(I argc,const t_atom *argv);

protected:

	virtual V Transform(I n,S *const *in);

	V ms_thresh(F thr) { _threshold = FromdB(_threshdB = thr); }

	F _threshold,_threshdB;
	BL _invert;

private:
	static V setup(t_classid c);

	FLEXT_ATTRGET_F(_threshdB)
	FLEXT_CALLSET_F(ms_thresh)
	FLEXT_ATTRVAR_B(_invert)
};

FLEXT_LIB_DSP_V("taint~",taint)


V taint::setup(t_classid c)
{
	FLEXT_CADDATTR_VAR(c,"thresh",_threshdB,ms_thresh);
	FLEXT_CADDATTR_VAR1(c,"invert",_invert);
}


taint::taint(I argc,const t_atom *argv):
	fftease(4,F_STEREO|F_BALANCED|F_BITSHUFFLE|F_NOPH2),
	_threshdB(-10),_invert(false)
{
	/* parse and set object's options given */
	if(argc >= 1) {
		if(CanbeFloat(argv[0]))
			_threshdB = GetAFloat(argv[0]);
		else
			post("%s - Threshold must be a float value - set to %f",thisName(),_threshdB);
	}

	if(argc >= 2) {
		if(CanbeBool(argv[1]))
			_invert = GetABool(argv[1]);
		else
			post("%s - Invert must be a boolean value - set to %0i",thisName(),_invert?1:0);
	}

	ms_thresh(_threshdB);

	AddInSignal("Messages and frequency reference signal");
	AddInSignal("Amplitude reference signal");
	AddOutSignal("Transformed signal");
}


V taint::Transform(I _N,S *const *)
{
	register const F thr = _threshold;
	
	if(_invert) {
		// use threshold for inverse filtering to avoid division by zero 
		for (I i = 0; i <= _N; i += 2) {
			const F magnitude = _channel2[i];

			if ( magnitude < thr )
				_channel1[i] = 0;
			else
				_channel1[i] /= magnitude;
		}
	}
	else {
		// simple multiplication of magnitudes 
		for (I i = 0; i <= _N; i += 2) {
			const F magnitude = _channel2[i];
			if (magnitude > thr) _channel1[i] *= magnitude;
		}
	}
}