Ovladanie_kamery.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Defaulting to user installation because normal site-packages is not writeable\n",
      "Looking in indexes: https://pypi.org/simple, https://www.piwheels.org/simple\n",
      "Requirement already satisfied: zwoasi in /home/pi/.local/lib/python3.7/site-packages (0.0.22)\n",
      "Requirement already satisfied: numpy>=1.7 in /usr/lib/python3/dist-packages (from zwoasi) (1.16.2)\n",
      "Requirement already satisfied: six in /usr/lib/python3/dist-packages (from zwoasi) (1.12.0)\n",
      "\u001b[33mWARNING: You are using pip version 20.2.1; however, version 20.2.2 is available.\n",
      "You should consider upgrading via the '/usr/bin/python3 -m pip install --upgrade pip' command.\u001b[0m\n"
     ]
    }
   ],
   "source": [
    "!pip3 install zwoasi\n",
    "import argparse\n",
    "import os\n",
    "import sys\n",
    "import time\n",
    "import zwoasi as asi"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def save_control_values(filename, settings):\n",
    "    filename += '.txt'\n",
    "    with open(filename, 'w') as f:\n",
    "        for k in sorted(settings.keys()):\n",
    "            f.write('%s: %s\\n' % (k, str(settings[k])))\n",
    "    print('Camera settings saved to %s' % filename)\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "asi.init(\"indigo_ccd_asi.so\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "cameras_found = asi.list_cameras()  # Models names of the connected cameras\n",
    "\n",
    "\n",
    "num_cameras = asi.get_num_cameras()\n",
    "if num_cameras == 0:\n",
    "    print('No cameras found')\n",
    "    sys.exit(0)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Found one camera: ZWO ASI533MC Pro\n"
     ]
    }
   ],
   "source": [
    "num_cameras = 1\n",
    "\n",
    "if num_cameras == 1:\n",
    "    camera_id = 0\n",
    "    print('Found one camera: %s' % cameras_found[0])\n",
    "else:\n",
    "    print('Found %d cameras' % num_cameras)\n",
    "    for n in range(num_cameras):\n",
    "        print('    %d: %s' % (n, cameras_found[n]))\n",
    "    # TO DO: allow user to select a camera\n",
    "    camera_id = 0\n",
    "    print('Using #%d: %s' % (camera_id, cameras_found[camera_id]))\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "camera = asi.Camera(camera_id)\n",
    "camera_info = camera.get_camera_property()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "Camera controls:\n",
      "    AutoExpMaxExpMS:\n",
      "        ControlType: 11\n",
      "        DefaultValue: 100\n",
      "        Description: 'Auto exposure maximum exposure value(unit ms)'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 60000\n",
      "        MinValue: 1\n",
      "        Name: 'AutoExpMaxExpMS'\n",
      "    AutoExpMaxGain:\n",
      "        ControlType: 10\n",
      "        DefaultValue: 300\n",
      "        Description: 'Auto exposure maximum gain value'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 600\n",
      "        MinValue: 0\n",
      "        Name: 'AutoExpMaxGain'\n",
      "    AutoExpTargetBrightness:\n",
      "        ControlType: 12\n",
      "        DefaultValue: 100\n",
      "        Description: 'Auto exposure target brightness value'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 160\n",
      "        MinValue: 50\n",
      "        Name: 'AutoExpTargetBrightness'\n",
      "    BandWidth:\n",
      "        ControlType: 6\n",
      "        DefaultValue: 50\n",
      "        Description: 'The total data transfer rate percentage'\n",
      "        IsAutoSupported: True\n",
      "        IsWritable: True\n",
      "        MaxValue: 100\n",
      "        MinValue: 40\n",
      "        Name: 'BandWidth'\n",
      "    CoolPowerPerc:\n",
      "        ControlType: 15\n",
      "        DefaultValue: 0\n",
      "        Description: 'Cooler power percent'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: False\n",
      "        MaxValue: 100\n",
      "        MinValue: 0\n",
      "        Name: 'CoolPowerPerc'\n",
      "    CoolerOn:\n",
      "        ControlType: 17\n",
      "        DefaultValue: 0\n",
      "        Description: 'turn on/off cooler(cool camera only)'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 1\n",
      "        MinValue: 0\n",
      "        Name: 'CoolerOn'\n",
      "    Exposure:\n",
      "        ControlType: 1\n",
      "        DefaultValue: 10000\n",
      "        Description: 'Exposure Time(us)'\n",
      "        IsAutoSupported: True\n",
      "        IsWritable: True\n",
      "        MaxValue: 2000000000\n",
      "        MinValue: 32\n",
      "        Name: 'Exposure'\n",
      "    Flip:\n",
      "        ControlType: 9\n",
      "        DefaultValue: 0\n",
      "        Description: 'Flip: 0->None 1->Horiz 2->Vert 3->Both'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 3\n",
      "        MinValue: 0\n",
      "        Name: 'Flip'\n",
      "    Gain:\n",
      "        ControlType: 0\n",
      "        DefaultValue: 200\n",
      "        Description: 'Gain'\n",
      "        IsAutoSupported: True\n",
      "        IsWritable: True\n",
      "        MaxValue: 600\n",
      "        MinValue: 0\n",
      "        Name: 'Gain'\n",
      "    HardwareBin:\n",
      "        ControlType: 13\n",
      "        DefaultValue: 0\n",
      "        Description: 'Is hardware bin2:0->No 1->Yes'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 1\n",
      "        MinValue: 0\n",
      "        Name: 'HardwareBin'\n",
      "    MonoBin:\n",
      "        ControlType: 18\n",
      "        DefaultValue: 0\n",
      "        Description: 'bin R G G B to one pixel for color camera, color will loss'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 1\n",
      "        MinValue: 0\n",
      "        Name: 'MonoBin'\n",
      "    Offset:\n",
      "        ControlType: 5\n",
      "        DefaultValue: 1\n",
      "        Description: 'offset'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 100\n",
      "        MinValue: 0\n",
      "        Name: 'Offset'\n",
      "    TargetTemp:\n",
      "        ControlType: 16\n",
      "        DefaultValue: 0\n",
      "        Description: 'Target temperature(cool camera only)'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: True\n",
      "        MaxValue: 30\n",
      "        MinValue: -40\n",
      "        Name: 'TargetTemp'\n",
      "    Temperature:\n",
      "        ControlType: 8\n",
      "        DefaultValue: 20\n",
      "        Description: 'Sensor temperature(degrees Celsius)'\n",
      "        IsAutoSupported: False\n",
      "        IsWritable: False\n",
      "        MaxValue: 1000\n",
      "        MinValue: -500\n",
      "        Name: 'Temperature'\n",
      "    WB_B:\n",
      "        ControlType: 4\n",
      "        DefaultValue: 95\n",
      "        Description: 'White balance: Blue component'\n",
      "        IsAutoSupported: True\n",
      "        IsWritable: True\n",
      "        MaxValue: 99\n",
      "        MinValue: 1\n",
      "        Name: 'WB_B'\n",
      "    WB_R:\n",
      "        ControlType: 3\n",
      "        DefaultValue: 52\n",
      "        Description: 'White balance: Red component'\n",
      "        IsAutoSupported: True\n",
      "        IsWritable: True\n",
      "        MaxValue: 99\n",
      "        MinValue: 1\n",
      "        Name: 'WB_R'\n",
      "Enabling stills mode\n",
      "Capturing a single 8-bit mono image\n",
      "Saved to image_mono.jpg\n",
      "Camera settings saved to image_mono.jpg.txt\n",
      "Capturing a single 16-bit mono image\n",
      "Saved to image_mono16.tiff\n",
      "Camera settings saved to image_mono16.tiff.txt\n",
      "Capturing a single, color image\n",
      "Saved to image_color.jpg\n",
      "Camera settings saved to image_color.jpg.txt\n",
      "Enabling video mode\n"
     ]
    },
    {
     "ename": "ZWO_IOError",
     "evalue": "Invalid control type",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mZWO_IOError\u001b[0m                               Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-19-626715c7c3ba>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[1;32m     77\u001b[0m     \u001b[0;32mif\u001b[0m \u001b[0mcontrols\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mc\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'ControlType'\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m==\u001b[0m \u001b[0masi\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mASI_BANDWIDTHOVERLOAD\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     78\u001b[0m         \u001b[0;32mcontinue\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 79\u001b[0;31m     \u001b[0mcamera\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mset_control_value\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mcontrols\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mc\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'ControlType'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcontrols\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mc\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'DefaultValue'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     80\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     81\u001b[0m \u001b[0;31m# Can autoexposure be used?\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/home/pi/.local/lib/python3.7/site-packages/zwoasi/__init__.py\u001b[0m in \u001b[0;36mset_control_value\u001b[0;34m(self, control_type, value, auto)\u001b[0m\n\u001b[1;32m    454\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    455\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0mset_control_value\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcontrol_type\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mvalue\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mauto\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mFalse\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 456\u001b[0;31m         \u001b[0m_set_control_value\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mid\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcontrol_type\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mvalue\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mauto\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    457\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    458\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0mget_bin\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/home/pi/.local/lib/python3.7/site-packages/zwoasi/__init__.py\u001b[0m in \u001b[0;36m_set_control_value\u001b[0;34m(id_, control_type, value, auto)\u001b[0m\n\u001b[1;32m     83\u001b[0m     \u001b[0mr\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mzwolib\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mASISetControlValue\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mid_\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcontrol_type\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mvalue\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mauto\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     84\u001b[0m     \u001b[0;32mif\u001b[0m \u001b[0mr\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 85\u001b[0;31m         \u001b[0;32mraise\u001b[0m \u001b[0mzwo_errors\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mr\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     86\u001b[0m     \u001b[0;32mreturn\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     87\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mZWO_IOError\u001b[0m: Invalid control type"
     ]
    }
   ],
   "source": [
    "\n",
    "# Get all of the camera controls\n",
    "print('')\n",
    "print('Camera controls:')\n",
    "controls = camera.get_controls()\n",
    "for cn in sorted(controls.keys()):\n",
    "    print('    %s:' % cn)\n",
    "    for k in sorted(controls[cn].keys()):\n",
    "        print('        %s: %s' % (k, repr(controls[cn][k])))\n",
    "\n",
    "\n",
    "# Use minimum USB bandwidth permitted\n",
    "camera.set_control_value(asi.ASI_BANDWIDTHOVERLOAD, camera.get_controls()['BandWidth']['MinValue'])\n",
    "\n",
    "# Set some sensible defaults. They will need adjusting depending upon\n",
    "# the sensitivity, lens and lighting conditions used.\n",
    "camera.disable_dark_subtract()\n",
    "\n",
    "camera.set_control_value(asi.ASI_GAIN, 150)\n",
    "camera.set_control_value(asi.ASI_EXPOSURE, 30000)\n",
    "camera.set_control_value(asi.ASI_WB_B, 99)\n",
    "camera.set_control_value(asi.ASI_WB_R, 75)\n",
    "camera.set_control_value(asi.ASI_GAMMA, 50)\n",
    "camera.set_control_value(asi.ASI_BRIGHTNESS, 50)\n",
    "camera.set_control_value(asi.ASI_FLIP, 0)\n",
    "\n",
    "\n",
    "print('Enabling stills mode')\n",
    "try:\n",
    "    # Force any single exposure to be halted\n",
    "    camera.stop_video_capture()\n",
    "    camera.stop_exposure()\n",
    "except (KeyboardInterrupt, SystemExit):\n",
    "    raise\n",
    "except:\n",
    "    pass\n",
    "\n",
    "print('Capturing a single 8-bit mono image')\n",
    "filename = 'image_mono.jpg'\n",
    "camera.set_image_type(asi.ASI_IMG_RAW8)\n",
    "camera.capture(filename=filename)\n",
    "print('Saved to %s' % filename)\n",
    "save_control_values(filename, camera.get_control_values())\n",
    "\n",
    "\n",
    "print('Capturing a single 16-bit mono image')\n",
    "filename = 'image_mono16.tiff'\n",
    "camera.set_image_type(asi.ASI_IMG_RAW16)\n",
    "camera.capture(filename=filename)\n",
    "print('Saved to %s' % filename)\n",
    "save_control_values(filename, camera.get_control_values())\n",
    "\n",
    "if camera_info['IsColorCam']:\n",
    "    filename = 'image_color.jpg'\n",
    "    camera.set_image_type(asi.ASI_IMG_RGB24)\n",
    "    print('Capturing a single, color image')\n",
    "    camera.capture(filename=filename)\n",
    "    print('Saved to %s' % filename)\n",
    "    save_control_values(filename, camera.get_control_values())\n",
    "else:\n",
    "    print('Color image not available with this camera')\n",
    "    \n",
    "# Enable video mode\n",
    "try:\n",
    "    # Force any single exposure to be halted\n",
    "    camera.stop_exposure()\n",
    "except (KeyboardInterrupt, SystemExit):\n",
    "    raise\n",
    "except:\n",
    "    pass\n",
    "\n",
    "print('Enabling video mode')\n",
    "camera.start_video_capture()\n",
    "\n",
    "# Restore all controls to default values except USB bandwidth\n",
    "for c in controls:\n",
    "    if controls[c]['ControlType'] == asi.ASI_BANDWIDTHOVERLOAD:\n",
    "        continue\n",
    "    camera.set_control_value(controls[c]['ControlType'], controls[c]['DefaultValue'])\n",
    "\n",
    "# Can autoexposure be used?\n",
    "k = 'Exposure'\n",
    "if 'Exposure' in controls and controls['Exposure']['IsAutoSupported']:\n",
    "    print('Enabling auto-exposure mode')\n",
    "    camera.set_control_value(asi.ASI_EXPOSURE,\n",
    "                             controls['Exposure']['DefaultValue'],\n",
    "                             auto=True)\n",
    "\n",
    "    if 'Gain' in controls and controls['Gain']['IsAutoSupported']:\n",
    "        print('Enabling automatic gain setting')\n",
    "        camera.set_control_value(asi.ASI_GAIN,\n",
    "                                 controls['Gain']['DefaultValue'],\n",
    "                                 auto=True)\n",
    "\n",
    "    # Keep max gain to the default but allow exposure to be increased to its maximum value if necessary\n",
    "    camera.set_control_value(controls['AutoExpMaxExpMS']['ControlType'], controls['AutoExpMaxExpMS']['MaxValue'])\n",
    "\n",
    "    print('Waiting for auto-exposure to compute correct settings ...')\n",
    "    sleep_interval = 0.100\n",
    "    df_last = None\n",
    "    gain_last = None\n",
    "    exposure_last = None\n",
    "    matches = 0\n",
    "    while True:\n",
    "        time.sleep(sleep_interval)\n",
    "        settings = camera.get_control_values()\n",
    "        df = camera.get_dropped_frames()\n",
    "        gain = settings['Gain']\n",
    "        exposure = settings['Exposure']\n",
    "        if df != df_last:\n",
    "            print('   Gain {gain:d}  Exposure: {exposure:f} Dropped frames: {df:d}'\n",
    "                  .format(gain=settings['Gain'],\n",
    "                          exposure=settings['Exposure'],\n",
    "                          df=df))\n",
    "            if gain == gain_last and exposure == exposure_last:\n",
    "                matches += 1\n",
    "            else:\n",
    "                matches = 0\n",
    "            if matches >= 5:\n",
    "                break\n",
    "            df_last = df\n",
    "            gain_last = gain\n",
    "            exposure_last = exposure\n",
    "\n",
    "# Set the timeout, units are ms\n",
    "timeout = (camera.get_control_value(asi.ASI_EXPOSURE)[0] / 1000) * 2 + 500\n",
    "camera.default_timeout = timeout\n",
    "\n",
    "if camera_info['IsColorCam']:\n",
    "    print('Capturing a single color frame')\n",
    "    filename = 'image_video_color.jpg'\n",
    "    camera.set_image_type(asi.ASI_IMG_RGB24)\n",
    "    camera.capture_video_frame(filename=filename)\n",
    "else:\n",
    "    print('Capturing a single 8-bit mono frame')\n",
    "    filename = 'image_video_mono.jpg'\n",
    "    camera.set_image_type(asi.ASI_IMG_RAW8)\n",
    "    camera.capture_video_frame(filename=filename)\n",
    "\n",
    "print('Saved to %s' % filename)\n",
    "save_control_values(filename, camera.get_control_values())"
   ]
  }
 ],
 "metadata": {
  "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.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}