-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #50 from c-dilks/range-based-it
refactor!: replace `hipo::bank::iterator` with `hipo::bank::rowlist`
- Loading branch information
Showing
12 changed files
with
562 additions
and
308 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,4 +1,7 @@ | ||
| *.o | ||
| *.exe | ||
| *.hipo | ||
| lib | ||
| slib | ||
| __pycache__ | ||
| build |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,174 @@ | ||
| //****************************************************************** | ||
| //* ██╗ ██╗██╗██████╗ ██████╗ ██╗ ██╗ ██████╗ | ||
| //* ██║ ██║██║██╔══██╗██╔═══██╗ ██║ ██║ ██╔═████╗ | ||
| //* ███████║██║██████╔╝██║ ██║ ███████║ ██║██╔██║ | ||
| //* ██╔══██║██║██╔═══╝ ██║ ██║ ╚════██║ ████╔╝██║ | ||
| //* ██║ ██║██║██║ ╚██████╔╝ ██║██╗╚██████╔╝ | ||
| //* ╚═╝ ╚═╝╚═╝╚═╝ ╚═════╝ ╚═╝╚═╝ ╚═════╝ | ||
| //************************ Jefferson National Lab (2017) *********** | ||
| //****************************************************************** | ||
| //* Example program for reading HIPO-4 Files.. | ||
| //*-- | ||
| //* Author: G.Gavalian | ||
| //* | ||
|
|
||
| #include <cstdlib> | ||
| #include <iostream> | ||
| #include <cassert> | ||
| #include "reader.h" | ||
| #include "twig.h" | ||
| #include "reaction.h" | ||
|
|
||
| // example using hipo::bank::rowlist to loop through a bank's rows | ||
| void example1(const char *file){ | ||
| printf("===== EXAMPLE 1 =====\n"); | ||
| hipo::reader r(file); | ||
| hipo::banklist list = r.getBanks({"REC::Particle","REC::Event"}); | ||
| int counter = 0; | ||
| while( r.next(list)&&counter<350){ | ||
| counter++; | ||
| int nrows = list[0].getRows(); | ||
| printf("\n>>> NEW EVENT <<<\n"); | ||
| printf(" rows = %d\n",nrows); | ||
|
|
||
| // first, loop over all the rows | ||
| printf(" loop over all rows:\n"); | ||
| for(auto const& row : list[0].getRowList()) | ||
| printf("\t pid [%d] = %d\n", row, list[0].getInt(0,row)); | ||
|
|
||
| // then, set the rowlist to contain only a few rows, and loop over those | ||
| if(nrows>6){ | ||
|
|
||
| // set the list of rows, then loop over those | ||
| printf(" there are more than 6 rows, let's loop over some specific rows: 0, 1, and 4\n"); | ||
| list[0].getMutableRowList().setList({0, 1, 4}); | ||
| for(auto const& row : list[0].getRowList()) | ||
| printf("\t pid [%d] = %d\n", row, list[0].getInt(0,row)); | ||
| assert((list[0].getRowList().size() == 3)); // test | ||
|
|
||
| // it's still possible to access the full set of rows | ||
| printf(" we can still loop over all the rows with `getFullRowList()`\n"); | ||
| for(auto const& row : list[0].getFullRowList()) | ||
| printf("\t pid [%d] = %d\n", row, list[0].getInt(0,row)); | ||
| printf(" or similarly with `getRows()`\n"); | ||
| for(int row=0; row<list[0].getRows(); row++) | ||
| printf("\t pid [%d] = %d\n", row, list[0].getInt(0,row)); | ||
| assert((static_cast<int>(list[0].getFullRowList().size()) == list[0].getRows())); // test | ||
|
|
||
| // you may also reset the list to its original, full state | ||
| printf(" resetting the rowlist restores the full row list\n"); | ||
| list[0].getMutableRowList().reset(); | ||
| for(auto const& row : list[0].getRowList()) | ||
| printf("\t pid [%d] = %d\n", row, list[0].getInt(0,row)); | ||
| assert((static_cast<int>(list[0].getRowList().size()) == list[0].getRows())); // test | ||
| } | ||
|
|
||
| } | ||
| } | ||
|
|
||
| // example showing how to filter a bank's rowlist with an expression | ||
| void example2(const char *file){ | ||
| printf("===== EXAMPLE 2 =====\n"); | ||
| hipo::reader r(file); | ||
| hipo::banklist list = r.getBanks({"REC::Particle","REC::Event"}); | ||
| int counter = 0; | ||
| while( r.next(list)&&counter<350){ | ||
| counter++; | ||
| printf("\n>>> NEW EVENT <<<\n"); | ||
| printf("=== BEFORE ==================================\n"); | ||
| list[0].show(); | ||
| list[0].getMutableRowList().filter("charge!=0"); | ||
| printf("=== AFTER ===================================\n"); | ||
| list[0].show(); | ||
| // list[0].show(true); // call `show(true)` if you still need to see the full (not filtered) bank | ||
| } | ||
| } | ||
|
|
||
| // example showing how to filter a bank's rowlist with an first-order (lambda) function | ||
| void example3(const char *file){ | ||
| printf("===== EXAMPLE 3 =====\n"); | ||
|
|
||
| hipo::reader r(file); | ||
| hipo::banklist list = r.getBanks({"REC::Particle","REC::Event"}); | ||
| std::function charged = [](hipo::bank &b, int row) { return b.getInt("charge",row)!=0 ? 1.0 : 0.0;}; | ||
|
|
||
| int counter = 0; | ||
| while( r.next(list)&&counter<350){ | ||
| counter++; | ||
| printf("\n>>> NEW EVENT <<<\n"); | ||
| printf("=== BEFORE ==================================\n"); | ||
| list[0].show(); | ||
| list[0].getMutableRowList().filter(charged); | ||
| printf("=== AFTER ===================================\n"); | ||
| list[0].show(); | ||
| // list[0].show(true); // call `show(true)` if you still need to see the full (not filtered) bank | ||
| } | ||
| } | ||
|
|
||
| // Example showing how rowlists can be used to get indicies of the reference bank. | ||
| // The link function will return a hipo::bank::rowlist for the bank "REC::Calorimeter" | ||
| // where the column number 1 (which is "pindex") has a value of 0. Then by itarating over the returned | ||
| // indicies the total energy can be calculated. | ||
| void example4(const char *file){ | ||
| printf("===== EXAMPLE 4 =====\n"); | ||
| hipo::reader r(file); | ||
| hipo::banklist list = r.getBanks({"REC::Particle","REC::Calorimeter"}); | ||
| int const pindex_column = 1; // of REC::Calorimeter | ||
| int const electron_row = 0; | ||
| int counter = 0; | ||
|
|
||
| while( r.next(list)&&counter<350){ | ||
| counter++; | ||
| int status = list[0].getInt("status",electron_row); | ||
| if(list[0].getInt("pid",electron_row)==11&&abs(status)>=2000&&abs(status)<3000){ | ||
| printf("\n>>> NEW EVENT <<<\n"); | ||
| printf("found electron in row %d\n", electron_row); | ||
| double energy = 0.0; | ||
| for(auto const& it : list[1].getRowListLinked(electron_row, pindex_column)){ | ||
| auto e = list[1].getFloat("energy",it); | ||
| printf(" => links REC::Calorimeter row=%d, which has energy %f\n", it, e); | ||
| energy += e; | ||
| } | ||
| printf("=========================\ntotal energy = %f\n=========================\n",energy); | ||
| list[0].show(); | ||
| list[1].show(); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| int main(int argc, char** argv) { | ||
|
|
||
| std::cout << " reading CLAS12 hipo file and plotting missing mass (ep->epi+pi-X) " << __cplusplus << std::endl; | ||
| //axis x(120,0.,1.0); | ||
| //printf("bin = %d\n",x.find(0.4)); | ||
|
|
||
| char inputFile[256]; | ||
| int example_num = -1; | ||
|
|
||
| if(argc>1) { | ||
| snprintf(inputFile,256,"%s",argv[1]); | ||
| //sprintf(outputFile,"%s",argv[2]); | ||
| } else { | ||
| std::cerr << " *** please provide a file name..." << std::endl; | ||
| exit(1); | ||
| } | ||
| if(argc>2) | ||
| example_num = std::stoi(argv[2]); | ||
|
|
||
| switch(example_num) { | ||
| case -1: | ||
| example1(inputFile); | ||
| example2(inputFile); | ||
| example3(inputFile); | ||
| example4(inputFile); | ||
| break; | ||
| case 1: example1(inputFile); break; | ||
| case 2: example2(inputFile); break; | ||
| case 3: example3(inputFile); break; | ||
| case 4: example4(inputFile); break; | ||
| default: | ||
| std::cerr << " *** ERROR: unknown example..." << std::endl; | ||
| exit(1); | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.