Generator.cpp

Go to the documentation of this file.

#include "joedb/compiler/generator/Generator.h"
#include "joedb/compiler/write_source_code.h"
#include "joedb/get_version.h"
#include "joedb/ui/get_time_string.h"
 
#include <iostream>
 
namespace joedb::generator
{
 ////////////////////////////////////////////////////////////////////////////
 bool Generator::db_has_values() const
 ////////////////////////////////////////////////////////////////////////////
 {
  for (const auto &[tid, tname]: options.db.get_tables())
  {
   if (options.db.get_freedom(tid).get_used_count() > Record_Id{0})
    return true;
  }
  return false;
 }
 
 ////////////////////////////////////////////////////////////////////////////
 bool Generator::table_has_field_type(Table_Id table_id, Type::Type_Id type_id) const
 ////////////////////////////////////////////////////////////////////////////
 {
  for (const auto &[fid, fname]: options.db.get_fields(table_id))
   if (options.db.get_field_type(table_id, fid).get_type_id() == type_id)
    return true;
  return false;
 }
 
 ////////////////////////////////////////////////////////////////////////////
 void Generator::write_initial_comment(std::ostream &out)
 ////////////////////////////////////////////////////////////////////////////
 {
  out << "/////////////////////////////////////////////////////////////////////////////\n";
  out << "//\n";
  out << "// This code was automatically generated by the joedb compiler\n";
  out << "// https://www.joedb.org/\n";
  out << "//\n";
  out << "// Path to compiler: " << options.exe_path << '\n';
  out << "// Version: " << joedb::get_version() << '\n';
  out << "// joedbc compilation time: " << __DATE__ " " __TIME__ "\n";
  out << "// Generation of this file: " << get_time_string_of_now() << '\n';
  out << "//\n";
  out << "/////////////////////////////////////////////////////////////////////////////\n";
 }
 
 ////////////////////////////////////////////////////////////////////////////
 void Generator::write_type
 ////////////////////////////////////////////////////////////////////////////
 (
  std::ostream &out,
  Type type,
  bool return_type,
  bool setter_type
 )
 {
  switch (type.get_type_id())
  {
   case Type::Type_Id::null:
    out << "void";
   break;
 
   case Type::Type_Id::reference:
    out << "id_of_" << options.db.get_table_name(type.get_table_id());
   break;
 
   case Type::Type_Id::string:
    if (return_type)
     out << "const std::string &";
    else if (setter_type)
     out << "std::string_view";
    else
     out << "std::string";
   break;
 
   #define TYPE_MACRO(storage_tt, return_tt, type_id, read, write)\
   case Type::Type_Id::type_id:\
    if (return_type || setter_type)\
     out << #return_tt;\
    else\
     out << #storage_tt;\
   break;
   #define TYPE_MACRO_NO_REFERENCE
   #define TYPE_MACRO_NO_STRING
   #include "joedb/TYPE_MACRO.h"
  }
 }
 
 ////////////////////////////////////////////////////////////////////////////
 void Generator::write_tuple_type
 ////////////////////////////////////////////////////////////////////////////
 (
  std::ostream &out,
  const Compiler_Options::Index &index,
  bool reference
 )
 {
  if (index.field_ids.size() > 1)
  {
   out << "std::tuple<";
 
   for (size_t i = 0; i < index.field_ids.size(); i++)
   {
    if (i > 0)
     out << ", ";
 
    const Type &type = options.db.get_field_type
    (
     index.table_id,
     index.field_ids[i]
    );
 
    write_type(out, type, false, false);
   }
 
   out << ">";
  }
  else
  {
   if (!reference)
   {
    write_type
    (
     out,
     options.db.get_field_type(index.table_id, index.field_ids[0]),
     false,
     false
    );
   }
  }
 }
 
 ////////////////////////////////////////////////////////////////////////////
 // type arrays
 ////////////////////////////////////////////////////////////////////////////
 #define STRINGIFY(X) #X
 #define EXPAND_AND_STRINGIFY(X) STRINGIFY(X)
 
 ////////////////////////////////////////////////////////////////////////////
 const char *Generator::get_type_string(Type type)
 ////////////////////////////////////////////////////////////////////////////
 {
  static constexpr char const * const types[] =
  {
   nullptr,
  #define TYPE_MACRO(a, b, type_id, d, e) EXPAND_AND_STRINGIFY(type_id),
  #include "joedb/TYPE_MACRO.h"
  };
 
  return types[int(type.get_type_id())];
 }
 
 ////////////////////////////////////////////////////////////////////////////
 const char *Generator::get_cpp_type_string(Type type)
 ////////////////////////////////////////////////////////////////////////////
 {
  static char const * const cpp_types[] =
  {
   nullptr,
  #define TYPE_MACRO(a, type, c, d, e) EXPAND_AND_STRINGIFY(type),
  #include "joedb/TYPE_MACRO.h"
  };
 
  return cpp_types[int(type.get_type_id())];
 }
 
 ////////////////////////////////////////////////////////////////////////////
 const char *Generator::get_storage_type_string(Type type)
 ////////////////////////////////////////////////////////////////////////////
 {
  static char const * const storage_types[] =
  {
   nullptr,
  #define TYPE_MACRO(storage, b, c, d, e) EXPAND_AND_STRINGIFY(storage),
  #include "joedb/TYPE_MACRO.h"
  };
 
  return storage_types[int(type.get_type_id())];
 }
 
 #undef EXPAND_AND_STRINGIFY
 #undef STRINGIFY
 
 ////////////////////////////////////////////////////////////////////////////
 Generator::Generator
 ////////////////////////////////////////////////////////////////////////////
 (
  std::string dir_name,
  std::string file_name,
  const Compiler_Options &options
 ):
  dir_name(std::move(dir_name)),
  file_name(std::move(file_name)),
  options(options)
 {
 }
 
 ////////////////////////////////////////////////////////////////////////////
 void Generator::generate()
 ////////////////////////////////////////////////////////////////////////////
 {
  const std::string dir_string =
   options.output_path + "/" +
   options.get_name_space_back() + "/" +
   std::string(dir_name);
 
  write_source_code(dir_string, file_name, [&](std::ostream &out)
  {
   write_initial_comment(out);
   write(out);
  });
 }
 
 ////////////////////////////////////////////////////////////////////////////
 Generator::~Generator() = default;
 ////////////////////////////////////////////////////////////////////////////
}