Joining-cfiles.ipynb

{
 "metadata": {
  "name": "",
  "signature": "sha256:acd67330cf5e5d84106d99dc7693c4d40882b2673ef61205e68b5533aa0f5bda"
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import pickle\n",
      "import numpy as np\n",
      "\n",
      "#Load:\n",
      "files = [ open( 'BEMSolver_cfiles/atrap_cct_simulation_{}.pkl'.format(j) ) for j in range(1,8) ]\n",
      "pkls = [ pickle.load(f) for f in files ]\n",
      "\n",
      "#Concatenate along the z axis:\n",
      "xlen = len(pkls[0].X)\n",
      "ylen = len(pkls[0].Y)\n",
      "pkl_output = pkls[0]\n",
      "#Concatenate z's of electrodes:\n",
      "for el in range(0,24):\n",
      "    el_name = 'EL_DC_{}'.format(el)\n",
      "    el_pot_tuple  = ( pkls[0][el_name],)\n",
      "    #initial z is repeated on pickle files after pikls[0]:\n",
      "    init_z  =  1\n",
      "    el_pot_tuple  = el_pot_tuple + tuple( [ pkl[el_name][:,:, init_z:] for pkl in pkls[1:] ] )\n",
      "    pkl_output[el_name] = np.concatenate( el_pot_tuple, axis=2 )\n",
      "\n",
      "#Concatenate Z's:\n",
      "#initial z is repeated on pickle files after pikls[0]:\n",
      "Zs_tuple  = (pkls[0]['Z'],)\n",
      "Zs_tuple  = Zs_tuple + tuple( [ pkl['Z'][1:] for pkl in pkls[1:] ] )\n",
      "pkl_output['Z'] = np.concatenate( Zs_tuple )\n",
      "\n",
      "#Save output:\n",
      "f = open('data.pkl','w')\n",
      "pickle.dump(pkl_output,f)\n",
      "\n",
      "f = open('data.pkl')\n",
      "pkl = pickle.load(f)"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 106
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "f = open('data.pkl')\n",
      "pkl = pickle.load(f)\n",
      "pkl.EL_DC_22"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "metadata": {},
       "output_type": "pyout",
       "prompt_number": 107,
       "text": [
        "array([[[ 0.00046175,  0.00046203,  0.0004623 , ...,  0.00050196,\n",
        "          0.00050226,  0.00050257],\n",
        "        [ 0.00046382,  0.00046409,  0.00046437, ...,  0.00050419,\n",
        "          0.0005045 ,  0.00050481],\n",
        "        [ 0.00046588,  0.00046616,  0.00046643, ...,  0.00050643,\n",
        "          0.00050674,  0.00050705],\n",
        "        ..., \n",
        "        [ 0.00049886,  0.00049916,  0.00049945, ...,  0.00054216,\n",
        "          0.00054249,  0.00054282],\n",
        "        [ 0.00050092,  0.00050122,  0.00050152, ...,  0.00054439,\n",
        "          0.00054472,  0.00054506],\n",
        "        [ 0.00050298,  0.00050328,  0.00050358, ...,  0.00054662,\n",
        "          0.00054695,  0.00054729]],\n",
        "\n",
        "       [[ 0.00046186,  0.00046214,  0.00046241, ...,  0.00050208,\n",
        "          0.00050239,  0.0005027 ],\n",
        "        [ 0.00046392,  0.0004642 ,  0.00046448, ...,  0.00050431,\n",
        "          0.00050462,  0.00050493],\n",
        "        [ 0.00046599,  0.00046626,  0.00046654, ...,  0.00050655,\n",
        "          0.00050686,  0.00050717],\n",
        "        ..., \n",
        "        [ 0.00049898,  0.00049927,  0.00049957, ...,  0.00054229,\n",
        "          0.00054262,  0.00054295],\n",
        "        [ 0.00050104,  0.00050134,  0.00050163, ...,  0.00054452,\n",
        "          0.00054485,  0.00054519],\n",
        "        [ 0.0005031 ,  0.0005034 ,  0.0005037 , ...,  0.00054675,\n",
        "          0.00054709,  0.00054742]],\n",
        "\n",
        "       [[ 0.00046197,  0.00046224,  0.00046252, ...,  0.0005022 ,\n",
        "          0.00050251,  0.00050282],\n",
        "        [ 0.00046403,  0.00046431,  0.00046459, ...,  0.00050444,\n",
        "          0.00050475,  0.00050506],\n",
        "        [ 0.0004661 ,  0.00046637,  0.00046665, ...,  0.00050667,\n",
        "          0.00050699,  0.0005073 ],\n",
        "        ..., \n",
        "        [ 0.00049909,  0.00049939,  0.00049969, ...,  0.00054242,\n",
        "          0.00054275,  0.00054309],\n",
        "        [ 0.00050115,  0.00050145,  0.00050175, ...,  0.00054465,\n",
        "          0.00054499,  0.00054532],\n",
        "        [ 0.00050321,  0.00050351,  0.00050381, ...,  0.00054689,\n",
        "          0.00054722,  0.00054756]],\n",
        "\n",
        "       ..., \n",
        "       [[ 0.00046369,  0.00046397,  0.00046425, ...,  0.00050416,\n",
        "          0.00050448,  0.00050479],\n",
        "        [ 0.00046576,  0.00046604,  0.00046632, ...,  0.00050641,\n",
        "          0.00050672,  0.00050703],\n",
        "        [ 0.00046783,  0.00046812,  0.00046839, ...,  0.00050865,\n",
        "          0.00050897,  0.00050928],\n",
        "        ..., \n",
        "        [ 0.00050095,  0.00050125,  0.00050154, ...,  0.00054453,\n",
        "          0.00054487,  0.0005452 ],\n",
        "        [ 0.00050301,  0.00050331,  0.00050362, ...,  0.00054677,\n",
        "          0.00054711,  0.00054744],\n",
        "        [ 0.00050508,  0.00050538,  0.00050568, ...,  0.00054901,\n",
        "          0.00054935,  0.00054969]],\n",
        "\n",
        "       [[ 0.0004638 ,  0.00046408,  0.00046436, ...,  0.00050429,\n",
        "          0.0005046 ,  0.00050491],\n",
        "        [ 0.00046587,  0.00046615,  0.00046643, ...,  0.00050653,\n",
        "          0.00050684,  0.00050716],\n",
        "        [ 0.00046794,  0.00046822,  0.0004685 , ...,  0.00050878,\n",
        "          0.00050909,  0.0005094 ],\n",
        "        ..., \n",
        "        [ 0.00050106,  0.00050136,  0.00050166, ...,  0.00054466,\n",
        "          0.000545  ,  0.00054533],\n",
        "        [ 0.00050313,  0.00050343,  0.00050373, ...,  0.0005469 ,\n",
        "          0.00054724,  0.00054758],\n",
        "        [ 0.0005052 ,  0.0005055 ,  0.0005058 , ...,  0.00054915,\n",
        "          0.00054948,  0.00054982]],\n",
        "\n",
        "       [[ 0.00046391,  0.00046418,  0.00046446, ...,  0.00050441,\n",
        "          0.00050472,  0.00050503],\n",
        "        [ 0.00046598,  0.00046626,  0.00046654, ...,  0.00050665,\n",
        "          0.00050697,  0.00050728],\n",
        "        [ 0.00046805,  0.00046833,  0.00046861, ...,  0.0005089 ,\n",
        "          0.00050921,  0.00050953],\n",
        "        ..., \n",
        "        [ 0.00050118,  0.00050148,  0.00050178, ...,  0.00054479,\n",
        "          0.00054513,  0.00054546],\n",
        "        [ 0.00050325,  0.00050355,  0.00050385, ...,  0.00054704,\n",
        "          0.00054737,  0.00054771],\n",
        "        [ 0.00050532,  0.00050562,  0.00050592, ...,  0.00054928,\n",
        "          0.00054962,  0.00054995]]])"
       ]
      }
     ],
     "prompt_number": 107
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "pkl.keys()"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "metadata": {},
       "output_type": "pyout",
       "prompt_number": 105,
       "text": [
        "['simCount',\n",
        " 'timeNow',\n",
        " 'baseDataName',\n",
        " 'EL_DC_22',\n",
        " 'EL_DC_23',\n",
        " 'EL_DC_20',\n",
        " 'EL_DC_21',\n",
        " 'EL_RF',\n",
        " 'simulationDirectory',\n",
        " 'perm',\n",
        " 'EL_DC_3',\n",
        " 'EL_DC_2',\n",
        " 'EL_DC_1',\n",
        " 'EL_DC_0',\n",
        " 'EL_DC_7',\n",
        " 'EL_DC_6',\n",
        " 'EL_DC_5',\n",
        " 'EL_DC_4',\n",
        " 'EL_DC_9',\n",
        " 'EL_DC_8',\n",
        " 'fileName',\n",
        " 'numElectrodes',\n",
        " 'Y',\n",
        " 'X',\n",
        " 'Z',\n",
        " 'useDate',\n",
        " 'EL_DC_17',\n",
        " 'EL_DC_16',\n",
        " 'EL_DC_15',\n",
        " 'EL_DC_14',\n",
        " 'EL_DC_13',\n",
        " 'EL_DC_12',\n",
        " 'EL_DC_11',\n",
        " 'EL_DC_10',\n",
        " 'dataPointsPerAxis',\n",
        " 'EL_DC_19',\n",
        " 'EL_DC_18',\n",
        " 'savePath']"
       ]
      }
     ],
     "prompt_number": 105
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "len(pkl.EL_DC_1[0][0])"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "metadata": {},
       "output_type": "pyout",
       "prompt_number": 102,
       "text": [
        "141"
       ]
      }
     ],
     "prompt_number": 102
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "pwd"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "metadata": {},
       "output_type": "pyout",
       "prompt_number": 108,
       "text": [
        "u'/home/trxw/Documents/dfr/codes/Fitting'"
       ]
      }
     ],
     "prompt_number": 108
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [],
     "language": "python",
     "metadata": {},
     "outputs": []
    }
   ],
   "metadata": {}
  }
 ]
}