INNER CODE UNIT · Python
write_strings_to_files
RQLuo/MixTeX-Latex-OCR · mixtex_data_gen/gen.py:134
def write_strings_to_files(strings, group_size, folder_name):
os.makedirs(folder_name, exist_ok=True)
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)