Skip to content

Commit

Permalink
Merge pull request #1104 from agnwinds/xband
Browse files Browse the repository at this point in the history
xband: windsave2fits, changes to radiation temperature and fixes to disk_diag file.
  • Loading branch information
jhmatthews authored Oct 17, 2024
2 parents b1728eb + 8d3e35a commit 9a9cf31
Show file tree
Hide file tree
Showing 14 changed files with 978 additions and 44 deletions.
5 changes: 5 additions & 0 deletions docs/sphinx/source/output/diagnostics.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,3 +10,8 @@ where the in a multiprocessor run, the number refers to the log of a specific th

Inspecting these logs is important for understanding whether a Python run is successful,
and for understanding why if failed if that is the case.

Additionally in this directory one will find a file **whatever.disk.diag** if the system contained a disk. This file contains information
about the structure of the disk as a function of disk radius. It also indicates how many photons hit the disk with how much total energy
as a function of radius. It provides several estimates of the effective temperature of the radiation that hits the disk, and determines
how the disk temperature of the disk would so that it would reradiate the sum of the viscous energy and the energy associated with illumination.
3 changes: 3 additions & 0 deletions docs/sphinx/source/py_progs/MakeMacro/MacroCombine.rst
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,11 @@
.. autosummary::

add_gtot
combine_collisions
plot_xsec
read_collisions
read_phot
redo_collisions
redo_lines
redo_phot
reweight
Expand Down
9 changes: 8 additions & 1 deletion source/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,7 @@ LDFLAGS+= -L$(LIB) -lm -lgsl -lgslcblas $(CUDA_LIBS)
#Note that version should be a single string without spaces.


VERSION = 88d
VERSION = 88f



Expand Down Expand Up @@ -264,6 +264,7 @@ rhf_objects = rad_hydro_files.o $(python_objects)
modify_wind_objects = modify_wind.o $(python_objects)
inspect_wind_objects = inspect_wind.o $(python_objects)
py_optd_obj = py_optd.o py_optd_output.o py_optd_trans.o py_optd_util.o py_optd_extern_init.o $(python_objects)
windsave2fits_objects = windsave2fits.o $(python_objects)

run_indent:
../py_progs/run_indent.py -all_no_headers
Expand All @@ -280,6 +281,12 @@ windsave2table: startup $(table_objects) $(CUDA_OBJECTS)
mv $@ $(BIN)/windsave2table$(VERSION)
@if [ $(INDENT) = yes ] ; then ../py_progs/run_indent.py -changed ; fi

windsave2fits: startup $(windsave2fits_objects) $(CUDA_OBJECTS)
$(CC) $(CFLAGS) $(windsave2fits_objects) $(CUDA_OBJECTS) $(LDFLAGS) -lcfitsio -o windsave2fits
cp $@ $(BIN)
mv $@ $(BIN)/windsave2fits$(VERSION)
@if [ $(INDENT) = yes ] ; then ../py_progs/run_indent.py -changed ; fi

rad_hydro_files: startup $(rhf_objects) $(CUDA_OBJECTS)
$(CC) $(CFLAGS) $(rhf_objects) $(CUDA_OBJECTS) $(LDFLAGS) -o rad_hydro_files
cp $@ $(BIN)
Expand Down
12 changes: 7 additions & 5 deletions source/communicate_plasma.c
Original file line number Diff line number Diff line change
Expand Up @@ -986,11 +986,11 @@ reduce_simple_estimators (void)
ion_helper2 = calloc (sizeof (double), NPLASMA * nions);
inner_ion_helper = calloc (sizeof (double), NPLASMA * n_inner_tot);
inner_ion_helper2 = calloc (sizeof (double), NPLASMA * n_inner_tot);
/* JM -- added routine to average the qdisk quantities. The 2 is because
we only have two doubles to worry about (heat and ave_freq) and
/* Routine to average the qdisk quantities. The 3 is because
we have three doubles to worry about (emit, heat and ave_freq) and
two integers (nhit and nphot) */
qdisk_helper = calloc (sizeof (double), NRINGS * 2);
qdisk_helper2 = calloc (sizeof (double), NRINGS * 2);
qdisk_helper = calloc (sizeof (double), NRINGS * 3);
qdisk_helper2 = calloc (sizeof (double), NRINGS * 3);

flux_helper = calloc (sizeof (double), NPLASMA * NFLUX_ANGLES * 3);
flux_helper2 = calloc (sizeof (double), NPLASMA * NFLUX_ANGLES * 3);
Expand Down Expand Up @@ -1068,6 +1068,7 @@ reduce_simple_estimators (void)
{
qdisk_helper[mpi_i] = qdisk.heat[mpi_i] / np_mpi_global;
qdisk_helper[mpi_i + NRINGS] = qdisk.ave_freq[mpi_i] / np_mpi_global;
qdisk_helper[mpi_i + 2 * NRINGS] = qdisk.emit[mpi_i] / np_mpi_global;
}


