diff --git a/Source/Data/Time/KTTimeSeriesFFTW.cc b/Source/Data/Time/KTTimeSeriesFFTW.cc index 3d20e779e..4811b5098 100644 --- a/Source/Data/Time/KTTimeSeriesFFTW.cc +++ b/Source/Data/Time/KTTimeSeriesFFTW.cc @@ -73,6 +73,20 @@ namespace Katydid return hist; } + TH1D* KTTimeSeriesFFTW::CreateHistogramImag(const std::string& name) const + { + unsigned nBins = GetNBins(); + TH1D* hist = new TH1D(name.c_str(), "Time Series Imag", (int)nBins, GetRangeMin(), GetRangeMax()); + for (unsigned iBin=0; iBinSetBinContent((int)iBin+1, (*this)(iBin).imag()); + } + hist->SetXTitle("Time (s)"); + hist->SetYTitle("Voltage (V)"); + return hist; + } + + TH1D* KTTimeSeriesFFTW::CreateAmplitudeDistributionHistogram(const std::string& name) const { double tMaxMag = -1.; diff --git a/Source/Data/Time/KTTimeSeriesFFTW.hh b/Source/Data/Time/KTTimeSeriesFFTW.hh index aeb59b8f8..0db9083a5 100644 --- a/Source/Data/Time/KTTimeSeriesFFTW.hh +++ b/Source/Data/Time/KTTimeSeriesFFTW.hh @@ -52,6 +52,8 @@ namespace Katydid public: virtual TH1D* CreateHistogram(const std::string& name = "hTimeSeries") const; + virtual TH1D* CreateHistogramImag(const std::string& name = "hTimeSeriesImag") const; + virtual TH1D* CreateAmplitudeDistributionHistogram(const std::string& name = "hTimeSeriesDist") const; #endif }; diff --git a/Source/IO/BasicROOTFileWriter/KTBasicROOTTypeWriterTime.cc b/Source/IO/BasicROOTFileWriter/KTBasicROOTTypeWriterTime.cc index 6785fe95d..3e3bc4f34 100644 --- a/Source/IO/BasicROOTFileWriter/KTBasicROOTTypeWriterTime.cc +++ b/Source/IO/BasicROOTFileWriter/KTBasicROOTTypeWriterTime.cc @@ -259,7 +259,7 @@ namespace Katydid convReal << "histTSReal_" << sliceNumber << "_" << iComponent; string histNameReal; convReal >> histNameReal; - TH1D* tsHistReal = ts->CreateHistogram(histNameReal); + TH1D* tsHistReal = tsFFTW->CreateHistogram(histNameReal); tsHistReal->SetDirectory(fWriter->GetFile()); tsHistReal->Write(); KTDEBUG(publog, "Histogram <" << histNameReal << "> written to ROOT file"); @@ -269,7 +269,7 @@ namespace Katydid convImag << "histTSImag_" << sliceNumber << "_" << iComponent; string histNameImag; convImag >> histNameImag; - TH1D* tsHistImag = ts->CreateHistogram(histNameImag); + TH1D* tsHistImag = tsFFTW->CreateHistogramImag(histNameImag); tsHistImag->SetDirectory(fWriter->GetFile()); tsHistImag->Write(); KTDEBUG(publog, "Histogram <" << histNameImag << "> written to ROOT file");