Line data Source code
1 : //# ProfileFit1D.cc: Class to fit Spectral components to vectors
2 : //# Copyright (C) 2004
3 : //# Associated Universities, Inc. Washington DC, USA.
4 : //#
5 : //# This library is free software; you can redistribute it and/or modify it
6 : //# under the terms of the GNU Library General Public License as published by
7 : //# the Free Software Foundation; either version 2 of the License, or (at your
8 : //# option) any later version.
9 : //#
10 : //# This library is distributed in the hope that it will be useful, but WITHOUT
11 : //# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12 : //# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
13 : //# License for more details.
14 : //#
15 : //# You should have received a copy of the GNU Library General Public License
16 : //# along with this library; if not, write to the Free Software Foundation,
17 : //# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
18 : //#
19 : //# Correspondence concerning AIPS++ should be addressed as follows:
20 : //# Internet email: aips2-request@nrao.edu.
21 : //# Postal address: AIPS++ Project Office
22 : //# National Radio Astronomy Observatory
23 : //# 520 Edgemont Road
24 : //# Charlottesville, VA 22903-2475 USA
25 : //#
26 : //# $Id: ProfileFit1D.tcc 20652 2009-07-06 05:04:32Z Malte.Marquarding $
27 :
28 : #include <components/SpectralComponents/ProfileFit1D.h>
29 :
30 : #include <casacore/casa/Arrays/ArrayLogical.h>
31 : #include <casacore/casa/Exceptions/Error.h>
32 : #include <components/SpectralComponents/SpectralEstimate.h>
33 : #include <components/SpectralComponents/SpectralElement.h>
34 : #include <casacore/casa/Utilities/DataType.h>
35 : #include <casacore/casa/Utilities/Assert.h>
36 :
37 : namespace casa {
38 :
39 : template <class T>
40 0 : ProfileFit1D<T>::ProfileFit1D()
41 : {
42 0 : checkType();
43 0 : }
44 :
45 : template <class T>
46 : ProfileFit1D<T>::ProfileFit1D(const ProfileFit1D& other)
47 : {
48 : checkType();
49 : copy(other);
50 : }
51 :
52 : template <class T>
53 0 : ProfileFit1D<T>& ProfileFit1D<T>::operator=(const ProfileFit1D& other)
54 : {
55 0 : if (this != &other) {
56 0 : copy(other);
57 : }
58 0 : return *this;
59 : }
60 :
61 : template <class T>
62 0 : ProfileFit1D<T>::~ProfileFit1D()
63 0 : {;}
64 :
65 : template <class T>
66 0 : bool ProfileFit1D<T>::setData (const casacore::Vector<casacore::Double>& x, const casacore::Vector<T>& y,
67 : const casacore::Vector<casacore::Bool>& mask, const casacore::Vector<casacore::Double>& weight)
68 : {
69 0 : if (x.nelements()==0) {
70 0 : itsError = "The X vector must have some elements";
71 0 : return false;
72 : }
73 0 : const casacore::uInt n = x.nelements();
74 : //
75 0 : if (y.nelements() != n) {
76 0 : itsError = "The Y vector must have the same number of elements as the X vector";
77 0 : return false;
78 : }
79 : //
80 0 : if (weight.nelements() != n && weight.nelements()!=0) {
81 0 : itsError = "The weights vector must have the same number of elements as the X vector";
82 0 : return false;
83 : }
84 : //
85 0 : if (mask.nelements() != n && mask.nelements() != 0) {
86 0 : itsError = "The mask vector must have the same number of elements (or zero) as the data";
87 0 : return false;
88 : }
89 : //
90 0 : itsX.resize(n);
91 0 : itsY.resize(n);
92 0 : itsDataMask.resize(n);
93 : //
94 0 : itsX = x;
95 0 : itsY = y;
96 : //
97 0 : if (weight.nelements()==0) {
98 0 : itsWeight.resize(0);
99 : } else {
100 0 : itsWeight.resize(n);
101 0 : itsWeight = weight;
102 : }
103 : //
104 0 : if (mask.nelements()==0) {
105 0 : itsDataMask = true;
106 : } else {
107 0 : itsDataMask = mask;
108 : }
109 0 : return true;
110 : }
111 :
112 :
113 : template <class T>
114 0 : bool ProfileFit1D<T>::setData (const casacore::Vector<casacore::Double>& x, const casacore::Vector<T>& y,
115 : const casacore::Vector<casacore::Bool>& mask)
116 : {
117 0 : casacore::Vector<casacore::Double> weight;
118 0 : return setData (x, y, mask, weight);
119 : }
120 :
121 : template <class T>
122 : bool ProfileFit1D<T>::setData (const casacore::Vector<casacore::Double>& x, const casacore::Vector<T>& y)
123 : {
124 : casacore::Vector<casacore::Bool> mask(x.nelements(), true);
125 : return setData (x, y, mask);
126 : }
127 :
128 :
129 : template <class T>
130 0 : void ProfileFit1D<T>::setElements (const SpectralList& list)
131 : {
132 0 : itsList = list;
133 0 : }
134 :
135 : template <class T>
136 0 : bool ProfileFit1D<T>::setGaussianElements (casacore::uInt nGauss)
137 : {
138 0 : if (nGauss==0) {
139 0 : itsError = "You must specify some Gaussian components";
140 0 : return false;
141 : }
142 : //
143 0 : if (itsY.nelements()==0) {
144 0 : itsError = "You must call function setData to set some data first";
145 0 : return false;
146 : }
147 :
148 : // Clear list
149 :
150 0 : itsList.clear();
151 :
152 : // Make estimate for Gaussians.
153 :
154 0 : SpectralEstimate estimator (nGauss);
155 0 : estimator.setQ(5);
156 0 : SpectralList listGauss = estimator.estimate (itsX, itsY); // Ignores masked data
157 0 : itsList.add (listGauss);
158 0 : return true;
159 : }
160 :
161 : template <class T>
162 : void ProfileFit1D<T>::addElements (const SpectralList& list)
163 : {
164 : itsList.add (list);
165 : }
166 :
167 : template <class T>
168 0 : void ProfileFit1D<T>::addElement (const SpectralElement& el)
169 : {
170 0 : itsList.add (el);
171 0 : }
172 :
173 :
174 : template <class T>
175 0 : void ProfileFit1D<T>::clearList ()
176 : {
177 0 : itsList.clear();
178 0 : }
179 :
180 :
181 : template <class T>
182 : bool ProfileFit1D<T>::setXRangeMask (const casacore::Vector<casacore::uInt>& start,
183 : const casacore::Vector<casacore::uInt>& end,
184 : casacore::Bool insideIsGood)
185 : {
186 : AlwaysAssert(start.nelements()==end.nelements(), casacore::AipsError);
187 : if (itsX.nelements()==0) {
188 : itsError = "You must call function setData to set some data first";
189 : return false;
190 : }
191 : //
192 : const casacore::uInt n = itsX.nelements();
193 : itsRangeMask.resize(n);
194 : casacore::Bool value = !insideIsGood;
195 : itsRangeMask = value;
196 : //
197 : for (casacore::uInt i=0; i<start.nelements(); i++) {
198 : if (start[i] > end[i]) {
199 : itsError = "The start index must be < the end index";
200 : return false;
201 : }
202 : if (start[i]>=n) {
203 : itsError = "The start index must be in the range 0->nElements-1";
204 : return false;
205 : }
206 : if (end[i]>=n) {
207 : itsError = "The end index must be in the range 0->nElements-1";
208 : return false;
209 : }
210 : //
211 : for (casacore::uInt j=start[i]; j<end[i]+1; j++) {
212 : itsRangeMask[j] = !value;
213 : }
214 : }
215 : return true;
216 : }
217 :
218 :
219 :
220 : template <class T>
221 : bool ProfileFit1D<T>::setXRangeMask (const casacore::Vector<T>& start,
222 : const casacore::Vector<T>& end,
223 : casacore::Bool insideIsGood)
224 : {
225 : if (start.nelements()!=end.nelements()) {
226 : itsError = "Start and end vectors must be the same length";
227 : return false;
228 : }
229 : if (itsX.nelements()==0) {
230 : itsError = "You must call function setData to set some data first";
231 : return false;
232 : }
233 : //
234 : const casacore::uInt n = itsX.nelements();
235 : itsRangeMask.resize(n);
236 : casacore::Bool value = !insideIsGood;
237 : itsRangeMask = value;
238 : //
239 : casacore::Vector<casacore::uInt> startIndex(start.nelements());
240 : casacore::Vector<casacore::uInt> endIndex(end.nelements());
241 :
242 : for (casacore::uInt i=0; i<start.nelements(); i++) {
243 : if (start[i] > end[i]) {
244 : itsError = "The start range must be < the end range";
245 : return false;
246 : }
247 : if (start[i]<itsX[0] || start[i]>itsX[n-1]) {
248 : itsError = "The start range must be in the X-range of the data";
249 : return false;
250 : }
251 : if (end[i]<itsX[0] || end[i]>itsX[n-1]) {
252 : itsError = "The end range must be in the X-range of the data";
253 : return false;
254 : }
255 :
256 : // Find the indices for this range
257 :
258 : casacore::Bool doneStart = false;
259 : casacore::Bool doneEnd = false;
260 : for (casacore::uInt j=0; j<n; j++) {
261 : if (!doneStart && itsX[j] >= start[i]) {
262 : startIndex[i] = j;
263 : doneStart = true;
264 : }
265 : if (!doneEnd && itsX[j] >= end[i]) {
266 : endIndex[i] = j;
267 : doneEnd = true;
268 : }
269 : if (!doneEnd) endIndex[i] = n-1;
270 : }
271 : }
272 : //
273 : return setXRangeMask (startIndex, endIndex);
274 : }
275 :
276 : template <class T>
277 0 : bool ProfileFit1D<T>::setXMask(const std::set<casacore::uInt>& indices, casacore::Bool specifiedPixelsAreGood) {
278 0 : const casacore::uInt n = itsX.nelements();
279 0 : ThrowIf(n == 0, "Logic Error: setData() must be called prior to setRangeMask()");
280 0 : itsRangeMask.resize(n);
281 0 : itsRangeMask = ! specifiedPixelsAreGood;
282 0 : if (indices.empty()) {
283 0 : return true;
284 : }
285 0 : std::set<casacore::uInt>::const_iterator iter = indices.begin();
286 0 : std::set<casacore::uInt>::const_iterator end = indices.end();
287 :
288 0 : while (iter != end && *iter < n) {
289 0 : itsRangeMask[*iter] = specifiedPixelsAreGood;
290 0 : ++iter;
291 : }
292 0 : return true;
293 : }
294 :
295 :
296 : template <class T>
297 0 : bool ProfileFit1D<T>::fit ()
298 : {
299 : /*
300 : if (itsX.nelements()==0) {
301 : itsError = "You must call function setData to set some data first";
302 : return false;
303 : }
304 : if (itsList.nelements()==0) {
305 : itsError = "You must call function setElements to set some fit components first";
306 : return false;
307 : }
308 : */
309 :
310 : // Set list in fitter
311 0 : itsFitter.clear();
312 0 : itsFitter.addFitElement (itsList);
313 : // Do the fit with the total mask
314 :
315 0 : casacore::Bool converged = itsWeight.empty()
316 0 : ? itsFitter.fit (itsY, itsX, makeTotalMask())
317 0 : : itsFitter.fit (itsWeight, itsY, itsX, makeTotalMask());
318 0 : return converged;
319 : }
320 :
321 : template <class T>
322 0 : const SpectralList& ProfileFit1D<T>::getList (casacore::Bool fit) const
323 : {
324 0 : if (fit) {
325 0 : return itsFitter.list();
326 : } else {
327 0 : return itsList;
328 : }
329 : }
330 :
331 :
332 : template <class T>
333 : casacore::Vector<T> ProfileFit1D<T>::getEstimate (casacore::Int which) const
334 : {
335 : casacore::Vector<T> tmp;
336 : if (itsX.nelements()==0) {
337 : itsError = "You must call function setData to set some data first";
338 : return tmp;
339 : }
340 : //
341 : if (which<0) {
342 : itsList.evaluate(tmp, itsX);
343 : } else {
344 : SpectralList list = getSubsetList (itsList, which);
345 : list.evaluate(tmp, itsX);
346 : }
347 : //
348 : return tmp;
349 : }
350 :
351 : template <class T>
352 0 : casacore::Vector<T> ProfileFit1D<T>::getFit (casacore::Int which) const
353 : {
354 0 : casacore::Vector<T> tmp;
355 0 : if (itsX.nelements()==0) {
356 0 : itsError = "You must call function setData to set some data first";
357 0 : return tmp;
358 : }
359 : //
360 0 : const SpectralList& fitList = itsFitter.list();
361 0 : if (fitList.nelements()==0) {
362 0 : itsError = "You must call function fit first";
363 0 : return tmp;
364 : }
365 : //
366 :
367 0 : if (which<0) {
368 0 : fitList.evaluate(tmp, itsX);
369 : } else {
370 0 : SpectralList list = getSubsetList (fitList, which);
371 0 : list.evaluate(tmp, itsX);
372 : }
373 : //
374 0 : return tmp;
375 : }
376 :
377 : template <class T>
378 0 : casacore::Vector<T> ProfileFit1D<T>::getResidual (casacore::Int which, casacore::Bool fit) const
379 : {
380 0 : casacore::Vector<T> tmp;
381 0 : if (itsX.nelements()==0) {
382 0 : itsError = "You must call function setData to set some data first";
383 0 : return tmp;
384 : }
385 : //
386 0 : SpectralList list;
387 0 : if (fit) {
388 0 : list = itsFitter.list();
389 0 : if (list.nelements()==0) {
390 0 : throw (casacore::AipsError("You must call function fit first"));
391 : }
392 : } else {
393 0 : list = itsList;
394 : }
395 : //
396 0 : tmp = itsY;
397 0 : if (which<0) {
398 0 : list.residual(tmp, itsX);
399 : } else {
400 0 : SpectralList subList = getSubsetList (list, which);
401 0 : subList.residual(tmp, itsX);
402 : }
403 : //
404 0 : return tmp;
405 : }
406 :
407 :
408 : // Private functions
409 :
410 :
411 : template <class T>
412 0 : SpectralList ProfileFit1D<T>::getSubsetList (const SpectralList& list, casacore::Int which) const
413 : {
414 0 : const casacore::Int n = list.nelements();
415 0 : if (which+1 > n) {
416 0 : throw casacore::AipsError("Illegal spectral element index");
417 : }
418 0 : SpectralList listOut;
419 0 : listOut.add(*list[which]);
420 0 : return listOut;
421 : }
422 :
423 : template <class T>
424 0 : casacore::Vector<casacore::Bool> ProfileFit1D<T>::makeTotalMask () const
425 : {
426 0 : casacore::Vector<casacore::Bool> mask;
427 0 : if (itsRangeMask.nelements()==0) {
428 0 : mask = itsDataMask;
429 : } else {
430 0 : mask = itsDataMask && itsRangeMask;
431 : }
432 0 : return mask;
433 : }
434 :
435 : template <class T>
436 0 : void ProfileFit1D<T>::copy(const ProfileFit1D& other)
437 : {
438 0 : itsX.resize(0);
439 0 : itsX = other.itsX;
440 : //
441 0 : itsY.resize(0);
442 0 : itsY = other.itsY;
443 : //
444 0 : itsWeight.resize(0);
445 0 : itsWeight = other.itsWeight;
446 : //
447 0 : itsDataMask.resize(0);
448 0 : itsDataMask = other.itsDataMask;
449 : //
450 0 : itsRangeMask.resize(0);
451 0 : itsRangeMask = other.itsRangeMask;
452 : //
453 0 : itsList = other.itsList;
454 : //
455 0 : itsFitter = other.itsFitter;
456 : //
457 0 : itsError = other.itsError;
458 0 : }
459 :
460 : template <class T>
461 0 : void ProfileFit1D<T>::checkType() const
462 : {
463 0 : AlwaysAssert(casacore::whatType<T>()==casacore::TpDouble,casacore::AipsError);
464 0 : }
465 :
466 : } //#End casa namespace
|
