# -*- coding: utf-8 -*-
"""
Created on Thu Sep 29 14:51:09 2022

@author: katja
"""

import numpy
import os
import pandas as pd
import nibabel
import csv
import matplotlib.pyplot as plt

def organ_percentile(img, mask, level, p):
#ORGAN_PERCENTILE - computes suv_x the pth percentile of the distribution of
#   image intensity values in img from within some ROI mask
#
#   Inputs:
#       img - image. ndarray
#       mask - ROI mask. Must be same dimension as img. Must be binary ndarray
#       p - percentile. Between 0-100
#
#   Outputs:
#       suv_x - percentile of distribution. Defined by:
#
#                 suv_x
#           p/100 = ∫ H(x) dx
#                   0
#
#       where H(x) is the normalized distribution of image values within mask
    #img = np.array(img)
    #mask = np.array(mask)

    h = img[mask == level]
    print("Extracting SUV% for organ", level)

    #histo=[]
    # col=["red", "green", "yellow","pink", "brown", "darkblue", "black", "purple"]
    # i=level
    # plt.hist(h, color= col[i-1],bins=50, range=(0,6))
    # plt.title("Histogram of SUV values for "+ str(level))
    # plt.show()
    suv_x = numpy.percentile(h, p)

    return suv_x


def main():

    # Input csv file that has the path to each patient imagey
    inputCSV = os.path.join('../data/cases_retroMManon.csv')

    # Returns concatinated sequnces of evenly spaced numbers over a specified interval - for SUV percentile
    pv=numpy.concatenate((numpy.linspace(10,50,5),
            numpy.linspace(55,80,6),
            numpy.linspace(82,90,5),
            numpy.linspace(91,100,10)))

    pvi=pv.astype(int)      #have to be indigers to be able to append to name
    col_names = ["SUV" + str(x) for x in pvi]

    flists = []

    # goes trough the whole csv file and makes an flist (list)
    with open(inputCSV, 'r') as inFile:
        cr = csv.DictReader(inFile, lineterminator='\n')
        flists = [row for row in cr]

    alldata=[]

    histo=[]
    for idx, entry in enumerate(flists, start=1):
        PET = entry['Image']
        Seg = entry['Mask']
        Visit = entry['Visit']
        ID = entry['ID']

        #open PET image and segmentation
        niPET=nibabel.load(PET)
        niSeg=nibabel.load(Seg)

        print(" Processing Patient, Visit , (Image: , Mask:)", idx, len(flists), Visit, entry['Image'], entry['Mask'])
        #In nnU-Net (organ segmentation are numbered):
        #1 liver  #DM
        #2 spleen  #DM
        #3 lungs #DM
        #4 thyroid #DM
        #5 bowel  # DM 16
        #6 pancreas
        #7 bladder
        #8 kidneys

        #extract SUV percentiles from PET image with mask - segmentation
        for level in [1,2,3,4,5,6,7,8]:
            v=organ_percentile(niPET.get_fdata(),niSeg.get_fdata(),level,pv)
            mat=[level,ID, Visit]
            mat=numpy.concatenate((mat,v))
            alldata.append(mat)

    # plt.hist(histo, bins=5, range=(0,6))
    # plt.title("Histogram of SUV values for Spleen")
    # plt.show()
            #print("Percentile, value: ", mat)
    #print(alldata)
    # plt.hist(h, bins=50, range=(0,6))
    # plt.show()
    Names=["Organ", "PatientID", "Visit"]
    Names=numpy.concatenate((Names,col_names))

    #make a dataframe to write the data than in excel - that can be read by another script
    df=pd.DataFrame(alldata,columns=Names)
    df.to_excel('../data/percentilesAnon.xlsx', index=None) #path to where you want to save excel
    print("Done.")

if __name__ == '__main__':
  main()