MambaToy/runner.cpp at master · adamdddave/MambaToy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
//c++
#include <iostream>
#include <cmath>
//root

//external
#include "ToyDataGenerator.h"
#include "FileWriter.h"

int main(int argc, char* const argv[]){
  std::cout<<"Toy Data Generator for Milano DAQ testing"<<std::endl;
  int seed =0;
  if(argc>1){
    seed = atoi(argv[1]);
    std::cout<<"using seed"<<atoi(argv[1])<<std::endl;
  }
  int nev = 100;
  if(argc >2){
    nev = atoi(argv[2]);
    std::cout<<"generating "<<atoi(argv[2])<<" events"<<std::endl;
  }
  ToyDataGenerator gen(seed,nev);
  gen.generateSignal();//sample everything first.

  try{
      FileWriter fileWriter("MambaMC.dat");
      for(int i=0; i<nev;++i){
        gen.generatePedestal();
        gen.generateEvent(i);
        gen.combineEvent();

        //write to file
        I_ToyDataGenerator::SignalMap signalMap=gen.getSignalMap();
        fileWriter.setData(signalMap);
        fileWriter.writeEventToFile();

        gen.clear_histograms();//start again
      }
      return 0;
  }catch(I_FileWriter::InputFileError& ex){
      cout<<ex.what()<<endl;
      return -1;
  }
}