ProteoWizard
Functions | Variables
MagnitudeLorentzianTest.cpp File Reference
#include "MagnitudeLorentzian.hpp"
#include "MagnitudeLorentzianTestData.hpp"
#include "pwiz/data/misc/FrequencyData.hpp"
#include "pwiz/utility/misc/unit.hpp"
#include <boost/filesystem/operations.hpp>
#include "pwiz/utility/misc/Std.hpp"
#include <cstring>

Go to the source code of this file.

Functions

void testBasic ()
 
void testFit ()
 
void testData ()
 
int main (int argc, char *argv[])
 

Variables

ostream * os_ = 0
 
double epsilon_ = numeric_limits<double>::epsilon()
 

Function Documentation

◆ testBasic()

void testBasic ( )

Definition at line 42 of file MagnitudeLorentzianTest.cpp.

43{
44 MagnitudeLorentzian m(1,0,1); // m(x) = 1/sqrt(x^2+1)
46 unit_assert_equal(m(1), 1/sqrt(2.), epsilon_);
47 unit_assert_equal(m(2), 1/sqrt(5.), epsilon_);
48 unit_assert_equal(m(3), 1/sqrt(10.), epsilon_);
49 unit_assert_equal(m(4), 1/sqrt(17.), epsilon_);
50
51 // center == 0, alpha == 2*pi, tau == 1/(2*pi)
52 unit_assert_equal(m.center(), 0, epsilon_);
53 unit_assert_equal(m.alpha(), 2*M_PI, epsilon_);
54 unit_assert_equal(m.tau(), 1/(2*M_PI), epsilon_);
55
56 if (os_) *os_ << "testBasic(): success!\n";
57}
double epsilon_
ostream * os_
#define unit_assert_equal(x, y, epsilon)
Definition unit.hpp:99

References pwiz::frequency::MagnitudeLorentzian::alpha(), pwiz::frequency::MagnitudeLorentzian::center(), epsilon_, os_, pwiz::frequency::MagnitudeLorentzian::tau(), and unit_assert_equal.

Referenced by main().

◆ testFit()

void testFit ( )

Definition at line 60 of file MagnitudeLorentzianTest.cpp.

61{
62 MagnitudeLorentzian ref(1,0,1);
63
64 // choose sample values near 1!
65 // weighting pow(y,6) gives big roundoff errors
66
67 vector< pair<double,double> > samples;
68 for (int i=-2; i<3; i++)
69 samples.push_back(make_pair(i/10.,ref(i/10.)));
70
71 MagnitudeLorentzian m(samples);
72
73 if (os_)
74 {
75 *os_ << "coefficients: " << setprecision(14);
76 copy(m.coefficients().begin(), m.coefficients().end(), ostream_iterator<double>(*os_, " "));
77 *os_ << endl;
78
79 *os_ << "error: " << m(0)-1 << endl;
80
81 for (int i=0; i<5; i++)
82 *os_ << i << ", " << m(i) << endl;
83 }
84
85 unit_assert_equal(m(0), 1, epsilon_*100);
86 unit_assert_equal(m(1), 1/sqrt(2.), epsilon_*100);
87 unit_assert_equal(m(2), 1/sqrt(5.), epsilon_*100);
88 unit_assert_equal(m(3), 1/sqrt(10.), epsilon_*100);
89 unit_assert_equal(m(4), 1/sqrt(17.), epsilon_*100);
90 if (os_) *os_ << "testFit(): success!\n";
91}

References pwiz::frequency::MagnitudeLorentzian::coefficients(), epsilon_, os_, and unit_assert_equal.

Referenced by main().

◆ testData()

void testData ( )

Definition at line 94 of file MagnitudeLorentzianTest.cpp.

