passes/cmds/chtype.cc

   1 /*
   2  *  yosys -- Yosys Open SYnthesis Suite
   3  *
   4  *  Copyright (C) 2012  Clifford Wolf <clifford@clifford.at>
   5  *
   6  *  Permission to use, copy, modify, and/or distribute this software for any
   7  *  purpose with or without fee is hereby granted, provided that the above
   8  *  copyright notice and this permission notice appear in all copies.
   9  *
  10  *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  11  *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  12  *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  13  *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  14  *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  15  *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  16  *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  *
  18  */
  19
  20 #include "kernel/yosys.h"
  21
  22 USING_YOSYS_NAMESPACE
  23 PRIVATE_NAMESPACE_BEGIN
  24
  25 struct ChtypePass : public Pass {
  26         ChtypePass() : Pass("chtype", "change type of cells in the design") { }
  27         void help() YS_OVERRIDE
  28         {
  29                 //   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
  30                 log("\n");
  31                 log("    chtype [options] [selection]\n");
  32                 log("\n");
  33                 log("Change the types of cells in the design.\n");
  34                 log("\n");
  35                 log("    -set <type>\n");
  36                 log("        set the cell type to the given type\n");
  37                 log("\n");
  38                 log("    -map <old_type> <new_type>\n");
  39                 log("        change cells types that match <old_type> to <new_type>\n");
  40                 log("\n");
  41                 log("\n");
  42         }
  43         void execute(std::vector<std::string> args, RTLIL::Design *design) YS_OVERRIDE
  44         {
  45                 IdString set_type;
  46                 dict<IdString, IdString> map_types;
  47
  48                 size_t argidx;
  49                 for (argidx = 1; argidx < args.size(); argidx++)
  50                 {
  51                         if (set_type == IdString() && args[argidx] == "-set" && argidx+1 < args.size()) {
  52                                 set_type = RTLIL::escape_id(args[++argidx]);
  53                                 continue;
  54                         }
  55                         if (args[argidx] == "-map" && argidx+2 < args.size()) {
  56                                 IdString old_type = RTLIL::escape_id(args[++argidx]);
  57                                 IdString new_type = RTLIL::escape_id(args[++argidx]);
  58                                 map_types[old_type] = new_type;
  59                                 continue;
  60                         }
  61                         break;
  62                 }
  63                 extra_args(args, argidx, design);
  64
  65                 for (auto module : design->selected_modules())
  66                 {
  67                         for (auto cell : module->selected_cells())
  68                         {
  69                                 if (map_types.count(cell->type)) {
  70                                         cell->type = map_types.at(cell->type);
  71                                         continue;
  72                                 }
  73
  74                                 if (set_type != IdString()) {
  75                                         cell->type = set_type;
  76                                         continue;
  77                                 }
  78                         }
  79                 }
  80         }
  81 } ChtypePass;
  82
  83 PRIVATE_NAMESPACE_END