Merge pull request #336 from mathoudebine/fix/335-331-screen-only-par…

…tially-drawing

library/lcd/lcd_comm.py

@@ -17,12 +17,12 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import copy
+import math
 import os
 import queue
 import sys
 import threading
 import time
-import math
 from abc import ABC, abstractmethod
 from enum import IntEnum
 from typing import Tuple
@@ -69,7 +69,6 @@ def __init__(self, com_port: str = "AUTO", display_width: int = 320, display_hei
         # Create a cache to store opened fonts, to avoid opening and loading from the filesystem every time
         self.font_cache = {}  # { key=(font, size), value=PIL.ImageFont }
 
-
     def get_width(self) -> int:
         if self.orientation == Orientation.PORTRAIT or self.orientation == Orientation.REVERSE_PORTRAIT:
             return self.display_width

library/lcd/lcd_comm_rev_a.py

@@ -151,6 +151,7 @@ def SetBrightness(self, level: int = 25):
         self.SendCommand(Command.SET_BRIGHTNESS, level_absolute, 0, 0, 0)
 
     def SetOrientation(self, orientation: Orientation = Orientation.PORTRAIT):
+        self.orientation = orientation
         width = self.get_width()
         height = self.get_height()
         x = 0