ana_NMR_poly_signal.cc

Go to the documentation of this file.

#include "NMRDataManager.h"
void DrawPolySignal(TH2* h2_vt, const int i_evt_b, const int i_evt_e, const int time_b, const int time_e);
 
void ana_NMR_poly_signal()
{
  NMRDataManager man;
  man.Verb(0); // 0 = silent, 1 = some, 2 = every event
  gSystem->mkdir("result", true);
 
  man.ReadEventFile("2024-02-08_19h02m57s");
  man.ReadEventFile("2024-02-08_22h13m47s");
 
  int ScanSweeps; // N of sweeps per measurement
  int ScanSteps;  // N of frequency steps per scan
  double RFFreq;  // Center frequency in MHz
  double RFMod;   // Half width of frequency range in kHz
  man.GetScanParam(ScanSweeps, ScanSteps, RFFreq, RFMod);
  RFMod /= 1000; // kHz -> MHz
 
  TTimeStamp ts0(2024, 2, 8, 18, 0, 0);
  int utime0 = ts0.GetSec() + TTimeStamp::GetZoneOffset();
 
  double RF_lo = RFFreq - RFMod;
  double RF_hi = RFFreq + RFMod;
  double sig_min, sig_max;
  man.GetPolySignalRange(sig_min, sig_max);
  TH2* h2_vt = new TH2D("h2_vt", "", ScanSteps, RF_lo, RF_hi,  100, sig_min, sig_max);
 
  const int n_draw_step = 50; // h2_vt is drawn every "n_draw_step" events.
  int i_evt_draw = -1;
  int utime_draw;
 
  int n_evt = man.GetNumEvent();
  int i_evt_ok = 0;
  for (int i_evt = 0; i_evt < n_evt; i_evt++) {
    NMREvent* evt = man.GetEvent(i_evt);
    int   evt_num = evt->GetInt("EventNum");
    if (utime0 == 0) utime0 = evt_num;
    if (evt_num < utime0) continue;
    NMRSignal* sig = man.GetPolySignal(evt_num);
    if (sig->GetNumPoint() != ScanSteps) continue; // The poly signal is sometimes empty.
    i_evt_ok++;
 
    if (i_evt_draw < 0) {
      i_evt_draw = i_evt_ok;
      utime_draw = evt_num;
    }
 
    for (int ii = 0; ii < ScanSteps; ii++) {
      double rf = RF_lo + 2.0 * RFMod / (ScanSteps - 1) * ii;
      h2_vt->Fill(rf, sig->GetPoint(ii));
    }
    if ((i_evt_ok + 1) % n_draw_step == 0 ||
        i_evt == n_evt - 1) {
      DrawPolySignal(h2_vt, i_evt_draw, i_evt_ok, utime_draw-utime0, evt_num-utime0);
      i_evt_draw = -1;
      h2_vt->Reset();
    }
  }
 
  // n/a
 
  exit(0);
}
 
void DrawPolySignal(TH2* h2_vt, const int i_evt_b, const int i_evt_e, const int time_b, const int time_e)
{
  TCanvas* c1 = new TCanvas("c1", "");
  c1->SetGrid(true);
  gStyle->SetOptStat(0);
  TGaxis::SetMaxDigits(4);
 
  ostringstream oss;
  oss << "Data #" << i_evt_b << "-" << i_evt_e << " ("
      << fixed << setprecision(2) << time_b/3600.0 << "-" << time_e/3600.0
      << " hours);Frequency;PolySignal";
  h2_vt->SetTitle(oss.str().c_str());
  h2_vt->Draw("colz");
  oss.str("");
  oss << setfill('0') << "result/h2_vt_" << setw(6) << i_evt_b << ".png";
  c1->SaveAs(oss.str().c_str());
  delete c1;
}