From 920e3024e35fb9f31d3614a96c45ae8245cd5b48 Mon Sep 17 00:00:00 2001
From: Christian Meesters <meesters@uni-mainz.de>
Date: Mon, 1 Apr 2024 14:39:32 +0200
Subject: [PATCH] feat: render now counts sections

---
 render | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 70 insertions(+), 1 deletion(-)

diff --git a/render b/render
index 80e278a..044243f 100755
--- a/render
+++ b/render
@@ -1,12 +1,15 @@
 #!/usr/bin/env python3
 
 import argparse
-from tempfile import TemporaryDirectory
+import math
 import logging
 import os
 from pathlib import Path
+import re
 import shutil
 import subprocess
+import sys
+from tempfile import TemporaryDirectory
 import yaml
 
 # non standard libraries
@@ -21,11 +24,71 @@ options = {
     "comment_end_string": "=}",
 }
 
+section_re = re.compile("\\\section\{")
+infile_re = re.compile(r"(\\input)|(\\include)")
+extract_re = re.compile(r"(?<=\{).+?(?=\})")
+
 PATH = os.path.split(os.path.realpath(__file__))[0]
 
 jinja_env = Environment(loader=FileSystemLoader(PATH), **options)
 
 
+def screen_master(masterdoc, tempdir):
+    prefix = os.path.join(tempdir, "slides")
+    fnames = list()
+    with open(os.path.join(prefix, masterdoc)) as infile:
+        for line in infile:
+            if infile_re.match(line):
+                fnames.extend(extract_re.findall(line))
+    return fnames
+
+
+def count_matching_lines(fp):
+    count = 0
+    for line in fp:
+        if section_re.match(line):
+            count += 1
+    return count
+
+
+def count_sections(file_list, tempdir):
+    """
+    counting '\section(*)' lines by calling
+    `count_matching_lines` - the function will
+    operate in the temporary directory, to ensure
+    sections introduced after parsing the configuration
+    are counted.
+
+    input:
+        - `file_list` - the list of files need to typeset
+          the master TeX document
+        - `tempdir` - the temporary directory
+    ouput:
+        - the total number of sections.
+    """
+    prefix = os.path.join(tempdir, "slides")
+
+    count = 0
+    for fname in file_list:
+        to_count = os.path.join(prefix, fname) + ".tex"
+        try:
+            with open(to_count) as fp:
+                count += count_matching_lines(fp)
+        except:
+            print(f"Error treating: '{fname}'")
+    return count
+
+
+def define_boundaries(section_count):
+    fhe = math.floor(section_count / 2)
+    return {
+        "lower": 1,
+        "upper": section_count,
+        "first_half_end": fhe,
+        "second_half_start": fhe + 1,
+    }
+
+
 def render_tex(env, template):
     template = template
     tpl = jinja_env.get_template(template)
@@ -93,5 +156,11 @@ if __name__ == "__main__":
                 logger.info(f"Attempt to render {fname}")
                 tex = render_tex(data, os.path.join(root, fname))
                 to_file(tex, fname, tempdir, root)
+        # we might need to adapt the chapter settings.
+        file_list = screen_master(args.master_tex, tempdir)
+        count     = count_sections(file_list, tempdir)
+        logger.info('Found %d sections.' % count)
+        boundaries = define_boundaries(count)
+        
         # now the typesetting needs to be triggered
         run_pdflatex(fname=args.master_tex, path=os.path.join(tempdir, "slides"))