Fraunhofer_1slit_FINAL.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## FRAUNHOFEROVA DIFRAKCIJA U 1 Dimenziji"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "import math\n",
    "from scipy.fftpack import fft\n",
    "from scipy.fftpack import ifftshift,  fftshift\n",
    "from mpl_toolkits.mplot3d import Axes3D\n",
    "plt.style.use('dark_background')\n",
    "plt.rcParams['image.cmap'] = 'plasma' "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "\n",
    "def Fraunhofer_input(lambd,w,z):\n",
    "    N=1048   #rezolucija\n",
    "    L=5.    # širina zaslona \n",
    "\n",
    "    lamb=lambd*10**(-9) #(mm)\n",
    "    z=z/1000.\n",
    "#    w=1.  #širina otvora (mm)\n",
    "#    z=20.    #udaljenost otvor - zaslon (mm)\n",
    "    k=2*np.pi/lamb   \n",
    "\n",
    "    delta_src=L/N     #frekvencija uzorkovanja\n",
    "    #critical_sampl_z=N*delta_src*z/lamb\n",
    "\n",
    "    xv=np.linspace(-N/2,N/2-1,N)\n",
    "    yv=np.linspace(-N/2,N/2-1,N)\n",
    "    [x1,y1]=np.meshgrid(xv,yv)\n",
    "    x1=x1*delta_src  #(mm)\n",
    "    y1=y1*delta_src  # (mm) \n",
    " \n",
    "    #definisati kvadratni otvor\n",
    "    X=np.where(np.abs(x1)<w/4,1,0)\n",
    "  #  Y=np.where(np.abs(y1)<3*w,1,0)\n",
    "    #vrijednosti sirine za Fresnel single slit: w/7, w/2, w/50\n",
    "    \n",
    "    #double slit\n",
    "    #double slit\n",
    "    #X[np.logical_and(x1<-0.1*w, x1>-0.25*w)]=1.0\n",
    "    #X[np.logical_and(x1<0.25*w, x1>0.1*w)]=1.0\n",
    "\n",
    "\n",
    "    u1=X\n",
    "    x1=x1/1000.\n",
    "#    return (u1,k,x1,X)\n",
    "\n",
    "\n",
    "    g=u1\n",
    "    G=fftshift(np.fft.fft(g))\n",
    "    h=np.exp(1j*k*z)/np.sqrt(1j*lamb*z)*np.exp(1j*k/(2*z)*x1**2)\n",
    "    u2=h*G\n",
    "    \n",
    "    identity=np.ones((N,N))\n",
    "    u2=u2*identity\n",
    "    \n",
    "    \n",
    "    \n",
    "    I=1./(lamb*z)*np.abs(u2)**2\n",
    "    I_1d=np.zeros(N)\n",
    "    for a in range (1,N):\n",
    "        I_1d[a]=I[0][a]\n",
    "        \n",
    "    \n",
    "    #I_1d=I_1d/I_1d.max()\n",
    "        \n",
    "    return(u2,I_1d,y1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "lamb=632.\n",
    "w=0.05\n",
    "z=20.\n",
    "\n",
    "u1,I1,x1=Fraunhofer_input(lamb,5*w,1000.)\n",
    "u2,I2,x2=Fraunhofer_input(lamb,w,100000.)\n",
    "u3,I3,x3=Fraunhofer_input(lamb,w/2,1000.)\n",
    "\n",
    "\n",
    "I1=np.abs(I1)\n",
    "I2=np.abs(I2)\n",
    "I3=np.abs(I3)\n",
    "\n",
    "I1=I1/I1.max()\n",
    "I2=I2/I2.max()\n",
    "I3=I3/I3.max()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "fig2=plt.figure(figsize=(11,10)) \n",
    "\n",
    "ax1=fig2.add_subplot(4,3,4)\n",
    "plt.title('(A) z = 0.5 m, w = 0.05 mm')\n",
    "plt.imshow(np.abs(u1))\n",
    "#plt.xlim(324,724)\n",
    "plt.ylim(460,588)\n",
    "plt.axis('off')\n",
    "#tick_locs = [131,393,524,655,786]\n",
    "#tick_lbls = [-2,-1,0,1,2]\n",
    "#plt.xticks(tick_locs, tick_lbls)\n",
    "#plt.yticks(tick_locs, tick_lbls)\n",
    "plt.xlabel('(A)')\n",
    "#plt.ylabel('Y(mm)')\n",
    "\n",
    "ax2=fig2.add_subplot(4,3,5)\n",
    "plt.title('(B) z = 1.0 m, w = 0.025 mm')\n",
    "plt.imshow(np.abs(u2))\n",
    "#plt.xlim(324,724)\n",
    "plt.ylim(460,588)\n",
    "plt.xlabel('(B)')\n",
    "plt.axis('off')\n",
    "\n",
    "ax3=fig2.add_subplot(4,3,6)\n",
    "plt.title('(C) z = 1.0 m, w = 0.1 mm')\n",
    "plt.imshow(np.abs(u3))\n",
    "#plt.xlim(324,724)\n",
    "plt.ylim(460,588)\n",
    "plt.xlabel('(C)')\n",
    "plt.axis('off')\n",
    "\n",
    "ax4=fig2.add_subplot(2,1,2)\n",
    "plt.title('Fraunhoferova difrakcija - jednostruka pukotina')\n",
    "\n",
    "ax4_1=ax4.plot(x1,np.abs(I1),'r-')\n",
    "ax4_2=ax4.plot(x2,np.abs(I2),'b-')\n",
    "ax4_3=ax4.plot(x3,np.abs(I3),'y:')\n",
    "\n",
    "\n",
    "ax4.text(-0.3,0.5,'(A)', color='red',fontsize=16)\n",
    "ax4.text(0.22,0.5,'(B)', color='blue',fontsize=16)\n",
    "ax4.text(0.85,0.5,'(C)', color='yellow',fontsize=16)\n",
    "\n",
    "plt.xlabel('X(mm)')\n",
    "plt.ylabel('Relativni intenzitet')\n",
    "fig2.savefig('Fraunhofer_1slit.png')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.14"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}