Ejemplo n.º 1
0
 private Map<Album, Map<ImageType, Resource>> collectAlbumTargets(
     Library library, final ResourceLocator resourceLocator, Collection<Album> changedAlbums) {
   Map<Album, Map<ImageType, Resource>> result = new HashMap<>();
   for (Album album : library.getAlbums()) {
     Resource artworkAssetResource = album.artworkAssetResource();
     if (artworkAssetResource != null) {
       Map<ImageType, Resource> targets = new HashMap<ImageType, Resource>();
       for (ImageType type : ImageType.values()) {
         String imagePath = resourceLocator.getAlbumImagePath(album, type);
         if (imagePath != null) {
           Resource resource = resourceLocator.getResource(imagePath);
           try {
             if (changedAlbums.contains(album) || !resource.exists()) {
               resource.getParent().mkdirs();
               targets.put(type, resource);
             }
           } catch (IOException e) {
             LOGGER.warning("Could not write image file: " + resource.getPath().toAbsolutePath());
           }
         }
       }
       if (!targets.isEmpty()) {
         result.put(album, targets);
       }
     }
   }
   return result;
 }
Ejemplo n.º 2
0
  public void formatImages(
      Library library,
      ResourceLocator resourceLocator,
      Collection<Album> changedAlbums,
      int maxThreads,
      final ProgressHandler progressHandler) {
    if (progressHandler != null) {
      progressHandler.beginTask(-1, "Preparing images...");
    }
    final Map<Album, Map<ImageType, Resource>> albumTargets =
        collectAlbumTargets(library, resourceLocator, changedAlbums);
    if (progressHandler != null) {
      progressHandler.endTask();
    }
    List<Album> albums = new ArrayList<>(albumTargets.keySet());

    int numberOfAlbumsPerTask = 10;
    int numberOfAlbums = albums.size();
    int numberOfThreads =
        Math.min(
            1 + (numberOfAlbums - 1) / numberOfAlbumsPerTask,
            Math.min(maxThreads, Runtime.getRuntime().availableProcessors()));
    if (progressHandler != null) {
      progressHandler.beginTask(numberOfAlbums, "Formatting images...");
    }
    final int progressModulo = numberOfAlbums < 200 ? 10 : numberOfAlbums < 1000 ? 50 : 100;
    if (numberOfThreads > 1) { // run on multiple threads
      if (LOGGER.isLoggable(Level.FINER)) {
        LOGGER.finer("Parallel: #threads = " + numberOfThreads);
      }
      ExecutorService executor = Executors.newFixedThreadPool(numberOfThreads);
      final AtomicInteger atomicCount = new AtomicInteger();
      for (int start = 0; start < numberOfAlbums; start += numberOfAlbumsPerTask) {
        final Collection<Album> albumsSlice =
            albums.subList(start, Math.min(numberOfAlbums, start + numberOfAlbumsPerTask));
        executor.execute(
            new Runnable() {
              @Override
              public void run() {
                for (Album album : albumsSlice) {
                  formatImages(album.artworkAssetResource(), albumTargets.get(album));
                  int count = atomicCount.getAndIncrement() + 1;
                  if (progressHandler != null && count % progressModulo == 0) {
                    progressHandler.progress(count, String.format("#albums = %4d", count));
                  }
                }
              }
            });
      }
      executor.shutdown();
      try {
        executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
      } catch (InterruptedException e) {
        LOGGER.warning("Interrupted: " + e.getMessage());
      }
    } else { // run on current thread
      int count = 0;
      for (Album album : albums) {
        formatImages(album.artworkAssetResource(), albumTargets.get(album));
        count++;
        if (progressHandler != null && count % progressModulo == 0) {
          progressHandler.progress(count, String.format("#albums = %4d", count));
        }
      }
    }
    if (progressHandler != null) {
      progressHandler.endTask();
    }
  }