Skip to content
Snippets Groups Projects
Commit 64c0b2eb authored by Pascal Engeler's avatar Pascal Engeler
Browse files

Added unit test for lattice generation

parent bdb23026
Branches
No related tags found
No related merge requests found
#include <lattice_generator.hpp>
#include <rbcomb_generator.hpp>
#include <drum.hpp>
#include <drum_parameters.hpp>
#include <drum_variables.hpp>
#include <rk4_buffer.hpp>
#include <vector>
#include <iostream>
int main()
{
using value_t = double;
using drum_t = Drum<value_t, DrumParameters, DrumVariables, RK4Buffer>;
using generator_t = LatticeGenerator<value_t, DrumParameters, DrumVariables, RK4Buffer>;
using rbgen_t = RbcombGenerator<value_t, DrumParameters, DrumVariables, RK4Buffer>;
using params_t = DrumVariables<value_t>;
DrumParameters<value_t> dp (800., 6.28318531E-03, 10., 9.64930905E-01, 10000., 1.0, 1.16883395E-06, Vec2<value_t>(0.,0.), 'A');
std::vector<generator_t*> gen_vec;
rbgen_t rbgen (50, 10);
gen_vec.push_back(&rbgen);
std::pair<std::vector<drum_t>, std::vector<std::vector<int> > > result_pair = gen_vec[0]->operator()(dp);
for(int i = 0; i < result_pair.second.size(); ++i){
for(int j = 0; j < result_pair.second[i].size(); ++j){
if(result_pair.second[i][j] != -1){
auto drum1 = result_pair.first[i];
auto drum2 = result_pair.first[result_pair.second[i][j]];
std::cout << drum1.get_parameters().position.x() << " " << drum1.get_parameters().position.y() << " " << drum2.get_parameters().position.x() << " " << drum2.get_parameters().position.y() << std::endl;
}
}
}
return 0;
}
import matplotlib.pyplot as plt
import numpy as np
lattice = np.loadtxt('lattice.txt', dtype=float)
#plot row by row
for i in range(len(lattice[:,0])):
plt.plot([lattice[i,0],lattice[i,2]],[lattice[i,1],lattice[i,3]])
plt.ylim([-22000.,47000.])
plt.savefig('lattice.pdf', dpi=500)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment