1 files changed, 30 insertions, 0 deletions

diff --git a/src/image.cpp b/src/image.cpp
index 384de9a..122df65 100644
--- a/src/image.cpp
+++ b/src/image.cpp
@@ -5,6 +5,7 @@
 #include <stdlib.h>

 #include <string.h>

 #include <png.h>

+#include <math.h>

 

 Image::Image() :

     iWidth( 0 ),

@@ -27,6 +28,16 @@ Image::~Image()
     delete[] pPix;

 }

 

+int Image::getWidth() const

+{

+    return iWidth;

+}

+

+int Image::getHeight() const

+{

+    return iHeight;

+}

+

 void Image::clear( uint8_t uColor )

 {

     memset( pPix, uColor, iWidth*iHeight );

@@ -139,5 +150,24 @@ void Image::drawLine( int32_t x1, int32_t y1, int32_t x2, int32_t y2, uint8_t uC
             set( j, y1, uColor );

         }

     }

+    else

+    {

+        int xd = x2-x1;

+        int yd = y2-y1;

+        float d = sqrt( xd*xd + yd*yd );

+        float ux = xd/d;

+        float uy = yd/d;

+

+        float x = x1;

+        float y = y1;

+

+        set( (int)x, (int)y, uColor );

+        for( float n = 0.0; n < d; n += 1.0 )

+        {

+            x += ux;

+            y += uy;

+            set( (int)x, (int)y, uColor );

+        }

+    }

 }