radii= getParam('radii','F','draw the radii',['T','F'] ) dsb= DataSetBuilder(2,100,2) # add the L-Shells for L in xrange(4,14,2): lamb= linspace( -PI/2,PI/2,100 ) r= L * cos(lamb)**2 for i in xrange(r.length()): dsb.putValue( -1, 0, cos( lamb[i] ) * r[i] ) dsb.putValue( -1, 1, sin( lamb[i] ) * r[i] ) dsb.nextRecord() dsb.putValue( -1, 0, -999 ) dsb.putValue( -1, 1, -999 ) dsb.nextRecord() if ( radii=='T' ): # add the radii. This doesn't work! for a in xrange( -90,91,15 ): dsb.putValue( -1, 0, cos( a*PI/180 ) * 1 ) dsb.putValue( -1, 1, sin( a*PI/180 ) * 1 ) dsb.nextRecord() dsb.putValue( -1, 0, cos( a*PI/180 ) * 15 ) dsb.putValue( -1, 1, sin( a*PI/180 ) * 15 ) dsb.nextRecord() dsb.putValue( -1, 0, -999 ) dsb.putValue( -1, 1, -999 ) dsb.nextRecord() dsb.putValue( -1, 0, -999 ) dsb.putValue( -1, 1, -999 ) dsb.nextRecord() dsb.putProperty( QDataSet.FILL_VALUE, -999 ) xy= dsb.getDataSet() result= link( xy[:,0], xy[:,1] )