com.moviejukebox.allocine.model.AbstractBaseInfos Maven / Gradle / Ivy

Go to download
/*
 *      Copyright (c) 2004-2015 YAMJ Members
 *      http://code.google.com/p/moviejukebox/people/list
 *
 *      This file is part of the Yet Another Movie Jukebox (YAMJ).
 *
 *      The YAMJ is free software: you can redistribute it and/or modify
 *      it under the terms of the GNU General Public License as published by
 *      the Free Software Foundation, either version 3 of the License, or
 *      any later version.
 *
 *      YAMJ is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *      You should have received a copy of the GNU General Public License
 *      along with the YAMJ.  If not, see .
 *
 *      Web: http://code.google.com/p/moviejukebox/
 *
 */
package com.moviejukebox.allocine.model;

import com.moviejukebox.allocine.tools.HtmlTools;
import java.util.Collections;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.Set;

@SuppressWarnings("serial")
public abstract class AbstractBaseInfos extends AbstractJsonUnknownHandleMapping {

    // Constants
    private static final double PERCENT_OUT_OF_5 = 5.0;
    private static final int PERCENT_OUT_OF_100 = 100;

    private Set genres;
    private Set nationalities;
    private Set posterURLS;

    protected int getCode(AbstractBaseMapping base) {
        return base == null ? -1 : base.getCode();
    }

    protected String getTitle(AbstractBaseMapping base) {
        return base == null ? null : base.getTitle();
    }

    protected String getOriginalTitle(AbstractBaseMapping base) {
        return base == null ? null : base.getOriginalTitle();
    }

    protected String getSynopsis(AbstractBaseMapping base) {
        return base == null ? null : HtmlTools.removeLineFeeds(base.getSynopsis());
    }

    protected String getSynopsisShort(AbstractBaseMapping base) {
        return base == null ? null : HtmlTools.removeLineFeeds(base.getSynopsisShort());
    }

    protected int getUserRating(AbstractBaseMapping base) {
        if (base == null || base.getStatistics() == null) {
            return -1;
        }

        double userRating = base.getStatistics().getUserRating();
        return (int) ((userRating / PERCENT_OUT_OF_5) * PERCENT_OUT_OF_100);
    }

    protected int getPressRating(AbstractBaseMapping base) {
        if (base == null || base.getStatistics() == null) {
            return -1;
        }

        double pressRating = base.getStatistics().getPressRating();
        return (int) ((pressRating / PERCENT_OUT_OF_5) * PERCENT_OUT_OF_100);
    }

    protected Set getGenres(AbstractBaseMapping base) {
        if (base == null) {
            return Collections.emptySet();
        }
        if (genres == null) {
            genres = new HashSet();
            if (base.getGenre() != null) {
                for (Genre genre : base.getGenre()) {
                    genres.add(genre.getName());
                }
            }
        }
        return genres;
    }

    protected Set getNationalities(AbstractBaseMapping base) {
        if (base == null) {
            return Collections.emptySet();
        }
        if (nationalities == null) {
            nationalities = new HashSet();
            if (base.getNationality() != null) {
                for (Nationality nationality : base.getNationality()) {
                    nationalities.add(nationality.getName());
                }
            }
        }
        return nationalities;
    }

    protected Set getActors(AbstractBaseMapping base) {
        Set set = new LinkedHashSet();
        if (base != null && base.getCastMember() != null) {
            for (CastMember member : base.getCastMember()) {
                if (member.isActor()) {
                    MoviePerson person = new MoviePerson();
                    person.setCode(member.getShortPerson().getCode());
                    person.setName(member.getShortPerson().getName());
                    person.setRole(member.getRole());
                    person.setLeadActor(member.isLeadActor());
                    if (member.getPicture() != null) {
                        person.setPhotoURL(member.getPicture().getHref());
                    }
                    set.add(person);
                }
            }
        }
        return set;
    }

    protected Set getDirectors(AbstractBaseMapping base) {
        Set set = new LinkedHashSet();
        if (base != null && base.getCastMember() != null) {
            for (CastMember member : base.getCastMember()) {
                if (member.isDirector()) {
                    addMember(member, set);
                }
            }
        }
        return set;
    }

    protected Set getWriters(AbstractBaseMapping base) {
        Set set = new LinkedHashSet();
        if (base != null && base.getCastMember() != null) {
            for (CastMember member : base.getCastMember()) {
                if (member.isWriter()) {
                    addMember(member, set);
                }
            }
        }
        return set;
    }

    protected Set getCamera(AbstractBaseMapping base) {
        Set set = new LinkedHashSet();
        if (base != null && base.getCastMember() != null) {
            for (CastMember member : base.getCastMember()) {
                if (member.isCamera()) {
                    addMember(member, set);
                }
            }
        }
        return set;
    }

    protected Set getProducers(AbstractBaseMapping base) {
        Set set = new LinkedHashSet();
        if (base != null && base.getCastMember() != null) {
            for (CastMember member : base.getCastMember()) {
                if (member.isProducer()) {
                    addMember(member, set);
                }
            }
        }
        return set;
    }

    /**
     * Add an cast member to a persons list
     *
     * @param member
     * @param persons
     */
    private void addMember(CastMember member, Set persons) {
        MoviePerson person = new MoviePerson();
        person.setCode(member.getShortPerson().getCode());
        person.setName(member.getShortPerson().getName());
        if (member.getPicture() != null) {
            person.setPhotoURL(member.getPicture().getHref());
        }
        persons.add(person);
    }

    private void parseMediaList(AbstractBaseMapping base) {
        if (posterURLS == null) {
            posterURLS = new LinkedHashSet();
        }

        if (base != null) {
            if (base.getPoster() != null) {
                posterURLS.add(base.getPoster().getHref());
            }
            if (base.getMedia() != null) {
                for (Medium medium : base.getMedia()) {
                    if (medium.isPoster() && medium.getThumbnail() != null) {
                        posterURLS.add(medium.getThumbnail().getHref());
                    }
                }
            }
        }
    }

    protected Set getPosterUrls(AbstractBaseMapping base) {
        if (posterURLS == null) {
            parseMediaList(base);
        }
        return posterURLS;
    }

    protected String getReleaseDate(AbstractBaseMapping base) {
        if (base == null) {
            return null;
        }
        if (base.getRelease() == null) {
            return null;
        }
        return base.getRelease().getReleaseDate();
    }

    protected String getReleaseState(AbstractBaseMapping base) {
        if (base == null) {
            return null;
        }
        if (base.getRelease() == null) {
            return null;
        }
        if (base.getRelease().getReleaseState() == null) {
            return null;
        }
        return base.getRelease().getReleaseState().getName();
    }
}