src/main/java/ftbsc/lll/mapper/impl/TSRGMapper.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
47
48
49
50

package ftbsc.lll.mapper.impl;

import com.google.auto.service.AutoService;
import ftbsc.lll.exceptions.MalformedMappingsException;
import ftbsc.lll.mapper.IMappingFormat;
import ftbsc.lll.mapper.tools.Mapper;
import ftbsc.lll.mapper.tools.data.ClassData;

import java.util.List;
import java.util.regex.Pattern;

/**
 * A {@link IMappingFormat} capable of parsing TSRG (an intermediary
 * format used by Forge) files.
 */

@AutoService(IMappingFormat.class)
public class TSRGMapper implements IMappingFormat {

   @Override
   public boolean claim(List<String> lines) {
      return Pattern.compile("tsrg2 [a-zA-Z]* [a-zA-Z]*")
         .matcher(lines.get(0)).matches();
   }

   @Override
   public Mapper getMapper(List<String> lines, boolean ignoreErrors) throws MalformedMappingsException {
      Mapper result = new Mapper();
      String currentClass = "";
      for(int i = 1; i < lines.size(); i++) { //start from 1 to skip header
         String currentLine = lines.get(i);
         boolean isMember = currentLine.startsWith("\t") || currentLine.startsWith(" ");
         String[] tokens = currentLine.trim().split(" ");
         if(isMember) {
            if(tokens.length == 2) //field
               result.getClassData(currentClass).addField(tokens[0], tokens[1]);
            else if(tokens.length == 3)//method
               result.getClassData(currentClass).addMethod(tokens[0], tokens[2], tokens[1]); //add child
            else if(!ignoreErrors) throw new MalformedMappingsException(i, "wrong number of space-separated tokens");
         } else {
            if(tokens.length == 2) {
               ClassData s = new ClassData(tokens[0], tokens[1]);
               currentClass = s.name;
               result.getRawMappings().put(s.name, s);
            } else if(!ignoreErrors) throw new MalformedMappingsException(i, "wrong number of space-separated tokens");
         }
      }
      return result;
   }
}