def grand_plot_all(o, grand):
    """ Make all the plots for the encompassing suite """
    factory = PF(int(o.sim_number),
                 snapshot_base=o.snap_base,
                 count=o.num_format,
                 grand=grand)
    factory.plot_all()
예제 #2
0
def plot_all(o):
    """ Make all the plots """
    factory = PF(int(o.sim_number),
                 snapshot_base=o.snap_base,
                 count=o.num_format)

    # Process snapshots ---> Make Plots
    factory.plot_all()  # <<<---
def plot_all(o):
  """ Make all the plots """
  factory = PF(int(o.sim_number), snapshot_base = o.snap_base, count = o.num_format)

  # Process snapshots ---> Make Plots
  factory.plot_all() # <<<--- 
    f.write("Inner Disk Radius: " + str(o.r_in) + "\n")
    f.write("Outer Disk Radius: " + str(o.r_out) + "\n")
    f.write("Number of Disk Particles: " + str(o.n_disk) + "\n")
    f.write("Disk Particle Power Law: " + str(o.power) + "\n")
    f.write("Orbit Pericenter: " + str(o.peri) + "\n")
    f.write("Orbit Eccentricity: " + str(o.ecc) + "\n")
    f.write("Orbit Inclination: " + str(o.incl) + "\n")
    f.write("Orbit Argument of Periapsis: " + str(o.omega) + "\n")
    f.write("Number of Steps: " + str(o.n_steps) + "\n")
    f.close()

    # Write 'o' dictionary into file (using Pickle)
    pickle_f = open(o.snap_dir + "/info.p", "wb")
    pickle.dump(o, pickle_f)
    pickle_f.close()

    r_step = o.r_in  ### set timestep based on period of innermost disk particles ###
    t_end = o.num_T * abs(time_peri)  # Current Target Period: 4.0 T
    #t_end = 1300.0 | units.yr

    integrate_disk_flyby(stars, planetesimals, t_end, time_peri, o.nb_end,
                         o.n_steps, r_step, o.snap_dir, o.fout, o.fredir,
                         o.br_dt, o.eta)

    factory = PF(int(o.sim_number),
                 snapshot_base=snap_base,
                 count=o.num_format)

    # Process snapshots ---> Make Plots
    factory.plot_all()  # <<<---
  f.write("Bridge Timestep: " + str(o.br_dt) + "\n")
  f.write("Huayno Timestep (eta): " + str(o.eta) + "\n")
  f.write("Inner Disk Radius: " + str(o.r_in) + "\n")
  f.write("Outer Disk Radius: " + str(o.r_out) + "\n")
  f.write("Number of Disk Particles: " + str(o.n_disk) + "\n")
  f.write("Disk Particle Power Law: " + str(o.power) + "\n")
  f.write("Orbit Pericenter: " + str(o.peri) + "\n")
  f.write("Orbit Eccentricity: " + str(o.ecc) + "\n")
  f.write("Orbit Inclination: " + str(o.incl) + "\n")
  f.write("Orbit Argument of Periapsis: " + str(o.omega) + "\n")
  f.write("Number of Steps: " + str(o.n_steps) + "\n")
  f.close()

  # Write 'o' dictionary into file (using Pickle)
  pickle_f = open(o.snap_dir + "/info.p", "wb")
  pickle.dump(o, pickle_f)
  pickle_f.close()

  r_step = o.r_in ### set timestep based on period of innermost disk particles ###
  t_end = o.num_T * abs(time_peri) # Current Target Period: 4.0 T
  #t_end = 1300.0 | units.yr
  
  integrate_disk_flyby(stars, planetesimals, t_end, time_peri, o.nb_end, o.n_steps,
                       r_step, o.snap_dir, o.fout, o.fredir, o.br_dt, o.eta)

  factory = PF(int(o.sim_number), snapshot_base = snap_base, count = o.num_format)

  # Process snapshots ---> Make Plots
  factory.plot_all() # <<<--- 
  
def grand_plot_all(o, grand):
  """ Make all the plots for the encompassing suite """
  factory = PF(int(o.sim_number), snapshot_base = o.snap_base, count = o.num_format, grand = grand)
  factory.plot_all()