95{
96 string filename = "MagnitudeLorentizianTest.cfd.temp.txt";
97 ofstream temp(filename.c_str());
98 temp << sampleData_;
99 temp.close();
100
101 FrequencyData fd(filename);
102 boost::filesystem::remove(filename);
103
104 FrequencyData::const_iterator max = fd.max();
105 if (os_) *os_ << "max: (" << max->x << ", " << abs(max->y) << ")\n";
106
107 // fit MagnitudeLorentzian to 3 points on unnormalized data
108
109 vector< pair<double,double> > samples1;
110 transform(fd.max()-1, fd.max()+2, back_inserter(samples1), FrequencyData::magnitudeSample);
111
112 if (os_)
113 {
114 *os_ << "raw data:\n";
115 for (unsigned int i=0; i<samples1.size(); i++)
116 *os_ << "sample " << i << ": (" << samples1[i].first << ", " << samples1[i].second << ")\n";
117 }
118
119 const MagnitudeLorentzian m1(samples1);
120
121 if (os_)
122 {
123 *os_ << "m1: ";
124 copy(m1.coefficients().begin(), m1.coefficients().end(), ostream_iterator<double>(*os_, " "));
125 *os_ << endl;
126 *os_ << "error: " << scientific << m1.leastSquaresError() << endl;
127
128 for (unsigned int i=0; i<samples1.size(); i++)
129 *os_ << "m1(" << i << ") == " << m1(samples1[i].first) << endl;
130 }
131
132 // now on normalized data
133
134
135 fd.normalize();
136
137 vector< pair<double,double> > samples2;
138 transform(fd.max()-1, fd.max()+2, back_inserter(samples2), FrequencyData::magnitudeSample);
139
140 if (os_)
141 {
142 *os_ << "normalized: \n";
143 for (unsigned int i=0; i<samples2.size(); i++)
144 *os_ << "sample " << i << ": (" << samples2[i].first << ", " << samples2[i].second << ")\n";
145 }
146
147 const MagnitudeLorentzian m2(samples2);
148
149 if (os_)
150 {
151 *os_ << "m2: ";
152 copy(m2.coefficients().begin(), m2.coefficients().end(), ostream_iterator<double>(*os_, " "));
153 *os_ << endl;
154 *os_ << "error: " << scientific << m2.leastSquaresError() << endl;
155
156 for (unsigned int i=0; i<samples2.size(); i++)
157 *os_ << "m2(" << i << ") == " << m2(samples2[i].first) << " [" << fd.scale()*m2(samples2[i].first) << "]\n";
158 }
159
160 unit_assert_equal(m2.leastSquaresError(), 0, 1e-15);
161}
const char * sampleData_
Class for binary storage of complex frequency data.
container::const_iterator const_iterator
static std::pair< double, double > magnitudeSample(const FrequencyDatum &datum)
Returns a <frequency,magnitude> pair.

References pwiz::frequency::MagnitudeLorentzian::coefficients(), pwiz::frequency::MagnitudeLorentzian::leastSquaresError(), pwiz::data::FrequencyData::magnitudeSample(), pwiz::data::FrequencyData::max(), pwiz::data::FrequencyData::normalize(), os_, sampleData_, pwiz::data::FrequencyData::scale(), and unit_assert_equal.

Referenced by main().

◆ main()

int main ( int  argc,
char *  argv[] 
)

Definition at line 164 of file MagnitudeLorentzianTest.cpp.

165{
166 TEST_PROLOG(argc, argv)
167
168 try
169 {
170 if (argc>1 && !strcmp(argv[1],"-v")) os_ = &cout;
171 if (os_) *os_ << "MagnitudeLorentzianTest\n";
172 testBasic();
173 testFit();
174 testData();
175 }
176 catch (exception& e)
177 {
178 TEST_FAILED(e.what())
179 }
180 catch (...)
181 {
182 TEST_FAILED("Caught unknown exception.")
183 }
184
186}
void testData()
void testBasic()
void testFit()
#define TEST_EPILOG
Definition unit.hpp:183
#define TEST_FAILED(x)
Definition unit.hpp:177
#define TEST_PROLOG(argc, argv)
Definition unit.hpp:175

References os_, TEST_EPILOG, TEST_FAILED, TEST_PROLOG, testBasic(), testData(), and testFit().

Variable Documentation

◆ os_

ostream* os_ = 0

Definition at line 38 of file MagnitudeLorentzianTest.cpp.

Referenced by main(), testBasic(), testData(), and testFit().

◆ epsilon_

double epsilon_ = numeric_limits<double>::epsilon()

Definition at line 39 of file MagnitudeLorentzianTest.cpp.

Referenced by testBasic(), and testFit().