package com.acumenvelocity.ath.mt.confidence;
   
   import java.util.List;
   
   import net.sf.okapi.common.resource.TextFragment;
   
   public class VertexDataStructures {
   
     public static class TranslationComparisonResult {
      private final List<ScoredSegmentTranslations> scoredSegments;
  
      public TranslationComparisonResult(List<ScoredSegmentTranslations> scoredSegments) {
        this.scoredSegments = scoredSegments;
      }
  
      public List<ScoredSegmentTranslations> getScoredSegments() {
        return scoredSegments;
      }
    }
  
    public static class ScoredSegmentTranslations {
      private final int segmentId;
      private final String sourceText;
      private final TextFragment sourceTf;
      private final List<ScoredTranslation> scoredTranslations;
  
      public ScoredSegmentTranslations(int segmentId, String sourceText, TextFragment sourceTf,
          List<ScoredTranslation> scoredTranslations) {
        this.segmentId = segmentId;
        this.sourceText = sourceText;
        this.sourceTf = sourceTf;
        this.scoredTranslations = scoredTranslations;
      }
  
      public int getSegmentId() {
        return segmentId;
      }
  
      public String getSourceText() {
        return sourceText;
      }
  
      public List<ScoredTranslation> getScoredTranslations() {
        return scoredTranslations;
      }
  
      public TextFragment getSourceTf() {
        return sourceTf;
      }
    }
  
    public static class ScoredTranslation {
      private final String modelId;
      private final String translation;
      private final Double metricXScore; // 0–100 (higher = better)
      private final Double fluencyScore; // 1.0–5.0
      private final Double coherenceScore; // 1.0–5.0
      private final Double groundednessScore; // 0.0–1.0
      private final Double lengthRatio;
      private final Double copyOverlap;
  
      public ScoredTranslation(String modelId, String translation, Double metricXScore,
          Double fluencyScore, Double coherenceScore, Double groundednessScore, Double lengthRatio,
          Double copyOverlap) {
        
        super();
        this.modelId = modelId;
        this.translation = translation;
        this.metricXScore = metricXScore;
        this.fluencyScore = fluencyScore;
        this.coherenceScore = coherenceScore;
        this.groundednessScore = groundednessScore;
        this.lengthRatio = lengthRatio;
        this.copyOverlap = copyOverlap;
      }
  
      public String getModelId() {
        return modelId;
      }
  
      public String getTranslation() {
        return translation;
      }
  
      public Double getMetricXScore() {
        return metricXScore;
      }
  
      public Double getFluencyScore() {
        return fluencyScore;
      }
  
      public Double getCoherenceScore() {
        return coherenceScore;
      }
  
      public Double getGroundednessScore() {
        return groundednessScore;
      }
 
     public Double getLengthRatio() {
       return lengthRatio;
     }
 
     public Double getCopyOverlap() {
       return copyOverlap;
     }
   }
 }