#! /usr/bin/python
# -*- coding: utf-8 -*-
# Author: Joker
# Copyright (c) 2019 OpenString. All rights reserved.
import cv2
import numpy as np
import imageio
from openstring.core import DoseParser
from openstring.config import ConfigManager
def convert(dcmDir, pngDir):
doseParser = DoseParser.DoseParser(dcmDir)
# doseLabelList = doseParser.getAllDoseLineLabel()
# for index, doseLabel in enumerate(doseLabelList):
# pngPath = '%s/%3.3d.png' % (pngDir, index + 1)
# cv2.imwrite(pngPath, doseLabel)
doseLabelList = doseParser.getAllDoseValueLabelWithFloat64()
for index, doseLabel in enumerate(doseLabelList):
pngPath = '%s/%3.3d.png' % (pngDir, index + 1)
cv2.imwrite(pngPath, doseLabel)
# pngPath = ConfigManager.getResourcePath('/png/test.png')
# label = doseParser.getOneDoseValueLabelWithFloat64(-634.4)
# cv2.imwrite(pngPath, label)
# imageio.imwrite(pngPath, label)
# imageio.imread(pngPath)
pass