diff --git a/utilities.py b/utilities.py index 36dd4a12ec388ed2892145fc17689754259aff7b..2fe0ad923a5976a329ad1573570b7e300f45c126 100644 --- a/utilities.py +++ b/utilities.py @@ -51,6 +51,22 @@ def multilayer_format_geo2net(net_dir,outname,w_inter=0.5,wheuristic=True,weight fout.write("%s %s %s %s %f \n" % (vals[1],vals[0],vals[3].strip(),vals[2],w_inter)) fout.close() + +def flatten_network(input_path,output_path,out_sl): + fout = open(output_path,'w') + fsl = open(out_sl,'w') + f = open(input_path, 'r') + for line in f: + vals = line.split(" ") + u = int(vals[1].strip()) + v = int(vals[3].strip()) + w = float(vals[4].strip()) + fout.write("%d %d %d %d %f\n" % (1,u,1,v,w)) + fsl.write("%d %d %f\n" % (u,v,w)) + fout.close() + fsl.close() + + # writes a qml style file starting from a infomap clustering solution in .clu format def qml_from_clu(clu_file, outname): fout = open(outname, 'w') @@ -84,3 +100,8 @@ def qml_from_clu(clu_file, outname): fout.write(r'</colorPalette></rasterrenderer><brightnesscontrast brightness="0" contrast="0"/><huesaturation colorizeGreen="128" colorizeOn="0" colorizeRed="255" colorizeBlue="128" grayscaleMode="0" saturation="0" colorizeStrength="100"/><rasterresampler maxOversampling="2"/></pipe><blendMode>0</blendMode></qgis>') fout.close() +if __name__=='__main__': + #multilayer_format_geo2net('nets_K10_flat_0_tps_ilweights','mlnet_K10_cmeans_ilweights.ncol',weighted_input=True) + qml_from_clu(".\\output\\infomap\\mlnet_K10_cmeans_pc50_sl.clu","mlnet_K10_cmeans_pc50_sl.qml") + #os.chdir("D:\\Mes Donnees\\Amoris\\geo2net-master\\geo2net-master\\ml_nets\\") + #flatten_network("mlnet_K10_cmeans_pc50.ncol","mlnet_K10_cmeans_pc50_flattened.ncol","mlnet_K10_cmeans_pc50_sl.ncol") \ No newline at end of file