opcodes.h

/* Jazz (c) 2018-2025 kaalam.ai (The Authors of Jazz), using (under the same license):
 
    1. Biomodelling - The AATBlockQueue class (c) Jacques Basaldúa, 2009-2012 licensed
      exclusively for the use in the Jazz server software.
 
      Copyright 2009-2012 Jacques Basaldúa
 
    2. BBVA - Jazz: A lightweight analytical web server for data-driven applications.
 
      Copyright 2016-2017 Banco Bilbao Vizcaya Argentaria, S.A.
 
      This product includes software developed at
 
      BBVA (https://www.bbva.com/)
 
    3. LMDB, Copyright 2011-2017 Howard Chu, Symas Corp. All rights reserved.
 
      Licensed under http://www.OpenLDAP.org/license.html
 
 
      Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at
 
      http://www.apache.org/licenses/LICENSE-2.0
 
      Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
*/
 
 
#include "src/jazz_bebop/space.h"
#include "src/onnx_proto/onnx.pb.h"
#include "onnxruntime_c_api.h"
 
#if defined CATCH_TEST
#ifndef INCLUDED_JAZZ_CATCH2
#define INCLUDED_JAZZ_CATCH2
 
#include "src/catch2/catch.hpp"
 
#endif
#endif
 
 
#ifndef INCLUDED_JAZZ_BEBOP_OPCODES
#define INCLUDED_JAZZ_BEBOP_OPCODES
 
 
namespace jazz_bebop
{
 
struct TensorType {
    int jazz_type;                                          
    onnx::TensorProto::DataType onnx_proto_type;            
    ONNXTensorElementDataType onnx_rt_type;                 
};
typedef std::map<stdName, TensorType>   TensorTypeDict;     
typedef std::vector<TensorType>         TensorTypes;        
typedef TensorTypes *pTensorTypes;                          
 
 
struct AttributeType {
    int jazz_type;                                          
    onnx::AttributeProto::AttributeType onnx_proto_type;    
    bool is_multi = false;                                  
};
typedef std::map<stdName, AttributeType> AttributeTypeDict; 
typedef AttributeType *pAttributeType;                      
 
 
class OnnxParameter {
 
    public:
 
        OnnxParameter(stdName nam, TensorTypes typ) {
            name    = nam;
            types   = typ;
        }
 
        stdName     name;                                       
        TensorTypes types;                                      
};
typedef std::vector<OnnxParameter>          OnnxParameters;     
 
 
class OnnxAttribute {
 
    public:
 
        OnnxAttribute(stdName nam, AttributeType typ) {
            name = nam;
            type = typ;
        }
 
        stdName       name;                                     
        AttributeType type;                                     
};
typedef std::vector<OnnxAttribute>          OnnxAttributes;     
 
 
class OnnxOpCode {
 
    public:
 
        OnnxOpCode(stdName nam, int version, OnnxParameters in, OnnxParameters out, OnnxAttributes attr) {
            name            = nam;
            opset_version   = version;
            inputs          = in;
            outputs         = out;
            attributes      = attr;
        }
 
        stdName name;                           
        int opset_version;                      
        OnnxParameters inputs;                  
        OnnxParameters outputs;                 
        OnnxAttributes attributes;              
};
typedef OnnxOpCode *pOnnxOpCode;                
typedef std::vector<OnnxOpCode> OnnxOpCodes;    
 
 
class stdNameVersion {
 
    public:
 
        stdNameVersion(stdName nam, int version) {
            name          = nam;
            opset_version = version;
        }
 
        bool operator==(const stdNameVersion &o) const {
            return (opset_version == o.opset_version) && (strcmp(name.name, o.name.name) == 0);
        }
 
        bool operator<(const stdNameVersion &o) const {
            return (opset_version < o.opset_version) || ((opset_version == o.opset_version) && (strcmp(name.name, o.name.name) < 0));
        }
 
        stdName name;               
        int opset_version;          
};
typedef std::map<stdNameVersion, int> OnnxOpCodeDict;       
 
 
class OpCodes : public Service {
 
    public:
 
        OpCodes(pLogger a_logger, pConfigFile a_config);
 
        virtual StatusCode start    ();
        virtual StatusCode shut_down();
 
        int latest_opset_version() {
            return op_vers_latest;
        }
 
        bool set_opset_version(int version) {
            if (version <= 0 || version > op_vers_latest)
                return false;
 
            op_vers_current = version;
            return true;
        }
 
        pOnnxOpCode get(stdName name) {
            OnnxOpCodeDict::iterator it = opcodes_idx.find(stdNameVersion(name, op_vers_current));
 
            if (it == opcodes_idx.end())
                return nullptr;
 
            return &opcodes.at(it->second);
        }
 
#ifndef CATCH_TEST
    private:
#endif
 
        bool build_opcode_dict     ();
        bool fill_op_code          (OnnxOpCode    &op);
        bool fill_all_dict_versions();
        bool fill_tensor_types     (TensorTypes   &types,
                                    std::string   &all_types);
        bool fill_attribute_type   (AttributeType &type,
                                    std::string   &type_name);
 
        ConfigFile onnx_conf = ConfigFile(nullptr);     
        int ir_vers = 0;                                
        int op_vers_latest = 0;                         
        int op_vers_current = 0;                        
        OnnxOpCodeDict opcodes_idx = {};                
        OnnxOpCodes opcodes = {};                       
};
typedef OpCodes *pOpCodes;      
 
} // namespace jazz_bebop
 
#endif // ifndef INCLUDED_JAZZ_BEBOP_OPCODES