]> git.sesse.net Git - casparcg/blobdiff - core/producer/scene/expression_parser.cpp
[scene] #563 Added some string functions to expression language
[casparcg] / core / producer / scene / expression_parser.cpp
index b4d07cc3c5d7dc81f7710694175df18369e1b71d..27156db264b63574428e00283938487224ece864 100644 (file)
 #include <functional>
 #include <typeinfo>
 #include <cstdint>
+#include <cmath>
 
 #include <boost/any.hpp>
+#include <boost/locale.hpp>
 
 #include <common/log.h>
 #include <common/except.h>
 #include <common/utf.h>
+#include <common/tweener.h>
 
 namespace caspar { namespace core { namespace scene {
 
@@ -69,6 +72,21 @@ std::wstring at_position(
                        + L" in " + str;
 }
 
+boost::any as_binding(const boost::any& value);
+
+template<typename T>
+binding<T> require(const boost::any& value)
+{
+       auto b = as_binding(value);
+
+       if (is<binding<T>>(b))
+               return as<binding<T>>(b);
+       else
+               CASPAR_THROW_EXCEPTION(user_error() << msg_info(
+                       L"Required binding of type " + u16(typeid(T).name())
+                       + L" but got " + u16(value.type().name())));
+}
+
 boost::any parse_expression(
                std::wstring::const_iterator& cursor,
                const std::wstring& str,
@@ -85,13 +103,171 @@ boost::any parse_parenthesis(
 
        auto expr = parse_expression(cursor, str, var_repo);
 
-       if (*cursor++ != L')')
+       if (next_non_whitespace(cursor, str, L"Expected )") != L')')
                CASPAR_THROW_EXCEPTION(user_error()
                                << msg_info(L"Expected )" + at_position(cursor, str)));
 
+       ++cursor;
+
        return expr;
 }
 
+boost::any create_animate_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 3)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"animate() function requires three parameters: to_animate, duration, tweener"));
+
+       auto to_animate         = require<double>(params.at(0));
+       auto frame_counter      = var_repo(L"frame").as<double>();
+       auto duration           = require<double>(params.at(1));
+       auto tw                         = require<std::wstring>(params.at(2)).transformed([](const std::wstring& s) { return tweener(s); });
+
+       return to_animate.animated(frame_counter, duration, tw);
+}
+
+boost::any create_sin_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"sin() function requires one parameters: angle"));
+
+       auto angle = require<double>(params.at(0));
+
+       return angle.transformed([](double a) { return std::sin(a); });
+}
+
+boost::any create_cos_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"cos() function requires one parameters: angle"));
+
+       auto angle = require<double>(params.at(0));
+
+       return angle.transformed([](double a) { return std::cos(a); });
+}
+
+boost::any create_abs_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"abs() function requires one parameters: value"));
+
+       auto val = require<double>(params.at(0));
+
+       return val.transformed([](double v) { return std::abs(v); });
+}
+
+boost::any create_floor_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"floor() function requires one parameters: value"));
+
+       auto val = require<double>(params.at(0));
+
+       return val.transformed([](double v) { return std::floor(v); });
+}
+
+std::locale create_utf_locale()
+{
+       boost::locale::generator gen;
+       gen.categories(boost::locale::codepage_facet);
+       gen.categories(boost::locale::convert_facet);
+
+       return gen("");
+}
+
+boost::any create_to_lower_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"to_lower() function requires one parameters: str"));
+
+       auto str        = require<std::wstring>(params.at(0));
+       auto locale     = create_utf_locale();
+
+       return str.transformed([=](std::wstring v) { return boost::locale::to_lower(v, locale); });
+}
+
+boost::any create_to_upper_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"to_upper() function requires one parameters: str"));
+
+       auto str        = require<std::wstring>(params.at(0));
+       auto locale     = create_utf_locale();
+
+       return str.transformed([=](std::wstring v) { return boost::locale::to_upper(v, locale); });
+}
+
+boost::any create_length_function(const std::vector<boost::any>& params, const variable_repository& var_repo)
+{
+       if (params.size() != 1)
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"length() function requires one parameters: str"));
+
+       auto str = require<std::wstring>(params.at(0));
+
+       return str.transformed([](std::wstring v) { return static_cast<double>(v.length()); });
+}
+
+boost::any parse_function(
+               const std::wstring& function_name,
+               std::wstring::const_iterator& cursor,
+               const std::wstring& str,
+               const variable_repository& var_repo)
+{
+       static std::map<std::wstring, std::function<boost::any (const std::vector<boost::any>& params, const variable_repository& var_repo)>> FUNCTIONS
+       {
+               { L"animate",   create_animate_function },
+               { L"sin",               create_sin_function },
+               { L"cos",               create_cos_function },
+               { L"abs",               create_abs_function },
+               { L"floor",             create_floor_function },
+               { L"to_lower",  create_to_lower_function },
+               { L"to_upper",  create_to_upper_function },
+               { L"length",    create_length_function }
+       };
+
+       auto function = FUNCTIONS.find(function_name);
+
+       if (function == FUNCTIONS.end())
+               CASPAR_THROW_EXCEPTION(user_error()
+                               << msg_info(function_name + L"() is an unknown function" + at_position(cursor, str)));
+
+       if (*cursor++ != L'(')
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"Expected (" + at_position(cursor, str)));
+
+       std::vector<boost::any> params;
+
+       while (cursor != str.end())
+       {
+               params.push_back(parse_expression(cursor, str, var_repo));
+
+               auto next = next_non_whitespace(cursor, str, L"Expected , or )");
+
+               if (next == L')')
+                       break;
+               else if (next != L',')
+                       CASPAR_THROW_EXCEPTION(user_error()
+                               << msg_info(L"Expected ) or ," + at_position(cursor, str)));
+
+               ++cursor;
+       }
+
+       if (next_non_whitespace(cursor, str, L"Expected , or )") != L')')
+               CASPAR_THROW_EXCEPTION(user_error()
+                       << msg_info(L"Expected ) " + at_position(cursor, str)));
+
+       ++cursor;
+
+       return function->second(params, var_repo);
+}
+
 double parse_constant(
                std::wstring::const_iterator& cursor, const std::wstring& str)
 {
@@ -184,7 +360,8 @@ boost::any parse_variable(
                if (ch == L'.'
                                || ch == L'_'
                                || (ch >= L'a' && ch <= L'z')
-                               || (ch >= L'A' && ch <= L'Z'))
+                               || (ch >= L'A' && ch <= L'Z')
+                               || (variable_name.length() > 0 && ch >= L'0' && ch <= L'9'))
                        variable_name += ch;
                else
                        break;
@@ -192,6 +369,9 @@ boost::any parse_variable(
                ++cursor;
        }
 
+       if (cursor != str.end() && *cursor == L'(')
+               return variable_name;
+
        if (variable_name == L"true")
                return true;
        else if (variable_name == L"false")
@@ -406,19 +586,6 @@ boost::any as_binding(const boost::any& value)
                                L"Couldn't detect type of " + u16(value.type().name())));
 }
 
