SchemeSQL.hpp

Go to the documentation of this file.

//'''
#pragma once
 
#include <string>
#include <sstream>
 
/*isigen:eval@utilityDeclarations'''
    attributesToIgnore = ['type', 'state']
    correctAttribute = [a for a in builderData['CellAttributesRecorded'] if a['value'] not in attributesToIgnore]
    correctGetters = builderData['CellFunctionsRecorded'] if 'CellFunctionsRecorded' in builderData else []
    nbDouble = len([a for a in correctAttribute if a['type'] == 'double']) + len([a for a in correctGetters if a['type'] == 'double'])
    nbInt = len([a for a in correctAttribute if (a['type'] == 'int') or ((a['type'] in builderData['enums']) if 'enums' in builderData else False)])
    nbInt += len([a for a in correctGetters if (a['type'] == 'int') or ((a['type'] in builderData['enums']) if 'enums' in builderData else False)])
    nbBool = len([a for a in correctAttribute if a['type'] == 'bool']) + len([a for a in correctGetters if a['type'] == 'bool'])
    def convert(type) :
        if type == 'bool':
            return "BOOLEAN"
        elif type == 'double':
            return "DOUBLE PRECISION"
        elif (type == "int") | ((type in builderData['enums']) if 'enums' in builderData else False):
            return "INTEGER"
        else:
            raise Exception(f'Unknown type {type} for cell attribute')
 
    def typeToFunction(type):
        if type == 'bool':
            return 'appendBool'
        elif type == 'double':
            return 'appendDouble'
        elif (type == 'int') or ((type in builderData['enums']) if 'enums' in builderData else False):
            return 'appendInt'
        else:
            raise Exception(f'Unknown type {type} for cell attribute')
 
    def cast_enum(a):
        if a['type'] not in builderData['enums']:
            return 'c->'+a['value']
        else:
            return 'static_cast<unsigned int>(c->'+a['value']+')'
    '''*/
 
namespace PluginSQL {
 
    struct Scheme {
 
    public:
        const std::string WorldTable;
        const std::string CellTable;
 
        std::string WorldTableCreationStatement;
        std::string CellTableCreationStatement;
 
        const std::string WorldTableDataLayout;
        const std::string CellTableDataLayout;
 
 
        Scheme(const std::string &simulation_id) :
            WorldTable(simulation_id + "_world"),
            CellTable(simulation_id + "_cells"),
            WorldTableCreationStatement("CREATE UNLOGGED TABLE " + simulation_id + "_world (" +
                                                                    "step           integer           NOT NULL," +
                                                                    "num_cells      integer           NOT NULL," +
                                                                    "hours          DOUBLE PRECISION  NOT NULL"  +
                                                                    "); "),
            CellTableCreationStatement("CREATE UNLOGGED TABLE " + simulation_id + "_cells (" +
                                                                    /*isigen:insert@CreateTableCells'''
                                                                        print('\n'.join(['"{value}\t{type}\t\tNOT NULL," +'.format(value=a['value'], type=convert(a['type'])) for a in correctAttribute]))
                                                                        print('\n'.join(['"{value}\t{type}\t\tNOT NULL," +'.format(value=a['value'][:-2], type=convert(a['type'])) for a in correctGetters]))
                                                                        '''*/
                                                                    "step        integer             NOT NULL," +
                                                                    "cell_id     integer             NOT NULL," +
                                                                    "type        integer             NOT NULL," +
                                                                    "state       integer             NOT NULL," +
                                                                    "x           DOUBLE PRECISION    NOT NULL," +
                                                                    "y           DOUBLE PRECISION    NOT NULL," +
                                                                    "z           DOUBLE PRECISION    NOT NULL," +
                                                                    "radius      DOUBLE PRECISION    NOT NULL);"),
            CellTableDataLayout(std::string("(") +
                                                    /*isigen:insert@CellTableDataLayout'''
                                                        print(('"'+', '.join([a['value'] for a in correctAttribute])+'," +') if len(correctAttribute)>0 else '')
                                                        print(('"'+', '.join([a['value'][:-2] for a in correctGetters])+'," +') if len(correctGetters)>0 else '')
                                                        '''*/
                                                    "step, cell_id, type, state, x, y, z, radius)"),
            WorldTableDataLayout("(step, num_cells, hours)")
        {}
 
        /*isigen:insert@CellValueStatement'''
            print(f'static constexpr size_t CellRowSize = {2+(nbDouble+4)*12+(nbInt+4)*8+nbBool*5};')
            '''*/
        static constexpr size_t WorldRowSize = 2 + 2*8 + 1*12;
 
        template<typename cell_t, typename world_t>
        static void CellValueBinary(Connection &conn, cell_t* c, world_t* w) {
            /*isigen:insert@CellValueStatement'''
                print(f'conn.beginRow({8+len(correctAttribute)+len(correctGetters)});')
                print('\n'.join(['conn.{addFunc}(c->{value});'.format(value=a['value'], addFunc=typeToFunction(a['type'])) for a in correctAttribute]))
                print('\n'.join(['conn.{addFunc}(c->{value});'.format(value=a['value'], addFunc=typeToFunction(a['type'])) for a in correctGetters]))
                '''*/
            conn.appendInt(w->getNbUpdates());
            conn.appendInt(c->getId());
            conn.appendInt(c->getType());
            conn.appendInt(c->getState());
            auto pos = c->getBody().getPosition();
            conn.appendDouble(pos.x());
            conn.appendDouble(pos.y());
            conn.appendDouble(pos.z());
            conn.appendDouble(c->getBody().getBoundingBoxRadius());
        }
 
        template<typename world_t>
        static void WorldValueBinary(Connection &conn, world_t *w) {
            conn.beginRow(3);
            conn.appendInt(w->getNbUpdates());
            conn.appendInt(w->cells.size());
            conn.appendDouble(w->getNbUpdates() * w->dt / 3600.);
        }
 
        template<typename cell_t, typename world_t>
        static void CellValueStatement(std::stringstream& ss, cell_t *c, world_t *w) {
            auto pos = c->getBody().getPosition();
            ss << "(";
            /*isigen:insert@CellValueStatement'''
                print('\n'.join(['ss << "CAST(" << {value} << " AS {type})" << ", ";'.format(value=cast_enum(a), type=convert(a['type'])) for a in correctAttribute]))
                print('\n'.join(['ss << "CAST(" << {value} << " AS {type})" << ", ";'.format(value=cast_enum(a), type=convert(a['type'])) for a in correctGetters]))
                '''*/
            ss << w->getNbUpdates() << ", ";
            ss << c->getId() << ", ";
            ss << static_cast<unsigned int>(c->getType()) << ",";
            ss << static_cast<unsigned int>(c->getState()) << ",";
            ss << "CAST(" << pos.x() << " AS DOUBLE PRECISION), ";
            ss << "CAST(" << pos.y() << " AS DOUBLE PRECISION), ";
            ss << "CAST(" << pos.z() << " AS DOUBLE PRECISION), ";
            ss << "CAST(" << c->getBody().getBoundingBoxRadius() << " AS DOUBLE PRECISION)";
            ss << " )";
        }
        template<typename world_t>
        static std::string WorldValueStatement(world_t *w) {
            std::stringstream ss;
            ss << "(";
            ss << w->getNbUpdates() << ", ";
            ss << w->cells.size() << ", ";
            ss << w->getNbUpdates() * w->dt / 3600;
            ss << " )";
 
            return ss.str();
        }
 
    };
}
//'''