Added error checking for Tao python floor_plan command. (#1063)

bmad-sim · Jul 15, 2024 · 98fcb63 · 98fcb63
1 parent fe42daf
commit 98fcb63
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tao/code/tao_python_cmd.f90 b/tao/code/tao_python_cmd.f90
@@ -4268,6 +4268,10 @@ subroutine tao_python_cmd (input_str)
  u => tao_pointer_to_universe(g%ix_universe, .true.)
  lat => u%model%lat
 
+ if (.not. allocated(g%floor_list)) then
+ call invalid ('Floor plan drawing not yet setup for this graph.')
+ endif
+
  do i = 1, size(g%floor_list)
  ele => pointer_to_ele(lat, g%floor_list(i)%ele_loc)
  ashape = g%floor_list(i)%shape