Doing wedge integrations

from __future__ import print_function

import sys
import os

import numpy as np
from matplotlib import pyplot as plt

sasview_directory = '/Users/rhf/git/sasview/build/lib.macosx-10.11-x86_64-2.7'
sys.path.append(sasview_directory)

from sas.sascalc.dataloader.readers.red2d_reader import Reader
from sas.sascalc.dataloader.manipulations import Sectorcut, SectorQ

filename =  "my_file_Iqxy.dat" # Reduced QxQy file
phi_center =  0  # Phi Center in degrees
phi_width = 30  # Phi Width in degrees

def plot_iqxqy(data):
    '''
    @param data :: Output of Sasview Reader
    '''
    qx = data.qx_data
    qy = data.qy_data
    iqxqy = data.data
    # Reshape the data as 2D
    plt.figure()
    plt.scatter(qx, qy, c=np.log(iqxqy), s=50, edgecolor='', marker='s')
    plt.colorbar()
    # plt.show()

def plot_iq(data):
    '''
    @param data :: is 1D
    '''
    # Plot in 1D log scale
    plt.figure()
    plt.yscale('log', nonposy='clip')
    plt.xscale('log', nonposx='clip')
    plt.errorbar(data.x, data.y, yerr=data.dy)

def integrate_wedge(data, phi_center, phi_width, nbins=100):
    '''
    @param base: must be a valid base for an algorithm, i.e., a positive number
    The code for integration is here:
    https://github.com/SasView/sasview/blob/master/src/sas/sascalc/dataloader/manipulations.py
    '''
    phi_min = phi_center - phi_width/2.0
    phi_max = phi_center + phi_width/2.0

    print("Integrating from {} to {}".format(phi_min, phi_max))

    phi_min = np.deg2rad(phi_min)
    phi_max = np.deg2rad(phi_max)
    
    sector_wedge = SectorQ(r_min=0.0001, r_max=1, phi_min=phi_min, 
                           phi_max=phi_max, nbins=nbins)
    iq_wedge = sector_wedge(data)
    return iq_wedge

r = Reader()
data = r.read(filename)
plot_iqxqy(data)
iq_wedge = integrate_wedge(data, phi_center, phi_width, nbins=50)
plot_iq(iq_wedge)
plt.show()