/***************************************************************************
        Copyright (C) 2002-2015 Kentaro Kitagawa
                           kitag@issp.u-tokyo.ac.jp

        This program is free software; you can redistribute it and/or
        modify it under the terms of the GNU Library General Public
        License as published by the Free Software Foundation; either
        version 2 of the License, or (at your option) any later version.

        You should have received a copy of the GNU Library General
        Public License and a list of authors along with this program;
        see the files COPYING and AUTHORS.
***************************************************************************/
#include "nmrrelax.h"
#include "nmrrelaxfit.h"
#include "ui_nmrrelaxform.h"
#include "nmrpulse.h"
#include "pulserdriver.h"
#include "analyzer.h"
#include "graph.h"
#include "graphwidget.h"
#include "rand.h"
#include "spectrumsolver.h"

REGISTER_TYPE(XDriverList, NMRT1, "NMR relaxation measurement");

#include <QPushButton>
#include <QComboBox>
#include <QCheckBox>

const char XNMRT1::P1DIST_LINEAR[] = "Linear";
const char XNMRT1::P1DIST_LOG[] = "Log";
const char XNMRT1::P1DIST_RECIPROCAL[] = "Reciprocal";

const char XNMRT1::P1STRATEGY_RANDOM[] = "Random";
const char XNMRT1::P1STRATEGY_FLATTEN[] = "Flatten";


class XRelaxFuncPlot : public XFuncPlot {
public:
    XRelaxFuncPlot(const char *name, bool runtime, Transaction &tr, const shared_ptr<XGraph> &graph
                   , const shared_ptr<XItemNode < XRelaxFuncList, XRelaxFunc > >  &item,
                   const shared_ptr<XNMRT1> &owner)
        : XFuncPlot(name, runtime, tr, graph), m_item(item), m_owner(owner)
    {}
    ~XRelaxFuncPlot() {}
    virtual double func(double t) const {
        shared_ptr<XNMRT1> owner = m_owner.lock();
        if( !owner) return 0;
        Snapshot shot( *owner);
        shared_ptr<XRelaxFunc> func1 = shot[ *m_item];
        if( !func1) return 0;
        double f, df;
        double it1 = shot[ *owner].m_params[0];
        double c = shot[ *owner].m_params[1];
        double a = shot[ *owner].m_params[2];
        func1->relax( &f, &df, t, it1);
        return c * f + a;
    }
private:
    shared_ptr<XItemNode < XRelaxFuncList, XRelaxFunc > > m_item;
    weak_ptr<XNMRT1> m_owner;
};

