noise_test.c

#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <dc1394/dc1394.h>
#include <math.h>
#include <fitsio.h>
#include <gsl/gsl_statistics.h>
#include <time.h>
#include <sys/time.h>
#include <string.h>
#include <inttypes.h>

#define REG_CAMERA_ABS_MIN                  0x000U
#define REG_CAMERA_ABS_MAX                  0x004U
#define REG_CAMERA_ABS_VALUE                0x008U

int grab_frame(dc1394camera_t *c, unsigned char *buf, int nbytes) {
  dc1394video_frame_t *frame=NULL;
  dc1394error_t err;
  
  err = dc1394_capture_dequeue(c, DC1394_CAPTURE_POLICY_WAIT, &frame);
  if (err != DC1394_SUCCESS) {
    dc1394_log_error("Unable to capture.");
    dc1394_capture_stop(c);
    dc1394_camera_free(c);
    exit(1);
  }
  
  memcpy(buf, frame->image, nbytes);
  dc1394_capture_enqueue(c, frame);
  return 1;
}

int main(int argc, char *argv[]) {
  
  long nelements, naxes[2];
  dc1394_t * dc;
  dc1394camera_t * camera;
  dc1394camera_list_t * list;
  dc1394error_t err;
  dc1394video_mode_t mode;
  unsigned int max_height, max_width, winleft, wintop;
  
  unsigned char *buffer, *buffer2;
  double *average, *diff;
  int j, f, status, nimages, xsize, ysize;
  double mean, var, exp;
  struct timeval start_time, end_time;
  time_t start_sec, end_sec;
  suseconds_t start_usec, end_usec;
  float elapsed_time, fps;
  
  stderr = freopen("noise_test.log", "w", stderr);
  
  srand48((unsigned)time(NULL));
  
  nimages = 0;
  status = 0;
  xsize = 320;
  ysize = 240;
  naxes[0] = xsize;
  naxes[1] = ysize;
  
  nelements = naxes[0]*naxes[1];
  
  mode = DC1394_VIDEO_MODE_FORMAT7_1;
  dc = dc1394_new();
  if (!dc)
    return -1;
  err = dc1394_camera_enumerate(dc, &list);
  DC1394_ERR_RTN(err, "Failed to enumerate cameras.");
  
  if (list->num == 0) {
    dc1394_log_error("No cameras found.");
    return -1;
  }
  
  camera = dc1394_camera_new(dc, list->ids[0].guid);
  if (!camera) {
    dc1394_log_error("Failed to initialize camera with guid %"PRIx64".", 
                     list->ids[0].guid);
    return -1;
  }
  dc1394_camera_free_list(list);
  
  printf("Using camera with GUID %"PRIx64"\n", camera->guid);
  
  // need to use legacy firewire400 mode for now.  800 not quite reliable.
  dc1394_video_set_iso_speed(camera, DC1394_ISO_SPEED_400);
  
  // configure camera for format7
  err = dc1394_video_set_mode(camera, mode);
  DC1394_ERR_CLN_RTN(err, dc1394_camera_free(camera), "Can't choose video mode.");
  
  err = dc1394_format7_get_max_image_size(camera, mode, &max_width, &max_height);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot get max image size.");
  
  winleft = 0;
  wintop = 0;
  
  err = dc1394_format7_set_roi(camera,
                               mode,
                               DC1394_COLOR_CODING_MONO8,
                               DC1394_USE_MAX_AVAIL,
                               //				 packet,
                               winleft, wintop, // left, top
                               xsize, ysize);
  DC1394_ERR_CLN_RTN(err, dc1394_camera_free(camera), "Can't set ROI.");
  
  err=dc1394_video_set_framerate(camera, DC1394_FRAMERATE_MAX);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set framerate");
    
  err = dc1394_feature_set_mode (camera, DC1394_FEATURE_SHUTTER, DC1394_FEATURE_MODE_MANUAL);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set shutter to manual");
  
  err = dc1394_feature_set_value (camera, DC1394_FEATURE_SHUTTER, 61);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set shutter");
  
  err = dc1394_feature_set_mode (camera, DC1394_FEATURE_GAIN, DC1394_FEATURE_MODE_MANUAL);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set gain to manual");
  
  err = dc1394_feature_set_value (camera, DC1394_FEATURE_GAIN, 48);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set gain");
  
  // set brightness manually.  use relative value in range 0 to 1023.
  err = dc1394_feature_set_mode(camera, DC1394_FEATURE_BRIGHTNESS, DC1394_FEATURE_MODE_MANUAL);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set brightness to manual");
  err = dc1394_feature_set_value(camera, DC1394_FEATURE_BRIGHTNESS, 500);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set brightness");
  printf ("I: brightness is %d\n", 500);
  
  err = dc1394_capture_setup(camera, 16, DC1394_CAPTURE_FLAGS_DEFAULT);
  DC1394_ERR_CLN_RTN(err, dc1394_camera_free(camera), "Error capturing.");
  
  // start the camera up
  err = dc1394_video_set_transmission(camera, DC1394_ON);
  if (err != DC1394_SUCCESS) {
    dc1394_log_error("Unable to start camera iso transmission.");
    dc1394_capture_stop(camera);
    dc1394_camera_free(camera);
    return -1;
  }
  
  printf("Camera successfully initialized.\n");
  
  if (!(buffer = malloc(nelements*sizeof(char)))) {
    printf("Couldn't Allocate Image Buffer\n");
    exit(-1);
  }
  
  if (!(buffer2 = malloc(nelements*sizeof(char)))) {
    printf("Couldn't Allocate Image2 Buffer\n");
    exit(-1);
  }
  
  if (!(average = calloc(nelements, sizeof(double)))) {
    printf("Couldn't Allocate Average Image Buffer\n");
    exit(-1);
  }
  
  if (!(diff = calloc(nelements, sizeof(double)))) {
    printf("Couldn't Allocate Diff Image Buffer\n");
    exit(-1);
  }
  
  gettimeofday(&start_time, NULL);
  
  f = 0;
  
  err = dc1394_feature_set_absolute_control(camera, DC1394_FEATURE_SHUTTER, DC1394_TRUE);
  DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set shutter to absolute mode");
  exp = 8.0*1.0e-5;
  err = dc1394_feature_set_absolute_value(camera, DC1394_FEATURE_SHUTTER, exp);
  grab_frame(camera, buffer, nelements*sizeof(char));
  grab_frame(camera, buffer, nelements*sizeof(char));
  
  for (f=8; f<320; f++) {
    exp = f*1.0e-5;
    err = dc1394_feature_set_absolute_value(camera, DC1394_FEATURE_SHUTTER, exp);
    DC1394_ERR_CLN_RTN(err,dc1394_camera_free (camera),"cannot set shutter");
    
    grab_frame(camera, buffer, nelements*sizeof(char));
    grab_frame(camera, buffer2, nelements*sizeof(char));
    for (j=0; j<nelements; j++) {
      average[j] = (buffer[j]+buffer2[j])/2.0;
    }
    for (j=0; j<nelements; j++) {
      diff[j] = buffer[j] - buffer2[j];
    }
    
    var = gsl_stats_variance(diff, 1, nelements);
    mean = gsl_stats_mean(average, 1, nelements);
    
    var = var/2.0;
    
    printf("%e %10.2f %10.3f\n", exp, mean, var);
    
  }

  gettimeofday(&end_time, NULL);
  printf("End capture.\n");
  
  start_sec = start_time.tv_sec;
  start_usec = start_time.tv_usec;
  end_sec = end_time.tv_sec;
  end_usec = end_time.tv_usec;
  
  elapsed_time = (float)((end_sec + 1.0e-6*end_usec) - (start_sec + 1.0e-6*start_usec));
  fps = nimages/elapsed_time;
  printf("Elapsed time = %g seconds.\n", elapsed_time);
  printf("Framerate = %g fps.\n", fps);
  
  /*-----------------------------------------------------------------------
   *  stop data transmission
   *-----------------------------------------------------------------------*/
  err=dc1394_video_set_transmission(camera,DC1394_OFF);
  DC1394_ERR_RTN(err,"couldn't stop the camera?");
  
  return (status);
}