@@ -101,9 +101,13 @@ <h1>Source code for magic.checker</h1><div class="highlight"><pre>
101101< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> magic</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> MagicTs</ span > < span class ="p "> ,</ span > < span class ="n "> avgField</ span > < span class ="p "> ,</ span > < span class ="n "> MagicSpectrum</ span > < span class ="p "> ,</ span > < span class ="n "> MagicRadial</ span > < span class ="p "> ,</ span > < span class ="n "> scanDir</ span > < span class ="p "> ,</ span > < span class ="n "> MagicSetup</ span >
102102< span class ="kn "> import</ span > < span class ="w "> </ span > < span class ="nn "> numpy</ span > < span class ="w "> </ span > < span class ="k "> as</ span > < span class ="w "> </ span > < span class ="nn "> np</ span >
103103< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> scipy.signal</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> argrelextrema</ span >
104- < span class ="kn "> import</ span > < span class ="w "> </ span > < span class ="nn "> matplotlib.pyplot</ span > < span class ="w "> </ span > < span class ="k "> as</ span > < span class ="w "> </ span > < span class ="nn "> plt</ span >
105104< span class ="kn "> import</ span > < span class ="w "> </ span > < span class ="nn "> os</ span >
106105
106+ < span class ="k "> try</ span > < span class ="p "> :</ span >
107+ < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> numpy</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> trapz</ span > < span class ="k "> as</ span > < span class ="n "> trapz</ span >
108+ < span class ="k "> except</ span > < span class ="p "> :</ span >
109+ < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> numpy</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> trapezoid</ span > < span class ="k "> as</ span > < span class ="n "> trapz</ span >
110+
107111< span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> bcolors</ span > < span class ="p "> :</ span >
108112 < span class ="n "> HEADER</ span > < span class ="o "> =</ span > < span class ="s1 "> '</ span > < span class ="se "> \033</ span > < span class ="s1 "> [95m'</ span >
109113 < span class ="n "> OKBLUE</ span > < span class ="o "> =</ span > < span class ="s1 "> '</ span > < span class ="se "> \033</ span > < span class ="s1 "> [34m'</ span >
@@ -177,7 +181,7 @@ <h1>Source code for magic.checker</h1><div class="highlight"><pre>
177181 < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="n "> bcolors</ span > < span class ="o "> .</ span > < span class ="n "> MODERATE</ span > < span class ="o "> +</ span > < span class ="s1 "> 'Sudden variations detected in power balance!'</ span > < span class ="o "> +</ span > < span class ="n "> bcolors</ span > < span class ="o "> .</ span > < span class ="n "> ENDC</ span > < span class ="p "> )</ span >
178182 < span class ="n "> ones</ span > < span class ="o "> =</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ones_like</ span > < span class ="p "> (</ span > < span class ="n "> ts</ span > < span class ="o "> .</ span > < span class ="n "> time</ span > < span class ="p "> [</ span > < span class ="n "> mask</ span > < span class ="p "> ])</ span >
179183 < span class ="n "> ones</ span > < span class ="p "> [</ span > < span class ="o "> ~</ span > < span class ="n "> mask_spike</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mf "> 0.</ span >
180- < span class ="n "> ttot_spikes</ span > < span class ="o "> =</ span > < span class ="n "> np </ span > < span class =" o " > . </ span > < span class =" n " > trapz</ span > < span class ="p "> (</ span > < span class ="n "> ones</ span > < span class ="p "> ,</ span > < span class ="n "> ts</ span > < span class ="o "> .</ span > < span class ="n "> time</ span > < span class ="p "> [</ span > < span class ="n "> mask</ span > < span class ="p "> ])</ span >
184+ < span class ="n "> ttot_spikes</ span > < span class ="o "> =</ span > < span class ="n "> trapz</ span > < span class ="p "> (</ span > < span class ="n "> ones</ span > < span class ="p "> ,</ span > < span class ="n "> ts</ span > < span class ="o "> .</ span > < span class ="n "> time</ span > < span class ="p "> [</ span > < span class ="n "> mask</ span > < span class ="p "> ])</ span >
181185 < span class ="n "> ttot</ span > < span class ="o "> =</ span > < span class ="n "> ts</ span > < span class ="o "> .</ span > < span class ="n "> time</ span > < span class ="p "> [</ span > < span class ="o "> -</ span > < span class ="mi "> 1</ span > < span class ="p "> ]</ span > < span class ="o "> -</ span > < span class ="n "> ts</ span > < span class ="o "> .</ span > < span class ="n "> time</ span > < span class ="p "> [</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
182186 < span class ="n "> ratio</ span > < span class ="o "> =</ span > < span class ="n "> ttot_spikes</ span > < span class ="o "> /</ span > < span class ="n "> ttot</ span >
183187 < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="s1 "> ' -Time fraction with spikes: </ span > < span class ="si "> {:.3f}</ span > < span class ="s1 "> %'</ span > < span class ="o "> .</ span > < span class ="n "> format</ span > < span class ="p "> (</ span > < span class ="mf "> 100.</ span > < span class ="o "> *</ span > < span class ="n "> ratio</ span > < span class ="p "> ))</ span >
0 commit comments