1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

// Copyright (c) 2020-2021 Thomas Kramer.
// SPDX-FileCopyrightText: 2022 Thomas Kramer
//
// SPDX-License-Identifier: AGPL-3.0-or-later

use super::layout_reference_access::*;
use super::netlist_reference_access::*;
use crate::prelude::L2NBase;

impl<'a, L: L2NBase> ShapeRef<'a, L> {
    /// Get the net which is connected to this shape, if any.
    pub fn net(&self) -> Option<NetRef<L>> {
        self.base.get_net_of_shape(&self.id).map(|id| NetRef {
            id,
            base: self.base,
        })
    }

    /// Get the pin which is connected to this shape, if any.
    pub fn pin(&self) -> Option<PinRef<L>> {
        self.base.get_pin_of_shape(&self.id).map(|id| PinRef {
            id,
            base: self.base,
        })
    }
}

impl<'a, L: L2NBase> NetRef<'a, L> {
    /// Iterate over all shapes attached to this net.
    pub fn each_shape(&self) -> impl Iterator<Item = ShapeRef<L>> {
        self.base.shapes_of_net(&self.id).map(move |id| ShapeRef {
            id,
            base: self.base,
        })
    }
}

impl<'a, L: L2NBase> PinRef<'a, L> {
    /// Iterate over all shapes attached to this pin.
    pub fn each_shape(&self) -> impl Iterator<Item = ShapeRef<L>> {
        self.base.shapes_of_pin(&self.id).map(move |id| ShapeRef {
            id,
            base: self.base,
        })
    }
}