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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

/*
 * Copyright (c) 2020-2021 Thomas Kramer.
 *
 * This file is part of LibrEDA
 * (see https://codeberg.org/libreda).
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

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

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
            })
    }
}