XNMRT1::XNMRT1(const char *name, bool runtime,
    Transaction &tr_meas, const shared_ptr<XMeasure> &meas)
    : XSecondaryDriver(name, runtime, ref(tr_meas), meas),
      m_relaxFuncs(create<XRelaxFuncList>("RelaxFuncs", true)),
      m_t1inv(create<XScalarEntry>("T1inv", false,
                                   dynamic_pointer_cast<XDriver>(shared_from_this()))),
      m_t1invErr(create<XScalarEntry>("T1invErr", false,
                                      dynamic_pointer_cast<XDriver>(shared_from_this()))),
      m_pulser(create<XItemNode < XDriverList, XPulser > >(
          "Pulser", false, ref(tr_meas), meas->drivers(), true)),
      m_pulse1(create<XItemNode < XDriverList, XNMRPulseAnalyzer > >(
          "NMRPulseAnalyzer1", false, ref(tr_meas), meas->drivers(), true)),
      m_pulse2(create<XItemNode < XDriverList, XNMRPulseAnalyzer > >(
          "NMRPulseAnalyzer2", false, ref(tr_meas), meas->drivers())),
      m_active(create<XBoolNode>("Active", false)),
      m_autoPhase(create<XBoolNode>("AutoPhase", false)),
      m_mInftyFit(create<XBoolNode>("MInftyFit", false)),
      m_absFit(create<XBoolNode>("AbsFit", false)),
      m_trackPeak(create<XBoolNode>("TrackPeakFreq", false)),
      m_p1Min(create<XDoubleNode>("P1Min", false)),
      m_p1Max(create<XDoubleNode>("P1Max", false)),
      m_p1Next(create<XDoubleNode>("P1Next", true)),
      m_p1AltNext(create<XDoubleNode>("P1Next", true)),
      m_phase(create<XDoubleNode>("Phase", false, "%.2f")),
      m_freq(create<XDoubleNode>("Freq", false, "%.3f")),
      m_windowFunc(create<XComboNode>("WindowFunc", false, true)),
      m_windowWidth(create<XComboNode>("WindowWidth", false, true)),
      m_autoWindow(create<XBoolNode>("AutoWindow", false)),
      m_mode(create<XComboNode>("Mode", false, true)),
      m_smoothSamples(create<XUIntNode>("SmoothSamples", false)),
      m_p1Strategy(create<XComboNode>("P1Strategy", false, true)),
      m_p1Dist(create<XComboNode>("P1Dist", false, true)),
      m_resetFit(create<XTouchableNode>("ResetFit", true)),
      m_clearAll(create<XTouchableNode>("ClearAll", true)),
      m_fitStatus(create<XStringNode>("FitStatus", true)),
      m_solver(create<SpectrumSolverWrapper>("SpectrumSolver", true, shared_ptr<XComboNode>(), m_windowFunc, shared_ptr<XDoubleNode>())),
      m_solverMapPulse(create<SpectrumSolverWrapper>("SpectrumSolverMapPulse", true, shared_ptr<XComboNode>(), m_mapWindowFunc, m_mapWindowWidth)),
      m_mapMode(create<XComboNode>("MapMode", false, true)),
      m_mapTikhonovMatrix(create<XComboNode>("MapTikhonovMatrix", false, true)),
      m_mapFreqRes(create<XDoubleNode>("MapFreqRes", false, "%.3f")),
      m_mapBandWidth(create<XDoubleNode>("MapBandWidth", false, "%.1f")),
      m_mapWindowFunc(create<XComboNode>("MapWindowFunc", false, true)),
      m_mapWindowWidth(create<XDoubleNode>("MapWindowWidth", false, "%.1f")),
      m_form(new FrmNMRT1),
      m_statusPrinter(XStatusPrinter::create(m_form.get())),
      m_wave(create<XWaveNGraph>("Wave", false, m_form->m_graph, m_form->m_edDump, m_form->m_tbDump, m_form->m_btnDump)),
      m_waveMap(create<XWaveNGraph>("WaveMap", false, m_form->m_graphRelaxMap, m_form->m_edRelaxMapDump, m_form->m_tbRelaxMapDump, m_form->m_btnRelaxMapDump)),
      m_waveAllRelaxCurves(create<XWaveNGraph>("WaveAllRelaxCurves", false, m_form->m_graphAllRelaxCurves, m_form->m_edAllRelaxCurvesDump, m_form->m_tbAllRelaxCurvesDump, m_form->m_btnAllRelaxCurvesDump))
      {

    iterate_commit([=](Transaction &tr){
        m_relaxFunc = create<XItemNode < XRelaxFuncList, XRelaxFunc > >(
                          tr, "RelaxFunc", false, tr, m_relaxFuncs, true);
    });

    m_form->m_btnClear->setIcon(QApplication::style()->standardIcon(QStyle::SP_DialogResetButton));
    m_form->m_btnResetFit->setIcon(QApplication::style()->standardIcon(QStyle::SP_BrowserReload));

    m_form->setWindowTitle(i18n("NMR Relaxation Measurement - ") + getLabel() );

    meas->scalarEntries()->insert(tr_meas, t1inv());
    meas->scalarEntries()->insert(tr_meas, t1invErr());

    connect(pulser());
    connect(pulse1());
    connect(pulse2());

    m_solverMapPulse->selectSolver(SpectrumSolverWrapper::SPECTRUM_SOLVER_ZF_FFT);

    iterate_commit([=](Transaction &tr){
        tr[ *m_windowFunc] = SpectrumSolverWrapper::WINDOW_FUNC_HAMMING;
        m_windowWidthList = {0.25, 0.5, 1.0, 1.5, 2.0};
        tr[ *m_windowWidth].add({"25%", "50%", "100%", "150%", "200%"});
        tr[ *m_windowWidth] = 2;

        const char *labels[] = {"P1 [ms] or 2Tau [us]", "Intens [V]",
                                "Weight [1/V]", "Abs [V]", "Re [V]", "Im [V]"};
        tr[ *m_wave].setColCount(6, labels);
        if( !tr[ *m_wave].insertPlot(tr, i18n("Relaxation"), 0, 1, -1, 2)) return;
        if( !tr[ *m_wave].insertPlot(tr, i18n("Out-of-Phase"), 0, 5, -1, 2)) return;
        shared_ptr<XAxis> axisx = tr[ *m_wave].axisx();
        shared_ptr<XAxis> axisy = tr[ *m_wave].axisy();
        tr[ *axisx->logScale()] = true;
        tr[ *axisy->label()] = i18n("Relaxation");
        tr[ *tr[ *m_wave].plot(0)->drawLines()] = false;
        tr[ *tr[ *m_wave].plot(1)->drawLines()] = false;
        tr[ *tr[ *m_wave].plot(1)->intensity()] = 1.0;
        shared_ptr<XFuncPlot> plot3 = m_wave->graph()->plots()->create<XRelaxFuncPlot>(
            tr, "FittedCurve", true, tr, m_wave->graph(),
            relaxFunc(), static_pointer_cast<XNMRT1>(shared_from_this()));
        if( !plot3) return;
        tr[ *plot3->label()] = i18n("Fitted Curve");
        tr[ *plot3->axisX()] = axisx;
        tr[ *plot3->axisY()] = axisy;
        tr[ *m_wave].clearPoints();

        tr[ *mode()].add({"T1 Measurement", "T2 Measurement", "St.E. Measurement", "T2 Multi-echo"});
        tr[ *mode()] = (int)MeasMode::T1;

        tr[ *p1Strategy()].add({P1STRATEGY_RANDOM, P1STRATEGY_FLATTEN});
        tr[ *p1Strategy()] = 1;

        tr[ *p1Dist()].add({P1DIST_LINEAR, P1DIST_LOG, P1DIST_RECIPROCAL});
        tr[ *p1Dist()] = 1;

        tr[ *relaxFunc()].str(XString("NMR I=1/2"));
        tr[ *p1Min()] = 1.0;
        tr[ *p1Max()] = 100.0;
        tr[ *autoPhase()] = true;
        tr[ *autoWindow()] = true;
        tr[ *mInftyFit()] = true;
        tr[ *trackPeak()] = false;
        tr[ *smoothSamples()] = 33;


        tr[ *mapMode()].add({"Off", "AllNonNegative", "Noise Analysis", "L Curve", "GCV"});
        tr[ *mapMode()] = (int)MapMode::Off;
        tr[ *mapTikhonovMatrix()].add({"Identity", "2nd Derivative Op."});
        tr[ *mapTikhonovMatrix()] = (int)TikhonovRegular::TikhonovMatrix::I;

        tr[ *mapBandWidth()] = 100.0;
        tr[ *mapFreqRes()] = 1.0;
        tr[ *m_mapWindowFunc] = SpectrumSolverWrapper::WINDOW_FUNC_DEFAULT;
        tr[ *m_mapWindowWidth] = 100.0;

        {
            const char *labels[] = {"Freq [kHz]", "P1 [ms] or 2Tau [us]", "Re [V]", "Im [V]", "Weight [1/V]"};
            tr[ *m_waveAllRelaxCurves].setColCount(5, labels);
            if( !tr[ *m_waveAllRelaxCurves].insertPlot(tr, i18n("Relaxation"), 0, 2, -1, 4, 1)) return;
            if( !tr[ *m_waveAllRelaxCurves].insertPlot(tr, i18n("Out-of-Phase"), 0, 3, -1, 4, 1)) return;
//            tr[ *m_waveAllRelaxCurves].insertPlot(labels[4], 0, 4, -1, 4, 1);
            shared_ptr<XAxis> axisx = tr[ *m_waveAllRelaxCurves].axisx();
            shared_ptr<XAxis> axisy = tr[ *m_waveAllRelaxCurves].axisy();
            shared_ptr<XAxis> axisz = tr[ *m_waveAllRelaxCurves].axisz();
            tr[ *axisx->label()] = i18n("Freq [kHz]");
            tr[ *axisz->logScale()] = true;
            tr[ *axisy->label()] = i18n("Intens [V]");
            tr[ *tr[ *m_waveAllRelaxCurves].plot(0)->drawLines()] = false;
            tr[ *tr[ *m_waveAllRelaxCurves].plot(1)->drawLines()] = false;
            tr[ *tr[ *m_waveAllRelaxCurves].plot(1)->intensity()] = 1.0;
//            tr[ *tr[ *m_waveAllRelaxCurves].plot(2)->lineColor()] = clLime; //QColor(0xa0, 0xa0, 0x00).rgb();
//            tr[ *tr[ *m_waveAllRelaxCurves].plot(2)->drawPoints()] = false;
//            tr[ *tr[ *m_waveAllRelaxCurves].plot(2)->intensity()] = 0.8;
        }
        {
            const char *labels[] = {"Freq [kHz]", "T1 [ms] or T2 [us]", "Density"};
            tr[ *m_waveMap].setColCount(3, labels);
            if( !tr[ *m_waveMap].insertPlot(tr, i18n("Density"), 0, 1, -1, -1, 2)) return;
            shared_ptr<XAxis> axisx = tr[ *m_waveMap].axisx();
            shared_ptr<XAxis> axisy = tr[ *m_waveMap].axisy();
            tr[ *axisy->logScale()] = true;
            tr[ *axisx->label()] = i18n("Freq [kHz]");
            tr[ *tr[ *m_waveMap].plot(0)->drawLines()] = false;
            tr[ *m_waveMap->graph()->backGround()] = QColor(0,0,0).rgb();
            tr[ *tr[ *m_waveMap].plot(0)->intensity()] = 2;
            tr[ *tr[ *m_waveMap].plot(0)->colorPlot()] = true;
            tr[ *tr[ *m_waveMap].plot(0)->colorPlotColorHigh()] = QColor(0xFF, 0xFF, 0x2F).rgb();
            tr[ *tr[ *m_waveMap].plot(0)->colorPlotColorLow()] = QColor(0x00, 0x00, 0xFF).rgb();
            tr[ *tr[ *m_waveMap].plot(0)->pointColor()] = QColor(0x00, 0xFF, 0x00).rgb();
            tr[ *tr[ *m_waveMap].plot(0)->majorGridColor()] = QColor(0x4A, 0x4A, 0x4A).rgb();
            tr[ *m_waveMap->graph()->titleColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisx()->ticColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisx()->labelColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisx()->ticLabelColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisy()->ticColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisy()->labelColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisy()->ticLabelColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisz()->ticColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisz()->labelColor()] = clWhite;
            tr[ *tr[ *m_waveMap].axisz()->ticLabelColor()] = clWhite;
            tr[ *m_waveMap].clearPoints();
        }
    });

    //Ranges should be preset in prior to connectors.
    m_form->m_dblPhase->setRange( -360.0, 360.0);
    m_form->m_dblPhase->setSingleStep(10.0);
    m_form->m_dbMapWindowWidth->setRange(3.0, 200.0);
    m_form->m_dbMapWindowWidth->setSingleStep(1.0);

    m_conUIs = {
        xqcon_create<XQLineEditConnector>(m_p1Min, m_form->m_edP1Min),
        xqcon_create<XQLineEditConnector>(m_p1Max, m_form->m_edP1Max),
        xqcon_create<XQLineEditConnector>(m_p1Next, m_form->m_edP1Next),
        xqcon_create<XQDoubleSpinBoxConnector>(m_phase, m_form->m_dblPhase, m_form->m_slPhase),
        xqcon_create<XQLineEditConnector>(m_freq, m_form->m_edFreq),
        xqcon_create<XQToggleButtonConnector>(m_autoWindow, m_form->m_ckbAutoWindow),
        xqcon_create<XQComboBoxConnector>(m_windowFunc, m_form->m_cmbWindowFunc, Snapshot( *m_windowFunc)),
        xqcon_create<XQComboBoxConnector>(m_windowWidth, m_form->m_cmbWindowWidth, Snapshot( *m_windowWidth)),
        xqcon_create<XQLineEditConnector>(m_smoothSamples, m_form->m_edSmoothSamples),
        xqcon_create<XQComboBoxConnector>(m_p1Strategy, m_form->m_cmbP1Strategy, Snapshot( *m_p1Strategy)),
        xqcon_create<XQComboBoxConnector>(m_p1Dist, m_form->m_cmbP1Dist, Snapshot( *m_p1Dist)),
        xqcon_create<XQButtonConnector>(m_clearAll, m_form->m_btnClear),
        xqcon_create<XQButtonConnector>(m_resetFit, m_form->m_btnResetFit),
        xqcon_create<XQToggleButtonConnector>(m_active, m_form->m_ckbActive),
        xqcon_create<XQToggleButtonConnector>(m_autoPhase, m_form->m_ckbAutoPhase),
        xqcon_create<XQToggleButtonConnector>(m_mInftyFit, m_form->m_ckbMInftyFit),
        xqcon_create<XQToggleButtonConnector>(m_absFit, m_form->m_ckbAbsFit),
        xqcon_create<XQToggleButtonConnector>(m_trackPeak, m_form->m_ckbTrackPeak),
        xqcon_create<XQTextBrowserConnector>(m_fitStatus, m_form->m_txtFitStatus),
        xqcon_create<XQComboBoxConnector>(m_relaxFunc, m_form->m_cmbRelaxFunc, Snapshot( *m_relaxFuncs)),
        xqcon_create<XQComboBoxConnector>(m_mode, m_form->m_cmbMode, Snapshot( *m_mode)),
        xqcon_create<XQComboBoxConnector>(m_pulser, m_form->m_cmbPulser, ref(tr_meas)),
        xqcon_create<XQComboBoxConnector>(m_pulse1, m_form->m_cmbPulse1, ref(tr_meas)),
        xqcon_create<XQComboBoxConnector>(m_pulse2, m_form->m_cmbPulse2, ref(tr_meas)),
        xqcon_create<XQComboBoxConnector>(m_mapMode, m_form->m_cmbRegularizationChoice, Snapshot( *m_mapMode)),
        xqcon_create<XQComboBoxConnector>(m_mapTikhonovMatrix, m_form->m_cmbTikhonovMatrix, Snapshot( *m_mapTikhonovMatrix)),
        xqcon_create<XQLineEditConnector>(m_mapFreqRes, m_form->m_edRegularizationResolution),
        xqcon_create<XQLineEditConnector>(m_mapBandWidth, m_form->m_edRegularizationBW),
        xqcon_create<XQComboBoxConnector>(m_mapWindowFunc, m_form->m_cmbMapWindowFunc, Snapshot( *m_mapWindowFunc)),
        xqcon_create<XQDoubleSpinBoxConnector>(m_mapWindowWidth, m_form->m_dbMapWindowWidth, m_form->m_slMapWindowWidth),
    };

    iterate_commit([=](Transaction &tr){
        m_lsnOnActiveChanged = tr[ *active()].onValueChanged().connectWeakly(
            shared_from_this(), &XNMRT1::onActiveChanged);
        m_lsnOnP1CondChanged = tr[ *p1Max()].onValueChanged().connectWeakly(
            shared_from_this(), &XNMRT1::onP1CondChanged);
        for(auto &&x: std::vector<shared_ptr<XValueNodeBase>>(
            {p1Min(), p1Strategy(), p1Dist(), smoothSamples()}))
            tr[ *x].onValueChanged().connect(m_lsnOnP1CondChanged);
        m_lsnOnCondChanged = tr[ *phase()].onValueChanged().connectWeakly(
            shared_from_this(), &XNMRT1::onCondChanged);
        for(auto &&x: std::vector<shared_ptr<XValueNodeBase>>(
            {mInftyFit(), absFit(), relaxFunc(), autoPhase(), freq(), autoWindow(),
            windowFunc(), windowWidth(), mode()}))
            tr[ *x].onValueChanged().connect(m_lsnOnCondChanged);
        m_lsnOnMapCondChanged = tr[ *mode()].onValueChanged().connectWeakly(
            shared_from_this(), &XNMRT1::onMapCondChanged);
        for(auto &&x: std::vector<shared_ptr<XValueNodeBase>>(
            {mapBandWidth(), mapFreqRes(),
            mapWindowFunc(), mapWindowWidth(), relaxFunc(),
            mode(), mapTikhonovMatrix()}))
            tr[ *x].onValueChanged().connect(m_lsnOnMapCondChanged);
        m_lsnOnMapClearCondRequested = tr[ *mode()].onValueChanged().connectWeakly(
            shared_from_this(), &XNMRT1::onMapClearCondRequested);
        for(auto &&x: std::vector<shared_ptr<XValueNodeBase>>(
            {mapMode(), smoothSamples(), p1Min(), p1Max(), p1Dist()}))
            tr[ *x].onValueChanged().connect(m_lsnOnMapClearCondRequested);

        m_lsnOnClearAll = tr[ *m_clearAll].onTouch().connectWeakly(
            shared_from_this(), &XNMRT1::onClearAll);
        m_lsnOnResetFit = tr[ *m_resetFit].onTouch().connectWeakly(
            shared_from_this(), &XNMRT1::onResetFit);
    });
}
void
XNMRT1::showForms() {
    m_form->showNormal();
    m_form->raise();
}
void
XNMRT1::onClearAll(const Snapshot &shot, XTouchableNode *) {
    trans( *this).m_timeClearRequested = XTime::now();
    trans( *this).m_timeMapClearRequested = XTime::now();
    requestAnalysis();
}
double
XNMRT1::distributeP1(const Snapshot &shot, double uniform_x_0_to_1) {
    double p1min = shot[ *p1Min()];
    double p1max = shot[ *p1Max()];
    double p1;
    if(shot[ *p1Dist()].to_str() == P1DIST_LINEAR)
        p1 = (1-(uniform_x_0_to_1)) * p1min + (uniform_x_0_to_1) * p1max;
    else if(shot[ *p1Dist()].to_str() == P1DIST_LOG)
        p1 = p1min * exp((uniform_x_0_to_1) * log(p1max/p1min));
    else
        //P1DIST_RECIPROCAL
        p1 =1/((1-uniform_x_0_to_1)/p1min + (uniform_x_0_to_1)/p1max);

    p1 = llrint(p1 * 1e4) / 1e4;//rounds.
    return p1;
}
void
XNMRT1::onResetFit(const Snapshot &shot, XTouchableNode *) {
    iterate_commit([=](Transaction &tr){
        const Snapshot &shot(tr);
        double x = randMT19937();
        double p1min = shot[ *p1Min()];
        double p1max = shot[ *p1Max()];
        if((p1min <= 0) || (p1min >= p1max)) {
            gErrPrint(i18n("Invalid P1Min or P1Max."));
            return;
        }
        tr[ *this].m_params[0] = 1.0 / distributeP1(shot, x);
        tr[ *this].m_params[1] = 0.1;
        tr[ *this].m_params[2] = 0.0;
    });
    requestAnalysis();
}
void
XNMRT1::obtainNextP1(Transaction &tr) {
    const Snapshot &shot(tr);
    double x_0_to_1;
    if(shot[ *p1Strategy()] .to_str() == P1STRATEGY_RANDOM) {
        x_0_to_1 = randMT19937();
    }
    else {
    //FLATTEN
        int samples = shot[ *this].m_sumpts.size();
        if( !samples) {
            x_0_to_1 = 0.5;
        }
        else {
            //binary search for area having small sum isigma.
            double p1min = shot[ *p1Min()];
            double p1max = shot[ *p1Max()];
            int lb = 0, ub = samples;
            double k_0 = (samples - 1) / log(p1max/p1min);
            int idx_p1next = lrint(log(shot[ *p1Next()] / p1min) * k_0);
            idx_p1next = std::min(std::max(idx_p1next, 0), samples - 1);
            bool p1dist_linear = (shot[ *p1Dist()].to_str() == P1DIST_LINEAR);
            bool p1dist_log = (shot[ *p1Dist()].to_str() == P1DIST_LOG);
            const auto &sumpts = shot[ *this].m_sumpts;
            for(;;) {
                int mid;
                if(p1dist_log)
                    mid = (lb + ub) / 2;  //m_sumpts has been stored in log scale.
                else {
                    double xlb = exp(lb / k_0) * p1min;
                    double xub = exp(ub / k_0) * p1min;
                    double xhalf;
                    if(p1dist_linear)
                        xhalf = (xlb + xub) / 2;
                    else
                        xhalf = 1.0/((1/xlb + 1/xub) / 2); //reciprocal
                    mid = lrint(log(xhalf / p1min) * k_0);
                }
                assert((mid >= lb) && (mid <= ub));
                int isigma_0 = 0;
                for(int idx = lb; idx < mid; ++idx)
                    isigma_0 += sumpts[idx].isigma;
                int isigma_1 = 0;
                for(int idx = mid; idx < ub; ++idx)
                    isigma_1 += sumpts[idx].isigma;
                if((lb <= idx_p1next) && (mid > idx_p1next))
                    isigma_0++;
                if((mid <= idx_p1next) && (ub > idx_p1next))
                    isigma_1++;
                if(isigma_0 == isigma_1) {
                    if(randMT19937() < 0.5)
                        ub = mid;
                    else
                        lb = mid;
                }
                else {
                    if(isigma_0 < isigma_1)
                        ub = mid;
                    else
                        lb = mid;
                }
                if(ub - lb <= 1) {
                    x_0_to_1 = (double)lb / (samples - 1);
                    break;
                }
            }
        }
    }
    tr[ *p1Next()] = distributeP1(shot, x_0_to_1);
    tr[ *p1AltNext()] = distributeP1(shot, 1 - x_0_to_1);
}
void
XNMRT1::onP1CondChanged(const Snapshot &shot, XValueNodeBase *node) {
    requestAnalysis();
    iterate_commit([=](Transaction &tr){
        const Snapshot &shot(tr);
        double p1min = shot[ *p1Min()];
        double p1max = shot[ *p1Max()];
        if((p1min <= 0) || (p1min >= p1max)) {
            gErrPrint(i18n("Invalid P1Min or P1Max."));
            return;
        }
        obtainNextP1(tr);
    });
}
void
XNMRT1::onCondChanged(const Snapshot &shot, XValueNodeBase *node) {
    if(
        (node == mode().get()) ||
        (node == freq().get())) {
        trans( *this).m_timeClearRequested = XTime::now();
    }
    requestAnalysis();
}
void
XNMRT1::onMapCondChanged(const Snapshot &shot, XValueNodeBase *node) {
    trans( *this).m_timeMapFTCalcRequested = XTime::now();
    requestAnalysis();
}
void
XNMRT1::onMapClearCondRequested(const Snapshot &shot, XValueNodeBase *node) {
    if(node == mapMode().get())
        if((MapMode)(int)shot[ *mapMode()] == MapMode::Off)
        trans( *this).m_timeMapClearRequested = XTime::now();
    requestAnalysis();
}
void
XNMRT1::analyzeSpectrum(Transaction &tr,
    const std::vector< std::complex<double> >&wave, int origin, double cf,
    std::deque<std::complex<double> > &value_by_cond) {
    const Snapshot &shot_this(tr);

    value_by_cond.clear();
    std::deque<FFT::twindowfunc> funcs;
    m_solver->windowFuncs(funcs);

    int idx = 0;
    for(std::deque<double>::const_iterator wit = m_windowWidthList.begin(); wit != m_windowWidthList.end(); wit++) {
        for(std::deque<FFT::twindowfunc>::iterator fit = funcs.begin(); fit != funcs.end(); fit++) {
            if(shot_this[ *this].m_convolutionCache.size() <= idx) {
                tr[ *this].m_convolutionCache.push_back(
                    std::make_shared<Payload::ConvolutionCache>());
            }
            shared_ptr<Payload::ConvolutionCache> cache = tr[ *this].m_convolutionCache[idx];
            if((cache->windowwidth != *wit) || (cache->origin != origin) ||
                (cache->windowfunc != *fit) || (cache->cfreq != cf) ||
                (cache->wave.size() != wave.size())) {
                cache->windowwidth = *wit;
                cache->origin = origin;
                cache->windowfunc = *fit;
                cache->cfreq = cf;
                cache->power = 0.0;
                cache->wave.resize(wave.size());
                double wk = 1.0 / FFTSolver::windowLength(wave.size(), -origin, *wit);
                for(int i = 0; i < (int)wave.size(); i++) {
                    double w = ( *fit)((i - origin) * wk) / (double)wave.size();
                    cache->wave[i] = std::polar(w, -2.0*M_PI*cf*(i - origin));
                    cache->power += w*w;
                }
            }

            std::complex<double> z(0.0);
            for(int i = 0; i < (int)cache->wave.size(); i++) {
                z += wave[i] * cache->wave[i];
            }

//			m_solver->solver()->exec(wave, fftout, -origin, 0.0, *fit, *wit);
//			value_by_cond.push_back(fftout[(cf + fftlen) % fftlen]);
            value_by_cond.push_back(z);
            idx++;
        }
    }
}
void
XNMRT1::ZFFFT(Transaction &tr,
              std::vector< std::complex<double> >&bufin, std::vector< std::complex<double> >&bufout,
              shared_ptr<Payload::Pulse> p, double interval){
    const Snapshot &shot_this(tr);
    FFT::twindowfunc wndfunc = m_solverMapPulse->windowFunc(shot_this);
    double wndwidth = shot_this[ *mapWindowWidth()] / 100.0;

    double mapfreqres = shot_this[ *mapFreqRes()] * 1e3; //[Hz]
    mapfreqres = std::min(mapfreqres, 1.0 / p->summedTrace.size() / interval);
    tr[ *this].m_mapFreqRes = mapfreqres;
    double mapbandwidth = shot_this[ *mapBandWidth()] * 1e3;
    tr[ *this].m_mapBandWidth = mapbandwidth;
    double startfreq = shot_this[ *this].mapStartFreq(); //[Hz]
    long xlen = shot_this[ *this].mapFreqCount();
    if(xlen < 4)
        throw XRecordError(i18n("Invalid div. for mapping."), __FILE__, __LINE__);
    if(xlen > 10000)
        throw XRecordError(i18n("Too many # of div. for mapping."), __FILE__, __LINE__);
    if(shot_this[ *this].m_mapTCount * xlen > 50000)
        throw XRecordError(i18n("Too many # of div. for mapping."), __FILE__, __LINE__);

    bufin.resize(p->summedTrace.size());
    Eigen::Map<Eigen::VectorXcd>( &bufin[0], bufin.size()) = p->summedTrace;

    int fftlen = std::max((long)p->summedTrace.size(), lrint(1.0 / mapfreqres / interval));
    double fftres = 1.0 / fftlen / interval; //[Hz]
    bufout.resize(fftlen);

    SpectrumSolver &solver(tr[ *m_solverMapPulse].solver());
    try {
        solver.exec(bufin, bufout, -p->ftOrigin, 0.3e-2, wndfunc, wndwidth);
    }
    catch (XKameError &e) {
        throw XSkippedRecordError(e.msg(), __FILE__, __LINE__);
    }
    double normalize = 1.0 / p->avgCount / p->summedTrace.size(); //for averaged voltage.
    p->ft.setZero(xlen);
    for(int i = 0; i < xlen; ++i) {
        double f = i * mapfreqres + startfreq; //Hz
        int j = (lrint(f / fftres) + fftlen) % fftlen;
        p->ft.coeffRef(i) = bufout[j] * normalize;
    }
}
void
XNMRT1::storePulseForMapping(Transaction &tr, double p1_or_2tau,
    const std::vector< std::complex<double> >&wave, const Snapshot &shot_pulse,
    const XNMRPulseAnalyzer &pulse) {
    const Snapshot &shot_this(tr);
    if((MapMode)(int)shot_this[ *mapMode()] == MapMode::Off)
        return;
    if((p1_or_2tau > shot_this[ *p1Max()]) || (p1_or_2tau < shot_this[ *p1Min()]))
        return;
    double min_reldiff = 1e10;
    std::shared_ptr<Payload::Pulse> p;
    for(auto x: shot_this[ *this].m_allPulses) {
        double reldiff = std::abs(x->p1 / p1_or_2tau - 1.0);
        if(reldiff < min_reldiff) {
            min_reldiff = reldiff;
            p = x;
        }
    }
    if(min_reldiff > shot_this[ *p1Max()]/shot_this[ *p1Min()] - 1.0)
        return;

    p->ftOrigin = shot_pulse[pulse].waveFTPos();
    if(p->summedTrace.size() != wave.size()) {
        p->avgCount = 0;
        p->summedDarkPSDSq = 0.0;
        p->summedTrace.setZero(wave.size());
    }
    p->summedTrace += Eigen::Map<Eigen::VectorXcd>(const_cast<std::complex<double>*>( &wave[0]), wave.size());
    p->avgCount++;
    const std::vector<double>& darkpsd = shot_pulse[pulse].darkPSD();
    auto vec_darkpsd = Eigen::Map<Eigen::VectorXd>(const_cast<double*>( &darkpsd[0]), darkpsd.size());

    p->summedDarkPSDSq += vec_darkpsd.sum() * shot_pulse[pulse].darkPSDFactorToVoltSq() / vec_darkpsd.size(); //[V^2]

    std::vector<std::complex<double> > fftout;
    std::vector<std::complex<double> > fftin;
    ZFFFT(tr, fftin, fftout, p, shot_pulse[pulse].interval());
}

