nodes: implemented more distortion nodes

2022-09-22 18:12:03 -07:00
parent f84cd8644d
commit b553d5afb7
20 changed files with 1051 additions and 3 deletions
@@ -0,0 +1,104 @@
+/***
+
+  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/>.
+
+***/
+
+#include "wavedistortnode.h"
+
+namespace olive {
+
+const QString WaveDistortNode::kTextureInput = QStringLiteral("tex_in");
+const QString WaveDistortNode::kFrequencyInput = QStringLiteral("frequency_in");
+const QString WaveDistortNode::kIntensityInput = QStringLiteral("intensity_in");
+const QString WaveDistortNode::kEvolutionInput = QStringLiteral("evolution_in");
+const QString WaveDistortNode::kVerticalInput = QStringLiteral("vertical_in");
+
+#define super Node
+
+WaveDistortNode::WaveDistortNode()
+{
+  AddInput(kTextureInput, NodeValue::kTexture, InputFlags(kInputFlagNotKeyframable));
+
+  AddInput(kFrequencyInput, NodeValue::kFloat, 10);
+  AddInput(kIntensityInput, NodeValue::kFloat, 10);
+  AddInput(kEvolutionInput, NodeValue::kFloat, 0);
+
+  AddInput(kVerticalInput, NodeValue::kCombo, false);
+
+  SetFlags(kVideoEffect);
+  SetEffectInput(kTextureInput);
+}
+
+QString WaveDistortNode::Name() const
+{
+  return tr("Wave");
+}
+
+QString WaveDistortNode::id() const
+{
+  return QStringLiteral("org.oliveeditor.Olive.wave");
+}
+
+QVector<Node::CategoryID> WaveDistortNode::Category() const
+{
+  return {kCategoryDistort};
+}
+
+QString WaveDistortNode::Description() const
+{
+  return tr("Distorts an image along a sine wave.");
+}
+
+void WaveDistortNode::Retranslate()
+{
+  super::Retranslate();
+
+  SetInputName(kTextureInput, tr("Input"));
+  SetInputName(kFrequencyInput, tr("Frequency"));
+  SetInputName(kIntensityInput, tr("Intensity"));
+  SetInputName(kEvolutionInput, tr("Evolution"));
+  SetInputName(kVerticalInput, tr("Direction"));
+  SetComboBoxStrings(kVerticalInput, {tr("Horizontal"), tr("Vertical")});
+}
+
+ShaderCode WaveDistortNode::GetShaderCode(const ShaderRequest &request) const
+{
+  Q_UNUSED(request)
+  return ShaderCode(FileFunctions::ReadFileAsString(":/shaders/wave.frag"));
+}
+
+void WaveDistortNode::Value(const NodeValueRow &value, const NodeGlobals &globals, NodeValueTable *table) const
+{
+  ShaderJob job;
+
+  job.Insert(value);
+
+  // If there's no texture, no need to run an operation
+  if (job.Get(kTextureInput).toTexture()) {
+    // Only run shader if at least one of flip or flop are selected
+    if (!qIsNull(job.Get(kIntensityInput).toDouble())) {
+      table->Push(NodeValue::kTexture, QVariant::fromValue(job), this);
+    } else {
+      // If we're not flipping or flopping just push the texture
+      table->Push(job.Get(kTextureInput));
+    }
+  }
+
+}
+
+}