#include "yosys.h"
#include <boost/python.hpp>
+struct YosysDesign;
+struct YosysModule;
+struct YosysCell;
+struct YosysWire;
+
void yosys_setup()
{
Yosys::log_streams.push_back(&std::cout);
struct YosysCell
{
- unsigned int hashid;
-
- YosysCell(unsigned int hashid)
- {
- this->hashid = hashid;
- }
+ Yosys::RTLIL::IdString name;
+ Yosys::RTLIL::IdString parent_name;
YosysCell(Yosys::RTLIL::Cell* ref)
{
- this->hashid = ref->hashidx_;
+ this->name = ref->name;
+ this->parent_name = ref->module->name;
}
Yosys::RTLIL::Cell* get_cpp_obj()
Yosys::RTLIL::Design* active_design = Yosys::yosys_get_design();
if(active_design == NULL)
return NULL;
- for(auto &mod_it : active_design->modules_)
- {
- for(auto &cell_it : mod_it.second->cells_)
- if(cell_it.second->hashidx_ == this->hashid)
- return cell_it.second;
- }
- return NULL;
+ if(active_design->modules_[this->parent_name] == NULL)
+ return NULL;
+ return active_design->modules_[this->parent_name]->cells_[this->name];
}
};
std::ostream &operator<<(std::ostream &ostr, const YosysCell &cell)
{
- ostr << "Cell with id " << cell.hashid;
+ ostr << "Cell with name " << cell.name.c_str();
return ostr;
}
struct YosysWire
{
- unsigned int hashid;
-
- YosysWire(unsigned int hashid)
- {
- this->hashid = hashid;
- }
+ Yosys::RTLIL::IdString name;
+ Yosys::RTLIL::IdString parent_name;
YosysWire(Yosys::RTLIL::Wire* ref)
{
- this->hashid = ref->hashidx_;
+ this->name = ref->name;
+ this->parent_name = ref->module->name;
}
Yosys::RTLIL::Wire* get_cpp_obj()
Yosys::RTLIL::Design* active_design = Yosys::yosys_get_design();
if(active_design == NULL)
return NULL;
- for(auto &mod_it : active_design->modules_)
- {
- for(auto &wire_it : mod_it.second->wires_)
- if(wire_it.second->hashidx_ == this->hashid)
- return wire_it.second;
- }
- return NULL;
+ if(active_design->modules_[this->parent_name] == NULL)
+ return NULL;
+ return active_design->modules_[this->parent_name]->wires_[this->name];
}
};
std::ostream &operator<<(std::ostream &ostr, const YosysWire &wire)
{
- ostr << "Wire with id " << wire.hashid;
+ ostr << "Wire with name " << wire.name.c_str();
return ostr;
}
-
struct YosysModule
{
- unsigned int hashid;
-
- YosysModule(unsigned int hashid)
- {
- this->hashid = hashid;
- }
+ Yosys::RTLIL::IdString name;
+ unsigned int parent_hashid;
YosysModule(Yosys::RTLIL::Module* ref)
{
- this->hashid = ref->hashidx_;
+ this->name = ref->name;
+ this->parent_hashid = ref->design->hashidx_;
}
Yosys::RTLIL::Module* get_cpp_obj()
Yosys::RTLIL::Design* active_design = Yosys::yosys_get_design();
if(active_design == NULL)
return NULL;
- for(auto &mod_it : active_design->modules_)
- {
- if(mod_it.second->hashidx_ == this->hashid)
- return mod_it.second;
- }
- return NULL;
+ if(active_design->hashidx_ != this->parent_hashid)
+ printf("Somehow the active design changed!\n");
+ return active_design->modules_[this->name];
}
boost::python::list get_cells()
std::ostream &operator<<(std::ostream &ostr, const YosysModule &module)
{
- ostr << "Module with id " << module.hashid;
+ ostr << "Module with name " << module.name.c_str();
return ostr;
}
{
printf("design is not null and has id %u\n", active_design->hashidx_);
this->hashid = active_design->hashidx_;
- /*
- for (auto &mod_it : active_design->modules_)
- {
- printf("found module in design!!!\n");
- //design->add(it.second->clone());
-
- for (auto &wire_it : mod_it.second->wires_)
- {
- printf("found wire in module!!!\n");
- }
-
- for (auto &cell_it : mod_it.second->cells_)
- {
- printf("found cell in module!!!\n");
- }
- }*/
}
}
.def("get_modules", &YosysDesign::get_modules)
;
- class_<YosysModule>("YosysModule", init<unsigned int>())
+ class_<YosysModule>("YosysModule", no_init)
.def(boost::python::self_ns::str(boost::python::self_ns::self))
.def(boost::python::self_ns::repr(boost::python::self_ns::self))
.def("get_cells", &YosysModule::get_cells)
.def("get_wires", &YosysModule::get_wires)
;
- class_<YosysCell>("YosysCell", init<unsigned int>())
+ class_<YosysCell>("YosysCell", no_init)
.def(boost::python::self_ns::str(boost::python::self_ns::self))
.def(boost::python::self_ns::repr(boost::python::self_ns::self))
;
- class_<YosysWire>("YosysWire", init<unsigned int>())
+ class_<YosysWire>("YosysWire", no_init)
.def(boost::python::self_ns::str(boost::python::self_ns::self))
.def(boost::python::self_ns::repr(boost::python::self_ns::self))
;