コード例 #1
0
ファイル: p_phaser.c プロジェクト: PaulBatchelor/Soundpipe
int main() {
    sp_data *sp;
    sp_create(&sp);
    sp_srand(sp, 12345);
    sp->sr = SR;
    sp->len = sp->sr * LEN;
    uint32_t t, u;
    SPFLOAT in1 = 0, out1 = 0;
    SPFLOAT in2 = 0, out2 = 0;

    sp_phaser *unit[NUM];

    for(u = 0; u < NUM; u++) { 
        sp_phaser_create(&unit[u]);
        sp_phaser_init(sp, unit[u]);
    }

    for(t = 0; t < sp->len; t++) {
        for(u = 0; u < NUM; u++) sp_phaser_compute(sp, unit[u], 
                &in1, &in2, &out1, &out2);
    }

    for(u = 0; u < NUM; u++) sp_phaser_destroy(&unit[u]);

    sp_destroy(&sp);
    return 0;
}
コード例 #2
0
ファイル: ex_phaser.c プロジェクト: PaulBatchelor/Soundpipe
int main() {
    UserData ud;
    sp_data *sp;
    sp_create(&sp);

    sp_phaser_create(&ud.phaser);
    sp_diskin_create(&ud.disk);

    sp_diskin_init(sp, ud.disk, "oneart.wav");
    sp_phaser_init(sp, ud.phaser);
    
    sp->len = 44100 * 5;
    sp_process(sp, &ud, process);

    sp_phaser_destroy(&ud.phaser);
    sp_diskin_destroy(&ud.disk);

    sp_destroy(&sp);
    return 0;
}
コード例 #3
0
ファイル: remove_modes.c プロジェクト: FXIhub/hawk
int main(int argc, char ** argv){
  if(argc < 5){
    printf("Usage: %s <object_with_modes> <masked_amplitudes> <threshold> <output> [use constraints]\n",argv[0]);
    exit(0);
  }
  Image * obj = sp_image_read(argv[1],0);
  Image * amp = sp_image_read(argv[2],0);
  real threshold = atof(argv[3]);
  Image * sup = sp_image_duplicate(obj,SP_COPY_ALL);
  real max = sp_image_max(obj,0,0,0,0);
  for(int i= 0;i<sp_image_size(sup);i++){
    if(sp_cabs(sup->image->data[i]) > threshold*max){
      sup->image->data[i] = sp_cinit(1,0);
    }else{
      sup->image->data[i] = sp_cinit(0,0);
    }
  }
  sp_image_write(sup,"input_support.h5",0);
  real beta = 0.9;
  SpPhasingAlgorithm * alg;
  if(argv > 5){
    alg = sp_phasing_er_alloc(SpPositiveRealObject);
  }else{
    alg = sp_phasing_er_alloc(SpNoConstraints);
  }
  //  SpPhasingAlgorithm * alg = sp_phasing_raar_alloc(beta,SpNoConstraints);
  SpSupportAlgorithm * sup_alg = NULL;
  SpPhaser * ph = sp_phaser_alloc();
  sp_phaser_init(ph,alg,sup_alg,SpEngineAutomatic);
  sp_phaser_set_objective(ph,SpRecoverAmplitudes);
  sp_phaser_init_support(ph,sup,0,0);
  Image * phased_amplitudes = sp_image_fft(obj);
  for(int i = 0;i<sp_image_size(obj);i++){
    phased_amplitudes->mask->data[i] = amp->mask->data[i];
    if(!amp->mask->data[i]){
      phased_amplitudes->image->data[i] = sp_cinit(0,0);
    }
  }
  sp_image_write(phased_amplitudes,"input_phased_amplitudes.h5",0);
  Image * zeroed_obj = sp_image_ifft(phased_amplitudes);
  sp_image_scale(zeroed_obj,1.0/sp_image_size(obj));
  sp_phaser_set_phased_amplitudes(ph,phased_amplitudes);
  if(sp_phaser_init_model(ph,zeroed_obj,0)){
    fprintf(stderr,"Error while initializing model!\n");
  }
  sp_image_write(zeroed_obj,"zeroed_obj.h5",0);
  int iter = 500;
  if(sp_phaser_iterate(ph,iter)){
    fprintf(stderr,"Error while iterating!\n");
  }
  Image * out = sp_image_duplicate(sp_phaser_model(ph),SP_COPY_ALL);
  Image * output_amplitudes = sp_image_fft(out);
  for(int i = 0;i<sp_image_size(obj);i++){
    if(amp->mask->data[i]){
      output_amplitudes->image->data[i] = phased_amplitudes->image->data[i];
    }
  }
  sp_image_write(output_amplitudes,"output_phased_amplitudes.h5",0);
  out = sp_image_ifft(output_amplitudes);
  sp_image_scale(out,1.0/sp_image_size(out));
  sp_image_write(out,argv[4],0);

}
コード例 #4
0
ファイル: uwrapc.c プロジェクト: FXIhub/hawk
void complete_reconstruction_clean(Image * amp, Image * initial_support,Options * opts){  

  SpPhasingAlgorithm * alg = NULL;
  Log log;
  init_log(&log);
  opts->flog = NULL;
  opts->cur_iteration = 0;

  SpPhasingConstraints phasing_constraints = SpNoConstraints;
  if(opts->enforce_real && opts->enforce_positivity){
    phasing_constraints |= SpPositiveRealObject;
  }else if(opts->enforce_real){
    phasing_constraints |= SpRealObject;
  }else if(opts->enforce_positivity){
    phasing_constraints |= SpPositiveComplexObject;
  }
  if(opts->enforce_centrosymmetry){
    phasing_constraints |= SpCentrosymmetricObject;
  }
  //  if (opts->enforce_ramp) {
  //    phasing_constraints |= SpRampObject;
  //  }
  if(get_algorithm(opts,&log) == HIO){
    alg = sp_phasing_hio_alloc(opts->beta_evolution,phasing_constraints);
  }
  if(get_algorithm(opts,&log) == RAAR){
    alg = sp_phasing_raar_alloc(opts->beta_evolution,phasing_constraints);
  }
  if(get_algorithm(opts,&log) == DIFF_MAP){
    alg = sp_phasing_diff_map_alloc(opts->beta_evolution,get_gamma1(opts),get_gamma2(opts),phasing_constraints);
  }
  SpSupportArray * sup_alg = NULL;
  if(opts->support_update_algorithm == FIXED){
    sup_alg = sp_support_array_init(sp_support_threshold_alloc(opts->support_blur_evolution,opts->threshold_evolution),
				    opts->iterations);
  }
  if(opts->support_update_algorithm == DECREASING_AREA){
    sup_alg = sp_support_array_init(sp_support_area_alloc(opts->support_blur_evolution,opts->object_area_evolution),
				    opts->iterations);;
  }
  if (opts->support_update_algorithm == TEMPLATE_AREA){
    sup_alg = sp_support_array_init(sp_support_template_alloc(opts->init_support,opts->template_blur_radius,
							      opts->template_area_evolution),opts->iterations);
  }
  if (opts->support_update_algorithm == STATIC){
    sup_alg = sp_support_array_init(sp_support_static_alloc(),opts->iterations);
  }
  if(!alg || !sup_alg){
    hawk_fatal("Algorithm is NULL!\nBlame the programmer!");
  }
  if (opts->support_closure_radius > 0) {
    sp_support_array_append(sup_alg,sp_support_close_alloc(opts->support_closure_radius));
  }
  // !CHANGE!
  if (opts->center_image){
    sp_support_array_append(sup_alg,sp_support_centre_image_alloc()); 
  }
  SpPhaser * ph = sp_phaser_alloc();
  sp_phaser_init(ph,alg,sup_alg,opts->phasing_engine);
  sp_phaser_set_amplitudes(ph,amp);
  sp_phaser_init_model(ph,opts->image_guess,0);
  //sp_phaser_init_support(ph,initial_support,0,0);
  if (opts->init_support == NULL) {
    sp_phaser_init_support(ph,initial_support,SpSupportFromPatterson,opts->init_level);
  } else {
    sp_phaser_init_support(ph,initial_support,0,0);
  }
  output_initial_images(sp_phaser_model(ph),initial_support);
  while(opts->max_iterations == 0 || opts->cur_iteration < opts->max_iterations){
    char buffer[1024];
    int to_output = opts->output_period-(ph->iteration)%opts->output_period;
    int to_log = opts->log_output_period-(ph->iteration)%opts->log_output_period;
    int to_iterate = sp_min(to_output,to_log);
//    int timer = sp_timer_start();
    sp_phaser_iterate(ph,to_iterate);
    //    printf("hawk - %d iterations in %d ms\n",to_iterate,(int)sp_timer_stop(timer));
    opts->cur_iteration = ph->iteration;
    if(to_iterate == to_log){
      output_from_phaser(ph,opts,&log);
    }
    if(to_iterate == to_output){
      sprintf(buffer,"real_space-%07d.h5",ph->iteration-1);
      //hawk_image_write(sp_phaser_model(ph),buffer,opts->output_precision);
      Image * rs = apply_output_projection(sp_phaser_model_with_support(ph),
                                           opts->output_projection,
                                           sp_phaser_amplitudes(ph));
      hawk_image_write(rs,buffer,opts->output_precision);
      sprintf(buffer,"support-%07d.h5",ph->iteration-1);
      hawk_image_write(sp_phaser_support(ph),buffer,opts->output_precision);
      sprintf(buffer,"fourier_space-%07d.h5",ph->iteration-1);
      hawk_image_write(sp_phaser_fmodel_with_mask(ph),buffer,opts->output_precision);
      // appending in cxidb file format is currently only implemented for 2D images
      if (amp->num_dimensions == SP_2D) {
        // initialize flag to include iteration number with the last 32 bits
        long long flag = ph->iteration;
        flag = (flag << 32) | 8;
        // copy support to data Image
        for (unsigned i = 0; i < sp_image_size(sp_phaser_support(ph)); i++) {
          rs->mask->data[i] = (int) sp_real(sp_phaser_support(ph)->image->data[i]);
        }
        // append data to cxidb file
        hawk_image_write(rs, "reconstruction.cxi", flag);
      }
      sp_image_free(rs);
    }
  }
  
}