Regression_LPE_H1.C

Go to the documentation of this file.

#define HCHANNEL "H1:LSC-STRAIN"
#define FRLIST_NAME "S5_H1_RDS_C03_L2.frl"
#define FRLIST_NAME2 "H-R-8159.frl"
#define START 815974300 
#define LENGHT 1200
#define F1 32
#define F2 2048

{

  using namespace CWB;
  int totalscratch=32;

  //GetHchannel
  wavearray<double> h;
  h.start(START-totalscratch);
  h.stop(LENGHT+START+totalscratch);
  frame frt(FRLIST_NAME,HCHANNEL);
  frt >> h;

  //Resample to 2048 Hz
  Meyer<double> B(1024);           // set wavelet for resampling
  WSeries<double> ww;
  ww.Forward(h,B,2);
  ww.getLayer(h,0);

  //Make WDM transform, resolution= 1Hz
  int lev=h.rate()/2;
  WDM<double> wdtf(lev, 2*lev, 6, 10);
  WSeries<double> tfmap;
  tfmap.Forward(h, wdtf);

  //Adding target channel
  regression r;
  r.add(tfmap,const_cast<char*>("hchannel"));
  r.mask(0);
  r.unmask(0,F1,F2);

  r.add(h,const_cast<char*>("hchannel"));

  //Calculate prediction
  r.setFilter(10);
  r.setMatrix(totalscratch,.95);
  r.solve(0.2, 0, 'h');
  r.apply(0.2);

  //Draw plot
  watplot plot;
  plot.goptions(const_cast<char*>("alp logy"), 1, START, START+LENGHT, true, F1, F2, true, 50);
  h >> plot;
  wavearray<double> hh = r.getClean();
  hh >> plot;
  TString ofile = "Regression_LPE_H1.png";
  plot >> ofile;

  wavearray<double> freq=r.vfreq;
  wavearray<double> rank=r.getRank(0);
  for (int i=0; i<freq.size(); i++) cout << freq.data[i] << " " << rank.data[i] << endl;
}