Component.hpp

 
#ifndef GYM_PCB_COMPONENT_H
#define GYM_PCB_COMPONENT_H
 
#include "Py.hpp"
#include "Object.hpp"
#include "Color.hpp"
#include <map>
#include <set>
#include <string>
#include <vector>
 
class CloneEnv;
class Pin;
class Connection;
 
class Component : public Object
{
public:
    Component(const std::string &name);
    ~Component();
    Object *clone(CloneEnv&) const override;

    void setLocation(const Point_2 &refPoint, int layer = 0);
 
    bool isPlaced() const { return mLayerRange[0] >= 0; }
    const Point_2& getRefPoint() const { return mRefPoint; }
 
    const std::vector<Object *>& getPins() const { return getChildren(); }
    uint numPins() const { return numChildren(); }
    Pin *getPin(const std::string &name) const;
    Pin& getPin(uint i) const { return *mChildren.at(i)->as<Pin>(); }

    Pin *addPin(const std::string &name);

    Pin *removeAndDisconnectPin(Pin&);

    void gatherConnections(std::set<Connection *>&) const;

    static uint connectivity(std::vector<std::pair<Pin *, Pin *>>&, const Component&, const Component&);

    void addOccludedObject(Object&) override;
 
    std::string str() const override;
    PyObject *getPy(uint depth) const override;
 
    // UI
    void setColors(const Color &line, const Color &fill) { mColorLine = line; mColorFill = fill; }
 
    const Color& getLineColor() const { return mColorLine; }
    const Color& getFillColor() const { return mColorFill; }
 
private:
    Point_2 mRefPoint;
    std::map<std::string, Pin *> mPinMap;
    Color mColorLine;
    Color mColorFill;
};
 
inline Pin *Component::getPin(const std::string &name) const
{
    auto I = mPinMap.find(name);
    if (I == mPinMap.end())
        return 0;
    return I->second;
}
 
#endif // GYM_PCB_COMPONENT_H