o ‚<Ôcã@sNddlZddlZddlmZddlmZddlmZm Z dd„Z d dd „ZdS)éN)Úcheck_reference)Ú timeseries)ÚreadfileÚ writefilecCs\tj|tjd}tj|tjd}||}|t |¡|t |¡<|t |¡|t |¡<|S)zSum of 2 input matrix©Údtype)ÚnpÚarrayÚfloat32Úisnan)Zdata1Údata2Údata©rúC/home/exouser/operations/rsmas_insar/tools/MintPy/src/mintpy/add.pyÚ add_matrixsrFc st|ƒ}td|›d|›ƒtj |d¡d}|sBtj |d¡d}td|ƒD]}|dtj tj ||¡¡d7}q)||7}dd„|Dƒ}td|›ƒ|dd krq|d|d}}t |¡} t |¡} t |ƒ ¡‰t |ƒ ¡‰‡fd d„ˆDƒ}tjtˆƒtj d}|ˆkrÁtd|›d |›ƒ|r½ttˆƒt|ƒƒ} tdƒtd| ›ƒd|t ‡fdd„| Dƒ¡<ntdƒ‚t| | ƒ\}}}td|›ƒtj||dd}|r|rtdtj |¡›d|›d|›ƒ|||d|df}tj|||dd}|t | ddd¡d|jd|jdf¡8}|rGtdtj |¡›d|›ƒ| |¡}|t ||dd…dd…f|jdddf¡8}td|›ƒt |¡d}|dk}|||7<d||<~tj||| |d|Sdd„|Dƒ}ttjtt|ƒŽƒ}tdd „|Dƒƒr|d}td!|ƒt|ƒdkr£td"|›ƒ‚i}|D]E}td#|›d$ƒtj|d|d\}}t|dd…ƒD]"\}}t||dƒdkrÕdn|}tj||dd}t||ƒ}qÄ|||<q§td%|d›ƒtj||||dd|S)&aLGenerate sum of all input files Parameters: fnames - list of str, path/name of input files to be added out_file - str, optional, path/name of output file Returns: out_file - str, path/name of output file Example: 'mask_all.h5' = add_file(['mask_1.h5','mask_2.h5','mask_3.h5'], 'mask_all.h5') zinput files to be added: (z) réÚ_plus_cSsg|] }t |¡d‘qS)Ú FILE_TYPE)rÚread_attribute©Ú.0ÚxrrrÚ /szadd_file..zinput file types: rcsg|]}|ˆvr|‘qSrr©rÚi)Ú dateList2rrr9srz WARNING: z does not contain all dates in zBContinue and enforce the differencing for their shared dates only.z4 with following dates are ignored for differencing: csg|]}ˆ |¡‘qSr)Úindexr)Ú dateList1rrrAóz7To enforce the differencing anyway, use --force option.zread from file: )ÚdatasetNamez* referencing data from z to y/x: ú/)rÚboxéÿÿÿÿéz to date: Ng)Úout_fileÚmetadataZref_filecSsg|]}t |¡‘qSr)rÚget_dataset_listrrrrrfrcss|] }t|ƒdkVqdS)rN)ÚlenrrrrÚ is€zadd_file..z&List of common datasets across files: z&No common datasets found among files: zadding z ...z use metadata from the 1st file: ) r'ÚprintÚosÚpathÚsplitextÚrangeÚbasenamerrrÚ get_date_listrÚonesÚbool_ÚlistÚsetr Ú ExceptionrÚreadÚtileÚreshapeÚshaperrÚwriteÚintersectionÚmapÚallÚ ValueErrorÚ enumerater)Úfnamesr$ÚforceÚnum_fileÚextrZftypesZfile1Zfile2Zatr1Zatr2ZdateListSharedZ dateSharedZ dateListExÚref_dateÚref_yÚref_xrZref_boxÚref_valZref_indr ÚmaskZ ds_names_listÚds_namesZdsDictÚds_nameÚatrÚfnameZds_name2readr)rrrÚadd_files‚& ", . ä rL)NF)r*ÚnumpyrZmintpy.diffrÚmintpy.objectsrÚmintpy.utilsrrrrLrrrrÚs