128 lines
3.4 KiB
C++
128 lines
3.4 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2022 Olive Team
|
|
Modifications Copyright (C) 2025 mikesolar
|
|
|
|
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/>.
|
|
|
|
***/
|
|
|
|
#include "math.h"
|
|
|
|
namespace olive
|
|
{
|
|
|
|
const QString MathNode::kMethodIn = QStringLiteral("method_in");
|
|
const QString MathNode::kParamAIn = QStringLiteral("param_a_in");
|
|
const QString MathNode::kParamBIn = QStringLiteral("param_b_in");
|
|
const QString MathNode::kParamCIn = QStringLiteral("param_c_in");
|
|
|
|
#define super MathNodeBase
|
|
|
|
MathNode::MathNode()
|
|
{
|
|
AddInput(kMethodIn, NodeValue::kCombo,
|
|
InputFlags(kInputFlagNotConnectable | kInputFlagNotKeyframable));
|
|
|
|
AddInput(kParamAIn, NodeValue::kFloat, 0.0);
|
|
SetInputProperty(kParamAIn, QStringLiteral("decimalplaces"), 8);
|
|
SetInputProperty(kParamAIn, QStringLiteral("autotrim"), true);
|
|
|
|
AddInput(kParamBIn, NodeValue::kFloat, 0.0);
|
|
SetInputProperty(kParamBIn, QStringLiteral("decimalplaces"), 8);
|
|
SetInputProperty(kParamBIn, QStringLiteral("autotrim"), true);
|
|
}
|
|
|
|
QString MathNode::Name() const
|
|
{
|
|
// Default to naming after the operation
|
|
if (parent()) {
|
|
QString op_name = GetOperationName(GetOperation());
|
|
if (!op_name.isEmpty()) {
|
|
return op_name;
|
|
}
|
|
}
|
|
|
|
return tr("Math");
|
|
}
|
|
|
|
QString MathNode::id() const
|
|
{
|
|
return QStringLiteral("org.olivevideoeditor.Olive.math");
|
|
}
|
|
|
|
QVector<Node::CategoryID> MathNode::Category() const
|
|
{
|
|
return { kCategoryMath };
|
|
}
|
|
|
|
QString MathNode::Description() const
|
|
{
|
|
return tr("Perform a mathematical operation between two values.");
|
|
}
|
|
|
|
void MathNode::Retranslate()
|
|
{
|
|
super::Retranslate();
|
|
|
|
SetInputName(kMethodIn, tr("Method"));
|
|
SetInputName(kParamAIn, tr("Value"));
|
|
SetInputName(kParamBIn, tr("Value"));
|
|
|
|
QStringList operations = { GetOperationName(kOpAdd),
|
|
GetOperationName(kOpSubtract),
|
|
GetOperationName(kOpMultiply),
|
|
GetOperationName(kOpDivide),
|
|
QString(),
|
|
GetOperationName(kOpPower) };
|
|
|
|
SetComboBoxStrings(kMethodIn, operations);
|
|
}
|
|
|
|
ShaderCode MathNode::GetShaderCode(const ShaderRequest &request) const
|
|
{
|
|
return GetShaderCodeInternal(request.id, kParamAIn, kParamBIn);
|
|
}
|
|
|
|
void MathNode::Value(const NodeValueRow &value, const NodeGlobals &globals,
|
|
NodeValueTable *table) const
|
|
{
|
|
// Auto-detect what values to operate with
|
|
// FIXME: Very inefficient
|
|
NodeValueTable at, bt;
|
|
at.Push(value[kParamAIn]);
|
|
bt.Push(value[kParamBIn]);
|
|
PairingCalculator calc(at, bt);
|
|
|
|
// Do nothing if no pairing was found
|
|
if (!calc.FoundMostLikelyPairing()) {
|
|
return;
|
|
}
|
|
|
|
return ValueInternal(GetOperation(), calc.GetMostLikelyPairing(), kParamAIn,
|
|
calc.GetMostLikelyValueA(), kParamBIn,
|
|
calc.GetMostLikelyValueB(), globals, table);
|
|
}
|
|
|
|
void MathNode::ProcessSamples(const NodeValueRow &values,
|
|
const SampleBuffer &input, SampleBuffer &output,
|
|
int index) const
|
|
{
|
|
return ProcessSamplesInternal(values, GetOperation(), kParamAIn, kParamBIn,
|
|
input, output, index);
|
|
}
|
|
|
|
}
|