You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
2.2 KiB
55 lines
2.2 KiB
#!/usr/bin/env python3 |
|
# -*- coding: utf-8 -*- |
|
""" |
|
Created on Sat Mar 14 12:47:24 2020 |
|
|
|
@author: roman |
|
""" |
|
from sympy import ccode |
|
from sympy.codegen.ast import float32, real |
|
|
|
class CodeGenerator: |
|
def __init__(self, file_name): |
|
self.file_name = file_name |
|
self.file = open(self.file_name, 'w') |
|
|
|
def print_string(self, string): |
|
self.file.write("// " + string + "\n") |
|
|
|
def get_ccode(self, expression): |
|
return ccode(expression, type_aliases={real:float32}) |
|
|
|
def write_subexpressions(self,subexpressions): |
|
write_string = "" |
|
for item in subexpressions: |
|
write_string = write_string + "const float " + str(item[0]) + " = " + self.get_ccode(item[1]) + ";\n" |
|
|
|
write_string = write_string + "\n\n" |
|
self.file.write(write_string) |
|
|
|
def write_matrix(self, matrix, identifier, is_symmetric=False): |
|
write_string = "" |
|
|
|
if matrix.shape[0] * matrix.shape[1] == 1: |
|
write_string = write_string + identifier + " = " + self.get_ccode(matrix[0]) + ";\n" |
|
elif matrix.shape[0] == 1 or matrix.shape[1] == 1: |
|
for i in range(0,len(matrix)): |
|
if (identifier == "Kfusion"): |
|
# Vector f format used by Kfusion |
|
write_string = write_string + identifier + "(" + str(i) + ") = " + self.get_ccode(matrix[i]) + ";\n" |
|
else: |
|
# legacy array format used by Hfusion |
|
write_string = write_string + identifier + "[" + str(i) + "] = " + self.get_ccode(matrix[i]) + ";\n" |
|
else: |
|
for j in range(0, matrix.shape[1]): |
|
for i in range(0, matrix.shape[0]): |
|
if j >= i or not is_symmetric: |
|
write_string = write_string + identifier + "(" + str(i) + "," + str(j) + ") = " + self.get_ccode(matrix[i,j]) + ";\n" |
|
# legacy array format |
|
# write_string = write_string + identifier + "[" + str(i) + "][" + str(j) + "] = " + self.get_ccode(matrix[i,j]) + ";\n" |
|
|
|
write_string = write_string + "\n\n" |
|
self.file.write(write_string) |
|
|
|
def close(self): |
|
self.file.close()
|
|
|