Expand All @@ -1080,7 +1081,7 @@ reduce_simple_estimators (void)
MPI_Allreduce (flux_helper, flux_helper2, NPLASMA * 3 * NFLUX_ANGLES, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
MPI_Allreduce (ion_helper, ion_helper2, NPLASMA * nions, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
MPI_Allreduce (inner_ion_helper, inner_ion_helper2, NPLASMA * n_inner_tot, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
MPI_Allreduce (qdisk_helper, qdisk_helper2, 2 * NRINGS, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
MPI_Allreduce (qdisk_helper, qdisk_helper2, 3 * NRINGS, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);

/* Unpacking stuff */
for (mpi_i = 0; mpi_i < NPLASMA; mpi_i++)
Expand Down Expand Up @@ -1157,6 +1158,7 @@ reduce_simple_estimators (void)
{
qdisk.heat[mpi_i] = qdisk_helper2[mpi_i];
qdisk.ave_freq[mpi_i] = qdisk_helper2[mpi_i + NRINGS];
qdisk.emit[mpi_i] = qdisk_helper2[mpi_i + 2 * NRINGS];
}

/* now we've done all the doubles so we can free their helper arrays */
Expand Down
133 changes: 114 additions & 19 deletions source/disk_init.c
Original file line number Diff line number Diff line change
Expand Up @@ -161,8 +161,8 @@ disk_init (rmin, rmax, m, mdot, freqmin, freqmax, ioniz_or_extract, ftot)
}
else if (spectype == SPECTYPE_BB_FCOL)
{
fcol = disk_colour_correction(t);
emit = emittance_bb (freqmin, freqmax, fcol * t) / pow(fcol, 4.0);
fcol = disk_colour_correction (t);
emit = emittance_bb (freqmin, freqmax, fcol * t) / pow (fcol, 4.0);
}
else
{
Expand Down Expand Up @@ -220,8 +220,8 @@ disk_init (rmin, rmax, m, mdot, freqmin, freqmax, ioniz_or_extract, ftot)
}
else if (spectype == SPECTYPE_BB_FCOL)
{
fcol = disk_colour_correction(t);
emit = emittance_bb (freqmin, freqmax, fcol * t) / pow(fcol, 4.0);
fcol = disk_colour_correction (t);
emit = emittance_bb (freqmin, freqmax, fcol * t) / pow (fcol, 4.0);
}
else
{
Expand Down Expand Up @@ -354,69 +354,164 @@ qdisk_init (rmin, rmax, m, mdot)
}


/**********************************************************/
/**
* @brief Reinitialize portions a structure (qdisk) for recording information about
* photons/energy impinging
* on the disk.
*
* @return Always return zero
*
*
* ###Notes###
*
* The structure qdisk records information about how many photons hit the disk
* and what the impact of these photons would be if these
* photons are used to modify the temperatures structure
* of the disk
*
* This routine records information about the
* photons which were emitted from the disk
* and zeros various arrays that will record information
* about photons that will hit the disk
*
*
**********************************************************/

int
qdisk_reinit (p)
PhotPtr p;
{
int nphot, i, n;
double rho;
struct photon pp;
for (n = 0; n < NRINGS; n++)
{
qdisk.emit[n] = qdisk.nhit[n] = qdisk.heat[n] = qdisk.nphot[n] = qdisk.w[n] = qdisk.ave_freq[n] = 0;
}

for (nphot = 0; nphot < NPHOT; nphot++)
{
stuff_phot (&p[nphot], &pp);
if (pp.origin_orig == PTYPE_DISK)
{
rho = sqrt (pp.x[0] * pp.x[0] + pp.x[1] * pp.x[1]);

i = 0;
while (rho > qdisk.r[i] && i < NRINGS - 1)
i++;
i--; /* So that the heating refers to the heating between i and i+1 */

qdisk.emit[i] += pp.w;
qdisk.nphot[i] += 1;


}
}

return (0);
}


/**********************************************************/
/**
* @brief Save the information in the qdisk structure to a file
*
* @param [in] char * diskfile Name of the file whihc is writteen
* @param [in] double ztot The total luminosity of the disk as
* calculate over multiple cycles
* @param [in] int ichoice A switch that just says whether the weighted
* frequency has been converted to a real frequency (0 = No, 1=Yes)
* @return Always returns 0
*
* The routine reformats the data about disk heating which has
* been accumulated and writes it to a file
*
* ###Notes###
*
* The data concerning heating by the disk is built up during
* a the ionization cycles
*
* The file that is produced should be readable as an astropy
* table
*
* Here is a definition of the columnts
*
* * r the radius of the wirng
* * v the velocity of the ring
* * zdisk the zheight ot the rign
* * t_disk the temperature derived from viscous heating or the input grid
* * g the log of the gravity
* * emit the luminosity of photons emitted by the ring in this cycle
* * nemit the number photons emitted by the ring in this cycle
* * heat the total luminosity of photons that hit the ring
* * nhit the number of photon bundles hitting a ring
* * ehit/emit the ratio of the energy of photons that hit the ring/to those
* to those that are emitted by the ring
* * t_heat the temperature calculated from the energy of the
* photons hitting a ring
* * t_freq The temperature calculated from the weighted frequency of
* photons hitting the ring
* * W_freq Assuming t_freq for the temperature, this gives ther
* ratio of the energy flux to LTE
* * t_rerad The temperature required to radiate away both the
* the viscous heating and the heating from photons
* in the next cycle
*
**********************************************************/

int
qdisk_save (diskfile, ztot)
qdisk_save (diskfile, ichoice)
char *diskfile;
double ztot;
int ichoice;
{
FILE *qptr;
int n;
double area, theat, ttot;
double ratio[NRINGS];
qptr = fopen (diskfile, "w");
fprintf (qptr,
" r v zdisk t_disk g heat nhit nhit/emit t_heat t_irrad W_irrad t_tot\n");
// fprintf (qptr,
// "---------- ---------- ---------- --------- --------- --------- ----- --------- --------- --------- --------- ---------\n");
" r v zdisk t_disk g emit nemit heat nhit ehit/emit t_heat t_freq W_freq t_rerad\n");

for (n = 0; n < NRINGS; n++)
{
if (n < NRINGS - 1)
{
// The factor of 2 comes from the fact that the disk has two sides
area = (2. * PI * (qdisk.r[n + 1] * qdisk.r[n + 1] - qdisk.r[n] * qdisk.r[n]));
}
else
{
area = 0;
}
theat = qdisk.heat[n] / area;
theat = pow (theat / STEFAN_BOLTZMANN, 0.25);
//theat is temperature if no internal energy production
if (qdisk.nhit[n] > 0)
{

qdisk.ave_freq[n] /= qdisk.heat[n];
/* During photon transport ave_freq is actually the w x frequency */
if (ichoice == 0)
qdisk.ave_freq[n] /= qdisk.heat[n];

qdisk.t_hit[n] = PLANCK * qdisk.ave_freq[n] / (BOLTZMANN * 3.832);
//Basic conversion from freq to T
qdisk.w[n] = qdisk.heat[n] / (4. * PI * STEFAN_BOLTZMANN * area * qdisk.t_hit[n] * qdisk.t_hit[n] * qdisk.t_hit[n] * qdisk.t_hit[n]);
ratio[n] = 99.;
if (qdisk.emit[n] > 0.0)
ratio[n] = qdisk.heat[n] / qdisk.emit[n];
theat = qdisk.heat[n] / area;
theat = pow (theat / STEFAN_BOLTZMANN, 0.25);
//theat is temperature if no internal energy production

}
else
{
qdisk.ave_freq[n] = 0.0;
qdisk.t_hit[n] = 0.0;
qdisk.w[n] = 0.0;
ratio[n] = 0.0;
theat = 0;
}
ttot = pow (qdisk.t[n], 4) + pow (theat, 4);
ttot = pow (ttot, 0.25);
fprintf (qptr,
"%9.4e %9.4e %0.4e %8.3e %8.3e %8.3e %5d %8.3e %8.3e %8.3e %8.3e %8.3e\n",
"%9.4e %9.4e %0.4e %8.3e %8.3e %8.3e %10d %8.3e %10d %8.3e %8.3e %8.3e %10.3e %8.3e\n",
qdisk.r[n], qdisk.v[n], zdisk (qdisk.r[n]), qdisk.t[n], qdisk.g[n],
qdisk.heat[n], qdisk.nhit[n], qdisk.heat[n] * NRINGS / ztot, theat, qdisk.t_hit[n], qdisk.w[n], ttot);
qdisk.emit[n], qdisk.nphot[n], qdisk.heat[n], qdisk.nhit[n], ratio[n], theat, qdisk.t_hit[n], qdisk.w[n], ttot);
}

fclose (qptr);
Expand Down
2 changes: 1 addition & 1 deletion source/disk_photon_gen.c
Original file line number Diff line number Diff line change
Expand Up @@ -189,7 +189,7 @@ photo_gen_disk (p, weight, f1, f2, spectype, istart, nphot)
}
else if (spectype == SPECTYPE_BB_FCOL)
{
fcol = disk_colour_correction(disk.t[nring]);
fcol = disk_colour_correction (disk.t[nring]);
p[i].freq = planck (fcol * disk.t[nring], freqmin, freqmax);
}
else if (spectype == SPECTYPE_UNIFORM)
Expand Down
Loading

0 comments on commit 9a9cf31

Please sign in to comment.