-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathsamplesModule.cc
57 lines (55 loc) · 1.77 KB
/
samplesModule.cc
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
45
46
47
48
49
50
51
52
53
54
55
56
57
#include "samples.h"
#include <string>
#include <iostream>
extern "C" {
double SC_fixed_lumi(){ return AnaSamples::luminosity; }
AnaSamples::SampleCollection* SC_new()
{
AnaSamples::SampleSet *ss = new AnaSamples::SampleSet();
return new AnaSamples::SampleCollection(*ss);
}
int SC_samples_size(AnaSamples::SampleCollection* sc, char *scn){ return (*sc)[std::string(scn)].size(); }
char const ** SC_samples(AnaSamples::SampleCollection* sc, char *scn)
{
auto& sampleVec = (*sc)[std::string(scn)];
const char **array = new const char*[sampleVec.size()];
int i = 0;
for(auto& sample : sampleVec)
{
array[i++] = sample.filePath.c_str();
}
return array;
}
char const ** SC_samples_names(AnaSamples::SampleCollection* sc, char *scn)
{
auto& sampleVec = sc->getSampleLabels(std::string(scn));
const char **array = new const char*[sampleVec.size()];
int i = 0;
for(auto& sample : sampleVec)
{
array[i++] = sample.c_str();
}
return array;
}
int SC_samplecollection_size(AnaSamples::SampleCollection* sc, char *scn){ return sc->size(); }
char const ** SC_samplecollection_names(AnaSamples::SampleCollection* sc)
{
const char **array = new const char*[sc->size()];
int i = 0;
for(auto& sample : *sc)
{
array[i++] = sample.first.c_str();
}
return array;
}
double const * SC_samplecollection_lumis(AnaSamples::SampleCollection* sc)
{
double *array = new double[sc->size()];
int i = 0;
for(auto& sample : *sc)
{
array[i++] = sc->getSampleLumi(sample.first);
}
return array;
}
}