30:T89c,function redshiftPlot(model, what, zmax, {nstep=31, height=500, label='f(z)', margin={top:20, right:30, bottom:50, left:50}}={}) { // Prepare the data to plot. const z = linspace(0, zmax, nstep, {dtype:Float64Array}); const f = model[what].bind(model); const fz = z.map(z => f(z)); const data = tidy({z:z, fz:fz}); // Create the graphics. const zscale = d3.scaleLinear() .domain([0, zmax]) .range([margin.left,width-margin.right]); const zaxis = g => g .attr("transform", `translate(0,${height-margin.bottom})`) .call(d3.axisBottom(zscale).ticks(width/80).tickSizeOuter(0)) .append('text') .text('Redshift z') .attr('text-anchor', 'middle') .attr('dominant-baseline', 'baseline') .attr('font-size', '18px') .attr('fill', '#000') .attr('x', (margin.left + width - margin.right)/2) .attr('y', margin.bottom); // relative to the axis. const zgrid = g => g .attr("stroke", "lightgray") .selectAll('line') .data(zscale.ticks()) .join('line') .attr('x1', d => zscale(d)) .attr('x2', d => zscale(d)) .attr('y1', margin.top) .attr('y2', height-margin.bottom); const fscale = d3.scaleLinear() .domain([0, d3.max(fz)]) .range([height-margin.bottom, margin.top]); const faxis = g => g .attr("transform", `translate(${margin.left},0)`) .call(d3.axisLeft(fscale).ticks(height/40)) .append('text') .text(label) .attr('text-anchor', 'middle') .attr('dominant-baseline', 'hanging') .attr('font-size', '18px') .attr('fill', '#000') .attr('transform', 'rotate(-90)') .attr('x', -(margin.top + height - margin.bottom)/2) // relative to the axis. .attr('y', -margin.left); const line = d3.line() .x(d => zscale(d.z)) .y(d => fscale(d.fz)); const plot = html` `; return plot; }31:Td86,function cosmicPie(model, z=0, pad=5) { // Based on https://observablehq.com/@d3/pie-chart const Otot = 1 - model.Ok(z); const data = [ {name: 'Dark Energy', value: model.Ode(z)/Otot, color: 'plum'}, {name: 'Dark Matter', value: model.Odm(z)/Otot, color: 'chocolate'}, {name: 'Atoms', value: model.Ob(z)/Otot, color: 'brown'}, {name: 'Neutrinos', value: model.Onu(z)/Otot, color: 'green'}, {name: 'Photons', value: model.Ogamma(z)/Otot, color: 'skyblue'} ]; const colors = Object.fromEntries(data.map(D => [D['name'], D['color']])); const radius = Math.floor(width/3 - pad); const pie = d3.pie().sort(null).value(d => d.value); const arc = d3.arc().innerRadius(0).outerRadius(radius); // Calculate label position for datum d. const labelPos = d => { const f = Math.min(1, d.value / 0.15); const r = radius * (1 - 0.5*f); return d3.arc().innerRadius(r).outerRadius(r).centroid(d); }; // Format percentage value for datum d. const fracFormat = frac => frac < 0.0001 ? '<0.01%' : Number(100*frac).toFixed(2) + '%'; const arcs = pie(data); const svg = d3.create("svg") .attr("viewBox", [-radius - pad, -radius - pad, width, 2 * (radius + pad)]) .attr("font-family", "sans-serif") .attr("text-anchor", "middle"); // Draw arcs svg.append("g") .attr("stroke", "white") .selectAll("path") .data(arcs) .join("path") .attr("fill", d => colors[d.data.name]) .attr("d", arc) .append("title") .text(d => `${d.data.name}: ${fracFormat(d.data.value)}`); // Draw arc labels svg.append("g") .attr("font-size", "16") .attr("font-weight", "bold") .selectAll("text") .data(arcs) .join("text") .attr("transform", d => `translate(${labelPos(d)})`) .call(text => text .filter(d => d.value > 0.04) // Hide labels when value < 4% .append("tspan") .text(d => d.data.name)); // Draw legend labels for the current t/a/z in the upper right. const when = model.lookback_time(0, z); const age = model.age; const aval = 1/(1 + z); const labels = [ {name: 't', value:`now ${when > 0 ? "-":"+"} ${autoprefix(Math.abs(when*1e9), {clip:0.1})}yr`}, {name: '', value:`bigbang + ${autoprefix((age-when)*1e9, {clip:0.1})}yr`}, {name: 'a(t)', value:autoprec(aval)}, {name: 'z(t)', value:Number(z).toFixed(1)} ]; svg.append("g") .attr("font-size", "24") .attr("transform", `translate(${3*width/8},${-radius+40})`) .selectAll("g") .data(labels) .join("g") .attr("transform", (d,i) => `translate(0,${40*i})`) .call(g => g .append("text") .attr("text-anchor", "end") .text(d => d.name + " =")) .call(g => g .append("text") .attr("text-anchor", "start") .attr("x", 8) .text(d => d.value)); // Draw legend labels for current fractions in the lower right. svg.append("g") .attr("font-size", "18") .attr("text-anchor", "end") .attr("transform", `translate(${7*width/16},${radius-30*data.length})`) .selectAll("g") .data(data) .join("g") .attr("transform", (d,i) => `translate(0,${30*i})`) .attr("fill", d => d.color) .call(g => g .append("text") .text(d => d.name + " :")) .call(g => g .append("text") .attr("x", 70) .text(d => fracFormat(d.value))); return svg.node(); }32:T57f,scaleInput = function({a0=1, amin=5e-5, amax=2, s=5, name='scale', showlabel=true}={}) { let [xmin,x0,xmax] = [amin,a0,amax].map(a => Math.asinh(s*(1-1/a))); // Ensure that amin,a0,amax correspond to a exact steps, increasing amax if needed. const xstep = (x0 - xmin)/500; const nstep = Math.ceil((xmax - xmin) / xstep); xmax = xmin + xstep * nstep; const slider=html``; const today=html``; const zout=html``, aout=html``; const label = showlabel ? html` z=${zout} , a=${aout}` : ''; const form=html`

`; today.addEventListener("click", () => { slider.value = x0; form.dispatchEvent(new CustomEvent("input")); }); const getz = () => { const z = -Math.sinh(slider.value)/s; const a = 1/(1+z); if(showlabel) { zout.innerText = Number(z).toFixed(2); aout.innerText = a > 0.01 ? Number(a).toFixed(3) : Number(a).toExponential(1); } return z; } form.addEventListener("submit", event => event.preventDefault()); form.addEventListener("change", () => form.dispatchEvent(new CustomEvent("input"))); form.addEventListener("input", () => form.value = getz()); form.value = getz(); return form; }33:T6d7,function modelInput(initial=Planck15Params, style=defaultInputStyle) { let form=html`

`; let inputs={}, params={}; // Define the parameter input sliders. for(const name of style['layout']) { const is_param = (name!='flat' && name!='lambda'); const {input, div} = is_param ? parameter(name, initial[name], style[name]) : checkbox(name, style[name]); inputs[name] = input; if(is_param) params[name] = input; form.appendChild(div); } // Extract model parameters from the inputs defined above and construct the model. const getModel = () => { const out = {}; for(const name in params) out[name] = params[name].valueAsNumber; return new w0waCDM(out); } // Initialize the input checkboxes. if(inputs['flat'].checked = (initial['Ok0'] == 0)) inputs['Ok0'].disabled = true; if(inputs['lambda'].checked = (initial['w0'] == -1 && initial['wa'] == 0)) inputs['w0'].disabled = inputs['wa'].disabled = true; // Implement the checkbox logic. inputs['flat'].addEventListener("change", () => { inputs['Ok0'].disabled = inputs['flat'].checked; if(inputs['flat'].checked) inputs['Ok0'].value = 0; }); inputs['lambda'].addEventListener("change", () => { inputs['wa'].disabled = inputs['w0'].disabled = inputs['lambda'].checked; if(inputs['lambda'].checked) [inputs['wa'].value, inputs['w0'].value] = [0,-1]; }); // Implement event handling following https://observablehq.com/@mbostock/form-input form.addEventListener("submit", event => event.preventDefault()); form.addEventListener("change", () => form.dispatchEvent(new CustomEvent("input"))); form.addEventListener("input", () => form.value = getModel()); form.value = getModel(); return form; }34:[] 2e:{"title":"CosmoViz","cells":[{"id":0,"value":"md`# CosmoViz\n\nVisualizations for cosmology building on the numerical calculations in \\`@dkirkby/cosmocalc\\`.\n\nSpecify a model, starting from Planck15 defaults:\n`","pinned":false,"hidden":false,"mode":"ojs"},{"id":16,"value":"viewof model=modelInput();","pinned":false,"hidden":false,"mode":"ojs"},{"id":360,"value":"md`Plot ${tex`\\Omega_m(z)`}:`","pinned":false,"hidden":false,"mode":"ojs"},{"id":196,"value":"redshiftPlot(model, 'Om', 3)","pinned":true,"hidden":false,"mode":"ojs"},{"id":363,"value":"md`Visualize a pie-chart of cosmic components for this model:`","pinned":false,"hidden":false,"mode":"ojs"},{"id":229,"value":"pie = cosmicPie(model, zpie)","pinned":true,"hidden":false,"mode":"ojs"},{"id":366,"value":"md`Specify what point during the expansion is displayed in the pie chart:`","pinned":false,"hidden":false,"mode":"ojs"},{"id":309,"value":"viewof zpie=scaleInput({showlabel:false})","pinned":true,"hidden":false,"mode":"ojs"},{"id":5,"value":"md`---`","pinned":false,"hidden":false,"mode":"ojs"},{"id":186,"value":"$30","pinned":false,"hidden":false,"mode":"ojs"},{"id":213,"value":"$31","pinned":false,"hidden":false,"mode":"ojs"},{"id":610,"value":"autoprefix = function(x, {clip=1e-10, digits=1}={}) {\n if(Math.abs(x) < clip) return \"0\";\n const mult = Math.max(-5, Math.min(5, Math.floor(Math.log10(Math.abs(x)) / 3)));\n const prefix = \"fpnμm kMGTP\".slice(5+mult, 6+mult).trim();\n return Number(x/10**(3*mult)).toFixed(digits) + prefix;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":647,"value":"autoprec = function(x, digits=2, maxprec=5, thresh=4) {\n let scaled=Math.abs(x);\n for(let prec=0; prec<=maxprec; prec++) if((scaled*=10) >= thresh) return Number(x).toFixed(prec+digits-1);\n return \"0\";\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":298,"value":"$32","pinned":false,"hidden":false,"mode":"ojs"},{"id":107,"value":"defaultInputStyle=({\n layout: ['H0','Om0','Ob0','Ok0','flat','w0','wa','lambda','Tcmb0','Neff'],\n H0: ({min:60, max:80, step:0.01, label:tex`H_0`}),\n Om0: ({min:0, max:1, step:0.001, label:tex`\\Omega_{\\text{m},0}`}),\n Ob0: ({min:0, max:1, step:0.001, label:tex`\\Omega_{\\text{b},0}`}),\n Ok0: ({min:-0.2, max:0.2, step:0.001, label:tex`\\Omega_{\\text{k},0}`}),\n w0: ({min:-1.2, max:-0.8, step:0.001, label:tex`w_0`}),\n wa: ({min:-0.5, max:0.5, step:0.001, label:tex`w_a`}),\n Tcmb0: ({min:2, max:4, step:0.001, label:tex`T_{\\text{cmb},0}`}),\n Neff: ({min:2.9, max:3.2, step:0.001, label:tex`N_{\\text{eff}}`}),\n flat: ({label:'Flat'}),\n lambda: ({label:tex`\\Lambda\\text{CDM}`})\n});","pinned":false,"hidden":false,"mode":"ojs"},{"id":11,"value":"$33","pinned":false,"hidden":false,"mode":"ojs"},{"id":115,"value":"function parameter(name, value, S) {\n const input=html``;\n const div=html`

${input}${S['label']}

`;\n return ({input:input, div:div});\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":147,"value":"function checkbox(name, S) {\n const input=html``;\n const div=html`

${input}${S['label']}

`;\n return ({input:input, div:div});\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":215,"value":"d3 = require(\"d3@6\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":191,"value":"import {plot} from '@dkirkby/pyplot'","pinned":false,"hidden":false,"mode":"ojs"},{"id":270,"value":"import {slider} from '@jashkenas/inputs'","pinned":false,"hidden":false,"mode":"ojs"},{"id":675,"value":"import {tidy} from '@dkirkby/array-utilities'","pinned":false,"hidden":false,"mode":"ojs"},{"id":187,"value":"import {linspace} from '@dkirkby/jumpy'","pinned":false,"hidden":false,"mode":"ojs"},{"id":8,"value":"import {w0waCDM,LambdaCDM,FlatLambdaCDM,Planck15Params} from '@dkirkby/cosmocalc'","pinned":false,"hidden":false,"mode":"ojs"}],"files":"$W34","stdlib":"1","titling":"auto"}