Improve file saving gui

1 files changed, 10 insertions, 3 deletions

diff --git a/src/RoundedRectangle.cpp b/src/RoundedRectangle.cpp
index 6b956e8..4b06ea8 100644
--- a/src/RoundedRectangle.cpp
+++ b/src/RoundedRectangle.cpp
@@ -1,6 +1,6 @@
 #include "../include/RoundedRectangle.hpp"
 #include <SFML/Graphics/Shader.hpp>
-#include <SFML/Graphics/RenderWindow.hpp>
+#include <SFML/Graphics/RenderTarget.hpp>
 #include <assert.h>
 
 namespace QuickMedia {
@@ -30,6 +30,13 @@ namespace QuickMedia {
         set_position(pos);
     }
 
+    void RoundedRectangle::set_color(sf::Color color) {
+        vertices[0].color = color;
+        vertices[1].color = color;
+        vertices[2].color = color;
+        vertices[3].color = color;
+    }
+
     sf::Vector2f RoundedRectangle::get_position() const {
         return pos;
     }
@@ -38,9 +45,9 @@ namespace QuickMedia {
         return size;
     }
 
-    void RoundedRectangle::draw(sf::RenderWindow &window) {
+    void RoundedRectangle::draw(sf::RenderTarget &target) {
         rounded_rectangle_shader->setUniform("resolution", size);
         rounded_rectangle_shader->setUniform("radius", radius);
-        window.draw(vertices, 4, sf::Quads, rounded_rectangle_shader);
+        target.draw(vertices, 4, sf::Quads, rounded_rectangle_shader);
     }
 }
\ No newline at end of file