Create_O1ChunkSlagRndFiles.C

Go to the documentation of this file.

#include <algorithm>    // std::random_shuffle
#include <vector>       // std::vector
#include <ctime>        // std::time
#include <cstdlib>      // std::rand, std::srand

#define SLAGS_DIR "/home/waveburst/O1/SLAGS"
#define SEED      150914

void Create_SlagRndFile(int chunkID, int slagMax);
void shuffle(std::vector<int>& vector, int seed=150914);

void Create_O1ChunkSlagRndFiles() {

  CWB::Toolbox TB;

  int nIFO = 2;

  char ifo[NIFO_MAX][8];
  strcpy(ifo[0],"L1");
  strcpy(ifo[1],"H1");

  vector<TString> ifos(nIFO);
  for(int n=0;n<nIFO;n++) ifos[n]=ifo[n];

  int slagMin    = 0;       // if slagMax=0 ->  slagMin must be < slagMax
  int slagOff    = 0;
  size_t* slagSite = NULL;  // site index starting with 0
  char*   slagFile = NULL;  // slag file list

  //jobs
  int segLen  = 1200;
  int segMLS  = 600;
  int segTHR  = 0;
  int segEdge = 10;

  // dq file list
  // {ifo, dqcat_file, dqcat[0/1/2], shift[sec], inverse[false/true], 4columns[true/false]}
  int nDQF=10;
  dqfile DQF[nDQF]={

                    {"L1" ,"/home/waveburst/O1/DQ/C02/ALL/L1_cat0.txt",   CWB_CAT0, 0., false, false},
                    {"H1" ,"/home/waveburst/O1/DQ/C02/ALL/H1_cat0.txt",   CWB_CAT0, 0., false, false},

                    {"L1" ,"/home/waveburst/O1/DQ/C02/ALL/L1_cat1.txt",   CWB_CAT1, 0., true,  false},
                    {"H1" ,"/home/waveburst/O1/DQ/C02/ALL/H1_cat1.txt",   CWB_CAT1, 0., true,  false},

                    {"L1" ,"/home/waveburst/O1/DQ/C02/ALL/L1_cat2.txt",   CWB_CAT2, 0., true,  false},
                    {"H1" ,"/home/waveburst/O1/DQ/C02/ALL/H1_cat2.txt",   CWB_CAT2, 0., true,  false},

                    {"L1" ,"/home/waveburst/O1/DQ/C02/ALL/L1_cat4.txt",   CWB_CAT1, 0., true,  false},
                    {"H1" ,"/home/waveburst/O1/DQ/C02/ALL/H1_cat4.txt",   CWB_CAT1, 0., true,  false},

                    {"L1" ,"                                         ",   CWB_CAT0, 0., false, false},
                    {"H1" ,"                                         ",   CWB_CAT0, 0., false, false}

                   };

  for(int k=1;k<=9;k++) {

    sprintf(DQF[8].file, "/home/waveburst/O1/CHUNKS/O1_chunk%02d.lst", k);
    sprintf(DQF[9].file, "/home/waveburst/O1/CHUNKS/O1_chunk%02d.lst", k);

    vector<waveSegment> cat1List=TB.readSegList(nDQF, DQF, CWB_CAT1);

    int slagMax=TB.getSlagJobList(cat1List, segLen).size();
     
    cout << k << " slagMax : " << slagMax << endl;

    Create_SlagRndFile(k, slagMax);
  }

  exit(0);
}

void Create_SlagRndFile(int chunkID, int slagMax) {

  std::vector<int> RND(2*slagMax);
  for(int n=0; n<slagMax; n++) {
    RND[2*n+0]=+(n+1);
    RND[2*n+1]=-(n+1);
  }

  shuffle(RND,SEED);    // shuffle data

  // check
  for(int n=0; n<slagMax; n++) {
    int A = +(n+1);
    int B = -(n+1);
    int nA=0;
    int nB=0;
    for(int n=0; n<2*slagMax; n++) {
      if(RND[n]==A) nA++;
      if(RND[n]==B) nB++;
    }
    if(nA!=1 && nB!=1) {
      cout << "ERROR !!!" << endl;
      exit(1);
    }
  }

  //for(int n=0; n<2*slagMax; n++) cout << n << "\t" << RND[n] << endl;;

  char ofile[1024];
  sprintf(ofile,"%s/slag_rnd_K%02d.lst",SLAGS_DIR,chunkID);

  ofstream out;
  out.open(ofile);
  out << 0 << "\t" << 0 << "\t" << 0 << endl;
  for(int n=0; n<2*slagMax; n++) {
    out << n+1 << "\t" << 0 << "\t" <<  RND[n] << endl;
  }
  out.close();

  return;
}

void shuffle(std::vector<int>& vector, int seed) {

  TRandom3 rnd3(seed);

  size_t size = vector.size();

  if(size > 1) {
    size_t i;
    for(size_t i=0; i<size; i++) {
      size_t j = int(rnd3.Uniform(0,size));
      int t = vector[j];
      vector[j] = vector[i];
      vector[i] = t;
    }
  }
}