Add argument --font

Author: JulienCoutault

TextRecognitionDataGenerator/run.py

@@ -2,6 +2,7 @@
 import os, errno
 import random
 import string
+import sys
 
 from tqdm import tqdm
 from string_generator import (
@@ -252,6 +253,13 @@ def parse_arguments():
         help="Apply a tight crop around the rendered text",
         default=False
     )
+    parser.add_argument(
+        "-ft",
+        "--font",
+        type=str,
+        nargs="?",
+        help="Define font to be used"
+    )
 
 
     return parser.parse_args()
@@ -295,7 +303,13 @@ def main():
     lang_dict = load_dict(args.language)
 
     # Create font (path) list
-    fonts = load_fonts(args.language)
+    if not args.font:
+        fonts = load_fonts(args.language)
+    else:
+        if os.path.isfile(args.font):
+            fonts = [args.font]
+        else:
+            sys.exit("Cannot open font")
 
     # Creating synthetic sentences (or word)
     strings = []

tests.py

@@ -664,6 +664,18 @@ def test_handwritten(self):
         self.assertTrue(len(os.listdir('tests/out/')) == 1)
         empty_directory('tests/out/')
 
+    def test_personalfont(self):
+        args = ['python3', 'run.py', '--font', 'fonts/latin/Aller_Bd.ttf' , '-c', '1', '--output_dir', '../tests/out/']
+        subprocess.Popen(args, cwd="TextRecognitionDataGenerator/").wait()
+        self.assertTrue(len(os.listdir('tests/out/')) == 1)
+        empty_directory('tests/out/')
+
+    def test_personalfont_unlocated(self):
+        args = ['python3', 'run.py', '--font', 'fonts/latin/unlocatedFont.ttf' , '-c', '1', '--output_dir', '../tests/out/']
+        subprocess.Popen(args, cwd="TextRecognitionDataGenerator/").wait()
+        self.assertTrue(len(os.listdir('tests/out/')) == 0)
+        empty_directory('tests/out/')
+
 #    def test_word_count(self):
 #        args = ['python3', 'run.py', '-c', '1', '-w', '5']
 #        subprocess.Popen(args, cwd="TextRecognitionDataGenerator/").wait()