nodes: implemented basic text node

2020-06-13 03:01:35 +10:00
parent 5c5f09a20c
commit 469b1e92d0
7 changed files with 212 additions and 1 deletions
@@ -27,6 +27,7 @@
 #include "generator/matrix/matrix.h"
 #include "generator/polygon/polygon.h"
 #include "generator/solid/solid.h"
+#include "generator/text/text.h"
 #include "filter/blur/blur.h"
 #include "filter/stroke/stroke.h"
 #include "input/media/video/video.h"
@@ -201,6 +202,8 @@ Node *NodeFactory::CreateInternal(const NodeFactory::InternalID &id)
    return new MergeNode();
  case kStrokeFilter:
    return new StrokeFilterNode();
+  case kTextGenerator:
+    return new TextGenerator();

  case kInternalNodeCount:
    break;
@@ -49,6 +49,7 @@ public:
    kSolidGenerator,
    kMerge,
    kStrokeFilter,
+    kTextGenerator,

    // Count value
    kInternalNodeCount
@@ -17,6 +17,7 @@
 add_subdirectory(matrix)
 add_subdirectory(polygon)
 add_subdirectory(solid)
+add_subdirectory(text)

 set(OLIVE_SOURCES
  ${OLIVE_SOURCES}
@@ -0,0 +1,22 @@
+# Olive - Non-Linear Video Editor
+# Copyright (C) 2019 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/>.
+
+set(OLIVE_SOURCES
+  ${OLIVE_SOURCES}
+  node/generator/text/text.h
+  node/generator/text/text.cpp
+  PARENT_SCOPE
+)
@@ -0,0 +1,119 @@
+/***
+
+  Olive - Non-Linear Video Editor
+  Copyright (C) 2019 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 "text.h"
+
+#include <QTextDocument>
+
+OLIVE_NAMESPACE_ENTER
+
+TextGenerator::TextGenerator()
+{
+  QString default_str = QStringLiteral("<html><div style='font-size:144pt; text-align: center;'>%1</div></html>").arg(tr("Sample Text"));
+
+  text_input_ = new NodeInput(QStringLiteral("text_in"),
+                              NodeParam::kText,
+                              default_str);
+  AddInput(text_input_);
+
+  color_input_ = new NodeInput(QStringLiteral("color_in"),
+                               NodeParam::kColor,
+                               QVariant::fromValue(Color(1.0f, 1.0f, 1.0)));
+  AddInput(color_input_);
+}
+
+Node *TextGenerator::copy() const
+{
+  return new TextGenerator();
+}
+
+QString TextGenerator::Name() const
+{
+  return tr("Text");
+}
+
+QString TextGenerator::id() const
+{
+  return QStringLiteral("org.olivevideoeditor.Olive.textgenerator");
+}
+
+QList<Node::CategoryID> TextGenerator::Category() const
+{
+  return {kCategoryGenerator};
+}
+
+QString TextGenerator::Description() const
+{
+  return tr("Generate rich text.");
+}
+
+void TextGenerator::Retranslate()
+{
+  text_input_->set_name(tr("Text"));
+  color_input_->set_name(tr("Color"));
+}
+
+NodeValueTable TextGenerator::Value(NodeValueDatabase &value) const
+{
+  GenerateJob job;
+  job.InsertValue(text_input_, value);
+  job.InsertValue(color_input_, value);
+  job.SetAlphaChannelRequired(true);
+
+  NodeValueTable table = value.Merge();
+
+  if (!job.GetValue(text_input_).data().toString().isEmpty()) {
+    table.Push(NodeParam::kGenerateJob, QVariant::fromValue(job), this);
+  }
+
+  return table;
+}
+
+void TextGenerator::GenerateFrame(FramePtr frame, const GenerateJob& job) const
+{
+  // This could probably be more optimized, but for now we use Qt to draw to a QImage.
+  // QImages only support integer pixels and we use float pixels, so what we do here is draw onto
+  // a single-channel QImage (alpha only) and then transplant that alpha channel to our float buffer
+  // with correct float RGB.
+  QImage img(frame->width(), frame->height(), QImage::Format_Grayscale8);
+  img.fill(0);
+
+  QTextDocument text_doc;
+  text_doc.setHtml(job.GetValue(text_input_).data().toString());
+  text_doc.setTextWidth(frame->video_params().width());
+
+  // Draw rich text onto image
+  QPainter p(&img);
+  p.scale(1.0 / frame->video_params().divider(), 1.0 / frame->video_params().divider());
+  text_doc.drawContents(&p);
+
+  // Transplant alpha channel to frame
+  Color rgb = job.GetValue(color_input_).data().value<Color>();
+  for (int x=0; x<frame->width(); x++) {
+    for (int y=0; y<frame->height(); y++) {
+      uchar src_alpha = img.bits()[img.bytesPerLine() * y + x];
+      float alpha = float(src_alpha) / 255.0f;
+
+      frame->set_pixel(x, y, Color(rgb.red() * alpha, rgb.green() * alpha, rgb.blue() * alpha, alpha));
+    }
+  }
+}
+
+OLIVE_NAMESPACE_EXIT
@@ -0,0 +1,55 @@
+/***
+
+  Olive - Non-Linear Video Editor
+  Copyright (C) 2019 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 TEXTGENERATOR_H
+#define TEXTGENERATOR_H
+
+#include "node/node.h"
+
+OLIVE_NAMESPACE_ENTER
+
+class TextGenerator : public Node
+{
+public:
+  TextGenerator();
+
+  virtual Node* copy() const override;
+
+  virtual QString Name() const override;
+  virtual QString id() const override;
+  virtual QList<CategoryID> Category() const override;
+  virtual QString Description() const override;
+
+  virtual void Retranslate() override;
+
+  virtual NodeValueTable Value(NodeValueDatabase& value) const override;
+
+  virtual void GenerateFrame(FramePtr frame, const GenerateJob &job) const override;
+
+private:
+  NodeInput* text_input_;
+
+  NodeInput* color_input_;
+
+};
+
+OLIVE_NAMESPACE_EXIT
+
+#endif // TEXTGENERATOR_H
@@ -124,8 +124,18 @@ void TimelineWidget::AddTool::MouseRelease(TimelineViewMouseEvent *event)
        new NodeEdgeAddCommand(solid->output(), clip->texture_input(), command);
        break;
      }
-      case OLIVE_NAMESPACE::Tool::kAddableBars:
      case OLIVE_NAMESPACE::Tool::kAddableTitle:
+      {
+        Node* text = NodeFactory::CreateFromID(QStringLiteral("org.olivevideoeditor.Olive.textgenerator"));
+
+        new NodeAddCommand(graph,
+                           text,
+                           command);
+
+        new NodeEdgeAddCommand(text->output(), clip->texture_input(), command);
+        break;
+      }
+      case OLIVE_NAMESPACE::Tool::kAddableBars:
      case OLIVE_NAMESPACE::Tool::kAddableTone:
        // Not implemented yet
        qWarning() << "Unimplemented add object:" << Core::instance()->selected_addable_object();