RuleEvaluator.h

Go to the documentation of this file.

 
#ifndef KIDS_RuleEvaluator_H
#define KIDS_RuleEvaluator_H
 
#include "kids/DataSet.h"
#include "kids/Einsum.h"
#include "kids/VariableDescriptor.h"
 
namespace PROJECT_NS {
 
struct RuleEvaluator {
    RuleEvaluator(const std::string& rule, std::shared_ptr<DataSet>& DS,  //
                  const std::string& mode, const std::string& save,       //
                  std::size_t totalFrameNumber);
 
    void calculateResult(int sampleIndex = 0);
 
    void collectResult();
 
    void writeTo(std::ofstream& ofs, void* data, int sampleIndex);
 
    std::size_t                           totalTermNumber;  
    std::size_t                           totalFrameNumber; 
    std::size_t                           numCollects;      
    std::string                           rule;             
    std::string                           mode;             
    std::string                           save;             
    std::shared_ptr<VariableDescriptor>   result;           
    std::shared_ptr<VariableDescriptor>   collect;          
    std::shared_ptr<VariableDescriptor>   reduced;          
    std::vector<VariableDescriptor>       variables;        
    std::vector<std::vector<std::size_t>> inputShapes;      
    std::vector<void*>                    inputData;        
    std::vector<kids_dtype>               inputDataTypes;   
    std::string                           expressionString; 
    std::shared_ptr<EinsumHelper>         einsumHelper;     
    std::string                           einsumString;     
    kids_dtype                            expressionType;   
    size_t                                expressionId;     
    int                                   expressionCheck;  
};
 
};  // namespace PROJECT_NS
 
#endif  // KIDS_RuleEvaluator_H