:

examples/manipulate_lattice/manipulate_lattice.py

@@ -201,116 +201,116 @@ idx_elem_bpm_y = [lat.lst[i].idx_elem for i in idx_bpm_y]
 
 ###########################################################################
 
-# -- Define indices for blocks of R-matrix from "step" (may be too much...)
-
-block_x = []
-for k in range(len(step_x) + 1):
-    if int(sum(step_x[:k])) >= len(idx_bpm_x):
-        block_x.append(len(idx_bpm_x))
-        break
-    else:
-        block_x.append(int(sum(step_x[:k])))
-if block_x[-1] < len(idx_bpm_x):
-    block.append(len(idx_bpm_x))
-block_y = []
-for k in range(len(step_y) + 1):
-    if int(sum(step_y[:k])) >= len(idx_bpm_y):
-        block_y.append(len(idx_bpm_y))
-        break
-    else:
-        block_y.append(int(sum(step_y[:k])))
-if block_y[-1] < len(idx_bpm_y):
-    block.append(len(idx_bpm_y))
-
-# -- R matrix
-
-Rx = Pool(Ncpu).map(Rx_column, range(len(idx_st_x)))
-Ry = Pool(Ncpu).map(Ry_column, range(len(idx_st_y)))
-
-Rx = np.array(Rx).transpose()
-Ry = np.array(Ry).transpose()
-
-if len(Rx) > len(Rx[0]):
-    print("[# of BPMs] > [# of steerers] for x, exiting...")
-    exit()
-if len(Ry) > len(Ry[0]):
-    print("[# of BPMs] > [# of steerers] for y, exiting...")
-    exit()
-if len(Rx) < len(Rx[0]):
-    print("[# of BPMs] < [# of steerers] for x, exiting...")
-    exit()
-if len(Ry) < len(Ry[0]):
-    print("[# of BPMs] < [# of steerers] for y, exiting...")
-    exit()
-
-# -- Main part
-
-data = Pool(Ncpu).map(job, range(Nrun))
-data = np.array(data).transpose()
-
-s = data[0][0]
-x = np.array(data[1].tolist()).transpose()  # array not applied to the 3rd level ??
-y = np.array(data[2].tolist()).transpose()  # array not applied to the 3rd level ??
-BLy = np.array(data[3].tolist()).transpose()  # array not applied to the 3rd level ??
-BLx = np.array(data[4].tolist()).transpose()  # array not applied to the 3rd level ??
-
-# -- Writing
-
-# x
-with open("x.out", "w") as file:
-    # Header
-    file.write("s  ")
-    for n in range(Nrun):
-        file.write("x%03d  " % n)
-    file.write("\n")
-    # Data
-    for k in range(len(s)):
-        file.write("%.4f  " % s[k])
-        for n in range(Nrun):
-            file.write("%.5f  " % x[k][n])
-        file.write("\n")
-
-# y
-with open("y.out", "w") as file:
-    # Header
-    file.write("s  ")
-    for n in range(Nrun):
-        file.write("y%03d  " % n)
-    file.write("\n")
-    # Data
-    for k in range(len(s)):
-        file.write("%.4f  " % s[k])
-        for n in range(Nrun):
-            file.write("%.5f  " % y[k][n])
-        file.write("\n")
-
-# x steering (BLy [Gm])
-with open("st.x.out", "w") as file:
-    # Header
-    file.write("##  s  ")
-    for n in range(Nrun):
-        file.write("BLy%03d  " % n)
-    file.write("\n")
-    # Data
-    for k in range(len(idx_st_x)):
-        file.write("%03d  " % (k + 1) + "%.4f  " % lat.lst[idx_st_x_elem[k]].s)
-        for n in range(Nrun):
-            file.write("%.4f  " % (1e4 * BLy[k][n]))
-        file.write("\n")
-
-# y steering (BLx [Gm])
-with open("st.y.out", "w") as file:
-    # Header
-    file.write("##  s  ")
-    for n in range(Nrun):
-        file.write("BLx%03d  " % n)
-    file.write("\n")
-    # Data
-    for k in range(len(idx_st_y)):
-        file.write("%03d  " % (k + 1) + "%.4f  " % lat.lst[idx_st_y_elem[k]].s)
-        for n in range(Nrun):
-            file.write("%.4f  " % (1e4 * BLx[k][n]))
-        file.write("\n")
+# # -- Define indices for blocks of R-matrix from "step" (may be too much...)
+
+# block_x = []
+# for k in range(len(step_x) + 1):
+#     if int(sum(step_x[:k])) >= len(idx_bpm_x):
+#         block_x.append(len(idx_bpm_x))
+#         break
+#     else:
+#         block_x.append(int(sum(step_x[:k])))
+# if block_x[-1] < len(idx_bpm_x):
+#     block.append(len(idx_bpm_x))
+# block_y = []
+# for k in range(len(step_y) + 1):
+#     if int(sum(step_y[:k])) >= len(idx_bpm_y):
+#         block_y.append(len(idx_bpm_y))
+#         break
+#     else:
+#         block_y.append(int(sum(step_y[:k])))
+# if block_y[-1] < len(idx_bpm_y):
+#     block.append(len(idx_bpm_y))
+
+# # -- R matrix
+
+# Rx = Pool(Ncpu).map(Rx_column, range(len(idx_st_x)))
+# Ry = Pool(Ncpu).map(Ry_column, range(len(idx_st_y)))
+
+# Rx = np.array(Rx).transpose()
+# Ry = np.array(Ry).transpose()
+
+# if len(Rx) > len(Rx[0]):
+#     print("[# of BPMs] > [# of steerers] for x, exiting...")
+#     exit()
+# if len(Ry) > len(Ry[0]):
+#     print("[# of BPMs] > [# of steerers] for y, exiting...")
+#     exit()
+# if len(Rx) < len(Rx[0]):
+#     print("[# of BPMs] < [# of steerers] for x, exiting...")
+#     exit()
+# if len(Ry) < len(Ry[0]):
+#     print("[# of BPMs] < [# of steerers] for y, exiting...")
+#     exit()
+
+# # -- Main part
+
+# data = Pool(Ncpu).map(job, range(Nrun))
+# data = np.array(data).transpose()
+
+# s = data[0][0]
+# x = np.array(data[1].tolist()).transpose()  # array not applied to the 3rd level ??
+# y = np.array(data[2].tolist()).transpose()  # array not applied to the 3rd level ??
+# BLy = np.array(data[3].tolist()).transpose()  # array not applied to the 3rd level ??
+# BLx = np.array(data[4].tolist()).transpose()  # array not applied to the 3rd level ??
+
+# # -- Writing
+
+# # x
+# with open("x.out", "w") as file:
+#     # Header
+#     file.write("s  ")
+#     for n in range(Nrun):
+#         file.write("x%03d  " % n)
+#     file.write("\n")
+#     # Data
+#     for k in range(len(s)):
+#         file.write("%.4f  " % s[k])
+#         for n in range(Nrun):
+#             file.write("%.5f  " % x[k][n])
+#         file.write("\n")
+
+# # y
+# with open("y.out", "w") as file:
+#     # Header
+#     file.write("s  ")
+#     for n in range(Nrun):
+#         file.write("y%03d  " % n)
+#     file.write("\n")
+#     # Data
+#     for k in range(len(s)):
+#         file.write("%.4f  " % s[k])
+#         for n in range(Nrun):
+#             file.write("%.5f  " % y[k][n])
+#         file.write("\n")
+
+# # x steering (BLy [Gm])
+# with open("st.x.out", "w") as file:
+#     # Header
+#     file.write("##  s  ")
+#     for n in range(Nrun):
+#         file.write("BLy%03d  " % n)
+#     file.write("\n")
+#     # Data
+#     for k in range(len(idx_st_x)):
+#         file.write("%03d  " % (k + 1) + "%.4f  " % lat.lst[idx_st_x_elem[k]].s)
+#         for n in range(Nrun):
+#             file.write("%.4f  " % (1e4 * BLy[k][n]))
+#         file.write("\n")
+
+# # y steering (BLx [Gm])
+# with open("st.y.out", "w") as file:
+#     # Header
+#     file.write("##  s  ")
+#     for n in range(Nrun):
+#         file.write("BLx%03d  " % n)
+#     file.write("\n")
+#     # Data
+#     for k in range(len(idx_st_y)):
+#         file.write("%03d  " % (k + 1) + "%.4f  " % lat.lst[idx_st_y_elem[k]].s)
+#         for n in range(Nrun):
+#             file.write("%.4f  " % (1e4 * BLx[k][n]))
+#         file.write("\n")
 
 # -- Ending