99 lines
2.6 KiB
C++
99 lines
2.6 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2021 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 "shapenode.h"
|
|
|
|
namespace olive {
|
|
|
|
#define super ShapeNodeBase
|
|
|
|
QString ShapeNode::kTypeInput = QStringLiteral("type_in");
|
|
|
|
ShapeNode::ShapeNode()
|
|
{
|
|
PrependInput(kTypeInput, NodeValue::kCombo);
|
|
}
|
|
|
|
QString ShapeNode::Name() const
|
|
{
|
|
return tr("Shape");
|
|
}
|
|
|
|
QString ShapeNode::id() const
|
|
{
|
|
return QStringLiteral("org.olivevideoeditor.Olive.shape");
|
|
}
|
|
|
|
QVector<Node::CategoryID> ShapeNode::Category() const
|
|
{
|
|
return {kCategoryGenerator};
|
|
}
|
|
|
|
QString ShapeNode::Description() const
|
|
{
|
|
return tr("Generate a 2D primitive shape.");
|
|
}
|
|
|
|
void ShapeNode::Retranslate()
|
|
{
|
|
super::Retranslate();
|
|
|
|
SetInputName(kTypeInput, tr("Type"));
|
|
|
|
// Coordinate with Type enum
|
|
SetComboBoxStrings(kTypeInput, {tr("Rectangle"), tr("Ellipse")});
|
|
}
|
|
|
|
ShaderCode ShapeNode::GetShaderCode(const QString &shader_id) const
|
|
{
|
|
if (shader_id == QStringLiteral("shape")) {
|
|
return ShaderCode(FileFunctions::ReadFileAsString(QStringLiteral(":/shaders/shape.frag")));
|
|
} else {
|
|
return super::GetShaderCode(shader_id);
|
|
}
|
|
}
|
|
|
|
void ShapeNode::Value(const NodeValueRow &value, const NodeGlobals &globals, NodeValueTable *table) const
|
|
{
|
|
ShaderJob job;
|
|
|
|
job.InsertValue(value);
|
|
job.InsertValue(QStringLiteral("resolution_in"), NodeValue(NodeValue::kVec2, globals.resolution(), this));
|
|
job.SetAlphaChannelRequired(GenerateJob::kAlphaForceOn);
|
|
job.SetShaderID(QStringLiteral("shape"));
|
|
|
|
if (!value[kBaseInput].data().isNull()) {
|
|
// Push as merge node
|
|
ShaderJob merge;
|
|
|
|
merge.SetShaderID(QStringLiteral("mrg"));
|
|
merge.InsertValue(MergeNode::kBaseIn, value[kBaseInput]);
|
|
merge.InsertValue(MergeNode::kBlendIn, NodeValue(NodeValue::kTexture, QVariant::fromValue(job), this));
|
|
merge.SetAlphaChannelRequired(GenerateJob::kAlphaForceOn);
|
|
|
|
table->Push(NodeValue::kTexture, QVariant::fromValue(merge), this);
|
|
} else {
|
|
// Just push generate job
|
|
table->Push(NodeValue::kTexture, QVariant::fromValue(job), this);
|
|
}
|
|
}
|
|
|
|
}
|