#!python
# -*- coding: utf-8 -*-
# Copyright(c) Ryuichiro Nakato <rnakato@iqb.u-tokyo.ac.jp>
# All rights reserved.

import sys
import os
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import pandas as pd
from custardpy.HiCmodule import *
from custardpy.generateCmap import *
cm = generate_cmap(['#FFFFFF', '#d10a3f'])

usage = 'Usage: \n    {} <matrixdir> <output prefix> <start> <end> <label>'.format(os.path.basename(__file__))
arguments = sys.argv
if len(arguments) == 1:
    print(usage)
    exit()

filename = sys.argv[1]
output = sys.argv[2]
start = int(sys.argv[3])
end = int(sys.argv[4])
label = sys.argv[5]

data = pd.read_csv(filename, delimiter='\t', index_col=0)
resolution = data.index[1] - data.index[0]

s = int(start / resolution)
e = int(end / resolution)

# Total read normalization
data = data * 1000000 / np.nansum(data)

fig = plt.figure(figsize=(8, 8))
ymax = np.sqrt(data.unstack().max())/2
plt.imshow(ExtractMatrix(data,s,e), clim=(0, ymax), cmap=cm)
plt.title(label)
plt.savefig(output)