bool
XNMRT1::checkDependency(const Snapshot &shot_this,
    const Snapshot &shot_emitter, const Snapshot &shot_others,
    XDriver *emitter) const {
    shared_ptr<XPulser> pulser__ = shot_this[ *pulser()];
    shared_ptr<XNMRPulseAnalyzer> pulse1__ = shot_this[ *pulse1()];
    shared_ptr<XNMRPulseAnalyzer> pulse2__ = shot_this[ *pulse2()];
    if( !pulser__ || !pulse1__) return false;
    if(emitter == this) return true;
    if(emitter == pulser__.get()) return false;
    assert((emitter == pulse1__.get()) || (emitter == pulse2__.get()));

    const Snapshot &shot_pulse1((emitter == pulse1__.get()) ? shot_emitter : shot_others);
    const Snapshot &shot_pulse2((emitter == pulse2__.get()) ? shot_emitter : shot_others);

    if(shot_others[ *pulser__].time() > shot_pulse1[ *pulse1__].time()) return false;

    switch(shot_others[ *pulser__].combMode()) {
    default:
        if(emitter == pulse2__.get())
            return false;
        return true;
    case XPulser::N_COMB_MODE_COMB_ALT:
    case XPulser::N_COMB_MODE_P1_ALT:
        if( !pulse2__) {
            m_statusPrinter->printError(i18n("2 Pulse Analyzers needed."));
            return false;
        }
        if(shot_pulse1[ *pulse1__].time() != shot_pulse2[ *pulse2__].time()) return false;
        return true;
    }
//    return (pulser__->time() < pulse1__->timeAwared()) && (pulser__->time() < pulse1__->time());
}

