aboutsummaryrefslogtreecommitdiff
path: root/manipulationhandler.cpp
blob: 707d4c786fdd4271d1567132c8321bf86e3dc667 (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
#include "manipulationhandler.h"

#include <algorithm>
#include <iostream>
#include <numeric>

#include <QDebug>

#include "mp.h"
#include "numericrange.h"

ManipulationHandler::ManipulationHandler(const arma::mat &X,
                                         const arma::uvec &cpIndices)
    : m_X(X)
    , m_Y(X.n_rows, 2)
    , m_cpIndices(cpIndices)
    , m_rpIndices(X.n_rows - cpIndices.n_elem)
    , m_technique(TECHNIQUE_LAMP)
{
    NumericRange<arma::uword> range(0, m_X.n_rows);
    std::set_symmetric_difference(range.cbegin(), range.cend(),
            m_cpIndices.cbegin(), m_cpIndices.cend(), m_rpIndices.begin());
}

void ManipulationHandler::setTechnique(ManipulationHandler::Technique technique)
{
    if (m_technique == technique)
        return;

    m_technique = technique;
}

void ManipulationHandler::setCP(const arma::mat &Ys)
{
    m_prevY = m_Y;

    switch (m_technique) {
    case TECHNIQUE_PLMP:
        // TODO?
        // mp::plmp(m_X, m_cpIndices, Ys, m_Y);
        break;
    case TECHNIQUE_LSP:
        // TODO?
        // mp::lsp(m_X, m_cpIndices, Ys, m_Y);
        break;
    case TECHNIQUE_LAMP:
        mp::lamp(m_X, m_cpIndices, Ys, m_Y);
        break;
    case TECHNIQUE_PEKALSKA:
        // TODO?
        // mp::pekalska(m_X, m_cpIndices, Ys, m_Y);
        break;
    }

    if (m_firstY.n_rows != m_Y.n_rows) {
        m_firstY = m_Y;
    }

    emit cpChanged(m_Y.rows(m_cpIndices));
    emit rpChanged(m_Y.rows(m_rpIndices));
    emit mapChanged(m_Y);
}

void ManipulationHandler::setRewind(double t)
{
    if (m_prevY.n_rows != m_Y.n_rows) {
        return;
    }

    arma::mat Y = m_Y * t + m_prevY * (1.0 - t);
    emit cpRewound(Y.rows(m_cpIndices));
    emit rpRewound(Y.rows(m_rpIndices));
    emit mapRewound(Y);
}