EncodingTranslator.java

package org.atlasapi.persistence.media.entity;

import java.util.List;
import java.util.Set;

import org.atlasapi.media.entity.Encoding;
import org.atlasapi.media.entity.Location;
import org.atlasapi.media.entity.Quality;
import org.atlasapi.persistence.ModelTranslator;

import com.google.common.collect.ComparisonChain;
import com.google.common.collect.Ordering;
import com.google.common.collect.Sets;
import com.metabroadcast.common.media.MimeType;
import com.metabroadcast.common.persistence.translator.TranslatorUtils;
import com.mongodb.BasicDBList;
import com.mongodb.DBObject;

public class EncodingTranslator implements ModelTranslator<Encoding> {
	
    private static final String SUBTITLED_KEY = "subtitled";
    static final String LOCATIONS_KEY = "availableAt";
    private static final String AUDIO_DESCRIBED_KEY = "audioDescribed";
    private static final String SIGNED_KEY = "signed";
    private static final String HIGH_DEFINITION_KEY = "highDefinition";
    private static final String QUALITY_KEY = "quality";
    private static final String QUALITY_DETAIL_KEY = "qualityDetail";

    private final IdentifiedTranslator descriptionTranslator = new IdentifiedTranslator();
    private final LocationTranslator locationTranslator = new LocationTranslator();

    @SuppressWarnings("unchecked")
    @Override
    public Encoding fromDBObject(DBObject dbObject, Encoding entity) {
        if (entity == null) {
            entity = new Encoding();
        }
        
        descriptionTranslator.fromDBObject(dbObject, entity);
        
        entity.setAdvertisingDuration((Integer) dbObject.get("advertisingDuration"));
        entity.setAudioBitRate((Integer) dbObject.get("audioBitRate"));
        entity.setAudioChannels((Integer) dbObject.get("audioChannels"));
        
        entity.setAudioCoding(readAudioCoding(dbObject));
        
        entity.setBitRate((Integer) dbObject.get("bitRate"));
        entity.setAudioChannels((Integer) dbObject.get("audioChannels"));
        entity.setContainsAdvertising((Boolean) dbObject.get("containsAdvertising"));
        entity.setDataContainerFormat(readContainerFormat(dbObject));
        entity.setDataSize((Long) dbObject.get("dataSize"));
        entity.setDistributor((String) dbObject.get("distributor"));
        entity.setHasDOG((Boolean) dbObject.get("hasDOG"));
        entity.setSource((String) dbObject.get("source"));
        entity.setVideoAspectRatio((String) dbObject.get("videoAspectRatio"));
        entity.setVideoBitRate((Integer) dbObject.get("videoBitRate"));
        entity.setSubtitled((Boolean) dbObject.get(SUBTITLED_KEY));
        if (dbObject.containsField(QUALITY_KEY)) {
            entity.setQuality(Quality.valueOf(TranslatorUtils.toString(dbObject, QUALITY_KEY).toUpperCase()));
        }
        entity.setQualityDetail(TranslatorUtils.toString(dbObject, QUALITY_DETAIL_KEY));
        
        entity.setVideoCoding(readVideoCoding(dbObject));
        
        entity.setVideoFrameRate(TranslatorUtils.toFloat(dbObject, "videoFrameRate"));
        
        entity.setVideoHorizontalSize((Integer) dbObject.get("videoHorizontalSize"));
        entity.setVideoProgressiveScan((Boolean) dbObject.get("videoProgressiveScan"));
        entity.setVideoVerticalSize((Integer) dbObject.get("videoVerticalSize"));

        if (dbObject.get(AUDIO_DESCRIBED_KEY) != null) {
            entity.setAudioDescribed(TranslatorUtils.toBoolean(dbObject, AUDIO_DESCRIBED_KEY));
        }

        if (dbObject.get(SIGNED_KEY) != null) {
            entity.setSigned(TranslatorUtils.toBoolean(dbObject, SIGNED_KEY));
        }
        if (dbObject.get(HIGH_DEFINITION_KEY) != null) {
            entity.setHighDefinition(TranslatorUtils.toBoolean(dbObject, HIGH_DEFINITION_KEY));
        }
        
        List<DBObject> list = (List<DBObject>) dbObject.get(LOCATIONS_KEY);
        if (list != null && ! list.isEmpty()) {
            Set<Location> locations = Sets.newHashSet();
            for (DBObject object: list) {
                Location location = locationTranslator.fromDBObject(object, null);
                locations.add(location);
            }
            entity.setAvailableAt(locations);
        }
        
        return entity;
    }

    private MimeType readVideoCoding(DBObject dbObject) {
    	String codingAsString = (String) dbObject.get("videoCoding");
    	if (codingAsString == null) {
    		return null;
    	}
    	return MimeType.fromString(codingAsString);
    }
    
