Refactor the rendering code

[netvis.git] / source / port.h

diff --git a/source/port.h b/source/port.h
index ebc9c22d98afee9e50035933a6568c2bc4c8a50d..d0511e96751c5ed6ca095a6a09a3677bb4fc9510 100644 (file)
--- a/source/port.h
+++ b/source/port.h
@@ -10,20 +10,27 @@ Distributed unter the GPL
 
 #include <string>
 #include <msp/gl/color.h>
+#include <msp/gl/mesh.h>
+
+class NetVis;
 
 class Port
 {
 private:
+       NetVis &netvis;
        unsigned number;
        std::string name;
        Msp::GL::Color color;
+       Msp::GL::Mesh mesh;
 
 public:
-       Port(unsigned, const Msp::GL::Color &);
+       Port(NetVis &, unsigned);
 
        unsigned get_number() const { return number; }
        const std::string &get_name() const { return name; }
        const Msp::GL::Color &get_color() const { return color; }
+
+       void render() const;
 };
 
 #endif