~jrjohansson/qutip/master

« back to all changes in this revision

Viewing changes to qutip/spost.py

  • Committer: Paul Nation
  • Date: 2011-04-21 04:46:56 UTC
  • Revision ID: git-v1:dd4c966b490aa468dfbd28cef66694df4bf235c8

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#This file is part of QuTIP.
 
2
#
 
3
#    QuTIP is free software: you can redistribute it and/or modify
 
4
#    it under the terms of the GNU General Public License as published by
 
5
#    the Free Software Foundation, either version 3 of the License, or
 
6
#   (at your option) any later version.
 
7
#
 
8
#    QuTIP is distributed in the hope that it will be useful,
 
9
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
#    GNU General Public License for more details.
 
12
#
 
13
#    You should have received a copy of the GNU General Public License
 
14
#    along with QuTIP.  If not, see <http://www.gnu.org/licenses/>.
 
15
#
 
16
# Copyright (C) 2011, Paul D. Nation & Robert J. Johansson
 
17
#
 
18
###########################################################################
 
19
import scipy
 
20
import scipy.linalg as la
 
21
import scipy.sparse as sp
 
22
from scipy import prod, transpose, reshape
 
23
from qobj import *
 
24
from istests import *
 
25
from operators import destroy
 
26
 
 
27
 
 
28
def spost(A,*args):
 
29
        if not isoper(A):
 
30
                raise TypeError('Input is not a quantum object')
 
31
 
 
32
        d=A.dims[0]
 
33
        S=qobj()
 
34
        S.dims=[[d[:],A.dims[1][:]],[d[:],A.dims[0][:]]]
 
35
        S.shape=[prod(S.dims[0][0])*prod(S.dims[0][1]),prod(S.dims[1][0])*prod(S.dims[1][1])]
 
36
        S.data=sp.kron(A.data.T,sp.identity(prod(d)))
 
37
        return S
 
38