package com.acumenvelocity.ath.mt.confidence;
   
   import java.util.List;
   import java.util.Map;
   
   /**
    * Data structures for the HeuristicQualityEstimator.
    */
   public class HeuristicDataStructures {
  
    /**
     * Represents the heuristic quality score for a single translation.
     */
    public static class HeuristicScore {
      private final String modelId;
      private final String translation;
      private final double overallScore; // 0.0 - 1.0 (higher is better)
      private final Map<String, Double> detailedMetrics;
      private final List<String> detectedIssues;
  
      public HeuristicScore(String modelId, String translation, double overallScore,
          Map<String, Double> detailedMetrics, List<String> detectedIssues) {
        
        this.modelId = modelId;
        this.translation = translation;
        this.overallScore = overallScore;
        this.detailedMetrics = detailedMetrics;
        this.detectedIssues = detectedIssues;
      }
  
      public String getModelId() {
        return modelId;
      }
  
      public String getTranslation() {
        return translation;
      }
  
      public double getOverallScore() {
        return overallScore;
      }
  
      public Map<String, Double> getDetailedMetrics() {
        return detailedMetrics;
      }
  
      public List<String> getDetectedIssues() {
        return detectedIssues;
      }
  
      @Override
      public String toString() {
        String metrics = detailedMetrics.entrySet().stream()
            .map(e -> String.format("%s=%.2f", e.getKey(), e.getValue()))
            .collect(java.util.stream.Collectors.joining(", "));
  
        StringBuilder sb = new StringBuilder();
        
        sb.append(String.format("Translation: '%s' | Score: %.3f | Metrics: %s",
            translation.substring(0, Math.min(translation.length(), 60))
                + (translation.length() > 60 ? "..." : ""),
            overallScore,
            metrics));
  
        if (!detectedIssues.isEmpty()) {
          sb.append("\n    Issues: ");
          sb.append(String.join("; ", detectedIssues));
        }
  
        return sb.toString();
      }
    }
  
    /**
     * Represents the collection of heuristic scores for all translations of a single segment.
     */
    public static class HeuristicResult {
      private final int segmentId;
      private final String sourceText;
      private final List<HeuristicScore> scores;
  
      public HeuristicResult(int segmentId, String sourceText, List<HeuristicScore> scores) {
        this.segmentId = segmentId;
        this.sourceText = sourceText;
        this.scores = scores;
      }
  
      public int getSegmentId() {
        return segmentId;
      }
  
      public String getSourceText() {
        return sourceText;
      }
  
      public List<HeuristicScore> getScores() {
        return scores;
      }
  
     public HeuristicScore getBestScore() {
       return scores.isEmpty() ? null : scores.get(0);
     }
 
     public void printReport() {
       System.out.printf("\n=== Segment %d ===%n", segmentId);
       System.out.printf("Source: %s%n", sourceText);
       System.out.println("\nHeuristic Scores:");
 
       for (int i = 0; i < scores.size(); i++) {
         System.out.printf("%d. %s%n", i + 1, scores.get(i));
       }
     }
   }
 }