Skip to content

Instantly share code, notes, and snippets.

@tsaoyu
Created March 11, 2020 12:14
Show Gist options
  • Save tsaoyu/8dbeb9a8cee332766a3a7a77cb87650c to your computer and use it in GitHub Desktop.
Save tsaoyu/8dbeb9a8cee332766a3a7a77cb87650c to your computer and use it in GitHub Desktop.
def read_and_process(filename):
with open(filename, 'r') as f:
raw_content = f.readline()
raw_content = re.sub(r'conj\(([a-z0-9]+)\)', r'\1', raw_content)
raw_content = re.sub(r'cos\(', r'ct::core::tpl::TraitSelector<SCALAR>::Trait::cos(', raw_content)
raw_content = re.sub(r'sin\(', r'ct::core::tpl::TraitSelector<SCALAR>::Trait::sin(', raw_content)
raw_content = re.sub(r'([a-zA-Z0-9]+)\^2', r'\1*\1', raw_content)
raw_content = re.sub(r'abs\(([a-zA-Z0-9]+)\)', r'ct::core::tpl::TraitSelector<SCALAR>::Trait::sqrt(\1 * \1)', raw_content)
raw_content = re.sub(r'\[', ' ', raw_content)
raw_content = re.sub(r'\]', ' ', raw_content)
raw_content = re.sub(r'matrix\(', ' ', raw_content)
raw_content_list = raw_content.split(', ')
print(raw_content_list)
with open(filename.split('.')[0] + "_output.txt", 'w') as f:
f.write('\n\n'.join(raw_content_list))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment