text: implement formatting arguments

app/node/generator/text/textv3.cpp

@@ -39,6 +39,9 @@ enum TextVerticalAlign {
 };
 
 const QString TextGeneratorV3::kTextInput = QStringLiteral("text_in");
+const QString TextGeneratorV3::kVerticalAlignmentInput = QStringLiteral("valign_in");
+const QString TextGeneratorV3::kUseArgsInput = QStringLiteral("use_args_in");
+const QString TextGeneratorV3::kArgsInput = QStringLiteral("args_in");
 
 TextGeneratorV3::TextGeneratorV3() :
   ShapeNodeBase(false)
@@ -48,8 +51,16 @@ TextGeneratorV3::TextGeneratorV3() :
 
   SetStandardValue(kSizeInput, QVector2D(400, 300));
 
+  AddInput(kVerticalAlignmentInput, NodeValue::kCombo);
+
+  AddInput(kUseArgsInput, NodeValue::kBoolean, true, InputFlags(kInputFlagHidden | kInputFlagStatic));
+
+  AddInput(kArgsInput, NodeValue::kText, InputFlags(kInputFlagArray));
+
   text_gizmo_ = new TextGizmo(this);
   text_gizmo_->SetInput(NodeInput(this, kTextInput));
+  connect(text_gizmo_, &TextGizmo::Activated, this, &TextGeneratorV3::GizmoActivated);
+  connect(text_gizmo_, &TextGizmo::Deactivated, this, &TextGeneratorV3::GizmoDeactivated);
 }
 
 QString TextGeneratorV3::Name() const
@@ -77,6 +88,9 @@ void TextGeneratorV3::Retranslate()
   super::Retranslate();
 
   SetInputName(kTextInput, tr("Text"));
+  SetInputName(kVerticalAlignmentInput, tr("Vertical Alignment"));
+  SetComboBoxStrings(kVerticalAlignmentInput, {tr("Top"), tr("Middle"), tr("Bottom")});
+  SetInputName(kArgsInput, tr("Arguments"));
 }
 
 void TextGeneratorV3::Value(const NodeValueRow &value, const NodeGlobals &globals, NodeValueTable *table) const
@@ -86,6 +100,21 @@ void TextGeneratorV3::Value(const NodeValueRow &value, const NodeGlobals &global
   job.SetAlphaChannelRequired(GenerateJob::kAlphaForceOn);
   job.SetRequestedFormat(VideoParams::kFormatUnsigned8);
 
+  if (value[kUseArgsInput].toBool()) {
+    auto args = value[kArgsInput].toArray();
+    if (!args.empty()) {
+      QStringList list;
+      list.reserve(args.size());
+      for (int i=0; i<args.size(); i++) {
+        list.append(args[i].toString());
+      }
+
+      NodeValue v = job.Get(kTextInput);
+      v.set_value(FormatString(v.toString(), list));
+      job.Insert(kTextInput, v);
+    }
+  }
+
   // FIXME: Provide user override for this
   job.SetColorspace(project()->color_manager()->GetDefaultInputColorSpace());
 
@@ -124,6 +153,18 @@ void TextGeneratorV3::GenerateFrame(FramePtr frame, const GenerateJob& job) cons
   p.translate(frame->video_params().width()/2, frame->video_params().height()/2);
   p.setClipRect(0, 0, size.x(), size.y());
 
+  switch (static_cast<VerticalAlignment>(job.Get(kVerticalAlignmentInput).toInt())) {
+  case kVAlignTop:
+    // Do nothing
+    break;
+  case kVAlignMiddle:
+    p.translate(0, size.y()/2-text_doc.size().height()/2);
+    break;
+  case kVAlignBottom:
+    p.translate(0, size.y()-text_doc.size().height());
+    break;
+  }
+
   // Ensure default text color is white
   QAbstractTextDocumentLayout::PaintContext ctx;
   ctx.palette.setColor(QPalette::Text, Qt::white);
@@ -140,4 +181,52 @@ void TextGeneratorV3::UpdateGizmoPositions(const NodeValueRow &row, const NodeGl
   text_gizmo_->SetHtml(row[kTextInput].toString());
 }
 
+QString TextGeneratorV3::FormatString(const QString &input, const QStringList &args)
+{
+  QString output;
+  output.reserve(input.size());
+
+  for (int i=0; i<input.size(); i++) {
+    const QChar &this_char = input.at(i);
+
+    if (i < input.size()-1 && this_char == '%') {
+      const QChar &next_char = input.at(i+1);
+      if (next_char == '%') {
+        // Double percent, append a single percent
+        output.append('%');
+        i++;
+      } else if (next_char.isDigit()) {
+        // Find length of number
+        QString num;
+        i++;
+        while (i < input.size() && input.at(i).isDigit()) {
+          num.append(input.at(i));
+          i++;
+        }
+        i--;
+        int index = num.toInt()-1;
+        if (index >= 0 && index < args.size()) {
+          output.append(args.at(index));
+        }
+      } else {
+        output.append(this_char);
+      }
+    } else {
+      output.append(this_char);
+    }
+  }
+
+  return output;
+}
+
+void TextGeneratorV3::GizmoActivated()
+{
+  SetStandardValue(kUseArgsInput, false);
+}
+
+void TextGeneratorV3::GizmoDeactivated()
+{
+  SetStandardValue(kUseArgsInput, true);
+}
+
 }