dataset.hpp 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. /**
  2. * @file
  3. * @author Caleb Fangmeier <caleb@fangmeier.tech>
  4. * @version 0.1
  5. *
  6. * @section LICENSE
  7. *
  8. *
  9. * MIT License
  10. *
  11. * Copyright (c) 2017 Caleb Fangmeier
  12. *
  13. * Permission is hereby granted, free of charge, to any person obtaining a copy
  14. * of this software and associated documentation files (the "Software"), to deal
  15. * in the Software without restriction, including without limitation the rights
  16. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  17. * copies of the Software, and to permit persons to whom the Software is
  18. * furnished to do so, subject to the following conditions:
  19. *
  20. * The above copyright notice and this permission notice shall be included in all
  21. * copies or substantial portions of the Software.
  22. *
  23. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  24. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  25. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  26. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  27. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  28. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  29. * SOFTWARE.
  30. */
  31. #ifndef dataset_hpp
  32. #define dataset_hpp
  33. #include <iostream>
  34. #include "value.hpp"
  35. #include "container.hpp"
  36. #include "log.hpp"
  37. namespace fv{
  38. typedef std::map<std::string, GenContainer*> ContainerSet;
  39. /*
  40. * A DataSet is a generic source of data that is used to populate
  41. * ObservedValues. For each ObservedValue, it is recommened that the DataSet
  42. * have a field whose value is updated when the load_next() method is called. A
  43. * pointer to this value is then passed during the creation of the
  44. * ObservedValue. It is important, therefore, that the location in memory of
  45. * the data not change from event to event.
  46. */
  47. class DataSet{
  48. private:
  49. void summary(){
  50. INFO(GenValue::summary());
  51. INFO(GenFunction::summary());
  52. }
  53. Value<int>* current_event_number;
  54. protected:
  55. ContainerSet containers;
  56. virtual bool load_next() = 0;
  57. virtual int get_events() = 0;
  58. virtual int get_current_event() = 0;
  59. std::map<std::string,std::string> get_container_name_value_map(){
  60. std::map<std::string, std::string> value_map;
  61. for(auto container : containers)
  62. value_map[container.first] = container.second->get_value_name();
  63. return value_map;
  64. }
  65. std::map<std::string,std::string> get_function_name_impl_map(){
  66. std::map<std::string, std::string> impl_map;
  67. for(auto fn : GenFunction::function_registry)
  68. if (fn.second != nullptr){
  69. impl_map[fn.first] = GenFunction::format_code(fn.second->get_impl());
  70. }
  71. return impl_map;
  72. }
  73. public:
  74. void process(bool silent=false){
  75. int events, current_event;
  76. summary();
  77. events = get_events();
  78. if (!silent) std::cout << std::endl;
  79. while( load_next() ){
  80. current_event = get_current_event();
  81. if (!silent) std::cout << "\rprocessing event: " << current_event+1 << "/" << events << std::flush;
  82. GenValue::reset();
  83. for(auto con : containers){
  84. con.second->fill();
  85. }
  86. }
  87. if (!silent) std::cout << " Finished!" << std::endl;
  88. }
  89. virtual void save_all(){
  90. for(auto container : containers)
  91. container.second->save();
  92. }
  93. template<typename C, typename... ArgTypes>
  94. C* register_container(ArgTypes... args){
  95. C* container = new C(args...);
  96. if (containers[container->get_name()] != nullptr){
  97. CRITICAL("Container with name \""+container->get_name()+"\" already exists.", -1);
  98. }
  99. containers[container->get_name()] = container;
  100. return container;
  101. }
  102. GenContainer* get_container(std::string container_name){
  103. GenContainer* c = containers[container_name];
  104. if (c == nullptr){
  105. CRITICAL("Request for container \"" << container_name << "\" failed. Doesn't exist.", -1);
  106. }
  107. return c;
  108. }
  109. DataSet(){
  110. auto& event_check = GenFunction::register_function<int()>("event_number",
  111. FUNC(([ds=this](){
  112. return ds->get_current_event();
  113. })));
  114. current_event_number = new BoundValue<int>(event_check);
  115. }
  116. Value<int>* get_current_event_number(){
  117. return current_event_number;
  118. }
  119. };
  120. }
  121. #endif // dataset_hpp