/*
 * Copyright (C) 2025 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <gtest/gtest.h>

#include "eiff/LineDictionary.h"
#include <private/filament/LineDictionaryUtils.h>

#include <string>

using namespace filamat;
using namespace ::filament::backend;

TEST(LineDictionary, splitString) {
    LineDictionary dictionary;
    const std::string text = "first line hp_copy_123456 second line";
    dictionary.addText(ShaderStage::FRAGMENT, text);
    EXPECT_EQ(dictionary.size(), 4);
    EXPECT_EQ(dictionary[0], "first line ");
    EXPECT_EQ(dictionary[1], "hp_copy");
    EXPECT_EQ(dictionary[2], "123456"); // Bypassing fails because 123456 > 16383
    EXPECT_EQ(dictionary[3], " second line");
}

TEST(LineDictionary, Empty) {
    LineDictionary const dictionary;
    EXPECT_TRUE(dictionary.empty());
    EXPECT_EQ(dictionary.size(), 0);
}

TEST(LineDictionary, AddTextSimple) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "Hello world\n");
    EXPECT_FALSE(dictionary.empty());
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "Hello world\n");
}

TEST(LineDictionary, AddTextMultipleLines) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "First line\nSecond line\n");
    EXPECT_EQ(dictionary.size(), 2);
    EXPECT_EQ(dictionary[0], "First line\n");
    EXPECT_EQ(dictionary[1], "Second line\n");
}

TEST(LineDictionary, AddTextDuplicateLines) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "Same line\nSame line\n");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "Same line\n");
}

TEST(LineDictionary, SplitLogicNoPattern) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "A simple line with no patterns.");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "A simple line with no patterns.");
}

TEST(LineDictionary, SplitLogicHpPattern) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "some_var = hp_copy_123;");
    EXPECT_EQ(dictionary.size(), 3);
    EXPECT_EQ(dictionary[0], "some_var = ");
    EXPECT_EQ(dictionary[1], "hp_copy");
    EXPECT_EQ(dictionary[2], ";"); // 123 is bypassed!
}

TEST(LineDictionary, SplitLogicMpPattern) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "another_var = mp_copy_4567;");
    EXPECT_EQ(dictionary.size(), 3);
    EXPECT_EQ(dictionary[0], "another_var = ");
    EXPECT_EQ(dictionary[1], "mp_copy");
    EXPECT_EQ(dictionary[2], ";");
}

TEST(LineDictionary, SplitLogicUnderscorePattern) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "var_1 = 0;");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "var_1 = 0;"); // Underscore preceded by a word char is not considered a pattern boundary
}

TEST(LineDictionary, SplitLogicMultiplePatterns) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1 mp_copy_2 _3");
    EXPECT_EQ(dictionary.size(), 3);
    EXPECT_EQ(dictionary[0], "hp_copy");
    EXPECT_EQ(dictionary[1], " ");
    EXPECT_EQ(dictionary[2], "mp_copy");
}

TEST(LineDictionary, SplitLogicInvalidPattern) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_ a_b_c");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "hp_copy_ a_b_c");
}

TEST(LineDictionary, SplitLogicPatternFollowedByWordChar) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_99rest");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "hp_copy_99rest"); // Invalid word boundary on right side
}

TEST(LineDictionary, SplitLogicPatternPrecededByWordChar) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "rest_of_it_hp_copy_99");
    EXPECT_EQ(dictionary.size(), 1);
    EXPECT_EQ(dictionary[0], "rest_of_it_hp_copy_99"); // Preceded by word char
}

TEST(LineDictionary, SplitLogicPatternNotFollowedByWordChar) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_99;");
    EXPECT_EQ(dictionary.size(), 2);
    EXPECT_EQ(dictionary[0], "hp_copy");
    EXPECT_EQ(dictionary[1], ";");
}

TEST(LineDictionary, AddEmptyText) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "");
    EXPECT_TRUE(dictionary.empty());
}

TEST(LineDictionary, GetIndicesMultiple) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "A _1 B _2");
    dictionary.addText(ShaderStage::FRAGMENT, "A _1");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("A _1"); // String is in dictionary
    ASSERT_EQ(indices.size(), 2);
    EXPECT_EQ(indices[0], 0); // "A "
    EXPECT_EQ(indices[1], filament::LineDictionaryUtils::DICTIONARY_NUMERIC_FLAG); // 1
    ASSERT_EQ(numerics.size(), 1);
    EXPECT_EQ(numerics[0], 1);
}

TEST(LineDictionary, GetIndicesMultiplePatternsInARow) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1 hp_copy_2");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("hp_copy_1 hp_copy_2");
    ASSERT_EQ(indices.size(), 5);
    EXPECT_EQ(indices[0], 0); // hp_copy
    EXPECT_EQ(indices[1], filament::LineDictionaryUtils::DICTIONARY_NUMERIC_FLAG);
    EXPECT_EQ(indices[2], 1); // " "
    EXPECT_EQ(indices[3], 0); // hp_copy
    EXPECT_EQ(indices[4], filament::LineDictionaryUtils::DICTIONARY_NUMERIC_FLAG);
    ASSERT_EQ(numerics.size(), 2);
    EXPECT_EQ(numerics[0], 1);
    EXPECT_EQ(numerics[1], 2);
}

TEST(LineDictionary, GetIndicesSamePatternMultipleTimes) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1 hp_copy_1");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("hp_copy_1 hp_copy_1");
    ASSERT_EQ(indices.size(), 5);
    EXPECT_EQ(indices[0], 0); // hp_copy
    EXPECT_EQ(indices[1], filament::LineDictionaryUtils::DICTIONARY_NUMERIC_FLAG);
    EXPECT_EQ(indices[2], 1); // " "
    EXPECT_EQ(indices[3], 0); // hp_copy
    EXPECT_EQ(indices[4], filament::LineDictionaryUtils::DICTIONARY_NUMERIC_FLAG);
    ASSERT_EQ(numerics.size(), 2);
    EXPECT_EQ(numerics[0], 1);
    EXPECT_EQ(numerics[1], 1);
}

TEST(LineDictionary, GetIndicesWithAdjacentPatterns) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1hp_copy_2");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("hp_copy_1hp_copy_2");
    ASSERT_EQ(indices.size(), 1);
    EXPECT_EQ(indices[0], 0);
}

TEST(LineDictionary, GetIndicesWithAdjacentPatternsNotInDictionary) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1");
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_2");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("hp_copy_1hp_copy_2");
    ASSERT_EQ(indices.size(), 0);
}

TEST(LineDictionary, GetIndicesWithMixedContent) {
    LineDictionary dictionary;
    dictionary.addText(ShaderStage::FRAGMENT, "hp_copy_1");
    dictionary.addText(ShaderStage::FRAGMENT, " ");
    dictionary.addText(ShaderStage::FRAGMENT, "mp_copy_2");
    dictionary.resolve();
    auto const [indices, numerics] = dictionary.tokenize("prefix hp_copy_1 mp_copy_2 suffix");
    ASSERT_EQ(indices.size(), 0);
}