INNER CODE UNIT · Python

num_files

RQLuo/MixTeX-Latex-OCR · mixtex_data_gen/gen.py:136

    num_files = (len(strings) + group_size - 1) // group_size
    
    for i in range(num_files):
        file_name = f"{folder_name}/{i + 1}.tex"
        start_index = i * group_size
        end_index = start_index + group_size
        current_group = strings[start_index:end_index]
        width = random.randint(12, 15)
        margin = random.randint(3, 4)
        line = random.randint(4, 20) / 10
        bg = f'\\documentclass{{ctexart}}\n\\usepackage[paperwidth={width}in, paperheight=36in, margin={margin}in]{{geometry}}\n'
        bg += '\\usepackage{amssymb}\n\\usepackage{amsmath}\n\\usepackage{stmaryrd}\n\\usepackage{color}\n'
        bg += '\\nonstopmode\n\\pagestyle{empty}\n\\renewcommand{\\baselinestretch}{' + str(line) + '}\n\n\\begin{document}\n \\newpage'
        ed = '\\end{document}'
        with open(file_name, 'w', encoding='utf-8') as file:
            file.write(bg)
            file.write(''.join(current_group))
            file.write(ed)

View source record →