コード例 #1
0
ファイル: solvers.c プロジェクト: hriskons/cirPar
void solve(sparse_matrix* matrix, sparse_vector* vector, gsl_vector* x,int vector_size)
{
	//char method;
	gsl_vector* x_gsl;

	sparse_vector* x_sparse;
	sparse_vector* x_sol;
	gsl_vector* vector_gsl;
	int i;
	int itol_value=1e-3;

	x_sparse = (sparse_vector*)safe_malloc(vector_size * sizeof(double));

	conjugate_gradient_sparse(matrix, vector, vector_size, x_sparse,itol_value);
	print_vector_to_file(x_sparse,vector_size,"solution_cg_gary");


	/* Added by hriskons */
	/* conversion of a double into a gsl
	safe_gsl_vector_calloc(&x_gsl, vector_size);
	safe_gsl_vector_calloc(&vector_gsl, vector_size);
	sparse_to_gsl_vector(vector,vector_gsl,vector_size);


	if( !sparse_solve_cg( matrix,vector_gsl,x_gsl) ){
		fprintf(stderr, "Solving Method Sparse LU failed\n" );
	}
	print_gsl_vector_to_file(x_gsl,"solution_cg_hriskons"); */
}
コード例 #2
0
ファイル: geometric_series.cpp プロジェクト: rahenry/3body
void three_body_system::generate_states_geometric(){
  vector<double> alphas_r, alphas_rho, alphas_com;
  int channel = 3;
  int cind = channel_index(channel);

  double minimum = cont->scale_min_to_r0_r ? cont->basis_min_r * r0_unlike : cont->basis_min_r;
  double maximum = cont->basis_max_r * pow(trapping[cind][0][0] / kinetic[cind][0][0], -0.125);
  cout << cont->basis_max_r << " " << pow(trapping[cind][0][0] / kinetic[cind][0][0], -0.125) << endl;
  geometric_series(alphas_r, minimum, maximum, 2);
  minimum = cont->scale_min_to_r0_rho ? cont->basis_min_rho * r0_unlike : cont->basis_min_rho;
  maximum = cont->basis_max_rho * pow(trapping[cind][1][1] / kinetic[cind][1][1], -0.125);
  geometric_series(alphas_rho, minimum, maximum, 2);

  if (1 == 1){
    //alphas_com.push_back(pow(trapping[2][2] / kinetic[2][2], 0.25));
    alphas_com.push_back(1.0/kinetic[cind][2][2]);
  }
  else{
    minimum = cont->scale_min_to_r0_com ? cont->basis_min_com * r0_unlike : cont->basis_min_com;
    maximum = cont->basis_max_com * pow(trapping[cind][2][2] / kinetic[cind][2][2], -0.125);
    geometric_series(alphas_com, minimum, maximum, 1);
  }

  states = vector<state_full>();
  int index = 0;
	for (auto & alpha_com : alphas_com){
      for (auto & alpha_rho : alphas_rho){
    for (auto & alpha_r : alphas_r){
  for (auto & angular: ang->states){
          states.push_back(
              state_full(index++, &angular, alpha_r, alpha_rho, alpha_com, 3));
        }
      }
    }
  }
  cout << "Generated geometric series states, total: " << alphas_r.size() << " x " << alphas_rho.size() << " x " << alphas_com.size() << " x " << ang->states.size() << " = " << states.size() << endl;

  if (true){
    print_vector_to_file(alphas_r, name + ".info.alphas_r");
    print_vector_to_file(alphas_rho, name + ".info.alphas_rho");
    print_vector_to_file(alphas_com, name + ".info.alphas_com");
  }

}