wirecell.img.clusters.ClusterMap Class Reference

Inheritance diagram for wirecell.img.clusters.ClusterMap:

Public Member Functions
def	__init__ (self, gr)
def	channel (self, key)
def	wire_chanseg (self, chan, seg)
def	wire_wip (self, wpid, wip)
def	wire_wid (self, wpid, wid)
def	find (self, typecode=None, kwds)
def	nodes_oftype (self, typecode)
def	neighbors_oftype (self, node, typecode)

Public Attributes
	gr

Private Attributes
	_id2ch
	_pi2ch
	_cs2wire
	_wip2wire
	_wid2wire

Detailed Description

Add some indexing and lookups to meta data on cluster graph vertices

Definition at line 20 of file clusters.py.

Constructor & Destructor Documentation

def wirecell.img.clusters.ClusterMap.__init__	(		self,
			gr
	)

Definition at line 25 of file clusters.py.

    def __init__(self, gr):
        self.gr = gr
        self._id2ch = dict()
        self._pi2ch = dict()
        self._cs2wire = dict()
        self._wip2wire = dict()
        self._wid2wire = dict()

        for node, data in gr.nodes.data():
            if data['code'] == 'c':
                self._id2ch[data['ident']] = node
                self._pi2ch[(data['wpid'], data['index'])] = node;
                continue;
            if data['code'] == 'w':
                self._cs2wire[(data['chid'], data['seg'])] = node
                self._wip2wire[(data['wpid'], data['index'])] = node;
                self._wid2wire[(data['wpid'], data['ident'])] = node;
                continue

Member Function Documentation

def wirecell.img.clusters.ClusterMap.channel	(		self,
			key
	)

Return a channel node by a key.  If key is scalar it is a
channel ident number, else assumed to be a pair of (wpid,
index).

Definition at line 44 of file clusters.py.

    def channel(self, key):
        '''
        Return a channel node by a key.  If key is scalar it is a
        channel ident number, else assumed to be a pair of (wpid,
        index).
        '''
        if type(key) == tuple:
            return self._pi2ch[key];
        return self._id2ch[key];
        

def wirecell.img.clusters.ClusterMap.find	(		self,
			typecode = None,
			kwds
	)

Return nodes with data matching kwds.  If typecode is given,
only consider nodes of that type.

Definition at line 73 of file clusters.py.

    def find(self, typecode=None, **kwds):
        '''
        Return nodes with data matching kwds.  If typecode is given,
        only consider nodes of that type.
        '''
        ret = list()
        for node,data in self.gr.nodes.data():
            if typecode and self.gr.nodes[node]['code'] != typecode:
                continue;
            if not match_dict(data,kwds):
                continue
            ret.append(node)
        return ret

def wirecell.img.clusters.ClusterMap.neighbors_oftype	(		self,
			node,
			typecode
	)

Return all connected nodes of the given node and given type.

Definition at line 94 of file clusters.py.

    def neighbors_oftype(self, node, typecode):
        '''
        Return all connected nodes of the given node and given type.
        '''
        ret = list()
        for nn in self.gr[node]:
            if self.gr.nodes[nn]['code'] == typecode:
                ret.append(nn)
        return ret

def wirecell.img.clusters.ClusterMap.nodes_oftype	(		self,
			typecode
	)

Return a list of nodes of given type code

Definition at line 88 of file clusters.py.

    def nodes_oftype(self, typecode):
        '''
        Return a list of nodes of given type code
        '''
        return [n for n,d in self.gr.nodes.data() if d['code'] == typecode]

def wirecell.img.clusters.ClusterMap.wire_chanseg	(		self,
			chan,
			seg
	)

Return a wire node by its channel and segment

Definition at line 54 of file clusters.py.

    def wire_chanseg(self, chan, seg):
        '''
        Return a wire node by its channel and segment
        '''
        return self._cs2wire[(chan,seg)]
    

def wirecell.img.clusters.ClusterMap.wire_wid	(		self,
			wpid,
			wid
	)

Return a wire node by its wire-ident and wire-plane ID.

Definition at line 66 of file clusters.py.

    def wire_wid(self, wpid, wid):
        '''
        Return a wire node by its wire-ident and wire-plane ID.
        '''
        return self._wid2wire[(wpid, wip)];

def wirecell.img.clusters.ClusterMap.wire_wip	(		self,
			wpid,
			wip
	)

Return a wire node by its wire-in-plane number in the given wire-plane ID

Definition at line 60 of file clusters.py.

    def wire_wip(self, wpid, wip):
        '''
        Return a wire node by its wire-in-plane number in the given wire-plane ID
        '''
        return self._wip2wire[(wpid, wip)];

Member Data Documentation

wirecell.img.clusters.ClusterMap._cs2wire

private

Definition at line 29 of file clusters.py.

wirecell.img.clusters.ClusterMap._id2ch

private

Definition at line 27 of file clusters.py.

wirecell.img.clusters.ClusterMap._pi2ch

private

Definition at line 28 of file clusters.py.

wirecell.img.clusters.ClusterMap._wid2wire

private

Definition at line 31 of file clusters.py.

wirecell.img.clusters.ClusterMap._wip2wire

private

Definition at line 30 of file clusters.py.

wirecell.img.clusters.ClusterMap.gr

Definition at line 26 of file clusters.py.

---

The documentation for this class was generated from the following file:

• wire-cell-build/python/wirecell/img/clusters.py

---

Generated by   1.8.11