COMPASS  5.4.4
End-to-end AO simulation tool using GPU acceleration
dm_standalone.py
1 
37 
38 # import cProfile
39 # import pstats as ps
40 
41 import sys
42 import os
43 # import numpy as np
44 import carmaWrap as ch
45 import shesha.config as conf
46 import time
47 import numpy as np
48 import matplotlib.pyplot as plt
49 plt.ion()
50 
51 from shesha.init.geom_init import geom_init_generic
52 from shesha.init.dm_init import dm_init_standalone
53 
54 #geom
55 p_geom = conf.Param_geom()
56 geom_init_generic(p_geom, 500)
57 
58 #dm
59 p_dm0 = conf.Param_dm()
60 p_dms = [p_dm0]
61 p_dm0.set_type("pzt")
62 # p_dm0.set_pattern("hexa")
63 p_dm0.set_nact(80)
64 p_dm0.set_alt(0.)
65 p_dm0.set_thresh(0.3)
66 p_dm0.set_coupling(0.2)
67 p_dm0.set_unitpervolt(0.01)
68 p_dm0.set_pzt_extent(0)
69 
70 # context
71 c = ch.context.get_instance_1gpu(0)
72 
73 # dm
74 print("->dm")
75 dms = dm_init_standalone(c, p_dms, p_geom)
76 
77 print("====================")
78 print("init done")
79 print("====================")
80 print("objects initialzed on GPU:")
81 print("--------------------------------------------------------")
82 print(dms)
83 
84 cmd = np.zeros(5268)
85 cmd[1111] = 1
86 dms.set_full_com(cmd)
87 plt.matshow(dms.d_dms[0].d_shape)
Parameter classes for COMPASS.
Initialization of a Dms object.
Definition: dm_init.py:1
Initialization of the system geometry and of the Telescope object.
Definition: geom_init.py:1