forked from bartvdbraak/blender
150 lines
3.8 KiB
C++
150 lines
3.8 KiB
C++
|
/*
|
||
|
* Copyright 2011-2016 Blender Foundation
|
||
|
*
|
||
|
* 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 "testing/testing.h"
|
||
|
#include "testing/mock_log.h"
|
||
|
|
||
|
#include "render/graph.h"
|
||
|
#include "render/scene.h"
|
||
|
#include "render/nodes.h"
|
||
|
#include "util/util_logging.h"
|
||
|
#include "util/util_string.h"
|
||
|
#include "util/util_vector.h"
|
||
|
|
||
|
using testing::AnyNumber;
|
||
|
using testing::HasSubstr;
|
||
|
using testing::ScopedMockLog;
|
||
|
using testing::_;
|
||
|
|
||
|
CCL_NAMESPACE_BEGIN
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
template<typename T>
|
||
|
class ShaderNodeBuilder {
|
||
|
public:
|
||
|
ShaderNodeBuilder(const string& name)
|
||
|
: name_(name)
|
||
|
{
|
||
|
node_ = new T();
|
||
|
}
|
||
|
|
||
|
const string& name() const {
|
||
|
return name_;
|
||
|
}
|
||
|
|
||
|
ShaderNode *node() const {
|
||
|
return node_;
|
||
|
}
|
||
|
|
||
|
template<typename V>
|
||
|
ShaderNodeBuilder& set(const string& input_name, V value)
|
||
|
{
|
||
|
ShaderInput *input_socket = node_->input(input_name.c_str());
|
||
|
EXPECT_NE((void*)NULL, input_socket);
|
||
|
input_socket->set(value);
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
string name_;
|
||
|
ShaderNode *node_;
|
||
|
};
|
||
|
|
||
|
class ShaderGraphBuilder {
|
||
|
public:
|
||
|
explicit ShaderGraphBuilder(ShaderGraph *graph)
|
||
|
: graph_(graph)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
ShaderNode *find_node(const string& name)
|
||
|
{
|
||
|
map<string, ShaderNode *>::iterator it = node_map_.find(name);
|
||
|
if(it == node_map_.end()) {
|
||
|
return NULL;
|
||
|
}
|
||
|
return it->second;
|
||
|
}
|
||
|
|
||
|
template<typename T>
|
||
|
ShaderGraphBuilder& add_node(const T& node)
|
||
|
{
|
||
|
EXPECT_EQ(NULL, find_node(node.name()));
|
||
|
graph_->add(node.node());
|
||
|
node_map_[node.name()] = node.node();
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
ShaderGraphBuilder& add_connection(const string& from,
|
||
|
const string& to)
|
||
|
{
|
||
|
vector<string> tokens_from, tokens_to;
|
||
|
string_split(tokens_from, from, "::");
|
||
|
string_split(tokens_to, to, "::");
|
||
|
EXPECT_EQ(2, tokens_from.size());
|
||
|
EXPECT_EQ(2, tokens_to.size());
|
||
|
ShaderNode *node_from = find_node(tokens_from[0]),
|
||
|
*node_to = find_node(tokens_to[0]);
|
||
|
EXPECT_NE((void*)NULL, node_from);
|
||
|
EXPECT_NE((void*)NULL, node_to);
|
||
|
EXPECT_NE(node_from, node_to);
|
||
|
ShaderOutput *socket_from = node_from->output(tokens_from[1].c_str());
|
||
|
ShaderInput *socket_to = node_to->input(tokens_to[1].c_str());
|
||
|
EXPECT_NE((void*)NULL, socket_from);
|
||
|
EXPECT_NE((void*)NULL, socket_to);
|
||
|
graph_->connect(socket_from, socket_to);
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
ShaderGraph *graph_;
|
||
|
map<string, ShaderNode *> node_map_;
|
||
|
};
|
||
|
|
||
|
} // namespace
|
||
|
|
||
|
#define DEFINE_COMMON_VARIABLES(builder_name, mock_log_name) \
|
||
|
util_logging_start(); \
|
||
|
util_logging_verbosity_set(1); \
|
||
|
ScopedMockLog mock_log_name; \
|
||
|
DeviceInfo device_info; \
|
||
|
SceneParams scene_params; \
|
||
|
Scene scene(scene_params, device_info); \
|
||
|
ShaderGraph graph; \
|
||
|
ShaderGraphBuilder builder(&graph); \
|
||
|
|
||
|
TEST(render_graph, constant_fold_rgb_to_bw)
|
||
|
{
|
||
|
DEFINE_COMMON_VARIABLES(builder, log);
|
||
|
|
||
|
EXPECT_CALL(log, Log(_, _, _)).Times(AnyNumber());
|
||
|
EXPECT_CALL(log, Log(google::INFO, _,
|
||
|
HasSubstr("Replacing rgb_to_bw with constant 0.8.")));
|
||
|
|
||
|
builder
|
||
|
.add_node(ShaderNodeBuilder<OutputNode>("OutputNode"))
|
||
|
.add_node(ShaderNodeBuilder<EmissionNode>("EmissionNode"))
|
||
|
.add_node(ShaderNodeBuilder<RGBToBWNode>("RGBToBWNodeNode")
|
||
|
.set("Color", make_float3(0.8f, 0.8f, 0.8f)))
|
||
|
.add_connection("RGBToBWNodeNode::Val", "EmissionNode::Color")
|
||
|
.add_connection("EmissionNode::Emission", "OutputNode::Surface");
|
||
|
|
||
|
graph.finalize(&scene);
|
||
|
}
|
||
|
|
||
|
CCL_NAMESPACE_END
|