    private MimeType readAudioCoding(DBObject dbObject) {
    	String codingAsString = (String) dbObject.get("audioCoding");
    	if (codingAsString == null) {
    		return null;
    	}
    	return MimeType.fromString(codingAsString);
    }

	private MimeType readContainerFormat(DBObject dbObject) {
    	String formatName = (String) dbObject.get("dataContainerFormat");
    	if (formatName == null) {
    		return null;
    	}
    	return MimeType.fromString(formatName);
    }

	@Override
    public DBObject toDBObject(DBObject dbObject, Encoding entity) {
        dbObject = descriptionTranslator.toDBObject(dbObject, entity);
        
        TranslatorUtils.from(dbObject, "advertisingDuration", entity.getAdvertisingDuration());
        TranslatorUtils.from(dbObject, "audioBitRate", entity.getAudioBitRate());
        TranslatorUtils.from(dbObject, "audioChannels", entity.getAudioChannels());
        
        if (entity.getAudioCoding() != null) {
        	TranslatorUtils.from(dbObject, "audioCoding", entity.getAudioCoding().toString());
        }
        
        TranslatorUtils.from(dbObject, "bitRate", entity.getBitRate());
        TranslatorUtils.from(dbObject, "audioChannels", entity.getAudioChannels());
        TranslatorUtils.from(dbObject, "containsAdvertising", entity.getContainsAdvertising());
        
        if (entity.getDataContainerFormat() != null) {
        	TranslatorUtils.from(dbObject, "dataContainerFormat", entity.getDataContainerFormat().toString());
        }
        
        TranslatorUtils.from(dbObject, "dataSize", entity.getDataSize());
        TranslatorUtils.from(dbObject, "distributor", entity.getDistributor());
        TranslatorUtils.from(dbObject, "hasDOG", entity.getHasDOG());
        TranslatorUtils.from(dbObject, "source", entity.getSource());
        TranslatorUtils.from(dbObject, "videoAspectRatio", entity.getVideoAspectRatio());
        TranslatorUtils.from(dbObject, "videoBitRate", entity.getVideoBitRate());
       
        if (entity.getVideoCoding() != null) {
        	TranslatorUtils.from(dbObject, "videoCoding", entity.getVideoCoding().toString());
        }
        
        if (entity.getQuality() != null) {
            TranslatorUtils.from(dbObject, QUALITY_KEY, entity.getQuality().toString().toLowerCase());
        }
        TranslatorUtils.from(dbObject, QUALITY_DETAIL_KEY, entity.getQualityDetail());
        TranslatorUtils.from(dbObject, "videoFrameRate", entity.getVideoFrameRate());
        TranslatorUtils.from(dbObject, "videoHorizontalSize", entity.getVideoHorizontalSize());
        TranslatorUtils.from(dbObject, "videoProgressiveScan", entity.getVideoProgressiveScan());
        TranslatorUtils.from(dbObject, "videoVerticalSize", entity.getVideoVerticalSize());
        TranslatorUtils.from(dbObject, AUDIO_DESCRIBED_KEY, entity.getAudioDescribed());
        TranslatorUtils.from(dbObject, SIGNED_KEY, entity.getSigned());
        TranslatorUtils.from(dbObject, SUBTITLED_KEY, entity.getSubtitled());
        TranslatorUtils.from(dbObject, HIGH_DEFINITION_KEY, entity.getHighDefinition());

        if (! entity.getAvailableAt().isEmpty()) {
            BasicDBList list = new BasicDBList();
            for (Location location: LOCATION_ORDERING.immutableSortedCopy(entity.getAvailableAt())) {
                list.add(locationTranslator.toDBObject(null, location));
            }
            dbObject.put(LOCATIONS_KEY, list);
        }
        
        return dbObject;
    }
	
    private static final Ordering<Location> LOCATION_ORDERING = new Ordering<Location>() {

        @Override
        public int compare(Location left, Location right) {
            ComparisonChain chain = ComparisonChain.start()
                    .compare(left.getCanonicalUri(), right.getCanonicalUri(), Ordering.natural().nullsLast())
                    .compare(left.getUri(), right.getUri(), Ordering.natural().nullsLast());
            
            if (left.getPolicy() != null && right.getPolicy() != null) {
                chain.compare(left.getPolicy().getPlatform(), right.getPolicy().getPlatform(), Ordering.natural().nullsLast())
                     .compare(left.getPolicy().getAvailabilityStart(), right.getPolicy().getAvailabilityStart(), Ordering.natural().nullsLast());
            }
            return chain.result();
        }
        
    };

}