From 195a472f152f5da0104ae21006fd88575e2f7a71 Mon Sep 17 00:00:00 2001
From: otya128 <otya128@users.noreply.github.com>
Date: Mon, 5 Dec 2016 22:54:30 +0900
Subject: [PATCH] Fix display()

---
 SMILEBASIC/petitcomputer.d | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/SMILEBASIC/petitcomputer.d b/SMILEBASIC/petitcomputer.d
index 1dd9d8c..fef2093 100644
--- a/SMILEBASIC/petitcomputer.d
+++ b/SMILEBASIC/petitcomputer.d
@@ -322,6 +322,12 @@ class PetitComputer
     {
         import std.exception : enforce;
         enforce(number >= 0);
+        if (xscreenmode == 2)
+        {
+            enforce(number == 0);
+            currentScreenWidth = screenWidthDisplay1;
+            currentScreenHeight = screenHeightDisplay1 * 2;
+        }
         if(xscreenmode == 0)
         {
             enforce(number == 0);
@@ -331,14 +337,16 @@ class PetitComputer
         else if(xscreenmode == 1)
         {
             enforce(number == 0 || number == 1);
-            currentScreenWidth = screenWidthDisplay1;
-            currentScreenHeight = screenHeightDisplay1;
-        }
-        else
-        {
-            enforce(number == 0);
-            currentScreenWidth = screenWidthDisplay1;
-            currentScreenHeight = screenHeightDisplay1 * 2;
+            if (number == 0)
+            {
+                currentScreenWidth = screenWidth;
+                currentScreenHeight = screenHeight;
+            }
+            else
+            {
+                currentScreenWidth = screenWidthDisplay1;
+                currentScreenHeight = screenHeightDisplay1;
+            }
         }
         displaynum = number;