void
XNMRT1::analyze(Transaction &tr, const Snapshot &shot_emitter, const Snapshot &shot_others,
    XDriver *emitter) {
    Snapshot &shot_this(tr);

    double p1min = shot_this[ *p1Min()];
    double p1max = shot_this[ *p1Max()];

    if((p1min <= 0) || (p1min >= p1max)) {
        throw XRecordError(i18n("Invalid P1Min or P1Max."), __FILE__, __LINE__);
    }

    int samples = shot_this[ *smoothSamples()];
    if(samples <= 10) {
        throw XRecordError(i18n("Invalid # of Samples."), __FILE__, __LINE__);
    }
    if(samples >= 100000) {
        m_statusPrinter->printWarning(i18n("Too many Samples."), true);
    }

    MeasMode mode__ = (MeasMode)(int)shot_this[ *mode()];
    shared_ptr<XNMRPulseAnalyzer> pulse1__ = shot_this[ *pulse1()];
    shared_ptr<XNMRPulseAnalyzer> pulse2__ = shot_this[ *pulse2()];
    const Snapshot &shot_pulse1((emitter == pulse1__.get()) ? shot_emitter : shot_others);
    const Snapshot &shot_pulse2((emitter == pulse2__.get()) ? shot_emitter : shot_others);

    shared_ptr<XPulser> pulser__ = shot_this[ *pulser()];
    const Snapshot &shot_pulser(shot_others);
    assert( pulser__ );
    if(shot_pulser[ *pulser__].time().isSet()) {
        //Check consitency.
        switch (mode__) {
        case MeasMode::T1:
            break;
        case MeasMode::T2:
            break;
        case MeasMode::ST_E:
            if((shot_pulser[ *pulser__].tau() != shot_pulser[ *pulser__].combPT()) ||
                    (shot_pulser[ *pulser__].combNum() != 2) ||
                    ( !shot_pulser[ *pulser__->conserveStEPhase()]) ||
                    (shot_pulser[ *pulser__].pw1() != 0.0) ||
                    (shot_pulser[ *pulser__].pw2() != shot_pulser[ *pulser__].combPW()))
                m_statusPrinter->printWarning(i18n("Strange St.E. settings."));
            break;
        case MeasMode::T2_Multi:
            break;
        }
    }


    //Prepares data for mapping.
    if(shot_this[ *this].m_allPulses.empty() || shot_this[ *this].m_timeMapClearRequested.isSet()) {
        tr[ *this].m_timeMapClearRequested = {};
        tr[ *this].m_allPulses.clear();
        if((MapMode)(int)shot_this[ *mapMode()] != MapMode::Off) {
            int samples = std::min(200u, shot_this[ *smoothSamples()] * 10);
            tr[ *this].m_mapTCount = samples;
            for(long i = 0; i < shot_this[ *smoothSamples()]; ++i) {
                tr[ *this].m_allPulses.push_back(std::make_shared<Payload::Pulse>());
                auto p = tr[ *this].m_allPulses.back();
                p->p1 = distributeP1(shot_this, (double)i / (shot_this[ *smoothSamples()] - 1));
            }
        }
        tr[ *m_waveMap].clearPoints();
        tr[ *m_waveAllRelaxCurves].clearPoints();
        m_regularization.reset();
    }

    //Reads spectra from NMRPulseAnalyzers
    if( emitter != this) {
        assert( pulse1__ );
        assert( shot_pulse1[ *pulse1__].time() );
        assert( shot_pulser[ *pulser__].time() );
        assert( emitter != pulser__.get() );

        if(shot_pulse1[ *pulse1__->exAvgIncr()] && mode__ != MeasMode::T2_Multi) {
            m_statusPrinter->printWarning(i18n("Do NOT use incremental avg. Skipping."));
            throw XSkippedRecordError(__FILE__, __LINE__);
        }

        std::deque<std::complex<double> > cmp1, cmp2;
        double cfreq = shot_this[ *freq()] * 1e3 * shot_pulse1[ *pulse1__].interval();
        if(shot_this[ *trackPeak()]) {
            if(((mode__ == MeasMode::T1) && (shot_pulser[ *pulser__].combP1() > distributeP1(shot_this, 0.66))) ||
               ((mode__ == MeasMode::T2) && (shot_pulser[ *pulser__].combP1() < distributeP1(shot_this, 0.33))) ||
               (mode__ == MeasMode::T2_Multi) ||
               shot_this[ *this].m_sumpts.empty()) {
                tr[ *freq()] = (double)shot_pulse1[ *pulse1__->entryPeakFreq()->value()];
                tr.unmark(m_lsnOnCondChanged); //avoiding recursive signaling.
            }
        }

        if(mode__ == MeasMode::T2_Multi){
            if(shot_pulser[ *pulser__].combMode() != XPulser::N_COMB_MODE_OFF)
                m_statusPrinter->printWarning(i18n("T2 mode with comb pulse!"));

            if(shot_pulse1[ *pulse1__->exAvgIncr()])
                tr[ *this].m_pts.clear();
            for(int i = 0; i < shot_pulser[ *pulser__].echoNum(); i++){
                Payload::RawPt pt1;
                analyzeSpectrum(tr, shot_pulse1[ *pulse1__].echoesT2()[i],
                        shot_pulse1[ *pulse1__].waveFTPos(), cfreq, cmp1);
                pt1.value_by_cond.resize(cmp1.size());

                double twotau = 2.0 * shot_pulser[ *pulser__].tau() * (i + 1);
                pt1.p1 = twotau;
                std::copy(cmp1.begin(), cmp1.end(), pt1.value_by_cond.begin());
                tr[ *this].m_pts.push_back(pt1);

                storePulseForMapping(tr, twotau, shot_pulse1[ *pulse1__].echoesT2()[i], shot_pulse1, *pulse1__);
            }
        }
        else {
            Payload::RawPt pt1, pt2;
            analyzeSpectrum(tr, shot_pulse1[ *pulse1__].wave(),
                    shot_pulse1[ *pulse1__].waveFTPos(), cfreq, cmp1);
            if(pulse2__) {
                analyzeSpectrum(tr, shot_pulse2[ *pulse2__].wave(),
                        shot_pulse2[ *pulse2__].waveFTPos(), cfreq, cmp2);
                pt2.value_by_cond.resize(cmp2.size());
            }
            pt1.value_by_cond.resize(cmp1.size());
            switch(shot_pulser[ *pulser__].combMode()) {
            default:
                throw XRecordError(i18n("Unknown Comb Mode!"), __FILE__, __LINE__);
            case XPulser::N_COMB_MODE_COMB_ALT: {
                if(mode__ != MeasMode::T1) throw XRecordError(i18n("Use T1 mode!"), __FILE__, __LINE__);
                assert(pulse2__);
                double p1 = shot_pulser[ *pulser__].combP1();
                pt1.p1 = p1;
                for(int i = 0; i < cmp1.size(); i++)
                    pt1.value_by_cond[i] = (cmp1[i] - cmp2[i]) / cmp1[i];
                tr[ *this].m_pts.push_back(pt1);
                if((MapMode)(int)shot_this[ *mapMode()] != MapMode::Off)
                    throw XRecordError(i18n("Unsupported Comb Mode for Mapping!"), __FILE__, __LINE__);
                break;
                }
            case XPulser::N_COMB_MODE_P1_ALT: {
                if(mode__ == MeasMode::T2)
                    throw XRecordError(i18n("Do not use T2 mode!"), __FILE__, __LINE__);
                assert(pulse2__);
                double p1 = shot_pulser[ *pulser__].combP1();
                pt1.p1 = p1;
                std::copy(cmp1.begin(), cmp1.end(), pt1.value_by_cond.begin());
                tr[ *this].m_pts.push_back(pt1);
                double p1_alt = shot_pulser[ *pulser__].combP1Alt();
                pt2.p1 = p1_alt;
                std::copy(cmp2.begin(), cmp2.end(), pt2.value_by_cond.begin());
                tr[ *this].m_pts.push_back(pt2);
                storePulseForMapping(tr, p1, shot_pulse1[ *pulse1__].wave(), shot_pulse1, *pulse1__);
                storePulseForMapping(tr, p1_alt, shot_pulse2[ *pulse2__].wave(), shot_pulse2, *pulse2__);
                break;
                }
            case XPulser::N_COMB_MODE_ON:
                if(mode__ != MeasMode::T2) {
                    double p1 = shot_pulser[ *pulser__].combP1();
                    pt1.p1 = p1;
                    std::copy(cmp1.begin(), cmp1.end(), pt1.value_by_cond.begin());
                    tr[ *this].m_pts.push_back(pt1);
                    storePulseForMapping(tr, p1, shot_pulse1[ *pulse1__].wave(), shot_pulse1, *pulse1__);
                    break;
                }
                m_statusPrinter->printWarning(i18n("T2 mode with comb pulse!"));
            case XPulser::N_COMB_MODE_OFF:
                if(mode__ != MeasMode::T2) {
                    m_statusPrinter->printWarning(i18n("Do not use T1 mode! Skipping."));
                    throw XSkippedRecordError(__FILE__, __LINE__);
                }
                //T2 measurement
                double twotau = 2.0 * shot_pulser[ *pulser__].tau();
                pt1.p1 = twotau;
                std::copy(cmp1.begin(), cmp1.end(), pt1.value_by_cond.begin());
                tr[ *this].m_pts.push_back(pt1);
                storePulseForMapping(tr, twotau, shot_pulse1[ *pulse1__].wave(), shot_pulse1, *pulse1__);
                break;
            }
        }
    }

    tr[ *this].m_sumpts.clear();

    if(shot_this[ *this].m_timeClearRequested.isSet()) {
        tr[ *this].m_timeClearRequested = {};
        tr[ *this].m_pts.clear();
        tr[ *m_wave].clearPoints();
        tr[ *m_fitStatus] = "";
        if( !shot_pulse1[ *pulse1__->exAvgIncr()] || mode__ != MeasMode::T2_Multi) {
            trans( *pulse1__->avgClear()).touch();
            if(pulse2__)
                trans( *pulse2__->avgClear()).touch();
        }
        throw XSkippedRecordError(__FILE__, __LINE__);
    }

    shared_ptr<XRelaxFunc> func = shot_this[ *relaxFunc()];
    if( !func) {
        throw XRecordError(i18n("Please select relaxation func."), __FILE__, __LINE__);
    }

    tr[ *this].m_sumpts.resize(samples);
    auto &sumpts(tr[ *this].m_sumpts);
    {
    //Building recovery curves after rounding log(P1) from all of aquirred points.
        Payload::Pt dummy = {};
        dummy.value_by_cond.resize(shot_this[ *this].m_convolutionCache.size());
        std::fill(tr[ *this].m_sumpts.begin(), tr[ *this].m_sumpts.end(), dummy);
        double k = (shot_this[ *this].m_sumpts.size() - 1) / log(p1max/p1min);
        auto pts_begin(shot_this[ *this].m_pts.begin());
        auto pts_end(shot_this[ *this].m_pts.end());
        int sum_size = (int)shot_this[ *this].m_sumpts.size();
        for(auto it = pts_begin; it != pts_end; it++) {
            int idx = lrint(log(it->p1 / p1min) * k);
            if((idx < 0) || (idx >= sum_size)) continue;
            double p1 = it->p1;
            //For St.E., T+tau = P1+3*tau.
            if(mode__ == MeasMode::ST_E)
                p1 += 3 * shot_pulser[ *pulser__].tau() * 1e-3;
            sumpts[idx].isigma += 1;
            sumpts[idx].p1 += p1;
            for(unsigned int i = 0; i < it->value_by_cond.size(); i++)
                sumpts[idx].value_by_cond[i] += it->value_by_cond[i];
        }
    }

    std::deque<std::complex<double> > sum_c(
        shot_this[ *this].m_convolutionCache.size()), corr(shot_this[ *this].m_convolutionCache.size());
    double sum_t = 0.0;
    int n = 0;
    for(auto it = shot_this[ *this].m_sumpts.begin();
        it != shot_this[ *this].m_sumpts.end(); it++) {
        if(it->isigma == 0) continue;
        double t = log10(it->p1 / it->isigma);
        for(unsigned int i = 0; i < it->value_by_cond.size(); i++) {
            sum_c[i] += it->value_by_cond[i];
            corr[i] += it->value_by_cond[i] * t;
        }
        sum_t += t * it->isigma;
        n += it->isigma;
    }
    if(n) {
        //correlation for y_i * log(t_i)
        for(unsigned int i = 0; i < corr.size(); i++) {
            corr[i] -= sum_c[i]*sum_t/(double)n;
            corr[i] *= ((mode__ == MeasMode::T1) ? 1 : -1);
        }

        bool absfit__ = shot_this[ *absFit()];

        std::deque<double> phase_by_cond(corr.size(), shot_this[ *phase()] / 180.0 * M_PI);
        int cond = -1;
        double maxsn2 = 0.0;
        for(unsigned int i = 0; i < corr.size(); i++) {
            if(shot_this[ *autoPhase()]) {
                phase_by_cond[i] = std::arg(corr[i]);
            }
            if(shot_this[ *autoWindow()]) {
                double sn2 = absfit__ ? std::abs(corr[i]) : std::real(corr[i] * std::polar(1.0, -phase_by_cond[i]));
                sn2 = sn2 * sn2 / shot_this[ *this].m_convolutionCache[i]->power;
                if(maxsn2 < sn2) {
                    maxsn2 = sn2;
                    cond = i;
                }
            }
        }
        if(cond < 0) {
            cond = 0;
            for(std::deque<shared_ptr<Payload::ConvolutionCache> >::const_iterator it
                = shot_this[ *this].m_convolutionCache.begin();
                it != shot_this[ *this].m_convolutionCache.end(); ++it) {
                if((m_windowWidthList[std::max(0, (int)shot_this[ *windowWidth()])] == ( *it)->windowwidth) &&
                    (m_solver->windowFunc(shot_this) == ( *it)->windowfunc)) {
                    break;
                }
                cond++;
            }
        }
        if(cond >= (shot_this[ *this].m_convolutionCache.size())) {
            throw XSkippedRecordError(__FILE__, __LINE__);
        }
        double ph = phase_by_cond[cond];
        if(shot_this[ *autoPhase()]) {
            tr[ *phase()] = ph / M_PI * 180;
            tr.unmark(m_lsnOnCondChanged); //avoiding recursive signaling.
        }
        if(shot_this[ *autoWindow()]) {
            for(unsigned int i = 0; i < m_windowWidthList.size(); i++) {
                if(m_windowWidthList[i] == shot_this[ *this].m_convolutionCache[cond]->windowwidth)
                    tr[ *windowWidth()] = i;
            }
            std::deque<FFT::twindowfunc> funcs;
            m_solver->windowFuncs(funcs);
            for(unsigned int i = 0; i < funcs.size(); i++) {
                if(funcs[i] == shot_this[ *this].m_convolutionCache[cond]->windowfunc)
                    tr[ *windowFunc()] = i;
            }
            tr.unmark(m_lsnOnCondChanged); //avoiding recursive signaling.
        }
        std::complex<double> cph(std::polar(1.0, -phase_by_cond[cond]));
        for(auto it = sumpts.begin(); it != sumpts.end(); it++) {
            if(it->isigma == 0) continue;
            it->p1 = it->p1 / it->isigma;
            it->c =  it->value_by_cond[cond] * cph / (double)it->isigma;
            it->var = (absfit__) ? std::abs(it->c) : std::real(it->c);
            it->isigma = sqrt(it->isigma);
        }

        tr[ *m_fitStatus] = iterate(tr, func, 4);

        t1inv()->value(tr, 1000.0 * shot_this[ *this].m_params[0]);
        t1invErr()->value(tr, 1000.0 * shot_this[ *this].m_errors[0]);
    }

    if(shot_this[ *this].m_timeMapFTCalcRequested.isSet()) {
        tr[ *this].m_timeMapFTCalcRequested = {};
        //recalculates all FT.
        std::vector<std::complex<double> > fftout;
        std::vector<std::complex<double> > fftin;
        for(auto &p: shot_this[ *this].m_allPulses) {
            if(p->avgCount)
                ZFFFT(tr, fftin, fftout, p, shot_pulse1[ *pulse1__].interval());
        }
        m_regularization.reset(); //for mode/relax fn. change.
    }

    m_isPulserControlRequested = (emitter != this);
}
void
XNMRT1::setNextP1(const Snapshot &shot) {
    shared_ptr<XPulser> pulser__ = shot[ *pulser()];
    if(pulser__ && shot[ *mode()] != (int)MeasMode::T2_Multi) {
        pulser__->iterate_commit([=](Transaction &tr){
            switch((MeasMode)(int)shot[ *mode()]) {
            case MeasMode::T1:
            case MeasMode::ST_E:
                tr[ *pulser__->combP1()] = (double)shot[ *p1Next()];
                tr[ *pulser__->combP1Alt()] = (double)shot[ *p1AltNext()];
                break;
            case MeasMode::T2:
                tr[ *pulser__->tau()] = shot[ *p1Next()] / 2.0;
                break;
            }
        });
        iterate_commit([=](Transaction &tr){
            obtainNextP1(tr);
        });
    }
}
void
XNMRT1::visualize(const Snapshot &shot) {
    if( !shot[ *this].time()) {
        iterate_commit([=](Transaction &tr){
            tr[ *m_wave].clearPoints();
        });
        return;
    }

    //set new P1s
    if(shot[ *active()] && m_isPulserControlRequested.compare_set_strong((int)true, (int)false)) {
        setNextP1(shot);
    }

    XString tlabel;
    switch((MeasMode)(int)shot[ *mode()]) {
    case MeasMode::T1:
        tlabel = "P1 [ms]";
        break;
    case MeasMode::T2:
    case MeasMode::T2_Multi:
        tlabel = "2tau [us]";
        break;
    case MeasMode::ST_E:
        tlabel = "T+tau [ms]";
        break;
    }
    m_wave->iterate_commit([=](Transaction &tr){

        tr[ *m_wave].setLabel(0, tlabel.c_str());
        tr[ *tr[ *m_wave].axisx()->label()] = tlabel;
        size_t length = shot[ *this].m_sumpts.size();
        tr[ *m_wave].setRowCount(length);
        std::vector<float> colp1(length, 0.0), colval(length, 0.0),
            colabs(length, 0.0), colre(length, 0.0), colim(length, 0.0),
            colisigma(length, 0.0);
        int i = 0;
        for(auto it = shot[ *this].m_sumpts.begin();
            it != shot[ *this].m_sumpts.end(); it++) {
            if(it->isigma != 0) {
                colval[i] = it->var;
                colabs[i] = std::abs(it->c);
                colre[i] = std::real(it->c);
                colim[i] = std::imag(it->c);
                colp1[i] = it->p1;
                colisigma[i] = it->isigma;
            }
            i++;
        }
        tr[ *m_wave].setColumn(0, std::move(colp1), 5);
        tr[ *m_wave].setColumn(1, std::move(colval), 5);
        tr[ *m_wave].setColumn(2, std::move(colisigma), 4);
        tr[ *m_wave].setColumn(3, std::move(colabs), 4);
        tr[ *m_wave].setColumn(4, std::move(colre), 4);
        tr[ *m_wave].setColumn(5, std::move(colim), 4);
        m_wave->drawGraph(tr);
    });

    auto mapmode = (MapMode)(int)shot[ *mapMode()];
    int pcount_stored = 0;
    for(auto p: shot[ *this].m_allPulses)
        if(p->avgCount)
            ++pcount_stored;

    if((mapmode != MapMode::Off) && pcount_stored) {
        Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> relax_fdep;
        relax_fdep.setZero(shot[ *this].mapFreqCount(), pcount_stored);

        m_waveAllRelaxCurves->iterate_commit([&](Transaction &tr){

            tr[ *m_waveAllRelaxCurves].setLabel(1, tlabel.c_str());
            tr[ *tr[ *m_waveAllRelaxCurves].axisz()->label()] = tlabel;
            size_t length = pcount_stored * shot[ *this].mapFreqCount();
            std::vector<float> colf(length, 0.0), colp1(length, 0.0),
                colre(length, 0.0), colim(length, 0.0), colisigma(length, 0.0);
            auto rot_ph = std::polar(1.0, -shot[ *phase()] / 180.0 * M_PI);
            int i = 0;
            int k = 0;
            for(auto p: shot[ *this].m_allPulses) {
                if(p->avgCount) {
                    for(int j = 0; j < shot[ *this].mapFreqCount(); ++j) {
                        double f = shot[ *this].mapStartFreq() + j * shot[ *this].m_mapFreqRes;
                        colp1[k] = p->p1;
                        colf[k] = f * 1e-3;
                        auto z = p->ft.coeff(j) * rot_ph;
                        colre[k] = std::real(z);
                        colim[k] = std::imag(z);
                        relax_fdep.coeffRef(j, i) = colre[k];
                        colisigma[k] = p->avgCount / sqrt(p->summedDarkPSDSq);
//                        colisigma[k] = sqrt(p->summedDarkPSDSq) / p->avgCount;
                        k++;
                    }
                    i++;
                }
            }
            assert(i == pcount_stored);
            tr[ *m_waveAllRelaxCurves].setRowCount(length);
            tr[ *m_waveAllRelaxCurves].setColumn(0, std::move(colf), 5);
            tr[ *m_waveAllRelaxCurves].setColumn(1, std::move(colp1), 5);
            tr[ *m_waveAllRelaxCurves].setColumn(2, std::move(colre), 4);
            tr[ *m_waveAllRelaxCurves].setColumn(3, std::move(colim), 4);
            tr[ *m_waveAllRelaxCurves].setColumn(4, std::move(colisigma), 3);
            m_waveAllRelaxCurves->drawGraph(tr);
        });
        double noisesq = 0.0;
        for(auto &p: shot[ *this].m_allPulses)
            if(p->avgCount)
                noisesq += p->summedDarkPSDSq / p->avgCount / p->avgCount;
        noisesq /= pcount_stored;

        auto mapT = [this](const Snapshot &shot, int i) {
            return shot[ *p1Min()] * exp(log(shot[ *p1Max()]/shot[ *p1Min()]) / (shot[ *this].m_mapTCount - 1) * i);
        };

        shared_ptr<XRelaxFunc> relax_fn = shot[ *relaxFunc()];
        if( !relax_fn) return;

        double relax_coeff = -1.0; //exp(-t/T1)
        if((MeasMode)(int)shot[ *mode()] == MeasMode::T1) {
            //ex. 1.0 - exp(-t/T1)
            relax_coeff = 1.0 / (shot[ *this].m_params[1] + shot[ *this].m_params[2]);
        }

        local_shared_ptr<TikhonovRegular> regularization = m_regularization;
        if( !regularization || (regularization->ylen() != pcount_stored)) {
            Eigen::MatrixXd mat_conv; //Matrix A; y = A x.
            mat_conv.setZero(pcount_stored, shot[ *this].m_mapTCount);
            for(int j = 0; j < shot[ *this].m_mapTCount; ++j) {
                double it1 = 1.0 / mapT(shot, j);
                int i = 0;
                double f, df;
                for(auto &p: shot[ *this].m_allPulses) {
                    if(p->avgCount) {
                        relax_fn->relax( &f, &df, p->p1, it1); //ex. f(t) = 1 - exp(-t/T1)
                        mat_conv.coeffRef(i, j) = relax_coeff * f + 1.0;
                        ++i;
                    }
                }
            }
            //very slow due to SVD.
            regularization.reset(new TikhonovRegular(mat_conv, (TikhonovRegular::TikhonovMatrix)(int)shot[ *mapTikhonovMatrix()]));
            m_regularization = regularization;
        }
        auto method = std::map<MapMode, TikhonovRegular::Method>{{MapMode::NoiseAnalysis, TikhonovRegular::Method::KnownError},
            {MapMode::GCV, TikhonovRegular::Method::MinGCV}, {MapMode::LCurve, TikhonovRegular::Method::L_Curve},
            {MapMode::AllNonNegative, TikhonovRegular::Method::AllNonNegative}}.at(mapmode);
        regularization->chooseLambda(method, relax_fdep.row(shot[ *this].mapFreqCount() / 2), noisesq);

        XString tlabel;
        switch((MeasMode)(int)shot[ *mode()]) {
        case MeasMode::T1:
            tlabel = "T1 [ms]";
            break;
        case MeasMode::T2:
        case MeasMode::T2_Multi:
            tlabel = "T2 [us]";
            break;
        case MeasMode::ST_E:
            tlabel = "Tste [ms]";
            break;
        }
        m_waveMap->iterate_commit([&](Transaction &tr){
            tr[ *m_waveMap].setLabel(1, tlabel.c_str());
            tr[ *tr[ *m_waveMap].axisy()->label()] = tlabel;
            size_t length = shot[ *this].m_mapTCount * shot[ *this].mapFreqCount();
            std::vector<float> colf(length, 0.0), colt(length, 0.0), colval(length, 0.0);
            int k = 0;
            for(int j = 0; j < shot[ *this].mapFreqCount(); ++j) {
                double f = shot[ *this].mapStartFreq() + j * shot[ *this].m_mapFreqRes;

                auto densities = regularization->solve(relax_fdep.row(j));

                for(int i = 0; i < shot[ *this].m_mapTCount; ++i) {
                    colt[k] = mapT(shot, i);
                    colf[k] = f * 1e-3;
                    colval[k] = densities[i];
                    k++;
                }
            }
            tr[ *m_waveMap].setRowCount(length);
            tr[ *m_waveMap].setColumn(0, std::move(colf), 5);
            tr[ *m_waveMap].setColumn(1, std::move(colt), 5);
            tr[ *m_waveMap].setColumn(2, std::move(colval), 4);
            m_waveMap->drawGraph(tr);
        });
    }
}

