casa6

//# UVContSubResult.h: implementation of the UVContSubResult class

//#

//#  CASA - Common Astronomy Software Applications (http://casa.nrao.edu/)

//#  Copyright (C) Associated Universities, Inc. Washington DC, USA 2021, All rights reserved.

//#  Copyright (C) European Southern Observatory, 2021, All rights reserved.

//#

//#  This library is free software; you can redistribute it and/or

//#  modify it under the terms of the GNU Lesser General Public

//#  License as published by the Free software Foundation; either

//#  version 2.1 of the License, or (at your option) any later version.

//#

//#  This library 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

//#  Lesser General Public License for more details.

//#

//#  You should have received a copy of the GNU Lesser General Public

//#  License along with this library; if not, write to the Free Software

//#  Foundation, Inc., 59 Temple Place, Suite 330, Boston,

//#  MA 02111-1307  USA

//# $Id: $

​

#include <mstransform/TVI/UVContSubResult.h>

​

using namespace casacore;

​

namespace casa { //# NAMESPACE CASA - BEGIN

​

namespace vi { //# NAMESPACE VI - BEGIN

​

void UVContSubResult::addOneFit(int field, int scan, int spw, int pol, Complex chiSquared)

{

    auto &polRes = accum[field][scan][spw];

    auto resIt = polRes.find(pol);

    if (resIt == polRes.end()) {

        FitResultAcc fitResult;

        fitResult.count = 1;

        fitResult.chiSqAvg = chiSquared;

        fitResult.chiSqMin = chiSquared;

        fitResult.chiSqMax = chiSquared;

        polRes.emplace(pol, fitResult);

    } else {

        auto &fitResult = resIt->second;

        fitResult.count++;

        fitResult.chiSqAvg += (chiSquared - fitResult.chiSqAvg) /

            static_cast<float>(fitResult.count);

        if (chiSquared < fitResult.chiSqMin)

            fitResult.chiSqMin = chiSquared;

        if (chiSquared > fitResult.chiSqMax)

            fitResult.chiSqMax = chiSquared;

    }

}

​

/*

 * Produces a record from information accumulated for every fit that

 * is calculated by the UVContSub TVI (addOneFit()). This is used at

 * the moment to hold basic goodness-of-fit information. The record

 * produced by this function is meant to be returned as result by the

 * UVContSub TVI and can be returned by the uvcontsub task. The result

 * record/dictionary would look like:

 *

 * {'description': 'summary of data fitting results in uv-continuum subtraction',

 *    'goodness_of_fit': {'field': {'0': {'scan': {'3': {'spw': {'0':