#!/usr/bin/env python
"""
Test script for EAMxx COSP histogram support.

This script tests the new EAMxx COSP histogram variables:
- isccp_ctptau (replaces FISCCP1_COSP)
- modis_ctptau (replaces CLMODIS)
- misr_cthtau (replaces CLD_MISR)
"""

import os
from e3sm_diags.parameter.core_parameter import CoreParameter
from e3sm_diags.run import runner

# Set up parameters
param = CoreParameter()

# Test case name
param.case_id = "EAMxx_cosp_histogram_test"
param.short_name = "EAMxx_COSP"

# Test data path - use PREPROCESSED data with added COSP coordinates
# Run preprocess_eamxx_data.py first to create this directory
test_data_path = "/pscratch/sd/c/chengzhu/EAMxx/ne256pg2_ne256pg2.F20TR-SCREAMv1.July-1.spanc800.2xauto.acc150.n0032.test2.1/rgr/climo"
test_data_path= "/pscratch/sd/t/terai/e3sm_scratch/pm-gpu/ne256pg2_ne256pg2.F20TR-SCREAMv1.copilot-fix.cosp-test/climo/rgr/"
param.test_data_path = test_data_path
param.test_name = "1ma_ne30pg2.AVERAGE"
param.test_name = "eamxx_ne256"

# Reference data (use obs data for COSP)
param.reference_data_path = "/global/cfs/cdirs/e3sm/diagnostics/observations/Atm/climatology"
param.results_dir = os.path.join(
    '/global/cfs/cdirs/e3sm/www/chengzhu/tests',
    'eamxx_cosp_histogram_try2'
)


# Run cosp_histogram set
param.sets = ["cosp_histogram"]
#param.seasons = ["ANN"]

# Run the diagnostics
runner.sets_to_run = param.sets
runner.run_diags([param])