# -*- coding: utf-8 -*-

import csv
import scipy.signal as signal
import numpy as np
import math
import os
import matplotlib.pyplot as plt

### Aufgabe 6
counteri=0
column0, column1, column2, column3=([]),([]),([]),([])

os.chdir('C:/Users/JoJo/Documents/Python Scripts/Daten')
file=csv.reader(open("johanna_hermannsgabner.csv", "r"), delimiter=",",quotechar='"')

for row in file:
    if counteri!=0:
        column0.append(float(row[0]))
        column1.append(float(row[1]))
        column2.append(float(row[2]))
        column3.append(float(row[3]))
    counteri+=1

#plt.plot(column3)
#plt.xlim(0,4400)
peaks=signal.find_peaks_cwt(column3, [11.94385])
ywert_peaks=[]

for i in peaks:
    ywert_peaks.append(column3[i])

#plt.plot(peaks, ywert_peaks, 'ro')
#plt.show()

n=peaks.size-1
periode=np.zeros(n)
for i in range(1,n+1):
    periode[i-1]=peaks[i]-peaks[i-1]
periode=periode/100

#hist=plt.hist(periode, bins='auto', density = False)



### Aufgabe 9
# 1a
mean0=np.mean(periode)
std=np.std(periode)
sigma_T=std/(math.sqrt(n))
l=0.5
sigma_l=0.05
print('_________________________________________________\n\n1a')
print('-Mittelwert der Periodendauer: <T> = {}.'.format(mean0))
print('-Statistischer Fehler der Periodendauer: sigma_T = {}.'.format(sigma_T))

pi=math.pi
g0=(l*4*pi**2)/mean0**2
sigma_g=math.sqrt(  (((-8*l*np.pi**2)/mean0**3 )**2) * (sigma_T**2)  +  (((4*np.pi**2)/mean0**2)**2) * (sigma_l**2)  )
print('-Erdbeschleunigung: g = {}m/s².\n\-Statistischer Fehler der Erdbeschneunigung: sigma_g = {}'.format(g0,sigma_g))


# 1b
a=6.1
b=2.8
sig_a=sig_b=0.1

kov=sig_a*sig_b
korr=math.sqrt( (b**2)*(sig_a**2)  +  (a**2)*(sig_b**2)  +  2*kov*a*b )
unkorr=math.sqrt((b**2)*(sig_a**2)  +  (a**2)*(sig_b**2))
flaeche=a*b
print('_________________________________________________\n\n1b')
print('-Fläche ohne Korrelation: {}+/-{}\n-Fläche mit 100% Korrelation: {}+/-{}'.format(flaeche,unkorr,flaeche,korr))


# 1c
n=2356/60       #### pro 1s
eps=0.1
sigma_eps=0.005
m=n/eps

def fehlerfortplanzung_produkt(m,n,eps,sigma_sys,sigma_stat):
    sigma_m=m*math.sqrt(((sigma_sys/eps)**2)+(sigma_stat/eps)**2)
    return sigma_m

sigma_m=fehlerfortplanzung_produkt(m,n,eps,sigma_eps,sigma_eps)
print('_________________________________________________\n\n1c')
print('-Aktivität in Becquerel {}+/-{}'.format(m,sigma_m))
n_h=2356*60
m_h=n_h/eps
sigma_mh=fehlerfortplanzung_produkt(m_h,n_h,eps,sigma_eps,sigma_eps)
print('-Aktivität pro Stunde {}+/-{}'.format(m_h,sigma_mh))


# 1d
stand=1000 ## in nV  mit 80%
sig_m=100  ## in nV
anzahl=[]
for t in range(30):
    kov=(stand**2)*np.exp(-t/5)
    n_=( 1.28*math.sqrt(kov)/sig_m )**2
    anzahl.append(n_)

plt.plot(anzahl)
plt.xlabel('Zeitabstand i-j')
plt.ylabel('Benoetigte Anzahl')
plt.show()