151 lines
3.8 KiB
C++
151 lines
3.8 KiB
C++
/*
|
|
* color_ramp.h
|
|
*/
|
|
|
|
#include "color_ramp.h"
|
|
|
|
//setter and getter names for property serialization
|
|
#define COLOR_RAMP_GET_OFFSETS "get_offsets"
|
|
#define COLOR_RAMP_GET_COLORS "get_colors"
|
|
#define COLOR_RAMP_SET_OFFSETS "set_offsets"
|
|
#define COLOR_RAMP_SET_COLORS "set_colors"
|
|
|
|
ColorRamp::ColorRamp() {
|
|
//Set initial color ramp transition from black to white
|
|
points.resize(2);
|
|
points[0].color = Color(0,0,0,1);
|
|
points[0].offset = 0;
|
|
points[1].color = Color(1,1,1,1);
|
|
points[1].offset = 1;
|
|
is_sorted = true;
|
|
}
|
|
|
|
ColorRamp::~ColorRamp() {
|
|
|
|
}
|
|
|
|
void ColorRamp::_bind_methods() {
|
|
|
|
|
|
|
|
|
|
|
|
ObjectTypeDB::bind_method(_MD("add_point","offset","color"),&ColorRamp::add_point);
|
|
ObjectTypeDB::bind_method(_MD("remove_point","offset","color"),&ColorRamp::remove_point);
|
|
|
|
ObjectTypeDB::bind_method(_MD("set_offset","point","offset"),&ColorRamp::set_offset);
|
|
ObjectTypeDB::bind_method(_MD("get_offset","point"),&ColorRamp::get_offset);
|
|
|
|
ObjectTypeDB::bind_method(_MD("set_color","point","color"),&ColorRamp::set_color);
|
|
ObjectTypeDB::bind_method(_MD("get_color","point"),&ColorRamp::get_color);
|
|
|
|
ObjectTypeDB::bind_method(_MD("interpolate","offset"),&ColorRamp::get_color_at_offset);
|
|
|
|
ObjectTypeDB::bind_method(_MD("get_point_count"),&ColorRamp::get_points_count);
|
|
|
|
ObjectTypeDB::bind_method(_MD(COLOR_RAMP_SET_OFFSETS,"offsets"),&ColorRamp::set_offsets);
|
|
ObjectTypeDB::bind_method(_MD(COLOR_RAMP_GET_OFFSETS),&ColorRamp::get_offsets);
|
|
|
|
ObjectTypeDB::bind_method(_MD(COLOR_RAMP_SET_COLORS,"colors"),&ColorRamp::set_colors);
|
|
ObjectTypeDB::bind_method(_MD(COLOR_RAMP_GET_COLORS),&ColorRamp::get_colors);
|
|
|
|
ADD_PROPERTY( PropertyInfo(Variant::REAL,"offsets"),_SCS(COLOR_RAMP_SET_OFFSETS),_SCS(COLOR_RAMP_GET_OFFSETS) );
|
|
ADD_PROPERTY( PropertyInfo(Variant::REAL,"colors"),_SCS(COLOR_RAMP_SET_COLORS),_SCS(COLOR_RAMP_GET_COLORS) );
|
|
}
|
|
|
|
Vector<float> ColorRamp::get_offsets() const {
|
|
Vector<float> offsets;
|
|
offsets.resize(points.size());
|
|
for(int i = 0; i < points.size(); i++)
|
|
{
|
|
offsets[i] = points[i].offset;
|
|
}
|
|
return offsets;
|
|
}
|
|
|
|
Vector<Color> ColorRamp::get_colors() const {
|
|
Vector<Color> colors;
|
|
colors.resize(points.size());
|
|
for(int i = 0; i < points.size(); i++)
|
|
{
|
|
colors[i] = points[i].color;
|
|
}
|
|
return colors;
|
|
}
|
|
|
|
void ColorRamp::set_offsets(const Vector<float>& p_offsets) {
|
|
points.resize(p_offsets.size());
|
|
for(int i = 0; i < points.size(); i++)
|
|
{
|
|
points[i].offset = p_offsets[i];
|
|
}
|
|
is_sorted = false;
|
|
}
|
|
|
|
void ColorRamp::set_colors(const Vector<Color>& p_colors) {
|
|
if(points.size()<p_colors.size())
|
|
is_sorted = false;
|
|
points.resize(p_colors.size());
|
|
for(int i = 0; i < points.size(); i++)
|
|
{
|
|
points[i].color = p_colors[i];
|
|
}
|
|
}
|
|
|
|
Vector<ColorRamp::Point>& ColorRamp::get_points() {
|
|
return points;
|
|
}
|
|
|
|
void ColorRamp::add_point(float p_offset, const Color& p_color) {
|
|
|
|
Point p;
|
|
p.offset=p_offset;
|
|
p.color=p_color;
|
|
is_sorted=false;
|
|
points.push_back(p);
|
|
|
|
}
|
|
|
|
void ColorRamp::remove_point(int p_index) {
|
|
|
|
ERR_FAIL_INDEX(p_index,points.size());
|
|
ERR_FAIL_COND(points.size()<=2);
|
|
points.remove(p_index);
|
|
}
|
|
|
|
void ColorRamp::set_points(Vector<ColorRamp::Point>& p_points) {
|
|
points = p_points;
|
|
is_sorted = false;
|
|
}
|
|
|
|
void ColorRamp::set_offset(int pos, const float offset) {
|
|
if(points.size() <= pos)
|
|
points.resize(pos + 1);
|
|
points[pos].offset = offset;
|
|
is_sorted = false;
|
|
}
|
|
|
|
float ColorRamp::get_offset(int pos) const {
|
|
if(points.size() > pos)
|
|
return points[pos].offset;
|
|
return 0; //TODO: Maybe throw some error instead?
|
|
}
|
|
|
|
void ColorRamp::set_color(int pos, const Color& color) {
|
|
if(points.size() <= pos)
|
|
{
|
|
points.resize(pos + 1);
|
|
is_sorted = false;
|
|
}
|
|
points[pos].color = color;
|
|
}
|
|
|
|
Color ColorRamp::get_color(int pos) const {
|
|
if(points.size() > pos)
|
|
return points[pos].color;
|
|
return Color(0,0,0,1); //TODO: Maybe throw some error instead?
|
|
}
|
|
|
|
int ColorRamp::get_points_count() const {
|
|
return points.size();
|
|
}
|