src/main/java/ftbsc/lll/mapper/writer/impl/SRGWriter.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

package ftbsc.lll.mapper.writer.impl;

import com.google.auto.service.AutoService;
import ftbsc.lll.mapper.tools.Mapper;
import ftbsc.lll.mapper.tools.MappingUtils;
import ftbsc.lll.mapper.tools.data.FieldData;
import ftbsc.lll.mapper.tools.data.MethodData;
import ftbsc.lll.mapper.writer.IWriter;

import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;

/**
 * An {@link IWriter} that writes in the SRG format.
 */
@AutoService(IWriter.class)
public class SRGWriter implements IWriter {
   @Override
   public String uniqueId() {
      return "srg";
   }

   @Override
   public void write(Mapper mapper, PrintWriter writer, String... ignored) {
      List<FieldData> fieldData = new ArrayList<>();
      List<MethodData> methodData = new ArrayList<>();

      //print classes and save rest for later
      mapper.getRawMappings().forEach((name, data) -> {
         writer.printf("CL: %s %s\n", name, data.nameMapped);
         fieldData.addAll(data.getFields().values());
         methodData.addAll(data.getMethods().values());
      });

      //print fields
      fieldData.forEach(data -> writer.printf("FD: %s/%s %s/%s\n",
         data.parentClass.name, data.name, data.parentClass.nameMapped, data.nameMapped));

      //print methods
      methodData.forEach(data -> writer.printf("MD: %s/%s %s %s/%s %s\n",
         data.parentClass.name, data.signature.name, data.signature.descriptor,
         data.parentClass.nameMapped, data.nameMapped, MappingUtils.mapMethodDescriptor(
            data.signature.descriptor, mapper, false)));
   }
}