First commit for TRES code in git

Author: spzwart

Makefile

@@ -0,0 +1,31 @@
+ifndef AMUSE_DIR
+    AMUSE_DIR=../..
+endif
+
+CODE_GENERATOR = $(AMUSE_DIR)/build.py
+
+CXXFLAGS = -Wall -g -DTOOLBOX  $(MUSE_INCLUDE_DIR)
+LDFLAGS = -lm $(MUSE_LD_FLAGS)
+
+OBJS = src/main_code.o src/ODE_system.o src/helper_routines.o src/tidal_friction_parameters.o src/cvode/cvode.o src/cvode/cvode_dense.o src/cvode/cvode_direct.o src/cvode/cvode_io.o src/cvode/cvode_diag.o src/cvode/cvode_spils.o src/cvode/cvode_spgmr.o src/cvode/cvode_spbcgs.o src/cvode/cvode_sptfqmr.o src/cvode/nvector_serial.o src/cvode/sundials_dense.o src/cvode/sundials_direct.o src/cvode/sundials_math.o src/cvode/sundials_nvector.o src/cvode/sundials_spgmr.o src/cvode/sundials_iterative.o src/cvode/sundials_spbcgs.o src/cvode/sundials_sptfqmr.o
+
+
+all: worker_code
+
+cleanall: clean
+	rm worker_code 
+
+clean:
+	rm -f *.so *.o *.pyc worker_code.cc src/*.o* src/cvode/*.o*
+
+worker_code.cc: interface.py
+	$(CODE_GENERATOR) --type=c interface.py SecularTripleInterface -o $@
+
+worker_code: worker_code.cc $(OBJS)
+	mpicxx $@.cc $(OBJS) -o $@
+
+.cc.o: $<
+	g++ $(CXXFLAGS) -c -o $@ $<
+
+.c.o: $<
+	g++ $(CXXFLAGS) -c -o $@ $<

__init__.py

@@ -0,0 +1,89 @@
+from amuse.units import units, constants
+from amuse.io import write_set_to_file
+from amuse.io import read_set_from_file
+from amuse.support.console import set_printing_strategy
+import numpy as np
+
+
+
+bin_type = {    
+                'unknown': 0,       
+                'merger': 1, 
+                'disintegrated': 2, 
+                'detached': 3,       
+                'contact': 4,    
+                'collision': 5,    
+                'semisecular': 6,      
+                'rlof': 7,   #only used for stopping conditions
+                'stable_mass_transfer': 8, 
+                'common_envelope': 9,     
+                'common_envelope_energy_balance': 10,     
+                'ce_e': 11,     
+                'ce_alpha': 12,     
+                'common_envelope_angular_momentum_balance': 13,
+                'ce_J': 14,
+                'ce_gamma': 15,
+                'double_common_envelope': 16,
+                'dce': 17,
+            }            
+
+
+
+
+       
+
+def rdc(file_name_root, file_type):
+    f_type = file_type
+    if file_type == "hdf5":
+        f_type = "hdf"
+        
+    file_name = file_name_root + "." + f_type            
+    if file_name_root[-4:]==".hdf":
+        file_name = file_name_root
+
+    triple=read_set_from_file(file_name , file_type)
+    counter = list(enumerate(triple.history))[0][1].number 
+
+    for i, triple in enumerate(triple.history):
+        if triple[0].number == counter:
+            counter += 1    
+            print('\n\n')
+             
+        print(' ')
+        print(i, triple[0].time, triple[0].number, triple[0].relative_inclination, triple[0].dynamical_instability, triple[0].kozai_type, triple[0].error_flag_secular)
+              
+        print( ' bs: ', triple[0].child2.bin_type, triple[0].child2.is_stable, triple[0].child2.semimajor_axis, triple[0].child2.eccentricity, triple[0].child2.argument_of_pericenter, triple[0].child2.longitude_of_ascending_node,)# triple[0].child2.mass_transfer_rate,
+        print( '|', triple[0].bin_type, triple[0].is_stable, triple[0].semimajor_axis, triple[0].eccentricity, triple[0].argument_of_pericenter, triple[0].longitude_of_ascending_node)#, triple[0].mass_transfer_rate
+        print( ' st: ',  triple[0].child2.child1.is_donor, triple[0].child2.child1.stellar_type, triple[0].child2.child1.mass,  triple[0].child2.child1.spin_angular_frequency, triple[0].child2.child1.radius, triple[0].child2.child1.core_mass,)
+        print( '|', triple[0].child2.child2.is_donor,  triple[0].child2.child2.stellar_type, triple[0].child2.child2.mass, triple[0].child2.child2.spin_angular_frequency, triple[0].child2.child2.radius,triple[0].child2.child2.core_mass, )
+        print( '|', triple[0].child1.is_donor, triple[0].child1.stellar_type, triple[0].child1.mass, triple[0].child1.spin_angular_frequency, triple[0].child1.radius, triple[0].child1.core_mass)
+
+
+
+
+def parse_arguments():
+    from amuse.units.optparse import OptionParser
+    parser = OptionParser()
+    parser.add_option("-f", dest="file_name_root", default = "TRES",
+                      help="file name [%default]")                      
+    parser.add_option("-F", dest="file_type", default = "hdf5",
+                      help="file type [%default]")  
+
+                      
+    options, args = parser.parse_args()
+    return options.__dict__
+
+
+if __name__ == '__main__':
+    options = parse_arguments()
+
+    set_printing_strategy("custom", 
+                          preferred_units = [units.MSun, units.RSun, units.Myr], 
+                          precision = 11, prefix = "", 
+                          separator = " [", suffix = "]")
+
+
+
+    print(' ')
+    rdc(**options)  
+