1 files changed, 40 insertions, 59 deletions

diff --git a/dmenu.c b/dmenu.c
index 175cbdc..f19e798 100644
--- a/dmenu.c
+++ b/dmenu.c
@@ -25,12 +25,9 @@ struct Item {
 };
 
 static void appenditem(Item *item, Item **list, Item **last);
-static void calcoffsetsh(void);
-static void calcoffsetsv(void);
+static void calcoffsets(void);
 static char *cistrstr(const char *s, const char *sub);
 static void drawmenu(void);
-static void drawmenuh(void);
-static void drawmenuv(void);
 static void grabkeyboard(void);
 static void insert(const char *s, ssize_t n);
 static void keypress(XKeyEvent *e);
@@ -63,7 +60,6 @@ static Window root, win;
 
 static int (*fstrncmp)(const char *, const char *, size_t) = strncmp;
 static char *(*fstrstr)(const char *, const char *) = strstr;
-static void (*calcoffsets)(void) = calcoffsetsh;
 
 void
 appenditem(Item *item, Item **list, Item **last) {
@@ -77,29 +73,25 @@ appenditem(Item *item, Item **list, Item **last) {
 }
 
 void
-calcoffsetsh(void) {
-        unsigned int w, x;
+calcoffsets(void)
+{
+        unsigned int h, i, n;
 
-        w = promptw + inputw + textw(dc, "<") + textw(dc, ">");
-        for(x = w, next = curr; next; next = next->right)
-                if((x += MIN(textw(dc, next->text), mw / 3)) > mw)
+        h = dc->font.height+2;
+        if(lines > 0)
+                n = lines * h;
+        else
+                n = mw - (promptw + inputw + textw(dc, "<") + textw(dc, ">"));
+
+        prev = next = curr;
+        for(i = 0; next; next = next->right)
+                if((i += (lines > 0) ? h : MIN(textw(dc, next->text), mw/3)) > n)
                         break;
-        for(x = w, prev = curr; prev && prev->left; prev = prev->left)
-                if((x += MIN(textw(dc, prev->left->text), mw / 3)) > mw)
+        for(i = 0; prev && prev->left; prev = prev->left)
+                if((i += (lines > 0) ? h : MIN(textw(dc, prev->left->text), mw/3)) > n)
                         break;
 }
 
-void
-calcoffsetsv(void) {
-        unsigned int i;
-
-        next = prev = curr;
-        for(i = 0; i < lines && next; i++)
-                next = next->right;
-        for(i = 0; i < lines && prev && prev->left; i++)
-                prev = prev->left;
-}
-
 char *
 cistrstr(const char *s, const char *sub) {
         size_t len;
@@ -112,6 +104,8 @@ cistrstr(const char *s, const char *sub) {
 
 void
 drawmenu(void) {
+        Item *item;
+
         dc->x = 0;
         dc->y = 0;
         drawrect(dc, 0, 0, mw, mh, BG(dc, normcol));
@@ -129,42 +123,31 @@ drawmenu(void) {
                 dc->w = inputw;
         drawtext(dc, text, normcol);
         drawrect(dc, textnw(dc, text, cursor) + dc->h/2 - 2, 2, 1, dc->h - 4, FG(dc, normcol));
-        if(lines > 0)
-                drawmenuv();
-        else if(curr && (dc->w == inputw || curr->next))
-                drawmenuh();
-        commitdraw(dc, win);
-}
-
-void
-drawmenuh(void) {
-        Item *item;
 
-        dc->x += inputw;
-        dc->w = textw(dc, "<");
-        if(curr->left)
-                drawtext(dc, "<", normcol);
-        for(item = curr; item != next; item = item->right) {
-                dc->x += dc->w;
-                dc->w = MIN(textw(dc, item->text), mw / 3);
-                drawtext(dc, item->text, (item == sel) ? selcol : normcol);
+        if(lines > 0) {
+                dc->y = topbar ? dc->h : 0;
+                dc->w = mw - dc->x;
+                for(item = curr; item != next; item = item->right) {
+                        drawtext(dc, item->text, (item == sel) ? selcol : normcol);
+                        dc->y += dc->h;
+                }
         }
-        dc->w = textw(dc, ">");
-        dc->x = mw - dc->w;
-        if(next)
-                drawtext(dc, ">", normcol);
-}
-
-void
-drawmenuv(void) {
-        Item *item;
-
-        dc->y = topbar ? dc->h : 0;
-        dc->w = mw - dc->x;
-        for(item = curr; item != next; item = item->right) {
-                drawtext(dc, item->text, (item == sel) ? selcol : normcol);
-                dc->y += dc->h;
+        else if(curr && (dc->w == inputw || curr->next)) {
+                dc->x += inputw;
+                dc->w = textw(dc, "<");
+                if(prev)
+                        drawtext(dc, "<", normcol);
+                for(item = curr; item != next; item = item->right) {
+                        dc->x += dc->w;
+                        dc->w = MIN(textw(dc, item->text), mw/3);
+                        drawtext(dc, item->text, (item == sel) ? selcol : normcol);
+                }
+                dc->w = textw(dc, ">");
+                dc->x = mw - dc->w;
+                if(next)
+                        drawtext(dc, ">", normcol);
         }
+        commitdraw(dc, win);
 }
 
 void
@@ -494,7 +477,6 @@ setup(void) {
                 y = topbar ? 0 : DisplayHeight(dc->dpy, screen) - mh;
                 mw = DisplayWidth(dc->dpy, screen);
         }
-
         /* input window */
         wa.override_redirect = True;
         wa.background_pixmap = ParentRelative;
@@ -543,8 +525,7 @@ main(int argc, char *argv[]) {
                         usage();
                 /* double flags */
                 else if(!strcmp(argv[i], "-l")) {
-                        if((lines = atoi(argv[++i])) > 0)
-                                calcoffsets = calcoffsetsv;
+                        lines = atoi(argv[++i]);
                 }
                 else if(!strcmp(argv[i], "-p"))
                         prompt = argv[++i];