{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# 3. C programming" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "The following code is needed only if you are using the notebook, because make command treats spaces and TAB characters differently. We need to be able to reconfigure Jupyter to treat TABS as characters, not replace them with 4 spaces, as it does by default." ] }, { "cell_type": "code", "execution_count": null, "metadata": { "scrolled": true }, "outputs": [], "source": [ "%%javascript\n", "IPython.tab_as_tab_everywhere = function(use_tabs) {\n", " if (use_tabs === undefined) {\n", " use_tabs = true; \n", " }\n", " // apply setting to all current CodeMirror instances\n", " IPython.notebook.get_cells().map(\n", " function(c) { return c.code_mirror.options.indentWithTabs=use_tabs; }\n", " );\n", " // make sure new CodeMirror instances created in the future also use this setting\n", " CodeMirror.defaults.indentWithTabs=use_tabs;\n", " };\n", "\n", "IPython.tab_as_tab_everywhere()" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## Gaussian packet evolution: a C project" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "This is the first homework to be handed in, a week-long project to graphically illustrate what happens to a Gaussian wave packet (we will make it up by adding a few individual waves) impacting on a barrier. We want to observe how some of the packet is reflected, and some undergoes quantum-mechanical tunneling, so both a reflected and a transmitted packet emerges after the impact. We will run the program repeatedly, changing the number of wave making up the packet, the number of points to plot, the initial energy of the incoming packet, etc. The physics of this scattering process will get discussed next, but first let's set up the mechanics of the new project:\n", "
\n",
"The second part of the homework is to try to figure out what is going on in an old Fortran program that does the same thing. We do not know either Fortran or C syntax, but hopefully the computational content, i.e. the algorithmic core, will be self-explanatory. The task is to \"port\", or convert this old code to C, and learn some basic C programming in the process."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%bash\n",
"## start with an empty directory for the project\n",
"if [ ! -d ~/5P10 ]; then\n",
" mkdir ~/5P10\n",
"fi\n",
"cd ~/5P10\n",
"rm -rf packet\n",
"mkdir packet"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%cd ~/5P10/packet"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%bash\n",
"pwd\n",
"ls -l"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Physics of tunnelling\n",
"\n",
"Review these notes from a QM course (PDF).\n",
"\n",
"Last session, we saw a working Fortran program. The relevant numerical part is this, and the notation is a straightforward translation of the expressions from the PDF notes. FORTRAN = \"FORmula TRANslation\"."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%file add_wave.f\n",
"C=======================================================================\t\n",
"\tsubroutine add_wave(x,P,A,E)\n",
"C..adds the specified wave to the the supplied complex storage array P\n",
"C the values of Psi are only calculated at points given in real array x \n",
"\treal*4 \t\tx(601),E\n",
"\tcomplex*8\tP(601),I,A,k1,K2,C1,b,c,d,f\n",
"\n",
"\tI = cmplx(0.,1.)\t!complex i\n",
"\n",
"\tk1 = sqrt(cmplx(E,0.))\n",
"\tK2 = sqrt(cmplx(E-1.,0.))\n",
"\n",
"\tC1 = -2*k1*K2-k1**2-2*k1*K2*exp(4*I*K2)+exp(4*I*K2)*k1**2\n",
" *\t-K2**2+K2**2*exp(4*I*K2)\n",
"\n",
"\tb = (k1**2-K2**2)/ C1 *(exp(2*I*(2*K2-k1))-exp(-2*I*k1))\n",
"\tc = -2*k1*(K2+k1)/ C1 *exp(I*(K2-k1))\n",
"\td = (-2*K2+2*k1)*k1/ C1 *exp(I*(-k1+3*K2))\n",
"\tf = -4*k1*K2/ C1 *exp(2*I*(K2-k1))\n",
"\n",
"\tdo j = 1,296\t\t! left of the barrier, x < -1\n",
"\t P(j) = P(j) + A * ( exp(I*k1*x(j)) + b*exp(-I*k1*x(j)) )\n",
"\tend do\n",
"\tdo j=297,304\t\t! inside the barrier, -1 < x < 1\n",
"\t P(j) = P(j) + A * ( c*exp(I*K2*x(j)) + d*exp(-I*K2*x(j)) )\n",
"\tend do\n",
"\tdo j=305,601\t\t! to the right of the barrier, x > 1\n",
"\t P(j) = P(j) + A * f*exp(I*k1*x(j))\n",
"\tend do\n",
"\t \n",
" \treturn\n",
"\tend"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## The beginnings of a C program\n",
"\n",
"Now we are ready to dive into C. We'll start with a simple skeleton C program, slightly enhanced compared to the version discussed the last time. The numeric heart of the program is still missing, and the output consists of meaningless numbers.\n",
"\n",
"We will need a Makefile and a gnuplot kernel loaded, for inline graphs."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%file Makefile\n",
"\n",
"OBJ = packet.o\n",
"\n",
"# these are pre-defined, but we can also change the default behaviour, if we have multiple compilers\n",
"CC = cc\n",
"#CC = icc\n",
"CFLAGS = -O -DVERSION=\"\\\"`date '+%Y-%b-%d-%H:%M'`\\\"\" \n",
"LIBS = -lm\n",
"\n",
"# special abbreviations: $@ is the file to be made; $? is the changed dependent(s).\n",
"\n",
"packet: $(OBJ)\n",
"\t$(CC) $(CFLAGS) -o $@ $(OBJ) $(LIBS)\n",
"\n",
"# this is the default implicit rule to convert xxx.c into xxx.o \n",
".c.o:\n",
"\t$(CC) $(CFLAGS) -c $<\n",
"\n",
"clean:\n",
"\trm -f packet *.o *~ core"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%file Vo.dat\n",
"-75\t0\n",
"-1\t0\n",
"-1\t1\n",
"1\t1\n",
"1\t0\n",
"75\t0"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%load_ext gnuplot_kernel"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%gnuplot inline pngcairo font \"Arial,16\" size 800,600"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%%file packet.c\n",
"/*\n",
" * packet.c\n",
" * packet - generate a Gaussian wavepacket impacting on an energy barrier.\n",
" *\n",
" * Completed: January.2018 (c) E.Sternin\n",
" * Revisions: \n",
" *\n",
" */\n",
"\n",
"#ifndef VERSION /* date-derived in Makefile */\n",
"#define VERSION \"2018.01\" /* default, that's when we first wrote the program */\n",
"#endif\n",
"\n",
"#include \n",
" real*4 x(601), ...\n",
"\tcomplex*8 P(601), ...\n",
"\t...\n",
"\tdo j = 1,601\n",
"\t x(j) = -75 + (j-1)*0.25\n",
"\t P(j) = cmplx(0.,0.)\n",
"\tend do\n",
"
\n",
"\n",
"C:\n",
"\n",
"...\n",
"#include <math.h>\n",
"#include <complex.h>\n",
"\n",
"#define NPTS 601 /* these cannot be changed via command-line switches */\n",
"#define X_MIN -75.0\n",
"#define X_MAX 75.0\n",
"...\n",
" double dx,x[NPTS], ...;\n",
" double complex Psi[NPTS];\n",
"...\n",
" dx = (X_MAX-X_MIN)/(double)(NPTS-1);\n",
" for (j=0; j < NPTS; j++) { \n",
" x[j]=X_MIN + j*dx;\n",
" Psi[j] = (double complex)0.0; /* start with a blank Psi(x) */\n",
" }\n",
"
"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Fortran:\n",
"\n",
"\tI = cmplx(0.,1.)\t!complex i\n",
"\n",
"\tk1 = sqrt(cmplx(E,0.))\n",
"\tK2 = sqrt(cmplx(E-1.,0.))\n",
"\n",
"\tC1 = -2*k1*K2-k1**2-2*k1*K2*exp(4*I*K2)+exp(4*I*K2)*k1**2\n",
" *\t-K2**2+K2**2*exp(4*I*K2)\n",
"\n",
"\tb = (k1**2-K2**2)/ C1 *(exp(2*I*(2*K2-k1))-exp(-2*I*k1))\n",
"\tc = -2*k1*(K2+k1)/ C1 *exp(I*(K2-k1))\n",
"\td = (-2*K2+2*k1)*k1/ C1 *exp(I*(-k1+3*K2))\n",
"\tf = -4*k1*K2/ C1 *exp(2*I*(K2-k1))\n",
"
\n",
"\n",
"C:\n",
"\n",
"#include <complex.h>\n",
"...\n",
" for (i=-n; i < n; i++) {\n",
" p_now = p0 + (double)i * p_step;\n",
" E = p_now*p_now;\n",
" A = A0*exp(-pow((p_now-p0)/(2*s),2))*cexp(-I*(p_now*x0 + E*t));\n",
"\n",
" k1 = csqrt((double complex)(E));\n",
" k2 = csqrt((double complex)(E-U));\n",
" ce4 = cexp(4*I*k2);\n",
" C1 = -2*k1*k2-k1*k1-2*k1*k2*ce4+ce4*(k1*k1+k2*k2)-k2*k2;\n",
"\n",
" b = (k1*k1-k2*k2)/ C1 *(cexp(2*I*(2*k2-k1))-cexp(-2*I*k1));\n",
" c = -2*k1*(k2+k1)/ C1 *cexp(I*(k2-k1));\n",
" d = (-2*k2+2*k1)*k1/ C1 *cexp(I*(-k1+3*k2));\n",
" f = -4*k1*k2/ C1 *cexp(2*I*(k2-k1));\n",
"\n",
" ...\n",
" }\n",
"
"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Fortran:\n",
"\n",
"\tdo j = 1,296\t\t! left of the barrier, x < -1\n",
"\t P(j) = P(j) + A * ( exp(I*k1*x(j)) + b*exp(-I*k1*x(j)) )\n",
"\tend do\n",
"\tdo j=297,304\t\t! inside the barrier, -1 < x < 1\n",
"\t P(j) = P(j) + A * ( c*exp(I*K2*x(j)) + d*exp(-I*K2*x(j)) )\n",
"\tend do\n",
"\tdo j=305,601\t\t! to the right of the barrier, x > 1\n",
"\t P(j) = P(j) + A * f*exp(I*k1*x(j))\n",
"\tend do\n",
"
\n",
"\n",
"C:\n",
"\n",
" for (j=0; j < NPTS; j++) {\n",
" if (x[j] < -1.) { /* to the left of the barrier, x < -1 */\n",
" Psi[j] += A*( cexp(I*k1*x[j]) + b*cexp(-I*k1*x[j]) );\n",
" }\n",
" else if (x[j] > 1.) { /* to the right of the barrier, x > 1 */\n",
" Psi[j] += A*f*cexp(I*k1*x[j]);\n",
" }\n",
" else { /* inside the barrier, -1 < x < 1 */\n",
" Psi[j] += A*( c*cexp(I*k2*x[j]) + d*cexp(-I*k2*x[j]) );\n",
" }\n",
" }\n",
"
"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"hide_input": false,
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.6.8"
},
"toc": {
"nav_menu": {},
"number_sections": false,
"sideBar": true,
"skip_h1_title": false,
"toc_cell": false,
"toc_position": {
"height": "1033.6px",
"left": "0px",
"right": "1440.97px",
"top": "79.4px",
"width": "156.033px"
},
"toc_section_display": "block",
"toc_window_display": true
}
},
"nbformat": 4,
"nbformat_minor": 2
}