codac2_FunctionArgsList.h

Go to the documentation of this file.

 
#pragma once
 
#include <list>
#include <vector>
#include <iostream>
#include <algorithm>
#include "codac2_VarBase.h"
 
namespace codac2
{
  class FunctionArgsList : public std::vector<std::shared_ptr<VarBase>>
  {
    public:

      FunctionArgsList()
      { }

      FunctionArgsList(const FunctionArgsList& args)
        : std::vector<std::shared_ptr<VarBase>>(args.size())
      {
        size_t i = 0;
        for(const auto& arg : args)
          (*this)[i++] = arg->arg_copy();
        compute_unique_arg_names();
      }

      FunctionArgsList(std::initializer_list<std::reference_wrapper<VarBase>> args)
        : FunctionArgsList(std::vector<std::reference_wrapper<VarBase>>(args))
      { }

      FunctionArgsList(const std::vector<std::reference_wrapper<VarBase>>& args)
      {
        for(const auto& arg : args)
          push_back(arg.get().arg_copy());
        compute_unique_arg_names();
      }

      Index total_size() const
      {
        Index n = 0;
        for(const auto& ai : *this)
          n += ai->size();
        return n;
      }
 
      void compute_unique_arg_names()
      {
        std::list<std::string> var_names;
        for(const auto& arg : *this)
          var_names.push_back(arg->name());
 
        int i = 23; // default first variable is x, then y,z, then starting from a...
        for(auto& arg : *this)
        {
          if(arg->name() == "?")
          {
            std::string new_name;
            do {
              new_name = 'a' + ((i++)%26);
            } while(std::find(var_names.begin(),var_names.end(),new_name) != var_names.end());
            arg->_name = new_name;
          }
        }
      }
  };
}