package com.acumenvelocity.ath.mt.confidence;
   
   import java.util.Map;
   
   import net.sf.okapi.common.resource.TextFragment;
   
   /**
    * Common data structures used across all quality estimators (Heuristic, Vertex, Hybrid).
    */
  public class CommonDataStructures {
  
    /**
     * Represents a single source segment and all its translations from various models.
     */
    public static class SegmentTranslations {
      private final int segmentId;
      private final String sourceText;
      private final TextFragment sourceTf;
      private final Map<String, String> translations;
  
      public SegmentTranslations(int segmentId, String sourceText, TextFragment sourceTf,
          Map<String, String> translations) {
        this.segmentId = segmentId;
        this.sourceText = sourceText;
        this.sourceTf = sourceTf;
        this.translations = translations;
      }
  
      public int getSegmentId() {
        return segmentId;
      }
  
      public String getSourceText() {
        return sourceText;
      }
  
      public Map<String, String> getTranslations() {
        return translations;
      }
  
      public TextFragment getSourceTf() {
        return sourceTf;
      }
    }
  
    /**
     * Represents the language pair support level from the Vertex AI Quality Estimation service.
     */
    public static class LanguagePairSupport {
      private final String sourceLang;
      private final String targetLang;
      private final String supportLevel; // FULL, COMET_ONLY, LIMITED
      private final boolean cometSupported;
      private final boolean metricXOptimal;
      private final String recommendation;
  
      public LanguagePairSupport(String sourceLang, String targetLang, String supportLevel,
          boolean cometSupported, boolean metricXOptimal, String recommendation) {
  
        this.sourceLang = sourceLang;
        this.targetLang = targetLang;
        this.supportLevel = supportLevel;
        this.cometSupported = cometSupported;
        this.metricXOptimal = metricXOptimal;
        this.recommendation = recommendation;
      }
  
      public String getSupportLevel() {
        return supportLevel;
      }
  
      public boolean isCometSupported() {
        return cometSupported;
      }
  
      public boolean isMetricXOptimal() {
        return metricXOptimal;
      }
  
      public String getRecommendation() {
        return recommendation;
      }
  
      @Override
      public String toString() {
        return String.format("%s→%s: %s [COMET: %s, MetricX: %s]\n%s", sourceLang, targetLang,
            supportLevel, cometSupported ? "✓" : "✗", metricXOptimal ? "✓" : "○", recommendation);
      }
    }
  }