batch_extraction_interp.py

import numpy as np
import h5py
import glob
import json
import os
import h5py
import math

from XRStools import tools_sequencer_interp

import os
def main():
    
    os.system("xz -dk ../mask.h5.xz --stdout > mask.h5 ")

    filter_path = "mask.h5:/FILTER_MASK/filter"

    roi_scan_num   = [245,246,247]
    reference_scan_list = [245, 246, 247]

    
    monitor_column = "izero/0.000001"
    
    first_scan_num = 651 
    Ydim        = 25
    Zdim        = 10
    Edim        = 7

    elastic_scan_for_peaks_shifts = 42

    datadir =  "/data/id20/inhouse/data/run3_20/run3_es949"

    # If reference_clip is not None, then a smaller part of the reference scan is considered
    # This may be usefule to obtain smaller volumes containing the interesting part
    # The used reference scan  will the correspond to the positions from reference_clip[0] to reference_clip[1]-1 included
    ###########
    #    reference_clip = None
    reference_clip = [ 90, 180 ]

    ## in the reference scan for each position there is a spot with a maximum. We set zero the background which is further than
    ## such radius from the maximum
    isolate_spot_by = 6


    #### For the fit of the response function based on reference scans
    response_fit_options = dict( [
        ["niter_optical" , 100],
        ["beta_optical"  , 0.1],
        ["niter_global"  , 3  ]
    ])
                                 

    selected_rois = list(range(0,24)) + list( range(36,60) ) 


    scal_prod_use_optional_solution = True 


    volume_retrieval_beta  = 6.0e-20
    volume_retrieval_niter = 100 
    

    
    steps_to_do = {
        "do_step_make_roi":                      False,
        "do_step_sample_extraction":             True,
        "do_step_interpolation":                 False,
        "do_step_extract_reference_scan":             False,
                
        "do_step_fit_reference_response":               False,
        "do_step_resynthetise_reference":               False,
        "do_step_scalars"               :    True,
        "do_step_volume_retrieval"      :    False,
    }

    
    os.system("mkdir results")


    scalar_products_and_volume_target_file  = "results/scalar_products_and_volume.h5"

    roi_target_path    = "results/myrois.h5:/ROIS"
        
    signals_target_file = "results/signals.h5"
    
    interpolated_signals_target_file = "results/interpolated_signals.h5"

    extracted_reference_target_file = "results/reference.h5"
    
    response_target_file = "results/response.h5"

    resynthetised_reference_and_roi_target_file = "results/resynthetised_roi_and_scan.h5"

    ###########################################################################################
    ######  LOADING PEAKS SHIFTS
    ###### peaks_shifts = h5py.File("../peaks_positions_for_analysers.h5","r")["peaks_positions"][()]
    ###### assert( len(peaks_shifts) == 72)
    
    roiob = xrs_rois.roi_object()
    roiob.loadH5( roi_target_path  )
    elastic = xrs_read.Hydra( datadir )
    elastic.set_roiObj( roiob )
    elastic.get_compensation_factor( elastic_scan_for_peaks_shifts , method='sum')
    el_dict = elastic.cenom_dict
    Enominal = np.median(  list( el_dict.values() ) )
    peaks_shift = np.array([    el_dict["ROI%02d"%i] if ("ROI%02d"%i) in  el_dict else nan    for i in range 72  ] ) 
    Enominal = np.median(peaks_shifts)
    peaks_shifts-= Enominal
    
    ##############################################################
    ##########################################################################
    
    tools_sequencer_interp.tools_sequencer(  peaks_shifts          = peaks_shifts          ,
                      datadir               = datadir               ,
                      filter_path           = filter_path           ,
                      roi_scan_num          = roi_scan_num          ,
                      roi_target_path       = roi_target_path       ,
                      
                      steps_to_do = steps_to_do,
                      
                      first_scan_num =  first_scan_num,
                      Ydim        =  Ydim       ,
                      Zdim        =  Zdim       ,
                      Edim        =  Edim       ,

                      monitor_column = monitor_column,
                      signals_target_file = signals_target_file,
                      interpolated_signals_target_file = interpolated_signals_target_file,

                      reference_scan_list = reference_scan_list,
                      reference_clip = reference_clip,
                      extracted_reference_target_file = extracted_reference_target_file ,
                      isolate_spot_by =  isolate_spot_by,
                      response_target_file = response_target_file,
                      response_fit_options = response_fit_options,
                      
                      resynthetised_reference_and_roi_target_file = resynthetised_reference_and_roi_target_file,

                      selected_rois = selected_rois,
                      
                      scal_prod_use_optional_solution = scal_prod_use_optional_solution ,
                      scalar_products_and_volume_target_file      = scalar_products_and_volume_target_file ,

                      volume_retrieval_beta  = volume_retrieval_beta ,
                      volume_retrieval_niter = volume_retrieval_niter 
                      
    )
    
    

            
            
main()