1 files changed, 112 insertions, 0 deletions
diff --git a/raphodo/filmstrip.py b/raphodo/filmstrip.py
new file mode 100644
index 0000000..1affd47
--- /dev/null
+++ b/raphodo/filmstrip.py
@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+
+# Copyright (C) 2011-2016 Damon Lynch <damonlynch@gmail.com>
+
+# This file is part of Rapid Photo Downloader.
+#
+# Rapid Photo Downloader is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Rapid Photo Downloader is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rapid Photo Downloader.  If not,
+# see <http://www.gnu.org/licenses/>.
+
+"""
+Overlays a filmstrip onto QImage, keeping the image's dimensions the
+same
+"""
+
+__author__ = 'Damon Lynch'
+__copyright__ = "Copyright 2011-2016, Damon Lynch"
+
+from PyQt5.QtGui import QImage, QPainter
+
+xpm_data = [
+    "12 10 27 1",
+    "   c #000000",
+    ".  c #232323",
+    "+  c #7A7A7A",
+    "@  c #838383",
+    "#  c #8C8C8C",
+    "$  c #909090",
+    "%  c #8E8E8E",
+    "&  c #525252",
+    "*  c #6E6E6E",
+    "=  c #939393",
+    "-  c #A3A3A3",
+    ";  c #ABABAB",
+    ">  c #A8A8A8",
+    ",  c #9B9B9B",
+    "'  c #727272",
+    ")  c #A4A4A4",
+    "!  c #BBBBBB",
+    "~  c #C4C4C4",
+    "{  c #C1C1C1",
+    "]  c #AFAFAF",
+    "^  c #3E3E3E",
+    "/  c #A6A6A6",
+    "(  c #BEBEBE",
+    "_  c #C8C8C8",
+    ":  c #070707",
+    "<  c #090909",
+    "[  c #0A0A0A",
+    "            ",
+    "            ",
+    "            ",
+    "    .+@#$%& ",
+    "    *@=-;>, ",
+    "    '%)!~{] ",
+    "    ^$/(_~% ",
+    "     :<[[[  ",
+    "            ",
+    "            "]
+
+
+def add_filmstrip(thumbnail: QImage) -> QImage:
+    """
+    Overlays a filmstrip onto the thumbnail.
+
+    Keeps the thumbnail's dimensions the same.
+
+    :param thumbnail: thumbnail on which to put the filmstrip
+    :return a copy of the thumbnail
+
+    """
+    format = thumbnail.format()
+
+    filmstrip = QImage(xpm_data)
+    filmstrip_width = filmstrip.width()
+    filmstrip_height = filmstrip.height()
+    filmstrip_right = filmstrip.mirrored(horizontal=True, vertical=False)
+
+    thumbnail_right_col = thumbnail.width() - filmstrip_width
+
+    painter = QPainter(thumbnail)
+
+    # add filmstrips to left and right
+    for i in range(thumbnail.height() // filmstrip_height):
+        painter.drawImage(0, i * filmstrip_height, filmstrip)
+        painter.drawImage(thumbnail_right_col,
+                          i * filmstrip_height, filmstrip_right)
+
+    #now do the remainder, at the bottom
+    remaining_height = thumbnail.height() % filmstrip_height
+    if remaining_height:
+        painter.drawImage(0, thumbnail.height() - remaining_height,
+                          filmstrip.copy(0, 0, filmstrip_width,
+                                         remaining_height))
+        painter.drawImage(thumbnail_right_col,
+                          thumbnail.height() - remaining_height,
+                          filmstrip_right.copy(0, 0, filmstrip_width,
+                                               remaining_height))
+
+    return thumbnail
+
+