-template<typename T>
-binding<T> require(const boost::any& value)
-{
-       auto b = as_binding(value);
-
-       if (is<binding<T>>(b))
-               return as<binding<T>>(b);
-       else
-               CASPAR_THROW_EXCEPTION(user_error() << msg_info(
-                               L"Required binding of type " + u16(typeid(T).name())
-                               + L" but got " + u16(value.type().name())));
-}
-
 boost::any negative(const boost::any& to_create_negative_of)
 {
        return -require<double>(to_create_negative_of);
@@ -547,7 +714,7 @@ boost::any ternary(
        auto cond = require<bool>(condition);
        auto t = as_binding(true_value);
        auto f = as_binding(false_value);
-       
+
        // double
        if (is<binding<double>>(t) && is<binding<double>>(f))
                return ternary(cond, as<binding<double>>(t), as<binding<double>>(f));
@@ -697,9 +864,17 @@ boost::any parse_expression(
                        tokens.push_back(parse_string_literal(cursor, str));
                        break;
                case L'(':
-                       tokens.push_back(parse_parenthesis(cursor, str, var_repo));
+                       if (!tokens.empty() && is<std::wstring>(tokens.back()))
+                       {
+                               auto function_name = as<std::wstring>(tokens.back());
+                               tokens.pop_back();
+                               tokens.push_back(parse_function(function_name, cursor, str, var_repo));
+                       }
+                       else
+                               tokens.push_back(parse_parenthesis(cursor, str, var_repo));
                        break;
                case L')':
+               case L',':
                        stop = true;
                        break;
                default:
@@ -713,7 +888,7 @@ boost::any parse_expression(
 
        if (tokens.empty())
                CASPAR_THROW_EXCEPTION(user_error()
-                               << msg_info(L"Expected expression"));
+                               << msg_info(L"Expected expression" + at_position(cursor, str)));
 
        int precedence = 1;