133 lines
3.7 KiB
C++
133 lines
3.7 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2022 Olive Team
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
***/
|
|
|
|
#ifndef MATHNODEBASE_H
|
|
#define MATHNODEBASE_H
|
|
|
|
#include "node/node.h"
|
|
|
|
namespace olive {
|
|
|
|
class MathNodeBase : public Node
|
|
{
|
|
public:
|
|
MathNodeBase() = default;
|
|
|
|
enum Operation {
|
|
kOpAdd,
|
|
kOpSubtract,
|
|
kOpMultiply,
|
|
kOpDivide,
|
|
kOpPower
|
|
};
|
|
|
|
static QString GetOperationName(Operation o);
|
|
|
|
protected:
|
|
enum Pairing {
|
|
kPairNone = -1,
|
|
|
|
kPairNumberNumber,
|
|
kPairVecVec,
|
|
kPairMatrixMatrix,
|
|
kPairColorColor,
|
|
kPairTextureTexture,
|
|
|
|
kPairVecNumber,
|
|
kPairMatrixVec,
|
|
kPairNumberColor,
|
|
kPairTextureNumber,
|
|
kPairTextureColor,
|
|
kPairTextureMatrix,
|
|
kPairSampleSample,
|
|
kPairSampleNumber,
|
|
|
|
kPairCount
|
|
};
|
|
|
|
class PairingCalculator {
|
|
public:
|
|
PairingCalculator(const NodeValueTable &table_a, const NodeValueTable &table_b);
|
|
|
|
bool FoundMostLikelyPairing() const;
|
|
Pairing GetMostLikelyPairing() const;
|
|
|
|
const NodeValue& GetMostLikelyValueA() const;
|
|
const NodeValue& GetMostLikelyValueB() const;
|
|
|
|
private:
|
|
static QVector<int> GetPairLikelihood(const NodeValueTable& table);
|
|
|
|
Pairing most_likely_pairing_;
|
|
|
|
NodeValue most_likely_value_a_;
|
|
|
|
NodeValue most_likely_value_b_;
|
|
|
|
};
|
|
|
|
template<typename T, typename U>
|
|
static T PerformAll(Operation operation, T a, U b);
|
|
|
|
template<typename T, typename U>
|
|
static T PerformMultDiv(Operation operation, T a, U b);
|
|
|
|
template<typename T, typename U>
|
|
static T PerformAddSub(Operation operation, T a, U b);
|
|
|
|
template<typename T, typename U>
|
|
static T PerformMult(Operation operation, T a, U b);
|
|
|
|
template<typename T, typename U>
|
|
static T PerformAddSubMult(Operation operation, T a, U b);
|
|
|
|
template<typename T, typename U>
|
|
static T PerformAddSubMultDiv(Operation operation, T a, U b);
|
|
|
|
static void PerformAllOnFloatBuffer(Operation operation, float *a, float b, int start, int end);
|
|
|
|
#if defined(Q_PROCESSOR_X86) || defined(Q_PROCESSOR_ARM)
|
|
static void PerformAllOnFloatBufferSSE(Operation operation, float *a, float b, int start, int end);
|
|
#endif
|
|
|
|
static QString GetShaderUniformType(const NodeValue::Type& type);
|
|
|
|
static QString GetShaderVariableCall(const QString& input_id, const NodeValue::Type& type, const QString &coord_op = QString());
|
|
|
|
static QVector4D RetrieveVector(const NodeValue& val);
|
|
|
|
static float RetrieveNumber(const NodeValue& val);
|
|
|
|
static bool NumberIsNoOp(const Operation& op, const float& number);
|
|
|
|
ShaderCode GetShaderCodeInternal(const QString &shader_id, const QString ¶m_a_in, const QString ¶m_b_in) const;
|
|
|
|
void PushVector(NodeValueTable* output, NodeValue::Type type, const QVector4D& vec) const;
|
|
|
|
void ValueInternal(Operation operation, Pairing pairing, const QString& param_a_in, const NodeValue &val_a, const QString& param_b_in, const NodeValue& val_b, const NodeGlobals &globals, NodeValueTable *output) const;
|
|
|
|
void ProcessSamplesInternal(const NodeValueRow &values, Operation operation, const QString& param_a_in, const QString& param_b_in, const SampleBuffer &input, SampleBuffer &output, int index) const;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif // MATHNODEBASE_H
|