| // Ceres Solver - A fast non-linear least squares minimizer |
| // Copyright 2019 Google Inc. All rights reserved. |
| // http://code.google.com/p/ceres-solver/ |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are met: |
| // |
| // * Redistributions of source code must retain the above copyright notice, |
| // this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // * Neither the name of Google Inc. nor the names of its contributors may be |
| // used to endorse or promote products derived from this software without |
| // specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
| // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| // POSSIBILITY OF SUCH DAMAGE. |
| // |
| // Author: darius.rueckert@fau.de (Darius Rueckert) |
| // |
| // This file tests the Expression class. For each member function one test is |
| // included here. |
| // |
| #include "ceres/codegen/internal/expression.h" |
| #include "gtest/gtest.h" |
| |
| namespace ceres { |
| namespace internal { |
| |
| TEST(Expression, ConstructorAndAccessors) { |
| Expression expr(ExpressionType::LOGICAL_NEGATION, |
| ExpressionReturnType::BOOLEAN, |
| 12345, |
| {1, 5, 8, 10}, |
| "TestConstructor", |
| 57.25); |
| EXPECT_EQ(expr.type(), ExpressionType::LOGICAL_NEGATION); |
| EXPECT_EQ(expr.return_type(), ExpressionReturnType::BOOLEAN); |
| EXPECT_EQ(expr.lhs_id(), 12345); |
| EXPECT_EQ(expr.arguments(), std::vector<ExpressionId>({1, 5, 8, 10})); |
| EXPECT_EQ(expr.name(), "TestConstructor"); |
| EXPECT_EQ(expr.value(), 57.25); |
| } |
| |
| TEST(Expression, CreateFunctions) { |
| // The default constructor creates a NOP! |
| EXPECT_EQ(Expression(), |
| Expression(ExpressionType::NOP, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {}, |
| "", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateCompileTimeConstant(72), |
| Expression(ExpressionType::COMPILE_TIME_CONSTANT, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {}, |
| "", |
| 72)); |
| |
| EXPECT_EQ(Expression::CreateInputAssignment("arguments[0][0]"), |
| Expression(ExpressionType::INPUT_ASSIGNMENT, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {}, |
| "arguments[0][0]", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateOutputAssignment(ExpressionId(5), "residuals[3]"), |
| Expression(ExpressionType::OUTPUT_ASSIGNMENT, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {5}, |
| "residuals[3]", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateAssignment(ExpressionId(3), ExpressionId(5)), |
| Expression(ExpressionType::ASSIGNMENT, |
| ExpressionReturnType::SCALAR, |
| 3, |
| {5}, |
| "", |
| 0)); |
| |
| EXPECT_EQ( |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)), |
| Expression(ExpressionType::BINARY_ARITHMETIC, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {3, 5}, |
| "+", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateUnaryArithmetic("-", ExpressionId(5)), |
| Expression(ExpressionType::UNARY_ARITHMETIC, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {5}, |
| "-", |
| 0)); |
| |
| EXPECT_EQ( |
| Expression::CreateBinaryCompare("<", ExpressionId(3), ExpressionId(5)), |
| Expression(ExpressionType::BINARY_COMPARISON, |
| ExpressionReturnType::BOOLEAN, |
| kInvalidExpressionId, |
| {3, 5}, |
| "<", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateLogicalNegation(ExpressionId(5)), |
| Expression(ExpressionType::LOGICAL_NEGATION, |
| ExpressionReturnType::BOOLEAN, |
| kInvalidExpressionId, |
| {5}, |
| "", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateScalarFunctionCall( |
| "pow", {ExpressionId(3), ExpressionId(5)}), |
| Expression(ExpressionType::FUNCTION_CALL, |
| ExpressionReturnType::SCALAR, |
| kInvalidExpressionId, |
| {3, 5}, |
| "pow", |
| 0)); |
| |
| EXPECT_EQ( |
| Expression::CreateLogicalFunctionCall("isfinite", {ExpressionId(3)}), |
| Expression(ExpressionType::FUNCTION_CALL, |
| ExpressionReturnType::BOOLEAN, |
| kInvalidExpressionId, |
| {3}, |
| "isfinite", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateIf(ExpressionId(5)), |
| Expression(ExpressionType::IF, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {5}, |
| "", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateElse(), |
| Expression(ExpressionType::ELSE, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {}, |
| "", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateEndIf(), |
| Expression(ExpressionType::ENDIF, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {}, |
| "", |
| 0)); |
| |
| EXPECT_EQ(Expression::CreateComment("Test"), |
| Expression(ExpressionType::COMMENT, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {}, |
| "Test", |
| 0)); |
| } |
| |
| TEST(Expression, IsArithmeticExpression) { |
| ASSERT_TRUE( |
| Expression::CreateCompileTimeConstant(5).IsArithmeticExpression()); |
| ASSERT_TRUE(Expression::CreateScalarFunctionCall("pow", {3, 5}) |
| .IsArithmeticExpression()); |
| // Logical expression are also arithmetic! |
| ASSERT_TRUE( |
| Expression::CreateBinaryCompare("<", 3, 5).IsArithmeticExpression()); |
| ASSERT_FALSE(Expression::CreateIf(5).IsArithmeticExpression()); |
| ASSERT_FALSE(Expression::CreateEndIf().IsArithmeticExpression()); |
| ASSERT_FALSE(Expression().IsArithmeticExpression()); |
| } |
| |
| TEST(Expression, IsControlExpression) { |
| // In the current implementation this is the exact opposite of |
| // IsArithmeticExpression. |
| ASSERT_FALSE(Expression::CreateCompileTimeConstant(5).IsControlExpression()); |
| ASSERT_FALSE(Expression::CreateScalarFunctionCall("pow", {3, 5}) |
| .IsControlExpression()); |
| ASSERT_FALSE( |
| Expression::CreateBinaryCompare("<", 3, 5).IsControlExpression()); |
| ASSERT_TRUE(Expression::CreateIf(5).IsControlExpression()); |
| ASSERT_TRUE(Expression::CreateEndIf().IsControlExpression()); |
| ASSERT_TRUE(Expression::CreateComment("Test").IsControlExpression()); |
| ASSERT_TRUE(Expression().IsControlExpression()); |
| } |
| |
| TEST(Expression, IsCompileTimeConstantAndEqualTo) { |
| ASSERT_TRUE( |
| Expression::CreateCompileTimeConstant(5).IsCompileTimeConstantAndEqualTo( |
| 5)); |
| ASSERT_FALSE( |
| Expression::CreateCompileTimeConstant(3).IsCompileTimeConstantAndEqualTo( |
| 5)); |
| ASSERT_FALSE(Expression::CreateBinaryCompare("<", 3, 5) |
| .IsCompileTimeConstantAndEqualTo(5)); |
| } |
| |
| TEST(Expression, IsReplaceableBy) { |
| // Create 2 identical expression |
| auto expr1 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| auto expr2 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| // They are idendical and of course replaceable |
| ASSERT_EQ(expr1, expr2); |
| ASSERT_EQ(expr2, expr1); |
| ASSERT_TRUE(expr1.IsReplaceableBy(expr2)); |
| ASSERT_TRUE(expr2.IsReplaceableBy(expr1)); |
| |
| // Give them different left hand sides |
| expr1.set_lhs_id(72); |
| expr2.set_lhs_id(42); |
| |
| // v_72 = v_3 + v_5 |
| // v_42 = v_3 + v_5 |
| // -> They should be replaceable by each other |
| |
| ASSERT_NE(expr1, expr2); |
| ASSERT_NE(expr2, expr1); |
| |
| ASSERT_TRUE(expr1.IsReplaceableBy(expr2)); |
| ASSERT_TRUE(expr2.IsReplaceableBy(expr1)); |
| |
| // A slightly differnt expression with the argument flipped |
| auto expr3 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(5), ExpressionId(3)); |
| |
| ASSERT_NE(expr1, expr3); |
| ASSERT_FALSE(expr1.IsReplaceableBy(expr3)); |
| } |
| |
| TEST(Expression, Replace) { |
| auto expr1 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| expr1.set_lhs_id(13); |
| |
| auto expr2 = |
| Expression::CreateAssignment(kInvalidExpressionId, ExpressionId(7)); |
| |
| // We replace the arithmetic expr1 by an assignment from the variable 7. This |
| // is the typical usecase in subexpression elimination. |
| expr1.Replace(expr2); |
| |
| // expr1 should now be an assignment from 7 to 13 |
| EXPECT_EQ(expr1, |
| Expression(ExpressionType::ASSIGNMENT, |
| ExpressionReturnType::SCALAR, |
| 13, |
| {7}, |
| "", |
| 0)); |
| } |
| |
| TEST(Expression, DirectlyDependsOn) { |
| auto expr1 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| ASSERT_TRUE(expr1.DirectlyDependsOn(ExpressionId(3))); |
| ASSERT_TRUE(expr1.DirectlyDependsOn(ExpressionId(5))); |
| ASSERT_FALSE(expr1.DirectlyDependsOn(ExpressionId(kInvalidExpressionId))); |
| ASSERT_FALSE(expr1.DirectlyDependsOn(ExpressionId(42))); |
| } |
| TEST(Expression, MakeNop) { |
| auto expr1 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| expr1.MakeNop(); |
| |
| EXPECT_EQ(expr1, |
| Expression(ExpressionType::NOP, |
| ExpressionReturnType::VOID, |
| kInvalidExpressionId, |
| {}, |
| "", |
| 0)); |
| } |
| |
| TEST(Expression, IsSemanticallyEquivalentTo) { |
| // Create 2 identical expression |
| auto expr1 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| auto expr2 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(5)); |
| |
| ASSERT_TRUE(expr1.IsSemanticallyEquivalentTo(expr1)); |
| ASSERT_TRUE(expr1.IsSemanticallyEquivalentTo(expr2)); |
| |
| auto expr3 = |
| Expression::CreateBinaryArithmetic("+", ExpressionId(3), ExpressionId(8)); |
| |
| ASSERT_TRUE(expr1.IsSemanticallyEquivalentTo(expr3)); |
| |
| auto expr4 = |
| Expression::CreateBinaryArithmetic("-", ExpressionId(3), ExpressionId(5)); |
| |
| ASSERT_FALSE(expr1.IsSemanticallyEquivalentTo(expr4)); |
| } |
| |
| } // namespace internal |
| } // namespace ceres |