def _generate_spectra(frequencies, testdir, cc_generator):
if os.path.isdir(testdir):
shutil.rmtree(testdir)
os.makedirs(testdir)
fin = np.hstack((frequencies, frequencies))
for nr, cc_set in enumerate(cc_generator()):
outdir = testdir + os.sep + '{0:03}_'.format(nr) + \
'{0}_{1}_{2}_{3}'.format(*cc_set) + os.sep
os.makedirs(outdir)
# generate data
rlogmag_rpha = colecole.cole_log(fin, cc_set)
data = np.vstack((np.exp(rlogmag_rpha[0, :]),
rlogmag_rpha[1, :])).flatten()
data_2d = np.atleast_2d(data)
# save data
np.savetxt(outdir + 'frequencies.dat', frequencies)
np.savetxt(outdir + 'data.dat', data_2d)
cc_set_2d = np.atleast_2d(cc_set)
np.savetxt(outdir + 'colecole_orig.dat', cc_set_2d)
#!/usr/bin/python
"""
Generate sample one-term Cole-Cole spectrum
Save to frequencies.dat and data.dat
"""
import numpy as np
import lib_cc_fit.colecole as CC
frequencies = np.logspace(-3, 3, 25)
fin = np.hstack((frequencies, frequencies))
# rho0 m1 tau1 c1 m2 tau2 c2
cc_pars = [np.log(50), 0.05, np.log(40), 0.6]
magpha = CC.cole_log(fin, cc_pars).flatten()[np.newaxis, :]
magpha[0, 0:magpha.size / 2] = np.exp(magpha[0, 0:magpha.size / 2])
np.savetxt('data.dat', magpha)
np.savetxt('frequencies.dat', frequencies)
