Skip to content

Commit 0520f59

Browse files
committed
Added update data functions
1 parent e0ecb10 commit 0520f59

File tree

1 file changed

+36
-36
lines changed

1 file changed

+36
-36
lines changed

qocogen/codegen.py

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -1065,12 +1065,12 @@ def generate_utils(
10651065
f.write(' "maximum iterations reached",\n')
10661066
f.write("};\n\n")
10671067

1068-
# f.write("void update_P(Workspace* work, double* P_new);\n")
1069-
# f.write("void update_A(Workspace* work, double* A_new);\n")
1070-
# f.write("void update_G(Workspace* work, double* G_new);\n")
1071-
# f.write("void update_c(Workspace* work, double* c_new);\n")
1072-
# f.write("void update_b(Workspace* work, double* b_new);\n")
1073-
# f.write("void update_h(Workspace* work, double* h_new);\n")
1068+
f.write("void update_P(Workspace* work, double* P_new);\n")
1069+
f.write("void update_A(Workspace* work, double* A_new);\n")
1070+
f.write("void update_G(Workspace* work, double* G_new);\n")
1071+
f.write("void update_c(Workspace* work, double* c_new);\n")
1072+
f.write("void update_b(Workspace* work, double* b_new);\n")
1073+
f.write("void update_h(Workspace* work, double* h_new);\n")
10741074

10751075
f.write("void load_data(Workspace* work);\n")
10761076
f.write("void set_default_settings(Workspace* work);\n")
@@ -1111,41 +1111,41 @@ def generate_utils(
11111111
f = open(solver_dir + "/utils.c", "a")
11121112
write_license(f)
11131113
f.write('#include "utils.h"\n\n')
1114-
# f.write("void update_P(Workspace* work, double* P_new) {\n")
1115-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % Pnnz)
1116-
# f.write(" work->P[i] = P_new[i];\n")
1117-
# f.write(" }\n")
1118-
# f.write("}\n\n")
1114+
f.write("void update_P(Workspace* work, double* P_new) {\n")
1115+
f.write(" for (int i = 0; i < %d; ++i) {\n" % Pnnz)
1116+
f.write(" work->P[i] = P_new[i];\n")
1117+
f.write(" }\n")
1118+
f.write("}\n\n")
11191119

1120-
# f.write("void update_A(Workspace* work, double* A_new) {\n")
1121-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % Annz)
1122-
# f.write(" work->A[i] = A_new[i];\n")
1123-
# f.write(" }\n")
1124-
# f.write("}\n\n")
1120+
f.write("void update_A(Workspace* work, double* A_new) {\n")
1121+
f.write(" for (int i = 0; i < %d; ++i) {\n" % Annz)
1122+
f.write(" work->A[i] = A_new[i];\n")
1123+
f.write(" }\n")
1124+
f.write("}\n\n")
11251125

1126-
# f.write("void update_G(Workspace* work, double* G_new) {\n")
1127-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % Gnnz)
1128-
# f.write(" work->G[i] = G_new[i];\n")
1129-
# f.write(" }\n")
1130-
# f.write("}\n\n")
1126+
f.write("void update_G(Workspace* work, double* G_new) {\n")
1127+
f.write(" for (int i = 0; i < %d; ++i) {\n" % Gnnz)
1128+
f.write(" work->G[i] = G_new[i];\n")
1129+
f.write(" }\n")
1130+
f.write("}\n\n")
11311131

1132-
# f.write("void update_c(Workspace* work, double* c_new) {\n")
1133-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % n)
1134-
# f.write(" work->c[i] = c_new[i];\n")
1135-
# f.write(" }\n")
1136-
# f.write("}\n\n")
1132+
f.write("void update_c(Workspace* work, double* c_new) {\n")
1133+
f.write(" for (int i = 0; i < %d; ++i) {\n" % n)
1134+
f.write(" work->c[i] = c_new[i];\n")
1135+
f.write(" }\n")
1136+
f.write("}\n\n")
11371137

1138-
# f.write("void update_b(Workspace* work, double* b_new) {\n")
1139-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % p)
1140-
# f.write(" work->b[i] = b_new[i];\n")
1141-
# f.write(" }\n")
1142-
# f.write("}\n\n")
1138+
f.write("void update_b(Workspace* work, double* b_new) {\n")
1139+
f.write(" for (int i = 0; i < %d; ++i) {\n" % p)
1140+
f.write(" work->b[i] = b_new[i];\n")
1141+
f.write(" }\n")
1142+
f.write("}\n\n")
11431143

1144-
# f.write("void update_h(Workspace* work, double* h_new) {\n")
1145-
# f.write(" for (int i = 0; i < %d; ++i) {\n" % m)
1146-
# f.write(" work->h[i] = h_new[i];\n")
1147-
# f.write(" }\n")
1148-
# f.write("}\n\n")
1144+
f.write("void update_h(Workspace* work, double* h_new) {\n")
1145+
f.write(" for (int i = 0; i < %d; ++i) {\n" % m)
1146+
f.write(" work->h[i] = h_new[i];\n")
1147+
f.write(" }\n")
1148+
f.write("}\n\n")
11491149

11501150

11511151
f.write("void load_data(Workspace* work) {\n")

0 commit comments

Comments
 (0)