[TEST]UPSTREAM: Pick some source changes from 48080d0a97
* Sync new folder structure
This commit is contained in:
@@ -60,7 +60,7 @@ def read_TwoPuncture_Output(Output_File_directory):
|
||||
|
||||
## Append the computed puncture information into the AMSS-NCKU input file
|
||||
|
||||
def append_AMSSNCKU_BSSN_input(File_directory, TwoPuncture_File_directory):
|
||||
def append_AMSSNCKU_BSSN_input(File_directory, TwoPuncture_File_directory, puncture_data=None):
|
||||
|
||||
charge_Q_BH = numpy.zeros( input_data.puncture_number ) ## initialize charge for each black hole
|
||||
|
||||
@@ -74,27 +74,32 @@ def append_AMSSNCKU_BSSN_input(File_directory, TwoPuncture_File_directory):
|
||||
|
||||
## If using another method for initial data, read parameters directly from input
|
||||
else:
|
||||
position_BH = input_data.position_BH
|
||||
momentum_BH = input_data.momentum_BH
|
||||
## angular_momentum_BH = input_data.angular_momentum_BH
|
||||
angular_momentum_BH = numpy.zeros( (input_data.puncture_number, 3) ) ## initialize spin angular momentum array
|
||||
mass_BH = numpy.zeros( input_data.puncture_number ) ## initialize mass array
|
||||
if puncture_data is None:
|
||||
position_BH = input_data.position_BH
|
||||
momentum_BH = input_data.momentum_BH
|
||||
angular_momentum_BH = numpy.zeros( (input_data.puncture_number, 3) ) ## initialize spin angular momentum array
|
||||
mass_BH = numpy.zeros( input_data.puncture_number ) ## initialize mass array
|
||||
|
||||
## Set charge and spin angular momentum for each puncture
|
||||
for i in range(input_data.puncture_number):
|
||||
|
||||
if ( input_data.Symmetry == "octant-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
charge_Q_BH[i] = mass_BH[i]* input_data.parameter_BH[i,1]
|
||||
angular_momentum_BH[i] = [ 0.0, 0.0, (mass_BH[i]**2) * input_data.parameter_BH[i,2] ]
|
||||
elif ( input_data.Symmetry == "equatorial-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
charge_Q_BH[i] = mass_BH[i]* input_data.parameter_BH[i,1]
|
||||
angular_momentum_BH[i] = [ 0.0, 0.0, (mass_BH[i]**2) * input_data.parameter_BH[i,2] ]
|
||||
elif ( input_data.Symmetry == "no-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
angular_momentum_BH[i] = (mass_BH[i]**2) * input_data.dimensionless_spin_BH[i]
|
||||
charge_Q_BH[i] = mass_BH[i] * input_data.parameter_BH[i,1]
|
||||
## Set charge and spin angular momentum for each puncture
|
||||
for i in range(input_data.puncture_number):
|
||||
if ( input_data.Symmetry == "octant-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
charge_Q_BH[i] = mass_BH[i]* input_data.parameter_BH[i,1]
|
||||
angular_momentum_BH[i] = [ 0.0, 0.0, (mass_BH[i]**2) * input_data.parameter_BH[i,2] ]
|
||||
elif ( input_data.Symmetry == "equatorial-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
charge_Q_BH[i] = mass_BH[i]* input_data.parameter_BH[i,1]
|
||||
angular_momentum_BH[i] = [ 0.0, 0.0, (mass_BH[i]**2) * input_data.parameter_BH[i,2] ]
|
||||
elif ( input_data.Symmetry == "no-symmetry" ):
|
||||
mass_BH[i] = input_data.parameter_BH[i,0]
|
||||
angular_momentum_BH[i] = (mass_BH[i]**2) * input_data.dimensionless_spin_BH[i]
|
||||
charge_Q_BH[i] = mass_BH[i] * input_data.parameter_BH[i,1]
|
||||
else:
|
||||
position_BH = puncture_data.position_BH
|
||||
momentum_BH = puncture_data.momentum_BH
|
||||
angular_momentum_BH = puncture_data.angular_momentum_BH
|
||||
mass_BH = puncture_data.dimensionless_mass_BH
|
||||
charge_Q_BH = puncture_data.charge_Q_BH
|
||||
|
||||
file1 = open( os.path.join(input_data.File_directory, "AMSS-NCKU.input"), "a") ## open file in append mode
|
||||
|
||||
|
||||
Reference in New Issue
Block a user