From ac4a8ce9d748a4d3b335ab73ed60777175616c34 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mattias=20Andr=C3=A9e?= <maandree@operamail.com>
Date: Fri, 2 Nov 2012 01:23:34 +0100
Subject: [PATCH] screen big ponies from random selection

---
 CHANGELOG               |  3 ++
 manuals/ponysay.texinfo |  3 ++
 ponysay.py              | 76 +++++++++++++++++++++++++++++++++++++++--
 3 files changed, 80 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG b/CHANGELOG
index 2a74d076..b107082e 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -26,6 +26,9 @@ Version 3.0
 
 	Balloons can be have and explicit minimum column span with placement justification.
 
+	Only ponies that fit the terminal will be randomly selected (for directory with pony
+	dimension files generated), however if no pony fits, any of the can be randomly selected.
+
 
 Version 2.9.1
 
diff --git a/manuals/ponysay.texinfo b/manuals/ponysay.texinfo
index 915316f6..a01554b9 100644
--- a/manuals/ponysay.texinfo
+++ b/manuals/ponysay.texinfo
@@ -2678,6 +2678,9 @@ height the balloon causes at the top and at the bottom of the pony.
 @file{${XDG_DATA_HOME}/ponysay/*} is allowed in favour of @file{${HOME}/.local/share/ponysay/*}
 @item
 Balloons can be have and explicit minimum column span with placement justification.
+@item
+Only ponies that fit the terminal will be randomly selected (for directory with pony
+dimension files generated), however if no pony fits, any of the can be randomly selected.
 @end itemize
 
 
diff --git a/ponysay.py b/ponysay.py
index a483e4d6..a542be56 100755
--- a/ponysay.py
+++ b/ponysay.py
@@ -339,6 +339,9 @@ class Ponysay():
             if args.opts['-o'] is not None:
                 self.mode += '$/= $$\\= $'
                 args.message = ''
+                self.ponyonly = True
+            else:
+                self.ponyonly = False
             
             ## The stuff
             if args.opts['-q'] is not None:
@@ -498,7 +501,7 @@ class Ponysay():
         ## List all pony files, without the .pony ending
         for ponydir in self.ponydirs:
             for ponyfile in os.listdir(ponydir):
-                if endswith(ponyfile, ".pony"):
+                if endswith(ponyfile, '.pony'):
                     pony = ponyfile[:-5]
                     if pony not in ponies:
                         ponies[pony] = ponydir + ponyfile
@@ -509,9 +512,78 @@ class Ponysay():
                 if os.path.exists(name):
                     ponies[name] = name
         
+        '''
+        Get ponies that fit the terminal
+        
+        @param  fitting      The set to fill
+        @param  requirement  The maximum allowed value
+        @param  file         The file with all data
+        '''
+        def getfitting(fitting, requirement, file):
+            data = file.read() # not too much data, can load everything at once
+            ptr = 0
+            while data[ptr] != 47: # 47 == ord('/')
+                ptr += 1
+            ptr += 1
+            size = 0
+            while data[ptr] != 47: # 47 == ord('/')
+                size = (size * 10) - (data[ptr] & 15)
+                ptr += 1
+            ptr += 1
+            jump = ptr - size
+            stop = 0
+            backjump = 0
+            while ptr < jump:
+                size = 0
+                while data[ptr] != 47: # 47 == ord('/')
+                    size = (size * 10) - (data[ptr] & 15)
+                    ptr += 1
+                ptr += 1
+                if -size > requirement:
+                    if backjump > 0:
+                        ptr = backjump
+                        while data[ptr] != 47: # 47 == ord('/')
+                            stop = (stop * 10) - (data[ptr] & 15)
+                            ptr += 1
+                        stop = -stop
+                    break
+                backjump = ptr
+                while data[ptr] != 47: # 47 == ord('/')
+                    ptr += 1
+                ptr += 1
+            if ptr == jump:
+                stop = len(data)
+            else:
+                ptr = jump
+                stop += ptr
+            passed = data[jump : stop].decode('utf8', 'replace').split('/')
+            for pony in passed:
+                fitting.add(pony)
+            
+        
         ## If there is not select ponies, choose all of them
         if (names is None) or (len(names) == 0):
-            names = list(ponies.keys())
+            oldponies = ponies
+            ponies = {}
+            (termh, termw) = self.__gettermsize()
+            for ponydir in self.ponydirs:
+                (fitw, fith) = (None, None)
+                if os.path.exists(ponydir + 'widths'):
+                    fitw = set()
+                    with open(ponydir + 'widths', 'rb') as file:
+                        getfitting(fitw, termw, file)
+                if os.path.exists(ponydir + ('onlyheights' if self.ponyonly else 'heights')):
+                    fith = set()
+                    with open(ponydir + ('onlyheights' if self.ponyonly else 'heights'), 'rb') as file:
+                        getfitting(fith, termh, file)
+                for ponyfile in os.listdir(ponydir):
+                    if endswith(ponyfile, '.pony'):
+                        pony = ponyfile[:-5]
+                        if pony not in ponies:
+                            if (fitw is None) or (pony in fitw):
+                                if (fith is None) or (pony in fith):
+                                    ponies[pony] = ponydir + ponyfile
+            names = list((oldponies if len(ponies) == 0 else ponies).keys())
         
         ## Select a random pony of the choosen onles
         pony = names[random.randrange(0, len(names))]