ana/pmt/gdmlplot.py

0001 #
0002 # Copyright (c) 2019 Opticks Team. All Rights Reserved.
0003 #
0004 # This file is part of Opticks
0005 # (see https://bitbucket.org/simoncblyth/opticks).
0006 #
0007 # Licensed under the Apache License, Version 2.0 (the "License");
0008 # you may not use this file except in compliance with the License.
0009 # You may obtain a copy of the License at
0010 #
0011 #   http://www.apache.org/licenses/LICENSE-2.0
0012 #
0013 # Unless required by applicable law or agreed to in writing, software
0014 # distributed under the License is distributed on an "AS IS" BASIS,
0015 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
0016 # See the License for the specific language governing permissions and
0017 # limitations under the License.
0018 #
0019
0020 import matplotlib.pyplot as plt
0021
0022 def do_plt(pc):
0023     fig = plt.figure()
0024     ax = fig.add_subplot(1,1,1, aspect='equal')
0025     polycone_plt(ax, pc)
0026     plt.show()
0027
0028
0029 def polycone_plt(fig, pcs, nx=4, ny=4):
0030     for i in range(len(pcs)):
0031         ax = fig.add_subplot(nx,ny,i+1, aspect='equal')
0032         pc = pcs[i]
0033         pc.plot(ax)
0034     pass
0035
0036
0037 if __name__ == "__main__":
0038
0039     pcs = gdml.findall_("solids//polycone")
0040     plt.ion()
0041     fig = plt.figure()
0042     polycone_plt(fig, pcs)
0043     fig.show()
0044
0045