void
XNMRT1::onActiveChanged(const Snapshot &shot, XValueNodeBase *) {
    Snapshot shot_this( *this);
    if(shot_this[ *active()]) {
        const shared_ptr<XPulser> pulser__ = shot_this[ *pulser()];
        const shared_ptr<XNMRPulseAnalyzer> pulse1__ = shot_this[ *pulse1()];
        const shared_ptr<XNMRPulseAnalyzer> pulse2__ = shot_this[ *pulse2()];

        onClearAll(shot_this, m_clearAll.get());
        if( !pulser__ || !pulse1__) {
            gErrPrint(getLabel() + ": " + i18n("No pulser or No NMR Pulse Analyzer."));
            return;
        }

        Snapshot shot_pulse1( *pulse1__);
        Snapshot shot_pulser( *pulser__);
        if(pulse2__ &&
           ((shot_pulser[ *pulser__->combMode()] == XPulser::N_COMB_MODE_COMB_ALT) ||
            (shot_pulser[ *pulser__->combMode()] == XPulser::N_COMB_MODE_P1_ALT))) {
            pulse2__->iterate_commit([=](Transaction &tr){
                tr[ *pulse2__->fromTrig()] =
                    shot_pulse1[ *pulse1__->fromTrig()] + shot_pulser[ *pulser__->altSep()];
                tr[ *pulse2__->width()] = (double)shot_pulse1[ *pulse1__->width()];
                tr[ *pulse2__->phaseAdv()] = (double)shot_pulse1[ *pulse1__->phaseAdv()];
                tr[ *pulse2__->bgPos()] =
                    shot_pulse1[ *pulse1__->bgPos()] + shot_pulser[ *pulser__->altSep()];
                tr[ *pulse2__->bgWidth()] = (double)shot_pulse1[ *pulse1__->bgWidth()];
                tr[ *pulse2__->fftPos()] =
                    shot_pulse1[ *pulse1__->fftPos()] + shot_pulser[ *pulser__->altSep()];
                tr[ *pulse2__->fftLen()] = (int)shot_pulse1[ *pulse1__->fftLen()];
                tr[ *pulse2__->windowFunc()] = (int)shot_pulse1[ *pulse1__->windowFunc()];
                tr[ *pulse2__->usePNR()] = (bool)shot_pulse1[ *pulse1__->usePNR()];
                tr[ *pulse2__->pnrSolverList()] = (int)shot_pulse1[ *pulse1__->pnrSolverList()];
                tr[ *pulse2__->solverList()] = (int)shot_pulse1[ *pulse1__->solverList()];
                tr[ *pulse2__->numEcho()] = (int)shot_pulse1[ *pulse1__->numEcho()];
                tr[ *pulse2__->echoPeriod()] = (double)shot_pulse1[ *pulse1__->echoPeriod()];
            });
        }
        shot_this = iterate_commit([=](Transaction &tr){
            obtainNextP1(tr);
        });
        setNextP1(shot_this);
        if(shot_this[ *mode()] == (int)MeasMode::T2_Multi){
            iterate_commit([=](Transaction &tr){
                tr[ *p1Min()] = 2.0 * shot_pulser[ *pulser__].tau();
                tr[ *p1Max()] = 2.0 * shot_pulser[ *pulser__].tau() * shot_pulser[ *pulser__].echoNum();
            });
        }

    }
}


