mirror of https://github.com/vladmandic/automatic
16 lines
763 B
Python
16 lines
763 B
Python
from modules import scripts_postprocessing, ui_sections, processing_grading
|
|
|
|
|
|
class ScriptPostprocessingColorGrading(scripts_postprocessing.ScriptPostprocessing):
|
|
name = "Color Grading"
|
|
|
|
def ui(self):
|
|
ui_controls = ui_sections.create_color_inputs('process')
|
|
ui_controls_dict = {control.label.replace(" ", "_").replace(".", "").lower(): control for control in ui_controls}
|
|
return ui_controls_dict
|
|
|
|
def process(self, pp: scripts_postprocessing.PostprocessedImage, *args, **kwargs): # pylint: disable=arguments-differ
|
|
grading_params = processing_grading.GradingParams(*args, **kwargs)
|
|
if processing_grading.is_active(grading_params):
|
|
pp.image = processing_grading.grade_image(pp.image, grading_params)
|